From e5f36b5c78c75ed5f1800e064a012f02b59a03d3 Mon Sep 17 00:00:00 2001
From: Iliyan Velichkov <velichkov.iliyan@gmail.com>
Date: Thu, 21 Nov 2024 16:24:14 +0200
Subject: [PATCH] add traces for the synchronizers

Signed-off-by: Iliyan Velichkov <velichkov.iliyan@gmail.com>
---
 .../base/synchronizer/BaseSynchronizer.java   | 64 ++++++++++++++++++-
 .../delegate/DirigibleCallDelegate.java       |  1 -
 2 files changed, 63 insertions(+), 2 deletions(-)

diff --git a/components/core/core-base/src/main/java/org/eclipse/dirigible/components/base/synchronizer/BaseSynchronizer.java b/components/core/core-base/src/main/java/org/eclipse/dirigible/components/base/synchronizer/BaseSynchronizer.java
index 5a55c82cb1a..d1284639c3d 100644
--- a/components/core/core-base/src/main/java/org/eclipse/dirigible/components/base/synchronizer/BaseSynchronizer.java
+++ b/components/core/core-base/src/main/java/org/eclipse/dirigible/components/base/synchronizer/BaseSynchronizer.java
@@ -9,6 +9,9 @@
  */
 package org.eclipse.dirigible.components.base.synchronizer;
 
+import io.opentelemetry.api.trace.Span;
+import io.opentelemetry.api.trace.Tracer;
+import io.opentelemetry.context.Scope;
 import org.eclipse.dirigible.components.base.artefact.Artefact;
 import org.eclipse.dirigible.components.base.artefact.ArtefactLifecycle;
 import org.eclipse.dirigible.components.base.artefact.ArtefactPhase;
@@ -16,6 +19,7 @@
 import org.eclipse.dirigible.components.base.spring.BeanProvider;
 import org.eclipse.dirigible.components.base.tenant.TenantContext;
 import org.eclipse.dirigible.components.base.tenant.TenantResult;
+import org.eclipse.dirigible.components.open.telemetry.OpenTelemetryProvider;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -25,7 +29,6 @@
 
 /**
  * The Class BaseSynchronizer.
- *
  */
 public abstract class BaseSynchronizer<A extends Artefact, ID> implements Synchronizer<A, ID> {
 
@@ -41,6 +44,41 @@ public abstract class BaseSynchronizer<A extends Artefact, ID> implements Synchr
      */
     @Override
     public final boolean complete(TopologyWrapper<A> wrapper, ArtefactPhase flow) {
+        Tracer tracer = OpenTelemetryProvider.get()
+                                             .getTracer("eclipse-dirigible");
+
+        String synchronizerClassName = this.getClass()
+                                           .getSimpleName();
+        Span span = tracer.spanBuilder("synchronizer_" + synchronizerClassName + "_complete_execution")
+                          .startSpan();
+
+        try (Scope scope = span.makeCurrent()) {
+            addSpanAttributes(span, wrapper, flow);
+            return completeInternal(wrapper, flow);
+
+        } catch (RuntimeException e) {
+            span.recordException(e);
+            span.setStatus(io.opentelemetry.api.trace.StatusCode.ERROR, "Exception occurred during synchronization");
+
+            throw e;
+        } finally {
+            span.end();
+        }
+    }
+
+    private void addSpanAttributes(Span span, TopologyWrapper<A> wrapper, ArtefactPhase phase) {
+        addSpanAttributes(wrapper.getArtefact(), span);
+
+        span.setAttribute("phase", phase.getValue());
+    }
+
+    private void addSpanAttributes(A artefact, Span span) {
+        span.setAttribute("synchronizer", this.getClass()
+                                              .getName());
+        span.setAttribute("artefactKey", artefact.getKey());
+    }
+
+    private boolean completeInternal(TopologyWrapper<A> wrapper, ArtefactPhase flow) {
         A artefact = wrapper.getArtefact();
         ArtefactLifecycle lifecycle = artefact.getLifecycle();
 
@@ -98,6 +136,30 @@ protected boolean isMultitenantArtefact(A artefact) {
      * @param artefact the artefact
      */
     public final void cleanup(A artefact) {
+        Tracer tracer = OpenTelemetryProvider.get()
+                                             .getTracer("eclipse-dirigible");
+
+        String synchronizerClassName = this.getClass()
+                                           .getSimpleName();
+        Span span = tracer.spanBuilder("synchronizer_" + synchronizerClassName + "_cleanup_execution")
+                          .startSpan();
+
+        try (Scope scope = span.makeCurrent()) {
+            addSpanAttributes(artefact, span);
+
+            cleanupInternal(artefact);
+
+        } catch (RuntimeException e) {
+            span.recordException(e);
+            span.setStatus(io.opentelemetry.api.trace.StatusCode.ERROR, "Exception occurred during synchronization");
+
+            throw e;
+        } finally {
+            span.end();
+        }
+    }
+
+    private void cleanupInternal(A artefact) {
         if (!multitenantExecution() || !isMultitenantArtefact(artefact)) {
             logger.debug("[{} will cleanup artefact [{}]", this, artefact);
             cleanupImpl(artefact);
diff --git a/components/engine/engine-bpm-flowable/src/main/java/org/eclipse/dirigible/components/engine/bpm/flowable/delegate/DirigibleCallDelegate.java b/components/engine/engine-bpm-flowable/src/main/java/org/eclipse/dirigible/components/engine/bpm/flowable/delegate/DirigibleCallDelegate.java
index 4a1d034d9bf..2609d1099b8 100644
--- a/components/engine/engine-bpm-flowable/src/main/java/org/eclipse/dirigible/components/engine/bpm/flowable/delegate/DirigibleCallDelegate.java
+++ b/components/engine/engine-bpm-flowable/src/main/java/org/eclipse/dirigible/components/engine/bpm/flowable/delegate/DirigibleCallDelegate.java
@@ -225,7 +225,6 @@ public void execute(DelegateExecution execution) {
         } finally {
             span.end();
         }
-
     }
 
     private void addSpanAttributes(DelegateExecution execution, Span span) {