pmat 3.15.0

PMAT - Zero-config AI context generation and code quality toolkit (CLI, MCP, HTTP)
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

// Resource manager coordinating all resource controllers
/// Resource manager.
pub struct ResourceManager {
    limits: Arc<RwLock<ResourceLimits>>,
    cpu_controller: Arc<dyn ResourceController>,
    memory_controller: Arc<dyn ResourceController>,
    gpu_controller: Option<Arc<dyn ResourceController>>,
    network_controller: Arc<dyn ResourceController>,
    io_controller: Arc<dyn ResourceController>,
    usage_history: Arc<RwLock<Vec<ResourceUsage>>>,
}

impl ResourceManager {
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Create a new instance.
    pub fn new(limits: ResourceLimits) -> Result<Self, ResourceError> {
        use cpu_limiter::CpuLimiter;
        use io_throttle::IoThrottle;
        use memory_limiter::MemoryLimiter;
        use network_throttle::NetworkThrottle;

        let cpu_controller = Arc::new(CpuLimiter::new(limits.cpu.clone())?);
        let memory_controller = Arc::new(MemoryLimiter::new(limits.memory.clone())?);
        let network_controller = Arc::new(NetworkThrottle::new(limits.network.clone())?);
        let io_controller = Arc::new(IoThrottle::new(limits.disk_io.clone())?);

        let gpu_controller = if let Some(gpu_limits) = &limits.gpu {
            use gpu_scheduler::GpuScheduler;
            Some(Arc::new(GpuScheduler::new(gpu_limits.clone())?) as Arc<dyn ResourceController>)
        } else {
            None
        };

        Ok(Self {
            limits: Arc::new(RwLock::new(limits)),
            cpu_controller,
            memory_controller,
            gpu_controller,
            network_controller,
            io_controller,
            usage_history: Arc::new(RwLock::new(Vec::new())),
        })
    }

    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Update limits.
    pub fn update_limits(&self, new_limits: ResourceLimits) -> Result<(), ResourceError> {
        self.cpu_controller.apply_limits(&new_limits)?;
        self.memory_controller.apply_limits(&new_limits)?;
        self.network_controller.apply_limits(&new_limits)?;
        self.io_controller.apply_limits(&new_limits)?;

        if let Some(gpu) = &self.gpu_controller {
            gpu.apply_limits(&new_limits)?;
        }

        *self.limits.write() = new_limits;
        Ok(())
    }

    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Get current usage.
    pub fn get_current_usage(&self) -> Result<ResourceUsage, ResourceError> {
        let cpu_usage = self.cpu_controller.get_usage()?;
        let memory_usage = self.memory_controller.get_usage()?;
        let network_usage = self.network_controller.get_usage()?;
        let io_usage = self.io_controller.get_usage()?;

        let gpu_usage = if let Some(gpu) = &self.gpu_controller {
            gpu.get_usage().ok()
        } else {
            None
        };

        let usage = ResourceUsage {
            cpu_percent: cpu_usage.cpu_percent,
            memory_bytes: memory_usage.memory_bytes,
            gpu_memory_bytes: gpu_usage.as_ref().and_then(|u| u.gpu_memory_bytes),
            gpu_compute_percent: gpu_usage.and_then(|u| u.gpu_compute_percent),
            network_ingress_bytes: network_usage.network_ingress_bytes,
            network_egress_bytes: network_usage.network_egress_bytes,
            disk_read_bytes: io_usage.disk_read_bytes,
            disk_write_bytes: io_usage.disk_write_bytes,
            timestamp: std::time::SystemTime::now(),
        };

        // Store in history
        let mut history = self.usage_history.write();
        history.push(usage.clone());

        // Keep only last 1000 samples
        let history_len = history.len();
        if history_len > 1000 {
            history.drain(0..history_len - 1000);
        }

        Ok(usage)
    }

    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Get usage history.
    pub fn get_usage_history(&self, duration: Duration) -> Vec<ResourceUsage> {
        let history = self.usage_history.read();
        let cutoff = std::time::SystemTime::now() - duration;

        history
            .iter()
            .filter(|u| u.timestamp >= cutoff)
            .cloned()
            .collect()
    }

    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Release all.
    pub fn release_all(&self) -> Result<(), ResourceError> {
        self.cpu_controller.release()?;
        self.memory_controller.release()?;
        self.network_controller.release()?;
        self.io_controller.release()?;

        if let Some(gpu) = &self.gpu_controller {
            gpu.release()?;
        }

        Ok(())
    }
}