From f065ad7bb9e4199e6570f883f18842256d0684e6 Mon Sep 17 00:00:00 2001
From: Gustav Karlsson <kagkarlsson@gmail.com>
Date: Wed, 3 Jul 2024 16:13:05 +0200
Subject: [PATCH] Poc OTEL listener

---
 db-scheduler/pom.xml                          | 12 ++++
 .../scheduler/event/OtelTracingListener.java  | 56 +++++++++++++++++++
 pom.xml                                       |  1 -
 3 files changed, 68 insertions(+), 1 deletion(-)
 create mode 100644 db-scheduler/src/main/java/com/github/kagkarlsson/scheduler/event/OtelTracingListener.java

diff --git a/db-scheduler/pom.xml b/db-scheduler/pom.xml
index b52bbb07..86009204 100644
--- a/db-scheduler/pom.xml
+++ b/db-scheduler/pom.xml
@@ -71,6 +71,13 @@
                 <scope>import</scope>
                 <type>pom</type>
             </dependency>
+          <dependency>
+            <groupId>io.opentelemetry</groupId>
+            <artifactId>opentelemetry-bom</artifactId>
+            <version>1.39.0</version>
+            <type>pom</type>
+            <scope>import</scope>
+          </dependency>
         </dependencies>
     </dependencyManagement>
 
@@ -128,6 +135,11 @@
             <artifactId>jackson-datatype-jsr310</artifactId>
             <optional>true</optional>
         </dependency>
+      <dependency>
+        <groupId>io.opentelemetry</groupId>
+        <artifactId>opentelemetry-api</artifactId>
+        <optional>true</optional>
+      </dependency>
 
 
         <!-- Test -->
diff --git a/db-scheduler/src/main/java/com/github/kagkarlsson/scheduler/event/OtelTracingListener.java b/db-scheduler/src/main/java/com/github/kagkarlsson/scheduler/event/OtelTracingListener.java
new file mode 100644
index 00000000..31e67b97
--- /dev/null
+++ b/db-scheduler/src/main/java/com/github/kagkarlsson/scheduler/event/OtelTracingListener.java
@@ -0,0 +1,56 @@
+/*
+ * Copyright (C) Gustav Karlsson
+ *
+ * <p>Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
+ * except in compliance with the License. You may obtain a copy of the License at
+ *
+ * <p>http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * <p>Unless required by applicable law or agreed to in writing, software distributed under the
+ * License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.github.kagkarlsson.scheduler.event;
+
+import com.github.kagkarlsson.scheduler.CurrentlyExecuting;
+import com.github.kagkarlsson.scheduler.task.ExecutionComplete;
+import com.github.kagkarlsson.scheduler.task.ExecutionComplete.Result;
+import com.github.kagkarlsson.scheduler.task.TaskInstance;
+import io.opentelemetry.api.OpenTelemetry;
+import io.opentelemetry.api.trace.Span;
+import io.opentelemetry.api.trace.Tracer;
+
+public class OtelTracingListener extends AbstractSchedulerListener {
+
+  private final OpenTelemetry openTelemetry;
+
+  public OtelTracingListener(OpenTelemetry openTelemetry) {
+    this.openTelemetry = openTelemetry;
+  }
+
+  @Override
+  public void onExecutionStart(CurrentlyExecuting currentlyExecuting) {
+    Tracer t = openTelemetry.getTracer("db-scheduler.execution");
+    TaskInstance<?> taskInstance = currentlyExecuting.getTaskInstance();
+    Span span =
+        t.spanBuilder("execute")
+            .setAttribute("task-name", taskInstance.getTaskName())
+            .setAttribute("task-instance-id", taskInstance.getId())
+            .startSpan();
+    span.makeCurrent();
+  }
+
+  @Override
+  public void onExecutionComplete(ExecutionComplete executionComplete) {
+    Span currentSpan = Span.current();
+
+    try {
+      if (executionComplete.getResult() == Result.FAILED) {
+        executionComplete.getCause().ifPresent(currentSpan::recordException);
+      }
+    } finally {
+      currentSpan.end();
+    }
+  }
+}
diff --git a/pom.xml b/pom.xml
index a85b0fc2..191e30fe 100644
--- a/pom.xml
+++ b/pom.xml
@@ -94,7 +94,6 @@
 				<version>3.12.0</version>
 				<scope>test</scope>
 			</dependency>
-
 		</dependencies>
 	</dependencyManagement>