From 3a15a9b72293dd2aee941669dd769b1d3ce39964 Mon Sep 17 00:00:00 2001
From: Giacomo Pasini <Zeegomo@users.noreply.github.com>
Date: Mon, 21 Nov 2022 15:35:52 +0100
Subject: [PATCH] Add log service (#4)

* add log service

* add ser/de to log config

* add futures dep
---
 nomos-services/Cargo.toml     |   4 ++
 nomos-services/src/lib.rs     |   1 +
 nomos-services/src/log/mod.rs | 119 ++++++++++++++++++++++++++++++++++
 3 files changed, 124 insertions(+)
 create mode 100644 nomos-services/src/log/mod.rs

diff --git a/nomos-services/Cargo.toml b/nomos-services/Cargo.toml
index c3f0d82e..5eaf0fc2 100644
--- a/nomos-services/Cargo.toml
+++ b/nomos-services/Cargo.toml
@@ -16,6 +16,10 @@ tokio = { version = "1", features = ["sync"] }
 thiserror = "1.0"
 tracing = "0.1"
 waku = { git = "https://github.com/waku-org/waku-rust-bindings" }
+tracing-appender = "0.2"
+tracing-subscriber =  { version = "0.3", features = ["json"] }
+tracing-gelf = "0.7"
+futures = "0.3"
 
 [dev-dependencies]
 tempfile = "3.3"
diff --git a/nomos-services/src/lib.rs b/nomos-services/src/lib.rs
index f00a4874..90b94acc 100644
--- a/nomos-services/src/lib.rs
+++ b/nomos-services/src/lib.rs
@@ -1,2 +1,3 @@
+pub mod log;
 pub mod network;
 pub mod storage;
\ No newline at end of file
diff --git a/nomos-services/src/log/mod.rs b/nomos-services/src/log/mod.rs
new file mode 100644
index 00000000..be7d4d8e
--- /dev/null
+++ b/nomos-services/src/log/mod.rs
@@ -0,0 +1,119 @@
+use overwatch::services::{
+    handle::ServiceStateHandle,
+    relay::NoMessage,
+    state::{NoOperator, NoState},
+    ServiceCore, ServiceData,
+};
+use serde::{Serialize, Deserialize};
+use std::net::SocketAddr;
+use std::path::PathBuf;
+use tracing::Level;
+use tracing_appender::non_blocking::WorkerGuard;
+use tracing_subscriber::{filter::LevelFilter, prelude::*};
+
+pub struct Logger(Option<WorkerGuard>);
+
+#[derive(Clone, Debug, Serialize, Deserialize)]
+pub enum LoggerBackend {
+    Gelf {
+        addr: SocketAddr,
+    },
+    File {
+        directory: PathBuf,
+        prefix: Option<PathBuf>,
+    },
+    Stdout,
+    Stderr,
+}
+
+#[derive(Clone, Debug, Serialize, Deserialize)]
+pub struct LoggerSettings {
+    backend: LoggerBackend,
+    format: LoggerFormat,
+    #[serde(with = "serde_level")]
+    level: Level,
+}
+
+#[derive(Clone, Debug, Serialize, Deserialize)]
+pub enum LoggerFormat {
+    Json,
+    Plain,
+}
+
+impl ServiceData for Logger {
+    const SERVICE_ID: &'static str = "Logger";
+    type State = NoState<Self::Settings>;
+    type StateOperator = NoOperator<Self::State>;
+    type Message = NoMessage;
+    type Settings = LoggerSettings;
+}
+
+// a macro and not a function because it's a bit of a type
+// mess with `Layer<S>`
+macro_rules! registry_init {
+    ($layer:expr, $format:expr, $level:expr) => {
+        if let LoggerFormat::Json = $format {
+            tracing_subscriber::registry()
+                .with(LevelFilter::from($level))
+                .with($layer)
+                .init();
+        } else {
+            tracing_subscriber::registry()
+                .with(LevelFilter::from($level))
+                .with($layer)
+                .init();
+        }
+    };
+}
+
+#[async_trait::async_trait]
+impl ServiceCore for Logger {
+    fn init(mut service_state: ServiceStateHandle<Self>) -> Self {
+        let config = service_state.settings_reader.get_updated_settings();
+        let (non_blocking, _guard) = match config.backend {
+            LoggerBackend::Gelf { addr } => {
+                let (layer, mut task) = tracing_gelf::Logger::builder().connect_tcp(addr).unwrap();
+                service_state
+                    .overwatch_handle
+                    .runtime()
+                    .spawn(async move { task.connect().await });
+                registry_init!(layer, config.format, config.level);
+                return Self(None);
+            }
+            LoggerBackend::File { directory, prefix } => {
+                let file_appender = tracing_appender::rolling::hourly(
+                    directory,
+                    prefix.unwrap_or_else(|| PathBuf::from("nomos.log")),
+                );
+                tracing_appender::non_blocking(file_appender)
+            }
+            LoggerBackend::Stdout => tracing_appender::non_blocking(std::io::stdout()),
+            LoggerBackend::Stderr => tracing_appender::non_blocking(std::io::stderr()),
+        };
+
+        let layer = tracing_subscriber::fmt::Layer::new()
+            .with_level(true)
+            .with_writer(non_blocking);
+        registry_init!(layer, config.format, config.level);
+        Self(Some(_guard))
+    }
+
+    async fn run(self) {
+        // keep the handle alive without stressing the runtime
+        futures::pending!()
+    }
+}
+
+
+mod serde_level {
+    use super::Level;
+    use serde::{Serializer, Serialize, Deserialize, Deserializer, de::Error};
+
+    pub fn deserialize<'de, D>(deserializer: D) -> Result<Level, D::Error> where D: Deserializer<'de> {
+        <String>::deserialize(deserializer).and_then(|v| v.parse().map_err(|e| D::Error::custom(format!("invalid log level {}", e))))
+    }
+
+    pub fn serialize<S>(value: &Level, serializer: S) -> Result<S::Ok, S::Error> where S: Serializer {
+        value.as_str().serialize(serializer)
+    }
+}
\ No newline at end of file