@autofleet/rabbit 3.0.0 → 3.0.1-alpha10

package/dist/index.d.ts

@@ -3,7 +3,7 @@ import { EventEmitter } from 'events';
 import { AmqpConnectionManager, ChannelWrapper } from 'amqp-connection-manager';
 import { ConfirmChannel, ConsumeMessage, Options, Replies } from 'amqplib';
 import { RedisConfig } from './lib/redis';
-import { TRACING_HEADER } from './lib/consts';
+import { TRACING_HEADER, USER_TRACING_HEADER } from './lib/consts';
 import { CallbackFunction, ConsumeMessageOrNull, ConsumeOptions, CustomMessageHeaders, QueuesCache, RedisLockType, ExchangesCache } from './lib/types';
 export interface IAfRabbitMq {
     ack: any;
@@ -44,6 +44,7 @@ declare class RabbitMq implements IAfRabbitMq {
     static getPublishOptions(customHeaders?: CustomMessageHeaders): {
         timestamp: number;
         headers: {
+            "x-af-user-id": any;
             "x-trace-id": any;
             redisTimestampValidationKey?: string | undefined;
             creationTimestamp: number;

package/dist/index.js

@@ -9,7 +9,7 @@ const util_1 = require("util");
 const moment_1 = __importDefault(require("moment"));
 const redis_lock_1 = __importDefault(require("redis-lock"));
 const amqp_connection_manager_1 = require("amqp-connection-manager");
-const outbreak_1 = require("@autofleet/outbreak");
+const zehut_1 = require("@autofleet/zehut");
 const logger_1 = __importDefault(require("./logger"));
 const rabbitError_1 = __importDefault(require("./lib/rabbitError"));
 const redis_1 = __importDefault(require("./lib/redis"));
@@ -124,12 +124,14 @@ class RabbitMq {
         }
     }
     static getPublishOptions(customHeaders = {}) {
-        const trace = outbreak_1.getCurrentContext();
+        const trace = zehut_1.getCurrentPayload();
+        const user = trace?.context?.get(consts_1.USER_OBJECT);
         return {
             timestamp: moment_1.default().unix(),
             headers: {
                 creationTimestamp: moment_1.default().valueOf(),
                 ...customHeaders,
+                [consts_1.USER_TRACING_HEADER]: user?.id,
                 [consts_1.TRACING_HEADER]: trace?.context?.get(consts_1.TRACING_HEADER),
             },
         };
@@ -319,7 +321,7 @@ class RabbitMq {
     async consumeFromRabbit(queue, callback, options) {
         const optionsWithDefaults = { ...consts_1.DEFAULT_OPTIONS, ...options };
         RabbitMq.validateName('queue', queue);
-        const { limit, deadMessageTtl, useConsumeWithLock, lockTimeout, } = optionsWithDefaults;
+        const { limit, deadMessageTtl, useConsumeWithLock, lockTimeout, auditContext, enableRabbitTrace, } = optionsWithDefaults;
         if (useConsumeWithLock) {
             if (!this.redisLock) {
                 throw new Error('Usage of consumeWithLock requires RedisInstance');
@@ -335,10 +337,23 @@ class RabbitMq {
                     return null;
                 }
                 const traceId = msg.properties.headers[consts_1.TRACING_HEADER];
+                const userId = msg.properties.headers[consts_1.USER_TRACING_HEADER];
+                logger_1.default.info('rabbit: consume', { traceId, userId, enableRabbitTrace });
                 if (traceId) {
-                    const trace = outbreak_1.newTrace(outbreak_1.traceTypes.RABBIT);
+                    const trace = zehut_1.newTrace(zehut_1.traceTypes.RABBIT);
                     trace?.context?.set(consts_1.TRACING_HEADER, traceId);
                 }
+                else if (userId && enableRabbitTrace) {
+                    try {
+                        await zehut_1.setRabbitTrace(userId);
+                    }
+                    catch (e) {
+                        logger_1.default.error('rabbit: failed to setRabbitTrace', { userId, e });
+                    }
+                }
+                if (auditContext) {
+                    auditContext(queue);
+                }
                 const parsedMessage = RabbitMq.parseMsg(msg);
                 const releaseLock = await this.lockRedisIfNeeded(parsedMessage, optionsWithDefaults);
                 const shouldConsume = await this.shouldConsumeMessageByTimestamp(parsedMessage);

package/dist/lib/consts.d.ts

@@ -2,6 +2,8 @@ export declare const DEFAULT_DEAD_TTL_TWO_DAYS: number;
 export declare const DEFAULT_LOCK_TIMEOUT: number;
 export declare const RETRY_HEADER = "x-retry-count";
 export declare const TRACING_HEADER = "x-trace-id";
+export declare const USER_TRACING_HEADER = "x-af-user-id";
+export declare const USER_OBJECT = "userObject";
 export declare const DEFAULT_USE_CONSUME_WITH_LOCK = false;
 export declare const CONNECTION_CREATED_CONST = "connectionCreated";
 export declare const DEFAULT_OPTIONS: {
@@ -10,4 +12,6 @@ export declare const DEFAULT_OPTIONS: {
     deadMessageTtl: number;
     lockTimeout: number;
     useConsumeWithLock: boolean;
+    auditContext: null;
+    enableRabbitTrace: boolean;
 };

package/dist/lib/consts.js

@@ -1,10 +1,12 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DEFAULT_OPTIONS = exports.CONNECTION_CREATED_CONST = exports.DEFAULT_USE_CONSUME_WITH_LOCK = exports.TRACING_HEADER = exports.RETRY_HEADER = exports.DEFAULT_LOCK_TIMEOUT = exports.DEFAULT_DEAD_TTL_TWO_DAYS = void 0;
+exports.DEFAULT_OPTIONS = exports.CONNECTION_CREATED_CONST = exports.DEFAULT_USE_CONSUME_WITH_LOCK = exports.USER_OBJECT = exports.USER_TRACING_HEADER = exports.TRACING_HEADER = exports.RETRY_HEADER = exports.DEFAULT_LOCK_TIMEOUT = exports.DEFAULT_DEAD_TTL_TWO_DAYS = void 0;
 exports.DEFAULT_DEAD_TTL_TWO_DAYS = 60000 * 60 * 12;
 exports.DEFAULT_LOCK_TIMEOUT = 1000 * 5;
 exports.RETRY_HEADER = 'x-retry-count';
 exports.TRACING_HEADER = 'x-trace-id';
+exports.USER_TRACING_HEADER = 'x-af-user-id';
+exports.USER_OBJECT = 'userObject';
 exports.DEFAULT_USE_CONSUME_WITH_LOCK = false;
 exports.CONNECTION_CREATED_CONST = 'connectionCreated';
 exports.DEFAULT_OPTIONS = {
@@ -13,4 +15,6 @@ exports.DEFAULT_OPTIONS = {
     deadMessageTtl: exports.DEFAULT_DEAD_TTL_TWO_DAYS,
     lockTimeout: exports.DEFAULT_LOCK_TIMEOUT,
     useConsumeWithLock: exports.DEFAULT_USE_CONSUME_WITH_LOCK,
+    auditContext: null,
+    enableRabbitTrace: false,
 };

package/dist/lib/types.d.ts

@@ -17,6 +17,8 @@ export interface ConsumeOptions {
     limit?: number;
     lockTimeout?: number;
     useConsumeWithLock?: boolean;
+    auditContext?: any;
+    enableRabbitTrace?: boolean;
 }
 export declare type CallbackFunction = (msg: ConsumeMessage, ack: any, nack: any) => Promise<any>;
 export declare type newChannelOpts = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autofleet/rabbit",
-  "version": "3.0.0",
+  "version": "3.0.1-alpha10",
   "description": "",
   "main": "dist/index.js",
   "scripts": {
@@ -16,6 +16,7 @@
   },
   "dependencies": {
     "@autofleet/logger": "^1.2.8",
+    "@autofleet/zehut": "^3.0.1-alpha3",
     "@types/amqplib": "0.8.2",
     "@types/uuid": "^8.3.3",
     "amqp-connection-manager": "4.1.9",
@@ -26,9 +27,6 @@
     "redis-lock": "^0.1.4",
     "uuid": "^8.3.2"
   },
-  "peerDependencies": {
-    "@autofleet/outbreak": "^0.2.1"
-  },
   "devDependencies": {
     "@types/jest": "^26.0.23",
     "@typescript-eslint/eslint-plugin": "^4.8.1",

package/src/index.ts

@@ -8,7 +8,9 @@ import { AmqpConnectionManager, ChannelWrapper, connect } from 'amqp-connection-
 import {
   ConfirmChannel, ConsumeMessage, Options, Replies,
 } from 'amqplib';
-import { getCurrentContext, newTrace, traceTypes } from '@autofleet/outbreak';
+import {
+  getCurrentPayload, newTrace, traceTypes, setRabbitTrace,
+} from '@autofleet/zehut';
 import logger from './logger';
 import RabbitError from './lib/rabbitError';
 import getRedisInstance, { RedisConfig } from './lib/redis';
@@ -19,6 +21,8 @@ import {
   DEFAULT_OPTIONS,
   RETRY_HEADER,
   TRACING_HEADER,
+  USER_OBJECT,
+  USER_TRACING_HEADER,
 } from './lib/consts';
 import {
   CallbackFunction,
@@ -108,12 +112,14 @@ class RabbitMq implements IAfRabbitMq {
   }
 
   static getPublishOptions(customHeaders: CustomMessageHeaders = {}) {
-    const trace = getCurrentContext();
+    const trace = getCurrentPayload();
+    const user = trace?.context?.get(USER_OBJECT);
     return {
       timestamp: moment().unix(),
       headers: {
         creationTimestamp: moment().valueOf(),
         ...customHeaders,
+        [USER_TRACING_HEADER]: user?.id,
         [TRACING_HEADER]: trace?.context?.get(TRACING_HEADER),
       },
     };
@@ -455,7 +461,7 @@ class RabbitMq implements IAfRabbitMq {
     const optionsWithDefaults = { ...DEFAULT_OPTIONS, ...options };
     RabbitMq.validateName('queue', queue);
     const {
-      limit, deadMessageTtl, useConsumeWithLock, lockTimeout,
+      limit, deadMessageTtl, useConsumeWithLock, lockTimeout, auditContext, enableRabbitTrace,
     } = optionsWithDefaults;
     if (useConsumeWithLock) {
       if (!this.redisLock) {
@@ -475,9 +481,21 @@ class RabbitMq implements IAfRabbitMq {
           }
 
           const traceId = msg.properties.headers[TRACING_HEADER];
+          const userId = msg.properties.headers[USER_TRACING_HEADER];
+          logger.info('rabbit: consume', { traceId, userId, enableRabbitTrace });
           if (traceId) {
             const trace = newTrace(traceTypes.RABBIT);
             (trace as any)?.context?.set(TRACING_HEADER, traceId);
+          } else if (userId && enableRabbitTrace) {
+            try {
+              await setRabbitTrace(userId);
+            } catch (e) {
+              logger.error('rabbit: failed to setRabbitTrace', { userId, e });
+            }
+          }
+
+          if (auditContext) {
+            auditContext(queue);
           }
 
           const parsedMessage = RabbitMq.parseMsg(msg);

package/src/lib/consts.ts

@@ -2,6 +2,8 @@ export const DEFAULT_DEAD_TTL_TWO_DAYS = 60000 * 60 * 12;
 export const DEFAULT_LOCK_TIMEOUT = 1000 * 5;
 export const RETRY_HEADER = 'x-retry-count';
 export const TRACING_HEADER = 'x-trace-id';
+export const USER_TRACING_HEADER = 'x-af-user-id';
+export const USER_OBJECT = 'userObject';
 export const DEFAULT_USE_CONSUME_WITH_LOCK = false;
 export const CONNECTION_CREATED_CONST = 'connectionCreated';
 export const DEFAULT_OPTIONS = {
@@ -10,4 +12,6 @@ export const DEFAULT_OPTIONS = {
   deadMessageTtl: DEFAULT_DEAD_TTL_TWO_DAYS,
   lockTimeout: DEFAULT_LOCK_TIMEOUT,
   useConsumeWithLock: DEFAULT_USE_CONSUME_WITH_LOCK,
+  auditContext: null,
+  enableRabbitTrace: false,
 };

package/src/lib/types.ts

@@ -21,6 +21,8 @@ export interface ConsumeOptions {
   limit?: number;
   lockTimeout?: number;
   useConsumeWithLock?: boolean;
+  auditContext?: any;
+  enableRabbitTrace?: boolean;
 }
 
 export type CallbackFunction = (msg: ConsumeMessage, ack: any, nack: any) => Promise<any>