feat(observability): trace Transaction

This change adds observability tracing for Transaction along with tests. Alsoo while here, added SessionPool.createSessions span. Updates googleapis#2079 Built from PR googleapis#2087 Updates googleapis#2114
odeke-em · Oct 3, 2024 · 310bf59 · 310bf59
1 parent 1f06871
commit 310bf59
Show file tree

Hide file tree

Showing 11 changed files with 1,360 additions and 450 deletions.
diff --git a/observability-test/batch-transaction.ts b/observability-test/batch-transaction.ts
@@ -139,7 +139,7 @@ describe('BatchTransaction', () => {
     batchTransaction = new BatchTransaction(SESSION as {} as Session);
     batchTransaction.session = SESSION as {} as Session;
     batchTransaction.id = ID;
-    batchTransaction.observabilityOptions = {tracerProvider: provider};
+    batchTransaction._observabilityOptions = {tracerProvider: provider};
     REQUEST.callsFake((_, callback) => callback(null, RESPONSE));
   });
 

diff --git a/observability-test/database.ts b/observability-test/database.ts
@@ -241,7 +241,7 @@ describe('Database', () => {
     database = new Database(INSTANCE, NAME, POOL_OPTIONS);
     database.parent = INSTANCE;
     database.databaseRole = 'parent_role';
-    database.observabilityConfig = {
+    database._observabilityOptions = {
       tracerProvider: provider,
       enableExtendedTracing: false,
     };