From 8b4d3e90dce19a074e1e64f415a210505d2a50e2 Mon Sep 17 00:00:00 2001 From: xy Date: Fri, 12 Jul 2024 18:03:20 +0900 Subject: [PATCH] Use tracing instead of log --- Cargo.lock | 198 +++++++++++++++---------------------------- Cargo.toml | 4 +- src/bin/bootstrap.rs | 10 +-- src/discord/mod.rs | 7 +- 4 files changed, 80 insertions(+), 139 deletions(-) diff --git a/Cargo.lock b/Cargo.lock index 95dba75..d645499 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -17,15 +17,6 @@ version = "1.0.2" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "f26201604c87b1e01bd3d98f8d5d9a8fcbb815e8cedb41ffccbeb4bf593a35fe" -[[package]] -name = "aho-corasick" -version = "1.1.2" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "b2969dcb958b36655471fc61f7e416fa76033bdd4bfed0678d8fee1e2d07a1f0" -dependencies = [ - "memchr", -] - [[package]] name = "android-tzdata" version = "0.1.1" @@ -41,54 +32,6 @@ dependencies = [ "libc", ] -[[package]] -name = "anstream" -version = "0.6.11" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "6e2e1ebcb11de5c03c67de28a7df593d32191b44939c482e97702baaaa6ab6a5" -dependencies = [ - "anstyle", - "anstyle-parse", - "anstyle-query", - "anstyle-wincon", - "colorchoice", - "utf8parse", -] - -[[package]] -name = "anstyle" -version = "1.0.6" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "8901269c6307e8d93993578286ac0edf7f195079ffff5ebdeea6a59ffb7e36bc" - -[[package]] -name = "anstyle-parse" -version = "0.2.3" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "c75ac65da39e5fe5ab759307499ddad880d724eed2f6ce5b5e8a26f4f387928c" -dependencies = [ - "utf8parse", -] - -[[package]] -name = "anstyle-query" -version = "1.0.2" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "e28923312444cdd728e4738b3f9c9cac739500909bb3d3c94b43551b16517648" -dependencies = [ - "windows-sys 0.52.0", -] - -[[package]] -name = "anstyle-wincon" -version = "3.0.2" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "1cd54b81ec8d6180e24654d0b371ad22fc3dd083b6ff8ba325b72e00c87660a7" -dependencies = [ - "anstyle", - "windows-sys 0.52.0", -] - [[package]] name = "arrayvec" version = "0.7.4" @@ -261,12 +204,6 @@ dependencies = [ "windows-targets 0.52.0", ] -[[package]] -name = "colorchoice" -version = "1.0.0" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "acbf1af155f9b9ef647e42cdc158db4b64a1b61f743629225fde6f3e0be2a7c7" - [[package]] name = "command_attr" version = "0.5.1" @@ -522,29 +459,6 @@ dependencies = [ "cfg-if", ] -[[package]] -name = "env_filter" -version = "0.1.0" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "a009aa4810eb158359dda09d0c87378e4bbb89b5a801f016885a4707ba24f7ea" -dependencies = [ - "log", - "regex", -] - -[[package]] -name = "env_logger" -version = "0.11.2" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "6c012a26a7f605efc424dd53697843a72be7dc86ad2d01f7814337794a12231d" -dependencies = [ - "anstream", - "anstyle", - "env_filter", - "humantime", - "log", -] - [[package]] name = "equivalent" version = "1.0.1" @@ -766,13 +680,13 @@ dependencies = [ "chatgpt_rs", "config", "dotenv", - "env_logger", "futures", - "log", "reqwest 0.12.2", "serenity", "thiserror", "tokio", + "tracing", + "tracing-subscriber", "url", ] @@ -900,12 +814,6 @@ version = "1.0.3" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "df3b46402a9d5adb4c86a0cf463f42e19994e3ee891101b1841f30a545cb49a9" -[[package]] -name = "humantime" -version = "2.1.0" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "9a3a5bfb195931eeb336b2a7b4d761daec841b97f947d34394601737a7bba5e4" - [[package]] name = "hyper" version = "0.14.28" @@ -1218,6 +1126,16 @@ dependencies = [ "minimal-lexical", ] +[[package]] +name = "nu-ansi-term" +version = "0.46.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "77a8165726e8236064dbb45459242600304b42a5ea24ee2948e18e023bf7ba84" +dependencies = [ + "overload", + "winapi", +] + [[package]] name = "num-conv" version = "0.1.0" @@ -1312,6 +1230,12 @@ dependencies = [ "hashbrown 0.13.2", ] +[[package]] +name = "overload" +version = "0.1.1" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "b15813163c1d831bf4a13c3610c05c0d03b39feb07f7e09fa234dac9b15aaf39" + [[package]] name = "parking_lot" version = "0.12.1" @@ -1510,35 +1434,6 @@ dependencies = [ "bitflags 1.3.2", ] -[[package]] -name = "regex" -version = "1.10.3" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "b62dbe01f0b06f9d8dc7d49e05a0785f153b00b2c227856282f671e0318c9b15" -dependencies = [ - "aho-corasick", - "memchr", - "regex-automata", - "regex-syntax", -] - -[[package]] -name = "regex-automata" -version = "0.4.5" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "5bb987efffd3c6d0d8f5f89510bb458559eab11e4f869acb20bf845e016259cd" -dependencies = [ - "aho-corasick", - "memchr", - "regex-syntax", -] - -[[package]] -name = "regex-syntax" -version = "0.8.2" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "c08c74e62047bb2de4ff487b251e4a92e24f48745648451635cec7d591162d9f" - [[package]] name = "reqwest" version = "0.11.24" @@ -1903,6 +1798,15 @@ dependencies = [ "digest", ] +[[package]] +name = "sharded-slab" +version = "0.1.7" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "f40ca3c46823713e0d4209592e8d6e826aa57e928f09752619fc696c499637f6" +dependencies = [ + "lazy_static", +] + [[package]] name = "signal-hook-registry" version = "1.4.1" @@ -2057,6 +1961,16 @@ dependencies = [ "syn 2.0.48", ] +[[package]] +name = "thread_local" +version = "1.1.8" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "8b9ef9bad013ada3808854ceac7b46812a6465ba368859a37e2100283d2d719c" +dependencies = [ + "cfg-if", + "once_cell", +] + [[package]] name = "time" version = "0.3.34" @@ -2283,6 +2197,32 @@ source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "c06d3da6113f116aaee68e4d601191614c9053067f9ab7f6edbcb161237daa54" dependencies = [ "once_cell", + "valuable", +] + +[[package]] +name = "tracing-log" +version = "0.2.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "ee855f1f400bd0e5c02d150ae5de3840039a3f54b025156404e34c23c03f47c3" +dependencies = [ + "log", + "once_cell", + "tracing-core", +] + +[[package]] +name = "tracing-subscriber" +version = "0.3.18" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "ad0f048c97dbd9faa9b7df56362b8ebcaa52adb06b498c050d2f4e32f90a7a8b" +dependencies = [ + "nu-ansi-term", + "sharded-slab", + "smallvec", + "thread_local", + "tracing-core", + "tracing-log", ] [[package]] @@ -2424,18 +2364,18 @@ version = "0.7.6" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "09cc8ee72d2a9becf2f2febe0205bbed8fc6615b7cb429ad062dc7b7ddd036a9" -[[package]] -name = "utf8parse" -version = "0.2.1" -source = "registry+https://github.com/rust-lang/crates.io-index" -checksum = "711b9620af191e0cdc7468a8d14e709c3dcdb115b36f838e601583af800a370a" - [[package]] name = "uwl" version = "0.6.0" source = "registry+https://github.com/rust-lang/crates.io-index" checksum = "f4bf03e0ca70d626ecc4ba6b0763b934b6f2976e8c744088bb3c1d646fbb1ad0" +[[package]] +name = "valuable" +version = "0.1.0" +source = "registry+https://github.com/rust-lang/crates.io-index" +checksum = "830b7e5d4d90034032940e4ace0d9a9a057e7a45cd94e6c007832e39edb82f6d" + [[package]] name = "vcpkg" version = "0.2.15" diff --git a/Cargo.toml b/Cargo.toml index da97f3d..f0e76a6 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -13,10 +13,10 @@ tokio = { version = "1", features = ["full"] } serenity = "0.12.0" async-trait = "0.1.77" futures = "0.3.30" -log = "0.4.20" dotenv = "0.15" -env_logger = "0.11.2" config = "0.14.0" url = "2.5.0" reqwest = "0.12.2" thiserror = "1.0.0" +tracing = "0.1.37" +tracing-subscriber = "0.3.16" diff --git a/src/bin/bootstrap.rs b/src/bin/bootstrap.rs index a2c56ac..867ed0e 100644 --- a/src/bin/bootstrap.rs +++ b/src/bin/bootstrap.rs @@ -3,12 +3,12 @@ use serenity::all::GatewayIntents; use serenity::Client; use std::env; use std::path::Path; +use tracing::{error, info}; #[tokio::main] async fn main() { - // Initialize the logger - - env_logger::init(); + // Initialize the logger with tracing + tracing_subscriber::fmt::init(); // Load environment variables load_environment_variables(); @@ -37,9 +37,9 @@ async fn main() { .await; // Start listening for events - println!("Bot is now running. Press Ctrl+C to stop."); + info!("Bot is now running. Press Ctrl+C to stop."); if let Err(why) = client.start().await { - log::error!("Client error: {:?}", why); + error!("Client error: {:?}", why); } } diff --git a/src/discord/mod.rs b/src/discord/mod.rs index 914a1e4..4a4ed39 100644 --- a/src/discord/mod.rs +++ b/src/discord/mod.rs @@ -9,6 +9,7 @@ use serenity::model::channel::Message; use std::time::Duration; use thiserror::Error; use tokio::time::interval; +use tracing::error; #[derive(Debug, Error)] pub enum Error { @@ -52,7 +53,7 @@ impl EventHandler for Handler { let mut stream = match self.gpt_client.send_message_streaming(prompt).await { Ok(stream) => stream, Err(e) => { - log::error!("Error sending message to ChatGPT: {:?}", e); + error!("Error sending message to ChatGPT: {:?}", e); return; } }; @@ -60,7 +61,7 @@ impl EventHandler for Handler { let processing_message = match msg.channel_id.say(&ctx.http, "Processing...").await { Ok(message) => message, Err(why) => { - log::error!("Error sending message: {:?}", why); + error!("Error sending message: {:?}", why); return; } }; @@ -84,7 +85,7 @@ impl EventHandler for Handler { if !result.is_empty() { let edit = EditMessage::default().content(&result); if let Err(why) = msg.channel_id.edit_message(&ctx.http, processing_message.id, edit).await { - log::error!("Error editing message: {:?}", why); + error!("Error editing message: {:?}", why); } } }