chore(memtrack): add prelude

Author: not-matthias

crates/memtrack/src/ebpf/memtrack.rs

@@ -1,11 +1,8 @@
-use anyhow::Context;
-use anyhow::Result;
-use anyhow::bail;
+use crate::prelude::*;
 use libbpf_rs::Link;
 use libbpf_rs::skel::OpenSkel;
 use libbpf_rs::skel::SkelBuilder;
 use libbpf_rs::{MapCore, UprobeOpts};
-use log::{debug, warn};
 use paste::paste;
 use std::mem::MaybeUninit;
 use std::path::Path;

crates/memtrack/src/ebpf/tracker.rs

@@ -1,9 +1,8 @@
+use crate::prelude::*;
 use crate::{
     AllocatorLib,
     ebpf::{Event, MemtrackBpf},
 };
-use anyhow::Result;
-use log::debug;
 use std::sync::mpsc::{self, Receiver};
 
 pub struct Tracker {

crates/memtrack/src/ipc.rs

@@ -1,4 +1,4 @@
-use anyhow::{Context, Result};
+use crate::prelude::*;
 use ipc_channel::ipc::{self, IpcOneShotServer, IpcSender};
 use serde::{Deserialize, Serialize};
 
@@ -86,8 +86,6 @@ impl MemtrackIpcClient {
 /// Handle incoming IPC messages in memtrack
 #[cfg(feature = "ebpf")]
 pub fn handle_ipc_message(msg: IpcMessage, tracker: &std::sync::Arc<std::sync::Mutex<Tracker>>) {
-    use log::debug;
-
     let response = match msg.command {
         IpcCommand::Enable => match tracker.lock() {
             Ok(mut t) => match t.enable() {

crates/memtrack/src/lib.rs

@@ -2,6 +2,7 @@ mod allocators;
 #[cfg(feature = "ebpf")]
 mod ebpf;
 mod ipc;
+pub mod prelude;
 
 pub use allocators::{AllocatorKind, AllocatorLib};
 pub use ipc::{

crates/memtrack/src/main.rs

@@ -1,7 +1,6 @@
-use anyhow::{Context, Result, anyhow};
 use clap::Parser;
-use ipc_channel::ipc::{self};
-use log::{debug, info};
+use ipc_channel::ipc;
+use memtrack::prelude::*;
 use memtrack::{MemtrackIpcMessage, Tracker, handle_ipc_message};
 use runner_shared::artifacts::{ArtifactExt, MemtrackArtifact, MemtrackEvent, MemtrackWriter};
 use std::path::{Path, PathBuf};

crates/memtrack/src/prelude.rs

@@ -0,0 +1,5 @@
+#[allow(unused_imports)]
+pub use anyhow::{Context, Error, Result, anyhow, bail, ensure};
+pub use itertools::Itertools;
+#[allow(unused_imports)]
+pub use log::{debug, error, info, trace, warn};