@autofleet/outbreak 0.1.0-beta-2 → 0.2.0-beta-2

package/dist/src/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/config.ts"],"names":[],"mappings":";;AAAA,MAAM,mBAAmB,GAAG,YAAY,CAAC;AAEzC,MAAM,kBAAkB,GAAG,CAAC,OAAO,EAAQ,EAAE;IAC3C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE;QAC3B,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;KAChD;IAED,iEAAiE;IACjE,MAAM,eAAe,GAAG,iBAAiB,CAAC;IAE1C,MAAM,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IAEhG,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE;QAC7B,MAAM,IAAI,KAAK,CAAC,yCAAyC,cAAc,EAAE,CAAC,CAAC;KAC5E;AACH,CAAC,CAAC;AASF,MAAM,IAAI,GAAG,CAAC,YAAuB,EAAE,EAAE,EAAE;IACzC,MAAM,EACJ,4BAA4B,GAAG,IAAI,EACnC,oBAAoB,GAAG,KAAK,EAC5B,kBAAkB,GAAG;QACnB,cAAc;QACd,cAAc;KACf,EACD,aAAa,GAAG,OAAO,GACxB,GAAG,SAAS,CAAC;IAEd,kBAAkB,CAAC,kBAAkB,CAAC,CAAC;IAEvC,MAAM,gBAAgB,GAAG,kBAAkB,CAAC;IAE5C,IAAI,eAAe,GAAG,gBAAgB,CAAC;IACvC,IAAI,4BAA4B,KAAK,IAAI,EAAE;QACzC,eAAe,GAAG,CAAC,mBAAmB,CAAC,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC;KAClE;IAED,OAAO;QACL,4BAA4B;QAC5B,oBAAoB;QACpB,mBAAmB;QACnB,gBAAgB;QAChB,eAAe;QACf,aAAa;KACd,CAAC;AACJ,CAAC,CAAC;AAEF,kBAAe;IACb,IAAI;IACJ,mBAAmB;CACpB,CAAC"}

package/dist/src/example.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/src/example.js

@@ -0,0 +1,46 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const express_1 = __importDefault(require("express"));
+const axios_1 = __importDefault(require("axios"));
+const winston_1 = __importDefault(require("winston"));
+const index_1 = __importDefault(require("./index"));
+axios_1.default.defaults.adapter = require('axios/lib/adapters/http');
+const logger = winston_1.default.createLogger({
+    transports: [
+        new winston_1.default.transports.Console(),
+    ],
+});
+index_1.default({
+    setAndPropagateCorrelationId: true,
+    headersToPropagate: [
+        'x-text',
+    ],
+    headersPrefix: 'x-af-',
+    winstonLogger: logger,
+});
+const app = express_1.default();
+app.all('/', (req, res) => __awaiter(void 0, void 0, void 0, function* () {
+    const network = axios_1.default.create({ baseURL: 'http://localhost:8080/' });
+    const { data } = yield network.get('/header-parrot');
+    res.json(data);
+}));
+app.all('/header-parrot', (req, res) => {
+    logger.info('Test message', { t: 1 });
+    res.json(req.headers);
+});
+app.listen(8080, () => {
+    logger.info('App listen');
+});
+//# sourceMappingURL=example.js.map

package/dist/src/example.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"example.js","sourceRoot":"","sources":["../../src/example.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAAA,sDAA8B;AAC9B,kDAA0B;AAC1B,sDAA8B;AAE9B,oDAAoC;AAEpC,eAAK,CAAC,QAAQ,CAAC,OAAO,GAAG,OAAO,CAAC,yBAAyB,CAAC,CAAC;AAE5D,MAAM,MAAM,GAAG,iBAAO,CAAC,YAAY,CAAC;IAClC,UAAU,EAAE;QACV,IAAI,iBAAO,CAAC,UAAU,CAAC,OAAO,EAAE;KACjC;CACF,CAAC,CAAC;AAEH,eAAa,CAAC;IACZ,4BAA4B,EAAE,IAAI;IAClC,kBAAkB,EAAE;QAClB,QAAQ;KACT;IACD,aAAa,EAAE,OAAO;IACtB,aAAa,EAAE,MAAM;CACtB,CAAC,CAAC;AAEH,MAAM,GAAG,GAAG,iBAAO,EAAE,CAAC;AAEtB,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAO,GAAG,EAAE,GAAG,EAAE,EAAE;IAC9B,MAAM,OAAO,GAAG,eAAK,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,wBAAwB,EAAE,CAAC,CAAC;IACpE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;IACrD,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACjB,CAAC,CAAA,CAAC,CAAC;AAEH,GAAG,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;IACrC,MAAM,CAAC,IAAI,CAAC,cAAc,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IACtC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;AACxB,CAAC,CAAC,CAAC;AAEH,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;IACpB,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;AAC5B,CAAC,CAAC,CAAC"}

