@rsdk/core 4.0.0-next.12 → 4.0.0-next.14

package/src/tracing/utils/create-span.ts

@@ -0,0 +1,20 @@
+import { getRandomBytes } from '@rsdk/common';
+
+/**
+ * Функция для генерации корректного рандомного traceId и spanId
+ * алгоритм взят из: https://github.com/openzipkin/zipkin-js/blob/ec89188cf6a07e184ab886c1dfb6c9dc276ddfa4/packages/zipkin/src/tracer/randomTraceId.js
+ * спецификация по traceId: https://www.w3.org/TR/trace-context/#considerations-for-trace-id-field-generation
+ * @returns {traceId, spanId}
+ */
+export function createSpan(): {
+  traceId: string;
+  spanId: string;
+} {
+  const rootSpanId = getRandomBytes(16);
+  const traceId = getRandomBytes(16) + rootSpanId;
+
+  return {
+    traceId,
+    spanId: rootSpanId,
+  };
+}

package/src/tracing/utils/save-async-hooks-context.ts

@@ -0,0 +1,62 @@
+import { SpanStatusCode } from '@opentelemetry/api';
+import { api } from '@opentelemetry/sdk-node';
+import type { ErrorLike } from '@rsdk/common';
+
+import { TraceInjector } from '../services';
+import { ActiveSpanStorage } from '../services/active-span.storage';
+
+/**
+ * Проблема: при передаче управления в rxjs теряется контекст асинк локал стораджа
+ *
+ * Решение: Утилита для проброса инфы из асинк локал стораджа в тело пайпа rxjs
+ * поиском "observableToAsyncIterable" можно найти тесты в которых есть пример использования
+ *
+ * Тут есть часть логик трейсер декоратора, суть простая:
+ * мы - находясь в контексте с актуальным асинк Metadata запускаем создание боди для пайпа rx и тем самым мы передаем контекст работы туда
+ * ```
+ * return this.booksStream.pipe(
+ *       concatMap(
+ *         saveAsyncHooksContext(async (book) => {
+ *           return { book: { ...book, pages: request.limit } };
+ *         }),
+ *       ),
+ *     );
+ * ```
+ */
+export function saveAsyncHooksContext<T>(
+  fn: (...args: any[]) => Promise<T>,
+): (...args: any[]) => Promise<T> {
+  const span = ActiveSpanStorage.getInstance()?.getActiveSpan();
+
+  return async function (...args): Promise<T> {
+    const activeSpan = ActiveSpanStorage.getInstance()?.getActiveSpan();
+    if (activeSpan) {
+      Object.assign(activeSpan.spanContext(), span?.spanContext());
+      return fn(...args);
+    } else {
+      if (!span) {
+        return fn(...args);
+      }
+      const spanContext = api.trace.setSpan(api.context.active(), span);
+
+      return api.context.with(spanContext, async () =>
+        fn(...args)
+          .then(async (result) => {
+            span.setAttribute('response', TraceInjector.toAttribute(result));
+            span.setStatus({ code: SpanStatusCode.OK });
+            span.end();
+            return result;
+          })
+          .catch(async (error) => {
+            span.recordException(error as ErrorLike);
+            span.setStatus({
+              code: SpanStatusCode.ERROR,
+              message: (error as ErrorLike).message,
+            });
+            span.end();
+            throw error;
+          }),
+      );
+    }
+  };
+}

package/dist/tracing/services/metadata.scanner.d.ts

@@ -1,11 +0,0 @@
-import type { Controller, Injectable as Provider } from '@nestjs/common/interfaces';
-import { ModulesContainer } from '@nestjs/core';
-import type { InstanceWrapper } from '@nestjs/core/injector/instance-wrapper';
-export declare class ExtendedMetadataScanner {
-    private readonly modules;
-    private readonly scanner;
-    constructor(modules: ModulesContainer);
-    getControllers(): Generator<InstanceWrapper<Controller>>;
-    getProviders(): Generator<InstanceWrapper<Provider>>;
-    getMethods(prototype: any): Generator<string>;
-}

package/dist/tracing/services/metadata.scanner.js

