feat: add attributes to pub/sub context

mdovhopo · Oct 17, 2022 · e675473 · e675473
1 parent bc2f216
commit e675473
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/pubsub-transport.spec.ts b/src/pubsub-transport.spec.ts
@@ -90,13 +90,18 @@ describe('PubSubTransport', () => {
       {
         a: 1,
       },
-      { deliveryAttempt: 1, id: 'messageId', pattern: 'operationId' }
+      {
+        deliveryAttempt: 1,
+        id: 'messageId',
+        pattern: 'operationId',
+        attributes: { operationId: 'operationId' },
+      }
     );
     expect(ackSpy).toBeCalled();
     expect(logInfoSpy).toBeCalled();
   });
 
-  it('handles incoming messages', async () => {
+  it('handles incoming messages if', async () => {
     const message = new Message({} as any, {
       ackId: 'ackId',
       message: {

diff --git a/src/pubsub-transport.ts b/src/pubsub-transport.ts
@@ -12,6 +12,8 @@ export interface PubSubContext {
   deliveryAttempt: number;
 
   pattern: string;
+
+  attributes: Record<string, string>;
 }
 export const PubSubTransportConfig = Symbol('PubSubTransportConfig');
 export type PubSubTransportConfig =
@@ -122,6 +124,7 @@ export class PubSubTransport extends Server implements CustomTransportStrategy {
       id: message.id,
       deliveryAttempt: message.deliveryAttempt,
       pattern,
+      attributes: message.attributes,
     };
 
     const handler = this.getHandlerByPattern(pattern);