cgroups-rs 0.5.0

Native Rust crate for managing control groups on Linux
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

// Copyright (c) 2020 And Group
//
// SPDX-License-Identifier: Apache-2.0 or MIT
//

//! Integration tests about the hugetlb subsystem
use cgroups_rs::fs::memory::{MemController, SetMemory};
use cgroups_rs::fs::{Cgroup, Controller, MaxValue};

#[test]
fn test_disable_oom_killer() {
    let h = cgroups_rs::fs::hierarchies::auto();
    let cg = Cgroup::new(h, String::from("test_disable_oom_killer")).unwrap();
    {
        let mem_controller: &MemController = cg.controller_of().unwrap();

        // before disable
        let m = mem_controller.memory_stat();
        assert!(!m.oom_control.oom_kill_disable);

        // now only v1
        if !mem_controller.v2() {
            // disable oom killer
            let r = mem_controller.disable_oom_killer();
            assert!(r.is_ok());

            // after disable
            let m = mem_controller.memory_stat();
            assert!(m.oom_control.oom_kill_disable);
        }
    }
    cg.delete().unwrap();
}

#[test]
fn set_kmem_limit_v1() {
    let h = cgroups_rs::fs::hierarchies::auto();
    if h.v2() {
        return;
    }

    let cg = Cgroup::new(h, String::from("set_kmem_limit_v1")).unwrap();
    {
        let mem_controller: &MemController = cg.controller_of().unwrap();
        mem_controller.set_kmem_limit(1).unwrap();
    }
    cg.delete().unwrap();
}

#[test]
fn set_mem_v2() {
    let h = cgroups_rs::fs::hierarchies::auto();
    if !h.v2() {
        return;
    }

    let cg = Cgroup::new(h, String::from("set_mem_v2")).unwrap();
    {
        let mem_controller: &MemController = cg.controller_of().unwrap();

        // before disable
        let m = mem_controller.get_mem().unwrap();
        // case 1: get default value
        assert_eq!(m.low, Some(MaxValue::Value(0)));
        assert_eq!(m.min, Some(MaxValue::Value(0)));
        assert_eq!(m.high, Some(MaxValue::Max));
        assert_eq!(m.max, Some(MaxValue::Max));

        // case 2: set parts
        let m = SetMemory {
            low: Some(MaxValue::Value(1024 * 1024 * 2)),
            high: Some(MaxValue::Value(1024 * 1024 * 1024 * 2)),
            min: Some(MaxValue::Value(1024 * 1024 * 3)),
            max: None,
        };
        let r = mem_controller.set_mem(m);
        assert!(r.is_ok());

        let m = mem_controller.get_mem().unwrap();
        // get
        assert_eq!(m.low, Some(MaxValue::Value(1024 * 1024 * 2)));
        assert_eq!(m.min, Some(MaxValue::Value(1024 * 1024 * 3)));
        assert_eq!(m.high, Some(MaxValue::Value(1024 * 1024 * 1024 * 2)));
        assert_eq!(m.max, Some(MaxValue::Max));

        // case 3: set parts
        let m = SetMemory {
            max: Some(MaxValue::Value(1024 * 1024 * 1024 * 2)),
            min: Some(MaxValue::Value(1024 * 1024 * 4)),
            high: Some(MaxValue::Max),
            low: None,
        };
        let r = mem_controller.set_mem(m);
        assert!(r.is_ok());

        let m = mem_controller.get_mem().unwrap();
        // get
        assert_eq!(m.low, Some(MaxValue::Value(1024 * 1024 * 2)));
        assert_eq!(m.min, Some(MaxValue::Value(1024 * 1024 * 4)));
        assert_eq!(m.max, Some(MaxValue::Value(1024 * 1024 * 1024 * 2)));
        assert_eq!(m.high, Some(MaxValue::Max));
    }

    cg.delete().unwrap();
}