feat: Write coordinator logs in GCP compatible format (#984)

near · Aug 7, 2024 · 7e7f6c0 · 7e7f6c0
1 parent 7e2b273
commit 7e7f6c0
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/coordinator/Cargo.lock b/coordinator/Cargo.lock
diff --git a/coordinator/Cargo.toml b/coordinator/Cargo.toml
@@ -12,6 +12,7 @@ tokio = "1.28"
 tonic = "0.10.2"
 tracing = "0.1.40"
 tracing-subscriber = { version = "0.3.18", features = ["env-filter"] }
+tracing-stackdriver = "0.10.0"
 serde = "1.0.195"
 serde_json = "1.0.108"
 

diff --git a/coordinator/src/main.rs b/coordinator/src/main.rs
@@ -33,10 +33,16 @@ async fn sleep(duration: Duration) -> anyhow::Result<()> {
 
 #[tokio::main]
 async fn main() -> anyhow::Result<()> {
-    tracing_subscriber::registry()
-        .with(tracing_subscriber::fmt::layer())
-        .with(tracing_subscriber::EnvFilter::from_default_env())
-        .init();
+    let subscriber =
+        tracing_subscriber::registry().with(tracing_subscriber::EnvFilter::from_default_env());
+
+    if std::env::var("GCP_LOGGING_ENABLED").is_ok() {
+        subscriber.with(tracing_stackdriver::layer()).init();
+    } else {
+        subscriber
+            .with(tracing_subscriber::fmt::layer().compact())
+            .init();
+    }
 
     let rpc_url = std::env::var("RPC_URL").expect("RPC_URL is not set");
     let registry_contract_id = std::env::var("REGISTRY_CONTRACT_ID")