Add metrics logs observer

ballerina-platform · Nov 19, 2024 · f37ec1d · f37ec1d
1 parent 265054d
commit f37ec1d
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/ballerina/init.bal b/ballerina/init.bal
@@ -31,6 +31,13 @@ function init() {
             externPrintError("failed to enable tracing");
         }
     }
+
+    if (observe:isMetricsLogsEnabled()) {
+        var err = externEnableMetricsLogging(observe:getTracingProvider());
+        if (err is error) {
+            externPrintError("failed to enable tracing");
+        }
+    }
 }
 
 function externEnableMetrics(string provider) returns error? = @java:Method {
@@ -43,6 +50,11 @@ function externEnableTracing(string provider) returns error? = @java:Method {
     name: "enableTracing"
 } external;
 
+function externEnableMetricsLogging(string provider) returns error? = @java:Method {
+    'class: "io.ballerina.stdlib.observe.internal.NativeFunctions",
+    name: "enableMetricsLogging"
+} external;
+
 function externPrintError(string message) = @java:Method {
     'class: "io.ballerina.stdlib.observe.internal.NativeFunctions",
     name: "printError"

diff --git a/native/src/main/java/io/ballerina/stdlib/observe/internal/NativeFunctions.java b/native/src/main/java/io/ballerina/stdlib/observe/internal/NativeFunctions.java
@@ -17,9 +17,11 @@
  */
 package io.ballerina.stdlib.observe.internal;
 
+import io.ballerina.runtime.api.Environment;
 import io.ballerina.runtime.api.values.BError;
 import io.ballerina.runtime.api.values.BString;
 import io.ballerina.runtime.observability.ObserveUtils;
+import io.ballerina.runtime.observability.metrics.BallerinaMetricsLogsObserver;
 import io.ballerina.runtime.observability.metrics.BallerinaMetricsObserver;
 import io.ballerina.runtime.observability.metrics.DefaultMetricRegistry;
 import io.ballerina.runtime.observability.metrics.MetricRegistry;
@@ -94,6 +96,15 @@ public static BError enableTracing(BString providerName) {
         }
     }
 
+    public static BError enableMetricsLogging(Environment env, BString providerName) {
+        try {
+            ObserveUtils.addObserver(new BallerinaMetricsLogsObserver(env));
+            return null;
+        } catch (BError e) {
+            return e;
+        }
+    }
+
     public static void printError(BString message) {
         errStream.println("error: " + message.getValue());
     }