feat: create(Blob) instrumentation

google-cloud-storage/src/main/java/com/google/cloud/storage/GrpcStorageImpl.java

@@ -248,8 +248,23 @@ public Blob create(BlobInfo blobInfo, byte[] content, BlobTargetOption... option
   public Blob create(
       BlobInfo blobInfo, byte[] content, int offset, int length, BlobTargetOption... options) {
     Opts<ObjectTargetOpt> opts = Opts.unwrap(options).resolveFrom(blobInfo);
-    return internalDirectUpload(blobInfo, opts, ByteBuffer.wrap(content, offset, length))
-        .asBlob(this);
+    // Start the otel span to retain information of the origin of the request
+    OpenTelemetryTraceUtil.Span otelSpan =
+        openTelemetryTraceUtil.startSpan("create(BlobInfo, BlobTargetOption");
+    try (OpenTelemetryTraceUtil.Scope unused = otelSpan.makeCurrent()) {
+      return internalDirectUpload(
+              blobInfo,
+              opts,
+              ByteBuffer.wrap(content, offset, length),
+              openTelemetryTraceUtil.currentContext())
+          .asBlob(this);
+    } catch (Exception e) {
+      otelSpan.recordException(e);
+      otelSpan.setStatus(io.opentelemetry.api.trace.StatusCode.ERROR, e.getClass().getSimpleName());
+      throw StorageException.coalesce(e);
+    } finally {
+      otelSpan.end();
+    }
   }
 
   @Override
@@ -799,38 +814,51 @@ public GrpcBlobWriteChannel writer(BlobInfo blobInfo, BlobWriteOption... options
 
   @Override
   public BlobInfo internalDirectUpload(
-      BlobInfo blobInfo, Opts<ObjectTargetOpt> opts, ByteBuffer buf) {
+      BlobInfo blobInfo,
+      Opts<ObjectTargetOpt> opts,
+      ByteBuffer buf,
+      OpenTelemetryTraceUtil.Context ctx) {
     requireNonNull(blobInfo, "blobInfo must be non null");
     requireNonNull(buf, "content must be non null");
+    OpenTelemetryTraceUtil.Span otelSpan =
+        openTelemetryTraceUtil.startSpan("internalDirectUpload(BlobInfo)", ctx);
     Opts<ObjectTargetOpt> optsWithDefaults = opts.prepend(defaultOpts);
     GrpcCallContext grpcCallContext =
         optsWithDefaults.grpcMetadataMapper().apply(GrpcCallContext.createDefault());
     WriteObjectRequest req = getWriteObjectRequest(blobInfo, optsWithDefaults);
     Hasher hasher = Hasher.enabled();
     GrpcCallContext merge = Utils.merge(grpcCallContext, Retrying.newCallContext());
     RewindableContent content = RewindableContent.of(buf);
-    return Retrying.run(
-        getOptions(),
-        retryAlgorithmManager.getFor(req),
-        () -> {
-          content.rewindTo(0);
-          UnbufferedWritableByteChannelSession<WriteObjectResponse> session =
-              ResumableMedia.gapic()
-                  .write()
-                  .byteChannel(storageClient.writeObjectCallable().withDefaultCallContext(merge))
-                  .setByteStringStrategy(ByteStringStrategy.noCopy())
-                  .setHasher(hasher)
-                  .direct()
-                  .unbuffered()
-                  .setRequest(req)
-                  .build();
-
-          try (UnbufferedWritableByteChannel c = session.open()) {
-            content.writeTo(c);
-          }
-          return session.getResult();
-        },
-        this::getBlob);
+    try (OpenTelemetryTraceUtil.Scope unused = otelSpan.makeCurrent()) {
+      return Retrying.run(
+          getOptions(),
+          retryAlgorithmManager.getFor(req),
+          () -> {
+            content.rewindTo(0);
+            UnbufferedWritableByteChannelSession<WriteObjectResponse> session =
+                ResumableMedia.gapic()
+                    .write()
+                    .byteChannel(storageClient.writeObjectCallable().withDefaultCallContext(merge))
+                    .setByteStringStrategy(ByteStringStrategy.noCopy())
+                    .setHasher(hasher)
+                    .direct()
+                    .unbuffered()
+                    .setRequest(req)
+                    .build();
+
+            try (UnbufferedWritableByteChannel c = session.open()) {
+              content.writeTo(c);
+            }
+            return session.getResult();
+          },
+          this::getBlob);
+    } catch (Exception e) {
+      otelSpan.recordException(e);
+      otelSpan.setStatus(io.opentelemetry.api.trace.StatusCode.ERROR, e.getClass().getSimpleName());
+      throw StorageException.coalesce(e);
+    } finally {
+      otelSpan.end();
+    }
   }
 
   @Override

google-cloud-storage/src/main/java/com/google/cloud/storage/ParallelCompositeUploadWritableByteChannel.java

@@ -219,7 +219,8 @@ public void close() throws IOException {
       // We never created any parts
       // create an empty object
       try {
-        BlobInfo blobInfo = storage.internalDirectUpload(ultimateObject, opts, Buffers.allocate(0));
+        BlobInfo blobInfo =
+            storage.internalDirectUpload(ultimateObject, opts, Buffers.allocate(0), null);
         finalObject.set(blobInfo);
         return;
       } catch (StorageException se) {
@@ -285,7 +286,8 @@ private void internalFlush(ByteBuffer buf) {
             ApiFutures.immediateFuture(partInfo),
             info -> {
               try {
-                return storage.internalDirectUpload(info, partOpts, buf);
+                // TODO: Add in Otel context when available
+                return storage.internalDirectUpload(info, partOpts, buf, null);
               } catch (StorageException e) {
                 // a precondition failure usually means the part was created, but we didn't get the
                 // response. And when we tried to retry the object already exists.

google-cloud-storage/src/main/java/com/google/cloud/storage/StorageImpl.java

@@ -50,6 +50,7 @@
 import com.google.cloud.storage.UnifiedOpts.ObjectSourceOpt;
 import com.google.cloud.storage.UnifiedOpts.ObjectTargetOpt;
 import com.google.cloud.storage.UnifiedOpts.Opts;
+import com.google.cloud.storage.otel.OpenTelemetryTraceUtil;
 import com.google.cloud.storage.spi.v1.StorageRpc;
 import com.google.cloud.storage.spi.v1.StorageRpc.RewriteRequest;
 import com.google.common.base.CharMatcher;
@@ -1737,7 +1738,11 @@ public BlobInfo internalCreateFrom(Path path, BlobInfo info, Opts<ObjectTargetOp
   }
 
   @Override
-  public BlobInfo internalDirectUpload(BlobInfo info, Opts<ObjectTargetOpt> opts, ByteBuffer buf) {
+  public BlobInfo internalDirectUpload(
+      BlobInfo info,
+      Opts<ObjectTargetOpt> opts,
+      ByteBuffer buf,
+      OpenTelemetryTraceUtil.Context ctx) {
 
     BlobInfo.Builder builder =
         info.toBuilder()

google-cloud-storage/src/main/java/com/google/cloud/storage/StorageInternal.java

@@ -20,6 +20,7 @@
 import com.google.cloud.storage.UnifiedOpts.ObjectSourceOpt;
 import com.google.cloud.storage.UnifiedOpts.ObjectTargetOpt;
 import com.google.cloud.storage.UnifiedOpts.Opts;
+import com.google.cloud.storage.otel.OpenTelemetryTraceUtil;
 import java.io.IOException;
 import java.nio.ByteBuffer;
 import java.nio.file.Path;
@@ -31,7 +32,11 @@ default BlobInfo internalCreateFrom(Path path, BlobInfo info, Opts<ObjectTargetO
     throw new UnsupportedOperationException("not implemented");
   }
 
-  default BlobInfo internalDirectUpload(BlobInfo info, Opts<ObjectTargetOpt> opts, ByteBuffer buf) {
+  default BlobInfo internalDirectUpload(
+      BlobInfo info,
+      Opts<ObjectTargetOpt> opts,
+      ByteBuffer buf,
+      OpenTelemetryTraceUtil.Context ctx) {
     throw new UnsupportedOperationException("not implemented");
   }
 

google-cloud-storage/src/main/java/com/google/cloud/storage/otel/OpenTelemetryInstance.java

@@ -156,24 +156,24 @@ public Scope makeCurrent() {
   public OpenTelemetryTraceUtil.Span startSpan(String methodName) {
     String formatSpanName = String.format("%s.%s/%s", "storage", "client", methodName);
     SpanBuilder spanBuilder = tracer.spanBuilder(formatSpanName).setSpanKind(SpanKind.CLIENT);
-    spanBuilder.setAttribute("rpc.system", transport);
     io.opentelemetry.api.trace.Span span =
         addSettingsAttributesToCurrentSpan(spanBuilder).startSpan();
     return new Span(span, formatSpanName);
   }
 
   @Override
   public OpenTelemetryTraceUtil.Span startSpan(
-      String spanName, OpenTelemetryTraceUtil.Context parent) {
+      String methodName, OpenTelemetryTraceUtil.Context parent) {
     assert (parent instanceof OpenTelemetryInstance.Context);
+    String formatSpanName = String.format("%s.%s/%s", "storage", "client", methodName);
     SpanBuilder spanBuilder =
         tracer
-            .spanBuilder(spanName)
+            .spanBuilder(formatSpanName)
             .setSpanKind(SpanKind.CLIENT)
             .setParent(((OpenTelemetryInstance.Context) parent).context);
     io.opentelemetry.api.trace.Span span =
         addSettingsAttributesToCurrentSpan(spanBuilder).startSpan();
-    return new Span(span, spanName);
+    return new Span(span, formatSpanName);
   }
 
   @Nonnull
@@ -196,6 +196,7 @@ private SpanBuilder addSettingsAttributesToCurrentSpan(SpanBuilder spanBuilder)
                 .put("gcp.client.version", GaxProperties.getLibraryVersion(this.getClass()))
                 .put("gcp.client.repo", "googleapis/java-storage")
                 .put("gcp.client.artifact", "com.google.cloud.google-cloud-storage")
+                .put("rpc.system", transport)
                 .build());
     return spanBuilder;
   }

google-cloud-storage/src/test/java/com/google/cloud/storage/ITGrpcOpenTelemetryTest.java

@@ -0,0 +1,107 @@
+/*
+ * Copyright 2024 Google LLC
+ *
+ * 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
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * 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.google.cloud.storage;
+
+import static java.nio.charset.StandardCharsets.UTF_8;
+
+import com.google.cloud.NoCredentials;
+import com.google.cloud.storage.it.runner.StorageITRunner;
+import com.google.cloud.storage.it.runner.annotations.Backend;
+import com.google.cloud.storage.it.runner.annotations.Inject;
+import com.google.cloud.storage.it.runner.annotations.SingleBackend;
+import com.google.cloud.storage.it.runner.registry.Generator;
+import com.google.cloud.storage.it.runner.registry.TestBench;
+import io.opentelemetry.api.common.AttributeKey;
+import io.opentelemetry.sdk.OpenTelemetrySdk;
+import io.opentelemetry.sdk.trace.SdkTracerProvider;
+import io.opentelemetry.sdk.trace.data.SpanData;
+import io.opentelemetry.sdk.trace.export.SimpleSpanProcessor;
+import io.opentelemetry.sdk.trace.export.SpanExporter;
+import java.util.List;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+@RunWith(StorageITRunner.class)
+@SingleBackend(Backend.TEST_BENCH)
+public class ITGrpcOpenTelemetryTest {
+  @Inject public TestBench testBench;
+  private StorageOptions options;
+  private SpanExporter exporter;
+  private Storage storage;
+  @Inject public Generator generator;
+  @Inject public BucketInfo testBucket;
+
+  @Before
+  public void setUp() {
+    exporter = new TestExporter();
+    OpenTelemetrySdk openTelemetrySdk =
+        OpenTelemetrySdk.builder()
+            .setTracerProvider(
+                SdkTracerProvider.builder()
+                    .addSpanProcessor(SimpleSpanProcessor.create(exporter))
+                    .build())
+            .build();
+    options =
+        StorageOptions.grpc()
+            .setHost(testBench.getGRPCBaseUri())
+            .setProjectId("projectId")
+            .setCredentials(NoCredentials.getInstance())
+            .setOpenTelemetrySdk(openTelemetrySdk)
+            .build();
+    storage = options.getService();
+  }
+
+  @Test
+  public void runCreateBucket() {
+    String bucket = "random-bucket";
+    storage.create(BucketInfo.of(bucket));
+    TestExporter testExported = (TestExporter) exporter;
+    SpanData spanData = testExported.getExportedSpans().get(0);
+    Assert.assertEquals("Storage", getAttributeValue(spanData, "gcp.client.service"));
+    Assert.assertEquals("googleapis/java-storage", getAttributeValue(spanData, "gcp.client.repo"));
+    Assert.assertEquals(
+        "com.google.cloud.google-cloud-storage",
+        getAttributeValue(spanData, "gcp.client.artifact"));
+    Assert.assertEquals("grpc", getAttributeValue(spanData, "rpc.system"));
+  }
+
+  @Test
+  public void runCreateBlob() {
+    byte[] content = "Hello, World!".getBytes(UTF_8);
+    BlobId toCreate = BlobId.of(testBucket.getName(), generator.randomObjectName());
+    storage.create(BlobInfo.newBuilder(toCreate).build(), content);
+    TestExporter testExported = (TestExporter) exporter;
+    List<SpanData> spanData = testExported.getExportedSpans();
+    // (1) Span when calling create
+    // (2) Span when passing call to internalDirectUpload
+    Assert.assertEquals(2, spanData.size());
+    for (SpanData span : spanData) {
+      Assert.assertEquals("Storage", getAttributeValue(span, "gcp.client.service"));
+      Assert.assertEquals("googleapis/java-storage", getAttributeValue(span, "gcp.client.repo"));
+      Assert.assertEquals(
+          "com.google.cloud.google-cloud-storage", getAttributeValue(span, "gcp.client.artifact"));
+      Assert.assertEquals("grpc", getAttributeValue(span, "rpc.system"));
+    }
+    Assert.assertEquals(spanData.get(1).getSpanContext(), spanData.get(0).getParentSpanContext());
+  }
+
+  private String getAttributeValue(SpanData spanData, String key) {
+    return spanData.getAttributes().get(AttributeKey.stringKey(key)).toString();
+  }
+}