package/dist/src/{http-wrapper/wrapper.js → http_wrapper.js}

@@ -2,11 +2,18 @@
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (Object.hasOwnProperty.call(mod, k)) result[k] = mod[k];
+    result["default"] = mod;
+    return result;
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 /* eslint-disable max-len */
 const http_1 = __importDefault(require("http"));
 const shortid_1 = __importDefault(require("shortid"));
-const tracer_1 = require("../tracer");
+const tracer_1 = __importStar(require("./tracer"));
 const originalHttpCreateServer = http_1.default.createServer;
 const originalRequest = http_1.default.request;
 const originalGet = http_1.default.get;
@@ -15,7 +22,7 @@ function setAndCollectCorrelationId(config, req, res) {
     if (typeof correlationId === 'undefined') {
         correlationId = shortid_1.default.generate();
     }
-    tracer_1.getCurrentTrace().context.set(config.correlationIdHeader, correlationId);
+    tracer_1.default.currentTrace.context.set(config.correlationIdHeader, correlationId);
     res.setHeader(config.correlationIdHeader, correlationId);
 }
 function collect(req, headers, headersPrefix) {
@@ -24,14 +31,14 @@ function collect(req, headers, headersPrefix) {
         : [];
     [...headers, ...prefixHeaders].forEach((header) => {
         if (typeof req.headers[header] !== 'undefined') {
-            tracer_1.getCurrentTrace().context.set(header, req.headers[header]);
+            tracer_1.default.currentTrace.context.set(header, req.headers[header]);
         }
     });
 }
 function injectInResponse(response, headers) {
-    if (tracer_1.getCurrentTrace()) {
+    if (tracer_1.default.currentTrace) {
         headers.forEach((header) => {
-            response.setHeader(header, tracer_1.getCurrentTrace().context.get(header));
+            response.setHeader(header, tracer_1.default.currentTrace.context.get(header));
         });
     }
 }
@@ -55,14 +62,14 @@ function wrapHttpCreateServer(config) {
     };
 }
 function inject(options) {
-    if (tracer_1.getCurrentTrace()) {
+    if (tracer_1.default.currentTrace) {
         if (!options.headers) {
             // eslint-disable-next-line no-param-reassign
             options.headers = {};
         }
-        [...(tracer_1.getCurrentTrace().context.keys())].forEach((header) => {
+        [...(tracer_1.default.currentTrace.context.keys())].forEach((header) => {
             // eslint-disable-next-line no-param-reassign
-            options.headers[header] = tracer_1.getCurrentTrace().context.get(header);
+            options.headers[header] = tracer_1.default.currentTrace.context.get(header);
         });
     }
 }
@@ -95,4 +102,4 @@ function wrapHttp(config) {
     http_1.default.get = wrapHttpRequest(originalGet);
 }
 exports.default = wrapHttp;
-//# sourceMappingURL=wrapper.js.map
+//# sourceMappingURL=http_wrapper.js.map

package/dist/src/http_wrapper.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http_wrapper.js","sourceRoot":"","sources":["../../src/http_wrapper.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,4BAA4B;AAC5B,gDAAwB;AACxB,sDAA8B;AAG9B,mDAA4C;AAE5C,MAAM,wBAAwB,GAAG,cAAI,CAAC,YAAY,CAAC;AACnD,MAAM,eAAe,GAAG,cAAI,CAAC,OAAO,CAAC;AACrC,MAAM,WAAW,GAAG,cAAI,CAAC,GAAG,CAAC;AAE7B,SAAS,0BAA0B,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG;IAClD,IAAI,aAAa,GAAG,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;IAC5D,IAAI,OAAO,aAAa,KAAK,WAAW,EAAE;QACxC,aAAa,GAAG,iBAAO,CAAC,QAAQ,EAAE,CAAC;KACpC;IACD,gBAAM,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,mBAAmB,EAAE,aAAa,CAAC,CAAC;IAC3E,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,mBAAmB,EAAE,aAAa,CAAC,CAAC;AAC3D,CAAC;AAED,SAAS,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,aAAa;IAC1C,MAAM,aAAa,GAAG,aAAa;QACjC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QACrE,CAAC,CAAC,EAAE,CAAC;IAEP,CAAC,GAAG,OAAO,EAAE,GAAG,aAAa,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;QAChD,IAAI,OAAO,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,WAAW,EAAE;YAC9C,gBAAM,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,EAAE,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;SAC9D;IACH,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CAAC,QAAQ,EAAE,OAAO;IACzC,IAAI,gBAAM,CAAC,YAAY,EAAE;QACvB,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YACzB,QAAQ,CAAC,SAAS,CAAC,MAAM,EAAE,gBAAM,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;KACJ;AACH,CAAC;AAED,SAAS,eAAe,CAAC,MAAW,EAAE,QAAiC;IACrE,OAAO,CAAC,GAAyB,EAAE,GAAwB,EAAE,IAAS,EAAE,KAAU,EAAQ,EAAE;QAC1F,iBAAQ,CAAC,aAAa,CAAC,CAAC;QAExB,IAAI,MAAM,CAAC,4BAA4B,KAAK,IAAI,EAAE;YAChD,0BAA0B,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;SAC9C;QAED,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,gBAAgB,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;QAE5D,IAAI,MAAM,CAAC,oBAAoB,EAAE;YAC/B,gBAAgB,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,CAAC;SACrD;QAED,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;IAClC,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,MAAM;IAClC,gGAAgG;IAChG,OAAO,SAAS,wBAAwB,CAAC,QAAQ;QAC/C,OAAO,wBAAwB,CAAC,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,MAAM,CAAC,OAAO;IACrB,IAAI,gBAAM,CAAC,YAAY,EAAE;QACvB,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;YACpB,6CAA6C;YAC7C,OAAO,CAAC,OAAO,GAAG,EAAE,CAAC;SACtB;QACD,CAAC,GAAG,CAAC,gBAAM,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;YAC3D,6CAA6C;YAC7C,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,gBAAM,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;KACJ;AACH,CAAC;AAED,SAAS,eAAe,CAAC,cAAc;IACrC,cAAc;IACd,SAAS,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;QAChC,MAAM,CAAC,OAAO,CAAC,CAAC;QAEhB,OAAO,cAAc,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,SAAS,YAAY,CAAC,OAAO,EAAE,EAAE;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC;QAEhB,OAAO,cAAc,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IACrC,CAAC;IAID,SAAS,kBAAkB,CAAC,GAAG,IAAI;QACjC,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,GAAG,CAAC;QACR,IAAI,QAAQ,CAAC;QAEb,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE;YACrB,CAAC,OAAO,EAAE,GAAG,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC;YAChC,OAAO,QAAQ,CAAC,GAAG,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;SACzC;QACD,CAAC,OAAO,EAAE,QAAQ,CAAC,GAAG,IAAI,CAAC;QAC3B,OAAO,YAAY,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACzC,CAAC;IACD,OAAO,kBAAkB,CAAC;AAC5B,CAAC;AAED,SAAwB,QAAQ,CAAC,MAAM;IACrC,cAAI,CAAC,YAAY,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;IACjD,cAAI,CAAC,OAAO,GAAG,eAAe,CAAC,eAAe,CAAC,CAAC;IAChD,cAAI,CAAC,GAAG,GAAG,eAAe,CAAC,WAAW,CAAC,CAAC;AAC1C,CAAC;AAJD,2BAIC"}

package/dist/src/index.d.ts

@@ -1,4 +1,17 @@
-import headersTracer from './http-wrapper';
-import { newTrace, getCurrentTrace } from './tracer';
-import { ApiUser, middleware, middlewareWithDecode, eagerLoadPermissionsMiddleware, getDecodedBearer } from './identity';
-export { headersTracer, getCurrentTrace, newTrace, ApiUser, middleware, middlewareWithDecode, eagerLoadPermissionsMiddleware, getDecodedBearer, };
+import winston from 'winston';
+interface Options {
+    winstonLogger?: winston.Logger;
+    loggerTraceKey?: string;
+    headersPrefix?: string;
+    setAndPropagateCorrelationId?: boolean;
+    headersToPropagate?: string[];
+}
+export default function (options: Options): void;
+export declare const getCurrentContext: () => any;
+export declare const newTrace: (type: any) => void;
+export declare const traceTypes: {
+    HTTP_REQUEST: string;
+    WEB_SOCKET: string;
+    RABBIT: string;
+};
+export {};

package/dist/src/index.js

@@ -2,16 +2,41 @@
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (Object.hasOwnProperty.call(mod, k)) result[k] = mod[k];
+    result["default"] = mod;
+    return result;
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-const http_wrapper_1 = __importDefault(require("./http-wrapper"));
-exports.headersTracer = http_wrapper_1.default;
-const tracer_1 = require("./tracer");
-exports.newTrace = tracer_1.newTrace;
-exports.getCurrentTrace = tracer_1.getCurrentTrace;
-const identity_1 = require("./identity");
-exports.ApiUser = identity_1.ApiUser;
-exports.middleware = identity_1.middleware;
-exports.middlewareWithDecode = identity_1.middlewareWithDecode;
-exports.eagerLoadPermissionsMiddleware = identity_1.eagerLoadPermissionsMiddleware;
-exports.getDecodedBearer = identity_1.getDecodedBearer;
+const winston_1 = require("winston");
+const http_wrapper_1 = __importDefault(require("./http_wrapper"));
+const config_1 = __importDefault(require("./config"));
+const tracer = __importStar(require("./tracer"));
+const addMetadataToLog = (loggerTraceKey) => winston_1.format((info) => {
+    const currentTrace = tracer.getCurrentTrace();
+    if (currentTrace && currentTrace.context && currentTrace.context.get) {
+        // eslint-disable-next-line no-param-reassign
+        info[loggerTraceKey] = currentTrace.context.get(config_1.default.correlationIdHeader);
+    }
+    return info;
+});
+function default_1(options) {
+    http_wrapper_1.default(config_1.default.load(options));
+    tracer.enable();
+    if (options.winstonLogger) {
+        const traceKey = options.loggerTraceKey || 'traceId';
+        // eslint-disable-next-line no-param-reassign
+        options.winstonLogger.format = winston_1.format.combine(addMetadataToLog(traceKey)(), options.winstonLogger.format);
+    }
+}
+exports.default = default_1;
+exports.getCurrentContext = () => tracer.getCurrentTrace();
+exports.newTrace = tracer.newTrace;
+exports.traceTypes = {
+    HTTP_REQUEST: 'httpRequest',
+    WEB_SOCKET: 'webSocket',
+    RABBIT: 'rabbit',
+};
 //# sourceMappingURL=index.js.map

package/dist/src/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;AAAA,kEAA2C;AAWzC,wBAXK,sBAAa,CAWL;AAVf,qCAAqD;AAYnD,mBAZO,iBAAQ,CAYP;AADR,0BAXiB,wBAAe,CAWjB;AAVjB,yCAMoB;AAMlB,kBAXA,kBAAO,CAWA;AACP,qBAXA,qBAAU,CAWA;AACV,+BAXA,+BAAoB,CAWA;AACpB,yCAXA,yCAA8B,CAWA;AAC9B,2BAXA,2BAAgB,CAWA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,qCAA0C;AAE1C,kEAAyC;AACzC,sDAA8B;AAC9B,iDAAmC;AAUnC,MAAM,gBAAgB,GAAG,CAAC,cAAc,EAAE,EAAE,CAAC,gBAAM,CAAC,CAAC,IAAI,EAAE,EAAE;IAC3D,MAAM,YAAY,GAAG,MAAM,CAAC,eAAe,EAAE,CAAC;IAC9C,IAAI,YAAY,IAAI,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE;QACpE,6CAA6C;QAC7C,IAAI,CAAC,cAAc,CAAC,GAAG,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAM,CAAC,mBAAmB,CAAC,CAAC;KAC7E;IAED,OAAO,IAAI,CAAC;AACd,CAAC,CAAC,CAAC;AAEH,mBAAyB,OAAgB;IACvC,sBAAW,CAAC,gBAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAClC,MAAM,CAAC,MAAM,EAAE,CAAC;IAEhB,IAAI,OAAO,CAAC,aAAa,EAAE;QACzB,MAAM,QAAQ,GAAG,OAAO,CAAC,cAAc,IAAI,SAAS,CAAC;QACrD,6CAA6C;QAC7C,OAAO,CAAC,aAAa,CAAC,MAAM,GAAG,gBAAM,CAAC,OAAO,CAC3C,gBAAgB,CAAC,QAAQ,CAAC,EAAE,EAC5B,OAAO,CAAC,aAAa,CAAC,MAAM,CAC7B,CAAC;KACH;AACH,CAAC;AAZD,4BAYC;AAEY,QAAA,iBAAiB,GAAG,GAAQ,EAAE,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC;AAErE,mCAAmC;AAEtB,QAAA,UAAU,GAAG;IACxB,YAAY,EAAE,aAAa;IAC3B,UAAU,EAAE,WAAW;IACvB,MAAM,EAAE,QAAQ;CACjB,CAAC"}

package/dist/src/tracer.d.ts

@@ -0,0 +1,10 @@
+/// <reference types="node" />
+import asyncHooks from 'async_hooks';
+declare const tracer: {
+    currentTrace: any;
+    traces: {};
+};
+export declare const newTrace: (type: any) => void;
+export declare const enable: () => asyncHooks.AsyncHook;
+export declare const getCurrentTrace: () => any;
+export default tracer;

package/dist/src/tracer.js

@@ -0,0 +1,55 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const async_hooks_1 = __importDefault(require("async_hooks"));
+const uuid_1 = __importDefault(require("uuid"));
+const prevStates = {};
+const tracer = {
+    currentTrace: null,
+    traces: {},
+};
+function init(asyncId, type, triggerAsyncId) {
+    if (tracer.traces[triggerAsyncId]) {
+        tracer.traces[asyncId] = tracer.traces[triggerAsyncId];
+    }
+}
+function before(asyncId) {
+    if (!tracer.traces[asyncId]) {
+        return;
+    }
+    prevStates[asyncId] = tracer.currentTrace;
+    tracer.currentTrace = tracer.traces[asyncId];
+}
+function after(asyncId) {
+    if (!tracer.traces[asyncId]) {
+        return;
+    }
+    tracer.currentTrace = prevStates[asyncId];
+}
+function destroy(asyncId) {
+    if (tracer.traces[asyncId]) {
+        delete tracer.traces[asyncId];
+        delete prevStates[asyncId];
+    }
+}
+const hook = async_hooks_1.default.createHook({
+    init, before, after, destroy,
+});
+class Trace {
+    constructor(type) {
+        this.id = uuid_1.default.v1();
+        this.type = type;
+        this.context = new Map();
+    }
+}
+exports.newTrace = (type) => {
+    tracer.currentTrace = new Trace(type);
+    tracer.traces[async_hooks_1.default.executionAsyncId()] = tracer.currentTrace;
+    return tracer.currentTrace;
+};
+exports.enable = () => hook.enable();
+exports.getCurrentTrace = () => tracer.currentTrace || {};
+exports.default = tracer;
+//# sourceMappingURL=tracer.js.map

package/dist/src/tracer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tracer.js","sourceRoot":"","sources":["../../src/tracer.ts"],"names":[],"mappings":";;;;;AAAA,8DAAqC;AACrC,gDAAwB;AAExB,MAAM,UAAU,GAAG,EAAE,CAAC;AAEtB,MAAM,MAAM,GAAG;IACb,YAAY,EAAE,IAAI;IAClB,MAAM,EAAE,EAAE;CACX,CAAC;AAEF,SAAS,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,cAAc;IACzC,IAAI,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,EAAE;QACjC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;KACxD;AACH,CAAC;AAED,SAAS,MAAM,CAAC,OAAO;IACrB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE;QAC3B,OAAO;KACR;IACD,UAAU,CAAC,OAAO,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC;IAC1C,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,KAAK,CAAC,OAAO;IACpB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE;QAC3B,OAAO;KACR;IACD,MAAM,CAAC,YAAY,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;AAC5C,CAAC;AAED,SAAS,OAAO,CAAC,OAAO;IACtB,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE;QAC1B,OAAO,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC9B,OAAO,UAAU,CAAC,OAAO,CAAC,CAAC;KAC5B;AACH,CAAC;AAED,MAAM,IAAI,GAAG,qBAAU,CAAC,UAAU,CAAC;IACjC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO;CAC7B,CAAC,CAAC;AAEH,MAAM,KAAK;IAOT,YAAY,IAAI;QACd,IAAI,CAAC,EAAE,GAAG,cAAI,CAAC,EAAE,EAAE,CAAC;QACpB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,OAAO,GAAG,IAAI,GAAG,EAAE,CAAC;IAC3B,CAAC;CACF;AAEY,QAAA,QAAQ,GAAG,CAAC,IAAI,EAAQ,EAAE;IACrC,MAAM,CAAC,YAAY,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC;IACtC,MAAM,CAAC,MAAM,CAAC,qBAAU,CAAC,gBAAgB,EAAE,CAAC,GAAG,MAAM,CAAC,YAAY,CAAC;IACnE,OAAO,MAAM,CAAC,YAAY,CAAC;AAC7B,CAAC,CAAC;AAEW,QAAA,MAAM,GAAG,GAAyB,EAAE,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;AAEnD,QAAA,eAAe,GAAG,GAAQ,EAAE,CAAC,MAAM,CAAC,YAAY,IAAI,EAAE,CAAC;AAEpE,kBAAe,MAAM,CAAC"}

package/dist/tests/http-wrapper.test.js

@@ -15,9 +15,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const express_1 = __importDefault(require("express"));
 const axios_1 = __importDefault(require("axios"));
 const winston_1 = __importDefault(require("winston"));
-const http_wrapper_1 = __importDefault(require("../src/http-wrapper"));
-const index_1 = require("../src/index");
-console.log(index_1.headersTracer.toString(), index_1.middlewareWithDecode.toString());
+const index_1 = __importDefault(require("../src/index"));
 const logger = winston_1.default.createLogger({
     transports: [
         new winston_1.default.transports.Console(),
@@ -37,7 +35,7 @@ const generateApp = (addEndpoints, port) => __awaiter(void 0, void 0, void 0, fu
 describe('E2E', () => {
     it('Basic functionality', () => __awaiter(void 0, void 0, void 0, function* () {
         let server2TraceId = null;
-        http_wrapper_1.default({
+        index_1.default({
             setAndPropagateCorrelationId: true,
             headersToPropagate: [
                 'x-test-header',
@@ -47,7 +45,7 @@ describe('E2E', () => {
         });
         const closeServer1 = yield generateApp((app) => {
             app.get('/', (req, res) => __awaiter(void 0, void 0, void 0, function* () {
-                const { data: res1 } = yield axios_1.default.post(`http://localhost:8082`);
+                const { data: res1 } = yield axios_1.default.post('http://localhost:8082');
                 res.json(res1);
             }));
         }, 8081);
@@ -56,15 +54,15 @@ describe('E2E', () => {
                 server2TraceId = req.headers['x-trace-id'];
                 res.json({
                     value: req.headers['x-test-header'],
-                    wkanda: req.headers['x-wakanda-id']
+                    wkanda: req.headers['x-wakanda-id'],
                 });
             });
         }, 8082);
-        const { data: res1, headers } = yield axios_1.default.get(`http://localhost:8081`, {
+        const { data: res1, headers } = yield axios_1.default.get('http://localhost:8081', {
             headers: {
                 'x-test-header': 'testHeader',
-                'x-wakanda-id': 'my-wakanda-id'
-            }
+                'x-wakanda-id': 'my-wakanda-id',
+            },
         });
         closeServer1();
         closeServer2();

package/dist/tests/http-wrapper.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"http-wrapper.test.js","sourceRoot":"","sources":["../../tests/http-wrapper.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAAA,sDAA8B;AAE9B,kDAA0B;AAC1B,sDAA8B;AAE9B,uEAAgD;AAChD,wCAA6E;AAE7E,OAAO,CAAC,GAAG,CAAC,qBAAC,CAAC,QAAQ,EAAE,EAAE,4BAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;AAExC,MAAM,MAAM,GAAG,iBAAO,CAAC,YAAY,CAAC;IAClC,UAAU,EAAE;QACV,IAAI,iBAAO,CAAC,UAAU,CAAC,OAAO,EAAE;KACjC;CACF,CAAC,CAAC;AAEH,MAAM,WAAW,GAAG,CAAO,YAAY,EAAE,IAAI,EAAE,EAAE;IAC/C,MAAM,GAAG,GAAG,iBAAO,EAAE,CAAC;IAEtB,YAAY,CAAC,GAAG,CAAC,CAAC;IAElB,MAAM,MAAM,GAAiB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QACzD,MAAM,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;YAC9B,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAA;YACnC,OAAO,CAAC,CAAC,CAAC,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAA;IAEF,OAAO,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;AAC9B,CAAC,CAAA,CAAA;AAED,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;IACnB,EAAE,CAAC,qBAAqB,EAAE,GAAS,EAAE;QACnC,IAAI,cAAc,GAAG,IAAI,CAAC;QAC1B,sBAAa,CAAC;YACZ,4BAA4B,EAAE,IAAI;YAClC,kBAAkB,EAAE;gBAClB,eAAe;aAChB;YACD,aAAa,EAAE,YAAY;YAC3B,aAAa,EAAE,MAAM;SACtB,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE;YAC7C,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAO,GAAG,EAAE,GAAG,EAAE,EAAE;gBAC9B,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,MAAM,eAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;gBACjE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAChB,CAAC,CAAA,CAAC,CAAA;QACJ,CAAC,EAAE,IAAI,CAAC,CAAA;QAER,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE;YAC7C,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;gBACzB,cAAc,GAAG,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC,CAAA;gBAC1C,GAAG,CAAC,IAAI,CAAC;oBACP,KAAK,EAAE,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC;oBACnC,MAAM,EAAE,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC;iBACpC,CAAC,CAAA;YACJ,CAAC,CAAC,CAAA;QACJ,CAAC,EAAE,IAAI,CAAC,CAAA;QAER,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,eAAK,CAAC,GAAG,CAAC,uBAAuB,EAAE;YACvE,OAAO,EAAE;gBACP,eAAe,EAAE,YAAY;gBAC7B,cAAc,EAAE,eAAe;aAChC;SACF,CAAC,CAAC;QACH,YAAY,EAAE,CAAC;QACf,YAAY,EAAE,CAAC;QAGf,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAA;QACrD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAA;QACxC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAA;IAC9C,CAAC,CAAA,CAAC,CAAC;AACL,CAAC,CAAC,CAAA"}
+{"version":3,"file":"http-wrapper.test.js","sourceRoot":"","sources":["../../tests/http-wrapper.test.ts"],"names":[],"mappings":";;;;;;;;;;;;;;AAAA,sDAA8B;AAE9B,kDAA0B;AAC1B,sDAA8B;AAE9B,yDAAyC;AAGzC,MAAM,MAAM,GAAG,iBAAO,CAAC,YAAY,CAAC;IAClC,UAAU,EAAE;QACV,IAAI,iBAAO,CAAC,UAAU,CAAC,OAAO,EAAE;KACjC;CACF,CAAC,CAAC;AAEH,MAAM,WAAW,GAAG,CAAO,YAAY,EAAE,IAAI,EAAE,EAAE;IAC/C,MAAM,GAAG,GAAG,iBAAO,EAAE,CAAC;IAEtB,YAAY,CAAC,GAAG,CAAC,CAAC;IAElB,MAAM,MAAM,GAAgB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;QACxD,MAAM,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;YAC9B,OAAO,CAAC,GAAG,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;YACpC,OAAO,CAAC,CAAC,CAAC,CAAC;QACb,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;AAC9B,CAAC,CAAA,CAAC;AAEF,QAAQ,CAAC,KAAK,EAAE,GAAG,EAAE;IACnB,EAAE,CAAC,qBAAqB,EAAE,GAAS,EAAE;QACnC,IAAI,cAAc,GAAG,IAAI,CAAC;QAC1B,eAAa,CAAC;YACZ,4BAA4B,EAAE,IAAI;YAClC,kBAAkB,EAAE;gBAClB,eAAe;aAChB;YACD,aAAa,EAAE,YAAY;YAC3B,aAAa,EAAE,MAAM;SACtB,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE;YAC7C,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAO,GAAG,EAAE,GAAG,EAAE,EAAE;gBAC9B,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,MAAM,eAAK,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;gBACjE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjB,CAAC,CAAA,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,CAAC;QAET,MAAM,YAAY,GAAG,MAAM,WAAW,CAAC,CAAC,GAAG,EAAE,EAAE;YAC7C,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;gBACzB,cAAc,GAAG,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;gBAC3C,GAAG,CAAC,IAAI,CAAC;oBACP,KAAK,EAAE,GAAG,CAAC,OAAO,CAAC,eAAe,CAAC;oBACnC,MAAM,EAAE,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC;iBACpC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC,EAAE,IAAI,CAAC,CAAC;QAET,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,eAAK,CAAC,GAAG,CAAC,uBAAuB,EAAE;YACvE,OAAO,EAAE;gBACP,eAAe,EAAE,YAAY;gBAC7B,cAAc,EAAE,eAAe;aAChC;SACF,CAAC,CAAC;QACH,YAAY,EAAE,CAAC;QACf,YAAY,EAAE,CAAC;QAGf,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACtD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QACzC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IAC/C,CAAC,CAAA,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autofleet/outbreak",
-  "version": "0.1.0-beta-2",
+  "version": "0.2.0-beta-2",
   "description": "",
   "main": "dist/src/index.js",
   "types": "dist/src/index.d.ts",

package/src/{http-wrapper/config.ts → config.ts}

@@ -30,7 +30,7 @@ const load = (overrides: Overrides = {}) => {
       'x-request-id',
       'x-variant-id',
     ],
-    headersPrefix = 'x-internal-',
+    headersPrefix = 'x-af-',
   } = overrides;
 
   validateHeaderList(headersToPropagate);

package/src/example.ts

@@ -0,0 +1,39 @@
+import express from 'express';
+import axios from 'axios';
+import winston from 'winston';
+
+import headersTracer from './index';
+
+axios.defaults.adapter = require('axios/lib/adapters/http');
+
+const logger = winston.createLogger({
+  transports: [
+    new winston.transports.Console(),
+  ],
+});
+
+headersTracer({
+  setAndPropagateCorrelationId: true,
+  headersToPropagate: [
+    'x-text',
+  ],
+  headersPrefix: 'x-af-',
+  winstonLogger: logger,
+});
+
+const app = express();
+
+app.all('/', async (req, res) => {
+  const network = axios.create({ baseURL: 'http://localhost:8080/' });
+  const { data } = await network.get('/header-parrot');
+  res.json(data);
+});
+
+app.all('/header-parrot', (req, res) => {
+  logger.info('Test message', { t: 1 });
+  res.json(req.headers);
+});
+
+app.listen(8080, () => {
+  logger.info('App listen');
+});

package/src/{http-wrapper/wrapper.ts → http_wrapper.ts}

@@ -3,7 +3,7 @@ import http from 'http';
 import shortid from 'shortid';
 import { URL } from 'url';
 import express from 'express';
-import { newTrace, getCurrentTrace } from '../tracer';
+import tracer, { newTrace } from './tracer';
 
 const originalHttpCreateServer = http.createServer;
 const originalRequest = http.request;
@@ -14,7 +14,7 @@ function setAndCollectCorrelationId(config, req, res): void {
   if (typeof correlationId === 'undefined') {
     correlationId = shortid.generate();
   }
-  getCurrentTrace().context.set(config.correlationIdHeader, correlationId);
+  tracer.currentTrace.context.set(config.correlationIdHeader, correlationId);
   res.setHeader(config.correlationIdHeader, correlationId);
 }
 
@@ -25,15 +25,15 @@ function collect(req, headers, headersPrefix): void {
 
   [...headers, ...prefixHeaders].forEach((header) => {
     if (typeof req.headers[header] !== 'undefined') {
-      getCurrentTrace().context.set(header, req.headers[header]);
+      tracer.currentTrace.context.set(header, req.headers[header]);
     }
   });
 }
 
 function injectInResponse(response, headers): void {
-  if (getCurrentTrace()) {
+  if (tracer.currentTrace) {
     headers.forEach((header) => {
-      response.setHeader(header, getCurrentTrace().context.get(header));
+      response.setHeader(header, tracer.currentTrace.context.get(header));
     });
   }
 }
@@ -64,14 +64,14 @@ function wrapHttpCreateServer(config) {
 }
 
 function inject(options): void {
-  if (getCurrentTrace()) {
+  if (tracer.currentTrace) {
     if (!options.headers) {
       // eslint-disable-next-line no-param-reassign
       options.headers = {};
     }
-    [...(getCurrentTrace().context.keys())].forEach((header) => {
+    [...(tracer.currentTrace.context.keys())].forEach((header) => {
       // eslint-disable-next-line no-param-reassign
-      options.headers[header] = getCurrentTrace().context.get(header);
+      options.headers[header] = tracer.currentTrace.context.get(header);
     });
   }
 }

package/src/index.ts

@@ -1,8 +1,47 @@
-import headersTracer from './http-wrapper';
-import { newTrace, getCurrentTrace } from './tracer';
+import winston, { format } from 'winston';
 
-export {
-  headersTracer,
-  getCurrentTrace,
-  newTrace,
+import httpWrapper from './http_wrapper';
+import config from './config';
+import * as tracer from './tracer';
+
+interface Options {
+    winstonLogger?: winston.Logger;
+    loggerTraceKey? : string;
+    headersPrefix?: string;
+    setAndPropagateCorrelationId?: boolean;
+    headersToPropagate?: string[];
+}
+
+const addMetadataToLog = (loggerTraceKey) => format((info) => {
+  const currentTrace = tracer.getCurrentTrace();
+  if (currentTrace && currentTrace.context && currentTrace.context.get) {
+    // eslint-disable-next-line no-param-reassign
+    info[loggerTraceKey] = currentTrace.context.get(config.correlationIdHeader);
+  }
+
+  return info;
+});
+
+export default function (options: Options): void {
+  httpWrapper(config.load(options));
+  tracer.enable();
+
+  if (options.winstonLogger) {
+    const traceKey = options.loggerTraceKey || 'traceId';
+    // eslint-disable-next-line no-param-reassign
+    options.winstonLogger.format = format.combine(
+      addMetadataToLog(traceKey)(),
+      options.winstonLogger.format,
+    );
+  }
+}
+
+export const getCurrentContext = (): any => tracer.getCurrentTrace();
+
+export const { newTrace } = tracer;
+
+export const traceTypes = {
+  HTTP_REQUEST: 'httpRequest',
+  WEB_SOCKET: 'webSocket',
+  RABBIT: 'rabbit',
 };

package/src/tracer.ts

@@ -0,0 +1,67 @@
+import asyncHooks from 'async_hooks';
+import uuid from 'uuid';
+
+const prevStates = {};
+
+const tracer = {
+  currentTrace: null,
+  traces: {},
+};
+
+function init(asyncId, type, triggerAsyncId): void {
+  if (tracer.traces[triggerAsyncId]) {
+    tracer.traces[asyncId] = tracer.traces[triggerAsyncId];
+  }
+}
+
+function before(asyncId): void {
+  if (!tracer.traces[asyncId]) {
+    return;
+  }
+  prevStates[asyncId] = tracer.currentTrace;
+  tracer.currentTrace = tracer.traces[asyncId];
+}
+
+function after(asyncId): void {
+  if (!tracer.traces[asyncId]) {
+    return;
+  }
+  tracer.currentTrace = prevStates[asyncId];
+}
+
+function destroy(asyncId): void {
+  if (tracer.traces[asyncId]) {
+    delete tracer.traces[asyncId];
+    delete prevStates[asyncId];
+  }
+}
+
+const hook = asyncHooks.createHook({
+  init, before, after, destroy,
+});
+
+class Trace {
+  id: string;
+
+  type: string;
+
+  context: Map<string, string>;
+
+  constructor(type) {
+    this.id = uuid.v1();
+    this.type = type;
+    this.context = new Map();
+  }
+}
+
+export const newTrace = (type): void => {
+  tracer.currentTrace = new Trace(type);
+  tracer.traces[asyncHooks.executionAsyncId()] = tracer.currentTrace;
+  return tracer.currentTrace;
+};
+
+export const enable = (): asyncHooks.AsyncHook => hook.enable();
+
+export const getCurrentTrace = (): any => tracer.currentTrace || {};
+
+export default tracer;