@@ -1,50 +0,0 @@
-"use strict";
-var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
-    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
-    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
-    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
-    return c > 3 && r && Object.defineProperty(target, key, r), r;
-};
-var __metadata = (this && this.__metadata) || function (k, v) {
-    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.ExtendedMetadataScanner = void 0;
-const common_1 = require("@nestjs/common");
-const core_1 = require("@nestjs/core");
-let ExtendedMetadataScanner = class ExtendedMetadataScanner {
-    modules;
-    scanner = new core_1.MetadataScanner();
-    constructor(modules) {
-        this.modules = modules;
-    }
-    *getControllers() {
-        for (const module of this.modules.values()) {
-            for (const controller of module.controllers.values()) {
-                if (controller && controller.metatype?.prototype) {
-                    yield controller;
-                }
-            }
-        }
-    }
-    *getProviders() {
-        for (const module of this.modules.values()) {
-            for (const provider of module.providers.values()) {
-                if (provider && provider.metatype?.prototype) {
-                    yield provider;
-                }
-            }
-        }
-    }
-    *getMethods(prototype) {
-        for (const name of this.scanner.getAllFilteredMethodNames(prototype)) {
-            yield name;
-        }
-    }
-};
-exports.ExtendedMetadataScanner = ExtendedMetadataScanner;
-exports.ExtendedMetadataScanner = ExtendedMetadataScanner = __decorate([
-    (0, common_1.Injectable)(),
-    __metadata("design:paramtypes", [core_1.ModulesContainer])
-], ExtendedMetadataScanner);
-//# sourceMappingURL=metadata.scanner.js.map

package/dist/tracing/services/metadata.scanner.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"metadata.scanner.js","sourceRoot":"","sources":["../../../src/tracing/services/metadata.scanner.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAK5C,uCAAiE;AAI1D,IAAM,uBAAuB,GAA7B,MAAM,uBAAuB;IAGL;IAFZ,OAAO,GAAG,IAAI,sBAAe,EAAE,CAAC;IAEjD,YAA6B,OAAyB;QAAzB,YAAO,GAAP,OAAO,CAAkB;IAAG,CAAC;IAE1D,CAAC,cAAc;QACb,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE;YAC1C,KAAK,MAAM,UAAU,IAAI,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,EAAE;gBACpD,IAAI,UAAU,IAAI,UAAU,CAAC,QAAQ,EAAE,SAAS,EAAE;oBAChD,MAAM,UAAU,CAAC;iBAClB;aACF;SACF;IACH,CAAC;IAED,CAAC,YAAY;QACX,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE;YAC1C,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE;gBAChD,IAAI,QAAQ,IAAI,QAAQ,CAAC,QAAQ,EAAE,SAAS,EAAE;oBAC5C,MAAM,QAAQ,CAAC;iBAChB;aACF;SACF;IACH,CAAC;IAED,CAAC,UAAU,CAAC,SAAc;QACxB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,OAAO,CAAC,yBAAyB,CAAC,SAAS,CAAC,EAAE;YACpE,MAAM,IAAI,CAAC;SACZ;IACH,CAAC;CACF,CAAA;AA9BY,0DAAuB;kCAAvB,uBAAuB;IADnC,IAAA,mBAAU,GAAE;qCAI2B,uBAAgB;GAH3C,uBAAuB,CA8BnC"}

package/src/tracing/services/metadata.scanner.ts

@@ -1,40 +0,0 @@
-import { Injectable } from '@nestjs/common';
-import type {
-  Controller,
-  Injectable as Provider,
-} from '@nestjs/common/interfaces';
-import { MetadataScanner, ModulesContainer } from '@nestjs/core';
-import type { InstanceWrapper } from '@nestjs/core/injector/instance-wrapper';
-
-@Injectable()
-export class ExtendedMetadataScanner {
-  private readonly scanner = new MetadataScanner();
-
-  constructor(private readonly modules: ModulesContainer) {}
-
-  *getControllers(): Generator<InstanceWrapper<Controller>> {
-    for (const module of this.modules.values()) {
-      for (const controller of module.controllers.values()) {
-        if (controller && controller.metatype?.prototype) {
-          yield controller;
-        }
-      }
-    }
-  }
-
-  *getProviders(): Generator<InstanceWrapper<Provider>> {
-    for (const module of this.modules.values()) {
-      for (const provider of module.providers.values()) {
-        if (provider && provider.metatype?.prototype) {
-          yield provider;
-        }
-      }
-    }
-  }
-
-  *getMethods(prototype: any): Generator<string> {
-    for (const name of this.scanner.getAllFilteredMethodNames(prototype)) {
-      yield name;
-    }
-  }
-}