manager loop auto stop servers
This commit is contained in:
@ -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);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user