mirror of
https://github.com/cloud-hypervisor/cloud-hypervisor.git
synced 2024-12-28 16:45:18 +00:00
3a0429c998
There is no need to include serde_derive separately, as it can be specified as serde feature instead. Signed-off-by: Maksym Pavlenko <pavlenko.maksym@gmail.com>
80 lines
2.1 KiB
Rust
80 lines
2.1 KiB
Rust
// Copyright © 2021 Intel Corporation
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
use serde::Serialize;
|
|
use std::borrow::Cow;
|
|
use std::collections::HashMap;
|
|
use std::fs::File;
|
|
use std::io::Write;
|
|
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();
|
|
|
|
let mut file = file;
|
|
file.write_all(b"\n\n").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))
|
|
}
|
|
};
|
|
|
|
}
|