manager loop auto stop servers
This commit is contained in:
@ -1,6 +1,7 @@
|
||||
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;
|
||||
|
||||
@ -44,12 +45,35 @@ async fn server_init(pool: &DbPool, docker: &Docker) -> Result<(), String> {
|
||||
|
||||
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 {
|
||||
Err(e) => { eprintln!("init failed , ignoring because idgaf : {}", e); },
|
||||
Ok(_) => {},
|
||||
};
|
||||
|
||||
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);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user