Add KVP Tracer and OpenTel

Cargo.toml

@@ -12,11 +12,19 @@ description = "A reference implementation for provisioning Linux VMs on Azure."
 exitcode = "1.1.2"
 anyhow = "1.0.81"
 tokio = { version = "1", features = ["full"] }
+opentelemetry_sdk = "0.21"
+opentelemetry = "0.21"
+tracing = "0.1"
+
 
 [dependencies.libazureinit]
 path = "libazureinit"
 version = "0.1.0"
 
+[dependencies.azurekvp]
+path = "azurekvp"
+version = "0.1.0"
+
 [profile.dev]
 incremental = true
 
@@ -31,4 +39,5 @@ path = "tests/functional_tests.rs"
 [workspace]
 members = [
     "libazureinit",
+    "azurekvp",
 ]

azurekvp/Cargo.toml

@@ -0,0 +1,26 @@
+[package]
+name = "azurekvp"
+version = "0.1.1"
+edition = "2021"
+repository = "https://github.com/Azure/azure-init/"
+homepage = "https://github.com/Azure/azure-init/"
+license = "MIT"
+description = "A binary library for implementing OpenTelemetry KVP for Linux VMs on Azure."
+
+
+[dependencies]
+once_cell = "1.12.0"
+opentelemetry = "0.18.0"
+opentelemetry_sdk = "0.22.0"
+opentelemetry-stdout = "0.2.0"
+tracing = "0.1"
+tracing-opentelemetry = "0.18.0"
+tracing-subscriber = "0.3.11"
+serde = { version = "1.0" }
+serde_json = "1.0.68"
+tokio = { version = "1", features = ["full"] }
+libc = "0.2"
+
+[lib]
+name = "azurekvp"
+path = "src/tracing.rs"

azurekvp/src/lib.rs

@@ -0,0 +1,4 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+pub mod main;

azurekvp/src/tracing.rs

@@ -0,0 +1,50 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT License.
+
+use opentelemetry::{ trace::{self, TracerProvider}, trace::Tracer as _};
+use opentelemetry::sdk::trace as sdktrace;
+use opentelemetry::sdk::export::trace::stdout;
+use opentelemetry::global;
+use tracing_opentelemetry::OpenTelemetryLayer;
+use once_cell::sync::Lazy;
+use std::fs::File;
+use std::os::unix::io::AsRawFd;
+use libc::dup2;
+use tracing_subscriber::{Registry, layer::SubscriberExt};
+
+
+pub static TRACER: Lazy<()> = Lazy::new(|| {   
+    // Redirect stdout to a file
+   let log_file_path = "spans.log";
+   let file = File::create(log_file_path).expect("Failed to create log file");
+   let stdout_fd = file.as_raw_fd();
+   unsafe {
+       dup2(stdout_fd, libc::STDOUT_FILENO);
+   }
+
+    // Set up the stdout exporter correctly
+    let exporter = stdout::Exporter::new(std::io::stdout(), true);
+
+    // Set up the TracerProvider with the stdout exporter
+    let provider = sdktrace::TracerProvider::builder()
+        .with_simple_exporter(exporter)
+        .build();
+
+   global::set_tracer_provider(provider.clone());
+
+   let tracer = provider.tracer("azure-kvp");
+   // Create the OpenTelemetry layer using the SDK tracer
+   let otel_layer = OpenTelemetryLayer::new(tracer);
+
+    // Create a `tracing` subscriber
+    let subscriber = Registry::default()
+        .with(otel_layer);
+
+    // Set the subscriber as the global default
+    tracing::subscriber::set_global_default(subscriber)
+        .expect("Setting default subscriber failed");
+});
+
+pub fn initialize_tracing() {
+    Lazy::force(&TRACER);
+}