feat: added daemon mode

src/bin/daemon.rs

@@ -24,14 +24,30 @@ async fn main() -> Result<()> {
 
     let config = Config::new()?;
 
-    let schedule = fabric::drivers::usage::schedule(config.clone().into());
-    let subscribe = fabric::drivers::monitor::subscribe(config.clone().into());
+    match config.mode {
+        Some(Mode::Usage) => {
+            fabric::drivers::usage::schedule(config.clone().into()).await?;
+        }
+        Some(Mode::Monitor) => {
+            fabric::drivers::monitor::subscribe(config.clone().into()).await?;
+        }
+        None => {
+            let schedule = fabric::drivers::usage::schedule(config.clone().into());
+            let subscribe = fabric::drivers::monitor::subscribe(config.clone().into());
 
-    try_join!(schedule, subscribe)?;
+            try_join!(schedule, subscribe)?;
+        }
+    };
 
     Ok(())
 }
 
+#[derive(Debug, Deserialize, Clone)]
+enum Mode {
+    Usage,
+    Monitor,
+}
+
 #[derive(Debug, Deserialize, Clone)]
 struct Prometheus {
     url: String,
@@ -47,6 +63,7 @@ struct Config {
     delay: Duration,
     topic: String,
     kafka: HashMap<String, String>,
+    mode: Option<Mode>,
 }
 impl Config {
     pub fn new() -> Result<Self> {

src/drivers/monitor/mod.rs

@@ -21,7 +21,7 @@ pub async fn subscribe(config: MonitorConfig) -> Result<()> {
     let consumer: StreamConsumer = client_config.create()?;
     consumer.subscribe(&[&config.topic])?;
 
-    info!("Subscriber running");
+    info!("Monitor subscribe running");
     loop {
         match consumer.recv().await {
             Err(error) => error!(?error, "kafka subscribe error"),

src/drivers/usage/mod.rs

@@ -16,6 +16,7 @@ pub async fn schedule(config: UsageConfig) -> Result<()> {
 
     let mut cursor = Utc::now();
 
+    info!("Usage schedule running");
     loop {
         sleep(config.delay).await;