manager loop auto stop servers

This commit is contained in:
2026-06-10 20:51:02 +02:00
parent 7bae17c525
commit fe5235a3ca

View File

@ -1,6 +1,7 @@
use bollard::{Docker, query_parameters::{ListContainersOptionsBuilder, StopContainerOptionsBuilder}}; use bollard::{Docker, query_parameters::{ListContainersOptionsBuilder, StopContainerOptionsBuilder}};
use chrono::{TimeDelta, Utc};
use crate::{ DbPool, dockermgr, models, schema, status::ServerStatus }; use crate::{ DbPool, dockermgr, models::{self, Servers}, schema, status::ServerStatus };
use async_std::task; use async_std::task;
@ -44,12 +45,35 @@ async fn server_init(pool: &DbPool, docker: &Docker) -> Result<(), String> {
pub async fn server_manager_loop(pool: DbPool, docker: Docker) pub async fn server_manager_loop(pool: DbPool, docker: Docker)
{ {
use schema::servers::dsl::*;
let conn = &mut pool.get().unwrap();
match server_init(&pool, &docker).await { match server_init(&pool, &docker).await {
Err(e) => { eprintln!("init failed , ignoring because idgaf : {}", e); }, Err(e) => { eprintln!("init failed , ignoring because idgaf : {}", e); },
Ok(_) => {}, Ok(_) => {},
}; };
loop { loop {
task::sleep(Duration::from_mins(5)).await; let a_servers = match servers.select(models::Servers::as_select()).filter(simple_redirect.eq(false)).load(conn) {
Ok( srv ) => { srv },
Err(error) => { eprintln!("Failed to retreive servers, ignoring : {}", error); vec![] },
};
for srv in a_servers {
let time_diff = Utc::now().naive_utc() - srv.last_login.unwrap();
should_stop(&docker, &pool, &time_diff, &srv).await;
}
task::sleep(Duration::from_secs(5)).await;
}
}
async fn should_stop(docker: &Docker, pool: &DbPool, diff: &TimeDelta, srv: &Servers) {
if srv.status != ServerStatus::Running { return ; }
if diff.num_days() >= 7 {
println!("last login time exceeded, stopping {}", srv.name);
let _ = dockermgr::stop_server(docker, pool, &srv.name).await;
println!("{} stopped", srv.name);
} }
} }