mirror of
https://github.com/cloud-hypervisor/cloud-hypervisor.git
synced 2024-08-21 15:31:15 +00:00
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))
|
||
|
}
|
||
|
};
|
||
|
|
||
|
}
|