xerbalind/planetwars.dev
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages 1 Wiki Activity

add resource limits to bots

Browse source
This commit is contained in:
Ilion Beyst 2022-05-21 16:44:58 +02:00
parent c873f3a1cb
commit b1151f6ac7
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
planetwars-matchrunner/src/docker_runner.rs
View file

@ -45,10 +45,19 @@ async fn spawn_docker_process(
let bot_code_dir = std::fs::canonicalize(&params.code_path).unwrap(); let bot_code_dir = std::fs::canonicalize(&params.code_path).unwrap();
let code_dir_str = bot_code_dir.as_os_str().to_str().unwrap(); let code_dir_str = bot_code_dir.as_os_str().to_str().unwrap();
let memory_limit = 512 * 1024 * 1024; // 512MB
let config = container::Config { let config = container::Config {
image: Some(params.image.clone()), image: Some(params.image.clone()),
host_config: Some(bollard::models::HostConfig { host_config: Some(bollard::models::HostConfig {
binds: Some(vec![format!("{}:{}", code_dir_str, "/workdir")]), binds: Some(vec![format!("{}:{}", code_dir_str, "/workdir")]),
network_mode: Some("none".to_string()),
memory: Some(memory_limit),
memory_swap: Some(memory_limit),
// TODO: this applies a limit to how much cpu one bot can use.
// when running multiple bots concurrently though, the server
// could still become resource-starved.
cpu_period: Some(100_000),
cpu_quota: Some(10_000),
..Default::default() ..Default::default()
}), }),
working_dir: Some("/workdir".to_string()), working_dir: Some("/workdir".to_string()),
@ -57,6 +66,7 @@ async fn spawn_docker_process(
attach_stdout: Some(true), attach_stdout: Some(true),
attach_stderr: Some(true), attach_stderr: Some(true),
open_stdin: Some(true), open_stdin: Some(true),
network_disabled: Some(true),
..Default::default() ..Default::default()
}; };