log_to_file

Author: dandavison

core/Cargo.toml

@@ -26,6 +26,7 @@ anyhow = "1.0"
 arc-swap = "1.3"
 async-trait = "0.1"
 base64 = "0.21"
+colored = "2"
 console-subscriber = { version = "0.1", optional = true }
 crossbeam = "0.8"
 dashmap = "5.5"

core/src/lib.rs

@@ -15,6 +15,7 @@ mod abstractions;
 #[cfg(feature = "ephemeral-server")]
 pub mod ephemeral_server;
 mod internal_flags;
+mod log_to_file;
 mod pollers;
 mod protosext;
 pub mod replay;

core/src/log_to_file.rs

@@ -0,0 +1,24 @@
+use colored::*;
+use std::fs::OpenOptions;
+use std::io::Write;
+
+pub fn log_to_file(msg: &str, prefix: &str, color: &str) {
+    let mut file = OpenOptions::new()
+        .create(true)
+        .write(true)
+        .append(true)
+        .open("/tmp/log")
+        .unwrap();
+
+    let msg = format!("{} {}", prefix, msg);
+    let msg = match color {
+        "red" => msg.red(),
+        "green" => msg.green(),
+        "blue" => msg.blue(),
+        _ => msg.black(),
+    };
+
+    if let Err(e) = writeln!(file, "{}", msg) {
+        eprintln!("Couldn't write to file: {}", e);
+    }
+}

core/src/worker/workflow/machines/update_state_machine.rs

@@ -1,5 +1,6 @@
 use super::{workflow_machines::MachineResponse, EventInfo, WFMachinesAdapter, WFMachinesError};
 use crate::{
+    log_to_file::log_to_file,
     protosext::protocol_messages::UpdateRequest,
     worker::workflow::machines::{Cancellable, HistEventData, NewMachineWithResponse},
 };
@@ -121,12 +122,27 @@ impl UpdateMachine {
                 )))
             }
             Some(update_response::Response::Accepted(_)) => {
+                log_to_file(
+                    "update_response::Accepted\n",
+                    "worker:sdk-core:fsm",
+                    "green",
+                );
                 self.on_event(UpdateMachineEvents::Accept)
             }
             Some(update_response::Response::Rejected(f)) => {
+                log_to_file(
+                    "update_response::Rejected\n",
+                    "worker:sdk-core:fsm",
+                    "green",
+                );
                 self.on_event(UpdateMachineEvents::Reject(f))
             }
             Some(update_response::Response::Completed(p)) => {
+                log_to_file(
+                    "update_response::Completed\n",
+                    "worker:sdk-core:fsm",
+                    "green",
+                );
                 self.on_event(UpdateMachineEvents::Complete(p))
             }
         }