Skip to content

Commit 9513b3a

Browse files
author
Jonathan Woollett-Light
committed
fix: tracing
Replaces custom logger with `tracing_subscriber`. Signed-off-by: Jonathan Woollett-Light <jcawl@amazon.co.uk>
1 parent fd72d1a commit 9513b3a

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

65 files changed

+538
-1183
lines changed

Cargo.lock

Lines changed: 129 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

src/api_server/Cargo.toml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,8 @@ mmds = { path = "../mmds" }
2121
seccompiler = { path = "../seccompiler" }
2222
utils = { path = "../utils" }
2323
vmm = { path = "../vmm" }
24+
tracing = { version = "0.1.37", features = ["attributes"] }
2425

2526
[dev-dependencies]
2627
libc = "0.2.117"
28+
log = "0.4.19"

src/api_server/src/lib.rs

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,15 +14,14 @@ use std::fmt::Debug;
1414
use std::path::PathBuf;
1515
use std::sync::mpsc;
1616

17-
use logger::{
18-
debug, error, info, update_metric_with_elapsed_time, warn, ProcessTimeReporter, METRICS,
19-
};
17+
use logger::{update_metric_with_elapsed_time, ProcessTimeReporter, METRICS};
2018
pub use micro_http::{
2119
Body, HttpServer, Method, Request, RequestError, Response, ServerError, ServerRequest,
2220
ServerResponse, StatusCode, Version,
2321
};
2422
use seccompiler::BpfProgramRef;
2523
use serde_json::json;
24+
use tracing::{debug, error, info, warn};
2625
use utils::eventfd::EventFd;
2726
use vmm::rpc_interface::{ApiRequest, ApiResponse, VmmAction, VmmData};
2827
use vmm::vmm_config::snapshot::SnapshotType;

src/api_server/src/parsed_request.rs

Lines changed: 2 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@
33

44
use std::fmt::Debug;
55

6-
use logger::{error, info, log_enabled, Level};
76
use micro_http::{Body, Method, Request, Response, StatusCode, Version};
87
use serde::ser::Serialize;
98
use serde_json::Value;
9+
use tracing::{error, info};
1010
use vmm::rpc_interface::{VmmAction, VmmActionError};
1111

1212
use super::VmmData;
@@ -229,19 +229,7 @@ fn log_received_api_request(api_description: String) {
229229
fn describe(method: Method, path: &str, body: Option<&Body>) -> String {
230230
match (path, body) {
231231
("/mmds", Some(_)) | (_, None) => format!("{:?} request on {:?}", method, path),
232-
("/cpu-config", Some(payload_value)) => {
233-
// If the log level is at Debug or higher, include the CPU template in
234-
// the log line.
235-
if log_enabled!(Level::Debug) {
236-
describe_with_body(method, path, payload_value)
237-
} else {
238-
format!(
239-
"{:?} request on {:?}. To view the CPU template received by the API, \
240-
configure log-level to DEBUG",
241-
method, path
242-
)
243-
}
244-
}
232+
("/cpu-config", Some(payload_value)) => describe_with_body(method, path, payload_value),
245233
(_, Some(payload_value)) => describe_with_body(method, path, payload_value),
246234
}
247235
}

src/api_server/src/request/logger.rs

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
// SPDX-License-Identifier: Apache-2.0
33

44
use logger::{IncMetric, METRICS};
5-
use vmm::vmm_config::logger::LoggerConfig;
5+
use vmm::vmm_config::LoggerConfig;
66

77
use super::super::VmmAction;
88
use crate::parsed_request::{Error, ParsedRequest};
@@ -22,7 +22,7 @@ pub(crate) fn parse_put_logger(body: &Body) -> Result<ParsedRequest, Error> {
2222
mod tests {
2323
use std::path::PathBuf;
2424

25-
use vmm::vmm_config::logger::LoggerLevel;
25+
use vmm::vmm_config::Level;
2626

2727
use super::*;
2828
use crate::parsed_request::tests::vmm_action_from_request;
@@ -38,9 +38,10 @@ mod tests {
3838

3939
let mut expected_cfg = LoggerConfig {
4040
log_path: PathBuf::from("log"),
41-
level: LoggerLevel::Warning,
42-
show_level: false,
43-
show_log_origin: false,
41+
level: Some(Level::Warn),
42+
show_level: Some(false),
43+
show_log_origin: Some(false),
44+
new_format: None,
4445
};
4546
match vmm_action_from_request(parse_put_logger(&Body::new(body)).unwrap()) {
4647
VmmAction::ConfigureLogger(cfg) => assert_eq!(cfg, expected_cfg),
@@ -56,9 +57,10 @@ mod tests {
5657

5758
expected_cfg = LoggerConfig {
5859
log_path: PathBuf::from("log"),
59-
level: LoggerLevel::Debug,
60-
show_level: false,
61-
show_log_origin: false,
60+
level: Some(Level::Debug),
61+
show_level: Some(false),
62+
show_log_origin: Some(false),
63+
new_format: None,
6264
};
6365
match vmm_action_from_request(parse_put_logger(&Body::new(body)).unwrap()) {
6466
VmmAction::ConfigureLogger(cfg) => assert_eq!(cfg, expected_cfg),

0 commit comments

Comments
 (0)