fix: updated span kind to CLIENT for ack and modack rpc spans

src/telemetry-tracing.ts

@@ -515,7 +515,7 @@ export class PubsubSpans {
     const span: Span = getTracer().startSpan(
       `${subInfo.subId ?? subInfo.subName} ack`,
       {
-        kind: SpanKind.CONSUMER,
+        kind: SpanKind.CLIENT,
         attributes: spanAttributes,
         links,
       },
@@ -563,7 +563,7 @@ export class PubsubSpans {
     const span: Span = getTracer().startSpan(
       `${subInfo.subId ?? subInfo.subName} ${type}`,
       {
-        kind: SpanKind.CONSUMER,
+        kind: SpanKind.CLIENT,
         attributes: spanAttributes,
         links,
       },

test/telemetry-tracing.ts

@@ -23,6 +23,7 @@
 import {SpanKind} from '@opentelemetry/api';
 import sinon = require('sinon');
 import {PubsubMessage} from '../src/publisher';
+import {Duration} from '../src/temporal'
 
 describe('OpenTelemetryTracer', () => {
   beforeEach(() => {
@@ -287,6 +288,7 @@
         firstSpan.attributes['messaging.destination.name'],
         tests.topicInfo.topicId,
       );
+      assert.strictEqual(firstSpan.attributes['messaging.operation'], 'create')
       assert.strictEqual(
         firstSpan.attributes['messaging.system'],
         'gcp_pubsub',
@@ -342,6 +344,7 @@
       assert.ok(parentReadSpan && childReadSpan);
 
       assert.strictEqual(childReadSpan.name, 'sub subscribe');
+      assert.strictEqual(childReadSpan.attributes['messaging.operation'], 'receive')
       assert.strictEqual(
         childReadSpan.attributes['messaging.destination.name'],
         'sub',
@@ -380,5 +383,84 @@
       assert.strictEqual(publishReadSpan.links.length, 1);
       assert.strictEqual(childReadSpan.links.length, 1);
     });
+
+    it('creates ack rpc span', () => {
+      const message: PubsubMessage = {};
+      const topicName = 'projects/test/topics/topicfoo';
+      const subName = 'subTest'
+      const producerSpan = otel.PubsubSpans.createPublisherSpan(
+        message,
+        topicName,
+        'test'
+      ) as trace.Span;
+      const span = otel.PubsubSpans.createAckRpcSpan(
+        [producerSpan],
+        subName,
+        'tests'
+      );
+      assert.ok(span);
+      producerSpan.end();
+      span.end();
+
+      const spans = exporter.getFinishedSpans();
+      assert.strictEqual(spans.length, 2);
+
+      const firstSpan = spans.pop();
+      assert.ok(firstSpan);
+      assert.strictEqual(firstSpan.kind, SpanKind.CLIENT);
+      assert.strictEqual(firstSpan.name, `${subName} ack`);
+      assert.strictEqual(
+        firstSpan.attributes['messaging.destination.name'],
+        subName
+      );
+      assert.strictEqual(firstSpan.attributes['messaging.operation'], 'receive');
+      assert.strictEqual(firstSpan.attributes['messaging.batch.message_count'], 1);
+      assert.strictEqual(
+        firstSpan.attributes['messaging.system'],
+        'gcp_pubsub'
+      );
+    });
+
+    it('creates modack rpc span', () => {
+      const message: PubsubMessage = {};
+      const topicName = 'projects/test/topics/topicfoo';
+      const subName = 'subTest'
+      const producerSpan = otel.PubsubSpans.createPublisherSpan(
+        message,
+        topicName,
+        'test'
+      ) as trace.Span;
+      const span = otel.PubsubSpans.createModackRpcSpan(
+        [producerSpan],
+        subName,
+        'modack',
+        'test',
+        Duration.from({ seconds: 1 }),
+        true
+      );
+      assert.ok(span);
+      producerSpan.end();
+      span.end();
+
+      const spans = exporter.getFinishedSpans();
+      assert.strictEqual(spans.length, 2);
+
+      const firstSpan = spans.pop();
+      assert.ok(firstSpan);
+      assert.strictEqual(firstSpan.kind, SpanKind.CLIENT);
+      assert.strictEqual(firstSpan.name, `${subName} modack`);
+      assert.strictEqual(
+        firstSpan.attributes['messaging.destination.name'],
+        subName
+      );
+      assert.strictEqual(firstSpan.attributes['messaging.operation'], 'receive');
+      assert.strictEqual(firstSpan.attributes['messaging.gcp_pubsub.message.ack_deadline_seconds'], 1);
+      assert.strictEqual(firstSpan.attributes['messaging.gcp_pubsub.is_receipt_modack'], true)
+      assert.strictEqual(
+        firstSpan.attributes['messaging.system'],
+        'gcp_pubsub'
+      );
+    });
   });
+
 });