feat(spanner): Update OpenTelemetry OTLP trace sample to use Google Cloud OTLP endpoint

Author: surbhigarg92

spanner/opentelemetry_traces/src/main/java/com/example/spanner/OpenTelemetryUsage.java

@@ -16,6 +16,7 @@
 
 package com.example.spanner;
 
+import com.google.auth.oauth2.GoogleCredentials;
 import com.google.cloud.opentelemetry.trace.TraceConfiguration;
 import com.google.cloud.opentelemetry.trace.TraceExporter;
 import com.google.cloud.spanner.DatabaseClient;
@@ -32,10 +33,13 @@
 import io.opentelemetry.sdk.trace.export.BatchSpanProcessor;
 import io.opentelemetry.sdk.trace.export.SpanExporter;
 import io.opentelemetry.sdk.trace.samplers.Sampler;
+import java.io.IOException;
+import java.util.Collections;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 
-/**
- * This sample demonstrates how to configure OpenTelemetry and inject via Spanner Options.
- */
+/** This sample demonstrates how to configure OpenTelemetry and inject via Spanner Options. */
 public class OpenTelemetryUsage {
 
   static SdkTracerProvider sdkTracerProvider;
@@ -46,20 +50,20 @@ public class OpenTelemetryUsage {
   static String instanceId = "my-instance";
   static String databaseId = "my-database";
 
-  // Replace these variables to use OTLP Exporter
-  static boolean useCloudTraceExporter = true; // Replace to false for OTLP
-  static String otlpEndpoint = "http://localhost:4317"; // Replace with your OTLP endpoint
+  static boolean useCloudTraceExporter = false; // Replace to true for Cloud Trace exporter
+  static String otlpEndpoint =
+      "https://telemetry.googleapis.com"; // Replace with your OTLP endpoint
 
-  public static void main(String[] args) {
+  public static void main(String[] args) throws IOException {
 
     if (useCloudTraceExporter) {
       spanner = getSpannerWithCloudTraceExporter();
     } else {
       spanner = getSpannerWithOtlpExporter();
     }
 
-    DatabaseClient dbClient = spanner
-        .getDatabaseClient(DatabaseId.of(projectId, instanceId, databaseId));
+    DatabaseClient dbClient =
+        spanner.getDatabaseClient(DatabaseId.of(projectId, instanceId, databaseId));
 
     try (ResultSet resultSet =
         dbClient
@@ -74,15 +78,37 @@ public static void main(String[] args) {
     sdkTracerProvider.forceFlush();
   }
 
-  public static Spanner getSpannerWithOtlpExporter() {
+  public static Spanner getSpannerWithOtlpExporter() throws IOException {
     // [START spanner_opentelemetry_traces_otlp_usage]
-    Resource resource = Resource
-        .getDefault().merge(Resource.builder().put("service.name", "My App").build());
-
+    Resource resource =
+        Resource.getDefault()
+            .merge(
+                Resource.builder()
+                    .put("service.name", "My App")
+                    .put("gcp.project_id", projectId)
+                    .build());
+
+    GoogleCredentials credentials =
+        GoogleCredentials.getApplicationDefault()
+            .createScoped(Collections.singleton("https://www.googleapis.com/auth/trace.append"));
     OtlpGrpcSpanExporter otlpGrpcSpanExporter =
-        OtlpGrpcSpanExporter
-            .builder()
-            .setEndpoint(otlpEndpoint) // Replace with your OTLP endpoint
+        OtlpGrpcSpanExporter.builder()
+            .setEndpoint(otlpEndpoint)
+            .setHeaders(
+                () -> {
+                  try {
+                    credentials.refreshIfExpired();
+                    Map<String, List<String>> metadata = credentials.getRequestMetadata();
+                    Map<String, String> headers = new HashMap<>();
+                    if (metadata != null) {
+                      metadata.forEach((key, values) -> headers.put(key, String.join(",", values)));
+                    }
+                    return headers;
+                  } catch (Exception e) {
+                    // Handle error fetching credentials
+                    return Collections.emptyMap();
+                  }
+                }) // Replace with your OTLP endpoint
             .build();
 
     // Using a batch span processor
@@ -92,24 +118,23 @@ public static Spanner getSpannerWithOtlpExporter() {
         BatchSpanProcessor.builder(otlpGrpcSpanExporter).build();
 
     // Create a new tracer provider
-    sdkTracerProvider = SdkTracerProvider.builder()
-        // Use Otlp exporter or any other exporter of your choice.
-        .addSpanProcessor(otlpGrpcSpanProcessor)
-        .setResource(resource)
-        .setSampler(Sampler.traceIdRatioBased(0.1))
-        .build();
+    sdkTracerProvider =
+        SdkTracerProvider.builder()
+            // Use Otlp exporter or any other exporter of your choice.
+            .addSpanProcessor(otlpGrpcSpanProcessor)
+            .setResource(resource)
+            .setSampler(Sampler.traceIdRatioBased(0.1))
+            .build();
 
     // Export to a collector that is expecting OTLP using gRPC.
-    OpenTelemetry openTelemetry = OpenTelemetrySdk.builder()
-        .setTracerProvider(sdkTracerProvider).build();
+    OpenTelemetry openTelemetry =
+        OpenTelemetrySdk.builder().setTracerProvider(sdkTracerProvider).build();
 
     // Enable OpenTelemetry traces before Injecting OpenTelemetry
     SpannerOptions.enableOpenTelemetryTraces();
 
     // Inject OpenTelemetry object via Spanner options or register as GlobalOpenTelemetry.
-    SpannerOptions options = SpannerOptions.newBuilder()
-        .setOpenTelemetry(openTelemetry)
-        .build();
+    SpannerOptions options = SpannerOptions.newBuilder().setOpenTelemetry(openTelemetry).build();
     Spanner spanner = options.getService();
     // [END spanner_opentelemetry_traces_otlp_usage]
 
@@ -118,40 +143,38 @@ public static Spanner getSpannerWithOtlpExporter() {
 
   public static Spanner getSpannerWithCloudTraceExporter() {
     // [START spanner_opentelemetry_traces_cloudtrace_usage]
-    Resource resource = Resource
-        .getDefault().merge(Resource.builder().put("service.name", "My App").build());
+    Resource resource =
+        Resource.getDefault().merge(Resource.builder().put("service.name", "My App").build());
 
-    SpanExporter traceExporter = TraceExporter.createWithConfiguration(
-        TraceConfiguration.builder().setProjectId(projectId).build()
-    );
+    SpanExporter traceExporter =
+        TraceExporter.createWithConfiguration(
+            TraceConfiguration.builder().setProjectId(projectId).build());
 
     // Using a batch span processor
     // You can use `.setScheduleDelay()`, `.setExporterTimeout()`,
     // `.setMaxQueueSize`(), and `.setMaxExportBatchSize()` to further customize.
-    BatchSpanProcessor otlpGrpcSpanProcessor =
-        BatchSpanProcessor.builder(traceExporter).build();
+    BatchSpanProcessor otlpGrpcSpanProcessor = BatchSpanProcessor.builder(traceExporter).build();
 
     // Create a new tracer provider
-    sdkTracerProvider = SdkTracerProvider.builder()
-        // Use Otlp exporter or any other exporter of your choice.
-        .addSpanProcessor(otlpGrpcSpanProcessor)
-        .setResource(resource)
-        .setSampler(Sampler.traceIdRatioBased(0.1))
-        .build();
+    sdkTracerProvider =
+        SdkTracerProvider.builder()
+            // Use Otlp exporter or any other exporter of your choice.
+            .addSpanProcessor(otlpGrpcSpanProcessor)
+            .setResource(resource)
+            .setSampler(Sampler.traceIdRatioBased(0.1))
+            .build();
 
     // Export to a collector that is expecting OTLP using gRPC.
-    OpenTelemetry openTelemetry = OpenTelemetrySdk.builder()
-        .setTracerProvider(sdkTracerProvider).build();
+    OpenTelemetry openTelemetry =
+        OpenTelemetrySdk.builder().setTracerProvider(sdkTracerProvider).build();
 
     // Enable OpenTelemetry traces before Injecting OpenTelemetry
     SpannerOptions.enableOpenTelemetryTraces();
 
     // Inject OpenTelemetry object via Spanner options or register it as global object.
     // To register as the global OpenTelemetry object,
     // use "OpenTelemetrySdk.builder()....buildAndRegisterGlobal()".
-    SpannerOptions options = SpannerOptions.newBuilder()
-        .setOpenTelemetry(openTelemetry)
-        .build();
+    SpannerOptions options = SpannerOptions.newBuilder().setOpenTelemetry(openTelemetry).build();
     Spanner spanner = options.getService();
     // [END spanner_opentelemetry_traces_cloudtrace_usage]