mirror of
https://github.com/cloud-hypervisor/cloud-hypervisor.git
synced 2024-09-12 01:54:55 +00:00
This crate exposes the abililty for the VMM to set a file that events should be written to. The event!() macro provides an interface to report those events allowing the specification of an event source, an event type and optional extra data. This will be written to the provided file descriptor as JSON data. Signed-off-by: Rob Bradford <robert.bradford@intel.com>
78 lines
2.0 KiB
Rust
78 lines
2.0 KiB
Rust
// Copyright © 2021 Intel Corporation
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#[macro_use]
|
|
extern crate serde_derive;
|
|
|
|
use std::borrow::Cow;
|
|
use std::collections::HashMap;
|
|
use std::fs::File;
|
|
use std::os::unix::io::AsRawFd;
|
|
use std::time::{Duration, Instant};
|
|
|
|
static mut MONITOR: Option<(File, Instant)> = None;
|
|
|
|
/// This function must only be called once from the main process before any threads
|
|
/// are created to avoid race conditions
|
|
pub fn set_monitor(file: File) -> Result<(), std::io::Error> {
|
|
assert!(unsafe { MONITOR.is_none() });
|
|
let fd = file.as_raw_fd();
|
|
let ret = unsafe {
|
|
let mut flags = libc::fcntl(fd, libc::F_GETFL);
|
|
flags |= libc::O_NONBLOCK;
|
|
libc::fcntl(fd, libc::F_SETFL, flags)
|
|
};
|
|
if ret < 0 {
|
|
return Err(std::io::Error::last_os_error());
|
|
}
|
|
unsafe {
|
|
MONITOR = Some((file, Instant::now()));
|
|
};
|
|
Ok(())
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
struct Event<'a> {
|
|
timestamp: Duration,
|
|
source: &'a str,
|
|
event: &'a str,
|
|
properties: Option<&'a HashMap<Cow<'a, str>, Cow<'a, str>>>,
|
|
}
|
|
|
|
pub fn event_log(source: &str, event: &str, properties: Option<&HashMap<Cow<str>, Cow<str>>>) {
|
|
if let Some((file, start)) = unsafe { MONITOR.as_ref() } {
|
|
let e = Event {
|
|
timestamp: start.elapsed(),
|
|
source,
|
|
event,
|
|
properties,
|
|
};
|
|
serde_json::to_writer_pretty(file, &e).ok();
|
|
}
|
|
}
|
|
|
|
/*
|
|
Through the use of Cow<'a, str> it is possible to use String as well as
|
|
&str as the parameters:
|
|
e.g.
|
|
event!("cpu_manager", "create_vcpu", "id", cpu_id.to_string());
|
|
*/
|
|
#[macro_export]
|
|
macro_rules! event {
|
|
($source:expr, $event:expr) => {
|
|
$crate::event_log($source, $event, None)
|
|
};
|
|
($source:expr, $event:expr, $($key:expr, $value:expr),*) => {
|
|
{
|
|
let mut properties = ::std::collections::HashMap::new();
|
|
$(
|
|
properties.insert($key.into(), $value.into());
|
|
)+
|
|
$crate::event_log($source, $event, Some(&properties))
|
|
}
|
|
};
|
|
|
|
}
|