code-executor 1.1.0

A code runner library for online judge system
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

use std::time::Duration;

use byte_unit::Byte;
use cgroups_rs::fs::{cpu::CpuController, memory::MemController};

const CPU_USAGE_PREFIX: &str = "usage_usec ";

pub trait CpuControllerExt {
    fn usage(&self) -> Duration;
}

impl CpuControllerExt for CpuController {
    fn usage(&self) -> Duration {
        let stats = self.cpu().stat;

        let usage = stats
            .lines()
            .find_map(|line| line.strip_prefix(CPU_USAGE_PREFIX))
            .unwrap();
        let usage = usage.parse().unwrap();
        Duration::from_micros(usage)
    }
}

pub trait MemControllerExt {
    fn usage(&self) -> Byte;
    fn limit(&self) -> Byte;
}

impl MemControllerExt for MemController {
    fn usage(&self) -> Byte {
        let stats = self.memory_stat();

        Byte::from_u64(stats.usage_in_bytes)
    }

    fn limit(&self) -> Byte {
        let stats = self.memory_stat();

        Byte::from_u64(stats.limit_in_bytes.max(0) as u64)
    }
}