Utility to create spans for synchronous blocks of code

modules/apm/src/main/java/org/elasticsearch/telemetry/apm/internal/tracing/APMTracer.java

@@ -105,6 +105,11 @@ public void setEnabled(boolean enabled) {
         }
     }
 
+    @Override
+    public boolean isEnabled() {
+        return enabled;
+    }
+
     public void setIncludeNames(List<String> includeNames) {
         this.includeNames = includeNames;
         this.filterAutomaton = buildAutomaton(includeNames, excludeNames);

server/src/main/java/org/elasticsearch/telemetry/Trace.java

@@ -0,0 +1,54 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+package org.elasticsearch.telemetry;
+
+import org.elasticsearch.common.UUIDs;
+import org.elasticsearch.core.Releasable;
+import org.elasticsearch.telemetry.tracing.Traceable;
+import org.elasticsearch.telemetry.tracing.Tracer;
+import org.elasticsearch.threadpool.ThreadPool;
+
+import java.util.Map;
+
+public class Trace {
+
+    public static Releasable trace(ThreadPool threadPool, Tracer tracer, String name) {
+        return trace(threadPool, tracer, name, Map.of());
+    }
+
+    /**
+     * Creates a new trace for a synchronous block of code.
+     * @return a span that needs to be released once block of code is completed
+     */
+    public static Releasable trace(ThreadPool threadPool, Tracer tracer, String name, Map<String, Object> attributes) {
+        if (tracer.isEnabled() == false) {
+            return () -> {};
+        }
+        var span = Span.create();
+        var ctx = threadPool.getThreadContext().newTraceContext();
+        tracer.startTrace(threadPool.getThreadContext(), span, name, attributes);
+
+        return () -> {
+            tracer.stopTrace(span);
+            ctx.restore();
+        };
+    }
+
+    private record Span(String spanId) implements Traceable {
+
+        @Override
+        public String getSpanId() {
+            return spanId;
+        }
+
+        public static Span create() {
+            return new Span(UUIDs.randomBase64UUID());
+        }
+    }
+}

server/src/main/java/org/elasticsearch/telemetry/tracing/Tracer.java

@@ -33,6 +33,11 @@
  */
 public interface Tracer {
 
+    /**
+     * @return {@code if service is enabled}
+     */
+    boolean isEnabled();
+
     /**
      * Called when a span starts.
      * @param traceContext the current context. Required for tracing parent/child span activity.
@@ -145,6 +150,12 @@ public interface Tracer {
      * in order to avoid null checks everywhere.
      */
     Tracer NOOP = new Tracer() {
+
+        @Override
+        public boolean isEnabled() {
+            return false;
+        }
+
         @Override
         public void startTrace(TraceContext traceContext, Traceable traceable, String name, Map<String, Object> attributes) {}