mirrors/krata
1
0
Fork 0
mirror of https://github.com/edera-dev/krata.git synced 2025-08-05 22:21:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: implement metrics support

Browse Source
This commit is contained in:
Alex Zenla
2024-04-10 12:58:21 +00:00
parent 183b4ddf73
commit 9b47cd648d
13 changed files with 341 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
crates/guest/src/background.rs
View File

@ -8,11 +8,12 @@ use krata::idm::{
client::IdmClient,
protocol::{
idm_event::Event, idm_request::Request, idm_response::Response, IdmEvent, IdmExitEvent,
IdmPingResponse, IdmRequest,
IdmMetricsResponse, IdmPingResponse, IdmRequest,
},
};
use log::debug;
use nix::unistd::Pid;
use sysinfo::System;
use tokio::{select, sync::broadcast};
pub struct GuestBackground {
@ -80,10 +81,26 @@ impl GuestBackground {
async fn handle_idm_request(&mut self, packet: IdmRequest) -> Result<()> {
let id = packet.id;
if let Some(Request::Ping(_)) = packet.request {
self.idm
.respond(id, Response::Ping(IdmPingResponse {}))
.await?;
match packet.request {
Some(Request::Ping(_)) => {
self.idm
.respond(id, Response::Ping(IdmPingResponse {}))
.await?;
}
Some(Request::Metrics(_)) => {
let mut sys = System::new();
sys.refresh_memory();
let response = IdmMetricsResponse {
total_memory_bytes: sys.total_memory(),
used_memory_bytes: sys.used_memory(),
};
self.idm.respond(id, Response::Metrics(response)).await?;
}
None => {}
}
Ok(())
}