manager loop WIP, add nix flake
This commit is contained in:
55
src/minecraft/manager_loop.rs
Normal file
55
src/minecraft/manager_loop.rs
Normal file
@ -0,0 +1,55 @@
|
||||
use bollard::{Docker, query_parameters::{ListContainersOptionsBuilder, StopContainerOptionsBuilder}};
|
||||
|
||||
use crate::{ DbPool, dockermgr, models, schema, status::ServerStatus };
|
||||
|
||||
use async_std::task;
|
||||
|
||||
use std::{collections::HashMap, time::Duration};
|
||||
|
||||
use diesel::prelude::*;
|
||||
|
||||
|
||||
async fn server_init(pool: &DbPool, docker: &Docker) -> Result<(), String> {
|
||||
use schema::servers::dsl::*;
|
||||
let conn = &mut pool.get().unwrap();
|
||||
|
||||
println!("init start");
|
||||
|
||||
let _ = diesel::update(servers).filter(simple_redirect.eq(false)).set(status.eq(ServerStatus::Stopped)).execute(conn);
|
||||
|
||||
let mut filters = HashMap::new();
|
||||
filters.insert("name", vec!["minecraft-"]);
|
||||
let container_list_options = ListContainersOptionsBuilder::new()
|
||||
.all(true)
|
||||
.filters(&filters)
|
||||
.build();
|
||||
let containers = docker.list_containers(Some(container_list_options)).await.map_err(|e| format!("Failed get current containers : {}", e))?;
|
||||
|
||||
if containers.len() != 0 {
|
||||
println!("stopping {} stray container(s)", containers.len())
|
||||
}
|
||||
|
||||
for container in containers {
|
||||
let container_name = &container.names.unwrap()[0][1..];
|
||||
println!("stopping {}", container_name);
|
||||
let stop_options = StopContainerOptionsBuilder::new()
|
||||
.t(60)
|
||||
.build();
|
||||
docker.stop_container(container_name, Some(stop_options)).await.map_err(|e| format!("Failed to stop container {}, : {}", container_name, e))?;
|
||||
println!("{} stopped", container_name);
|
||||
}
|
||||
println!("init done");
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn server_manager_loop(pool: DbPool, docker: Docker)
|
||||
{
|
||||
match server_init(&pool, &docker).await {
|
||||
Err(e) => { eprintln!("init failed , ignoring because idgaf : {}", e); },
|
||||
Ok(_) => {},
|
||||
};
|
||||
|
||||
loop {
|
||||
task::sleep(Duration::from_mins(5)).await;
|
||||
}
|
||||
}
|
||||
@ -5,3 +5,6 @@ mod client;
|
||||
mod socket;
|
||||
|
||||
pub use socket::mc_socket_listen;
|
||||
|
||||
mod manager_loop;
|
||||
pub use manager_loop::server_manager_loop;
|
||||
|
||||
@ -1,4 +1,4 @@
|
||||
use crate::minecraft::client::client::Client;
|
||||
use crate::minecraft::{client::client::Client, manager_loop};
|
||||
use bollard::Docker;
|
||||
use tokio::net::TcpStream;
|
||||
|
||||
@ -8,8 +8,20 @@ use tokio::net::TcpListener;
|
||||
|
||||
use std::io;
|
||||
|
||||
use crate::minecraft::server_manager_loop;
|
||||
|
||||
pub async fn mc_socket_listen(pool: DbPool, docker: Docker) -> io::Result<()> {
|
||||
let mc_listener = TcpListener::bind("0.0.0.0:25565").await?;
|
||||
|
||||
{
|
||||
let cloned_pool = pool.clone();
|
||||
let cloned_docker = docker.clone();
|
||||
|
||||
tokio::spawn(async move {
|
||||
server_manager_loop(cloned_pool, cloned_docker).await;
|
||||
});
|
||||
}
|
||||
|
||||
loop {
|
||||
if let Ok((socket, _)) = mc_listener.accept().await {
|
||||
let cloned_pool = pool.clone();
|
||||
|
||||
Reference in New Issue
Block a user