fix(spanner): end spans for read-write methods (#3629)

googleapis · Feb 10, 2025 · 4a1f99c · 4a1f99c
1 parent 40568de
commit 4a1f99c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/google-cloud-spanner/src/main/java/com/google/cloud/spanner/DatabaseClientImpl.java b/google-cloud-spanner/src/main/java/com/google/cloud/spanner/DatabaseClientImpl.java
@@ -294,6 +294,8 @@ public TransactionManager transactionManager(TransactionOption... options) {
       span.setStatus(e);
       span.end();
       throw e;
+    } finally {
+      span.end();
     }
   }
 

diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/DatabaseClientImplTest.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/DatabaseClientImplTest.java
@@ -4006,7 +4006,10 @@ public void testTransactionManager_usesOptions() {
     when(pool.getSession()).thenReturn(session);
     TransactionOption option = mock(TransactionOption.class);
 
-    DatabaseClientImpl client = new DatabaseClientImpl(pool, mock(TraceWrapper.class));
+    TraceWrapper traceWrapper =
+        new TraceWrapper(Tracing.getTracer(), OpenTelemetry.noop().getTracer(""), false);
+
+    DatabaseClientImpl client = new DatabaseClientImpl(pool, traceWrapper);
     try (TransactionManager ignore = client.transactionManager(option)) {
       verify(session).transactionManager(option);
     }