[GIT]

Author: subhrashisdas

org.eclipse.jgit.telemetry/src/main/java/org/eclipse/jgit/telemetry/JGitTelemetryAspect.java

@@ -0,0 +1,65 @@
+package com.appsmith.server.aspect;
+
+import io.opentelemetry.api.GlobalOpenTelemetry;
+import io.opentelemetry.api.trace.Span;
+import io.opentelemetry.api.trace.Tracer;
+import io.opentelemetry.context.Scope;
+import org.aspectj.lang.ProceedingJoinPoint;
+import org.aspectj.lang.annotation.Around;
+import org.aspectj.lang.annotation.Aspect;
+import org.aspectj.lang.annotation.Pointcut;
+
+@Aspect
+public class TelemetryAspect {
+
+    private static final Tracer tracer = GlobalOpenTelemetry.getTracer("appsmith");
+    private static final long DEFAULT_THRESHOLD_NS = 1_000_000L;
+    private static final long thresholdNanoseconds = Long.parseLong(
+        System.getenv().getOrDefault("TELEMETRY_THRESHOLD_NANOSECONDS", String.valueOf(DEFAULT_THRESHOLD_NS))
+    );
+
+    @Pointcut("execution(* org.eclipse.jgit..*(..))")
+    public void monitoredAppsmithMethods() {}
+
+    @Around("monitoredAppsmithMethods()")
+    public Object traceAppsmithBusinessLogic(ProceedingJoinPoint joinPoint) throws Throwable {
+        long startTime = System.nanoTime();
+
+        Object result;
+        Throwable caughtException = null;
+        try {
+            result = joinPoint.proceed();
+        } catch (Throwable t) {
+            caughtException = t;
+            throw t;
+        } finally {
+            long duration = System.nanoTime() - startTime;
+            if (duration > thresholdNanoseconds) {
+                String spanName = getSpanName(joinPoint);
+
+                Span span = tracer.spanBuilder(spanName).startSpan();
+
+                try (Scope scope = span.makeCurrent()) {
+                    span.setAttribute("execution.time.ns", duration);
+                    if (caughtException != null) {
+                        span.recordException(caughtException);
+                    }
+                    logSpan(span, spanName, duration);
+                } finally {
+                    span.end();
+                }
+            }
+        }
+
+        return result;
+    }
+
+    private String getSpanName(ProceedingJoinPoint joinPoint) {
+        return joinPoint.getSignature().getDeclaringTypeName() + "." + joinPoint.getSignature().getName();
+    }
+
+    private void logSpan(Span span, String spanName, long duration) {
+        System.out.println("🚀 " + span.getSpanContext().getTraceId() + " > "
+                + span.getSpanContext().getSpanId() + " > " + spanName + " took: " + duration + " ns");
+    }
+}