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,
  };