@rsdk/nats.producer.direct 5.3.0-next.4

package/CHANGELOG.md

@@ -0,0 +1,10 @@
+# Change Log
+
+All notable changes to this project will be documented in this file.
+See [Conventional Commits](https://conventionalcommits.org) for commit guidelines.
+
+## [5.3.0-next.4](https://github.com/R-Vision/rsdk/compare/v5.3.0-next.3...v5.3.0-next.4) (2024-11-05)
+
+### Features
+
+* **nats.producer:** direct nats producer alpha implementation ([#301](https://github.com/R-Vision/rsdk/issues/301)) ([d173b51](https://github.com/R-Vision/rsdk/commit/d173b51c664c718a214cdc695708cfdd40fc4dcb))

package/dist/direct/direct.autodoc.d.ts

@@ -0,0 +1,7 @@
+import type { DocumentResolver } from '@rsdk/autodoc.protocol';
+import { Composite } from '@rsdk/autodoc.protocol';
+import type { RsdkMetadataProvider } from '@rsdk/metadata';
+export declare class NatsProducerDirectAutodocResolver implements DocumentResolver {
+    getNode(rsdkMetadataProvider: RsdkMetadataProvider): Composite | null;
+}
+export declare const natsProducerDirectAutodocResolver: NatsProducerDirectAutodocResolver;

package/dist/direct/direct.autodoc.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.natsProducerDirectAutodocResolver = exports.NatsProducerDirectAutodocResolver = void 0;
+const autodoc_protocol_1 = require("@rsdk/autodoc.protocol");
+const nats_common_1 = require("@rsdk/nats.common");
+const direct_constants_1 = require("./direct.constants");
+class NatsProducerDirectAutodocResolver {
+    getNode(rsdkMetadataProvider) {
+        const resources = rsdkMetadataProvider.get(direct_constants_1.NATS_PRODUCED_RSDK_METADATA_SCOPE);
+        if (!resources) {
+            return null;
+        }
+        return new autodoc_protocol_1.Composite(new autodoc_protocol_1.Table(resources
+            .filter((resource) => resource.value.type === nats_common_1.TopicType.PRODUCE)
+            .map((resource) => {
+            const { topicName, group: group, partitionKeyField, eventType, } = resource.value;
+            return {
+                TopicName: topicName,
+                Group: group ?? '-',
+                PartitionKeyField: partitionKeyField ?? '-',
+                EventType: eventType?.$type ?? '-',
+            };
+        })), 'Nats JetStream producer direct');
+    }
+}
+exports.NatsProducerDirectAutodocResolver = NatsProducerDirectAutodocResolver;
+exports.natsProducerDirectAutodocResolver = new NatsProducerDirectAutodocResolver();
+//# sourceMappingURL=direct.autodoc.js.map

package/dist/direct/direct.autodoc.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct.autodoc.js","sourceRoot":"","sources":["../../src/direct/direct.autodoc.ts"],"names":[],"mappings":";;;AACA,6DAA0D;AAM1D,mDAA8C;AAE9C,yDAAuE;AAEvE,MAAa,iCAAiC;IAC5C,OAAO,CAAC,oBAA0C;QAChD,MAAM,SAAS,GAAG,oBAAoB,CAAC,GAAG,CACxC,oDAAiC,CAClC,CAAC;QAEF,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,IAAI,4BAAS,CAClB,IAAI,wBAAK,CACP,SAAS;aACN,MAAM,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,uBAAS,CAAC,OAAO,CAAC;aAC/D,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YAChB,MAAM,EACJ,SAAS,EACT,KAAK,EAAE,KAAK,EACZ,iBAAiB,EACjB,SAAS,GACV,GAAiC,QAAQ,CAAC,KAAK,CAAC;YAEjD,OAAO;gBACL,SAAS,EAAE,SAAS;gBACpB,KAAK,EAAE,KAAK,IAAI,GAAG;gBACnB,iBAAiB,EAAE,iBAAiB,IAAI,GAAG;gBAC3C,SAAS,EAAE,SAAS,EAAE,KAAK,IAAI,GAAG;aACnC,CAAC;QACJ,CAAC,CAAC,CACL,EACD,gCAAgC,CACjC,CAAC;IACJ,CAAC;CACF;AAjCD,8EAiCC;AAEY,QAAA,iCAAiC,GAC5C,IAAI,iCAAiC,EAAE,CAAC"}

package/dist/direct/direct.constants.d.ts

@@ -0,0 +1,3 @@
+import type { EventType } from '@rsdk/nats.common';
+export declare const NATS_PRODUCED_RSDK_METADATA_SCOPE = "nats-producer-direct";
+export declare const createDirectProducerToken: (eventType: EventType<any>) => string;

package/dist/direct/direct.constants.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createDirectProducerToken = exports.NATS_PRODUCED_RSDK_METADATA_SCOPE = void 0;
+exports.NATS_PRODUCED_RSDK_METADATA_SCOPE = 'nats-producer-direct';
+const createDirectProducerToken = (eventType) => `DIRECT_NATS_PRODUCER_${eventType.$type}`;
+exports.createDirectProducerToken = createDirectProducerToken;
+//# sourceMappingURL=direct.constants.js.map

package/dist/direct/direct.constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct.constants.js","sourceRoot":"","sources":["../../src/direct/direct.constants.ts"],"names":[],"mappings":";;;AAEa,QAAA,iCAAiC,GAAG,sBAAsB,CAAC;AAEjE,MAAM,yBAAyB,GAAG,CAAC,SAAyB,EAAU,EAAE,CAC7E,wBAAwB,SAAS,CAAC,KAAK,EAAE,CAAC;AAD/B,QAAA,yBAAyB,6BACM"}

package/dist/direct/direct.decorator.d.ts

@@ -0,0 +1,2 @@
+import type { EventType } from '@rsdk/nats.common';
+export declare const InjectDirectProducer: (eventType: EventType) => ParameterDecorator;

package/dist/direct/direct.decorator.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InjectDirectProducer = void 0;
+const common_1 = require("@nestjs/common");
+const metadata_1 = require("@rsdk/metadata");
+const direct_constants_1 = require("./direct.constants");
+const direct_module_generator_1 = require("./direct.module-generator");
+const InjectDirectProducer = (eventType) => {
+    return (target, propertyKey, parameterIndex) => {
+        const directProducerToken = (0, direct_constants_1.createDirectProducerToken)(eventType);
+        (0, common_1.Inject)(directProducerToken)(target, propertyKey, parameterIndex);
+        metadata_1.RsdkMetadata.setWithScope(target, metadata_1.PLATFORM_RAW_GLOBAL_METADATA_SCOPE, directProducerToken, direct_module_generator_1.DirectModuleGenerator.createModule(eventType), true);
+    };
+};
+exports.InjectDirectProducer = InjectDirectProducer;
+//# sourceMappingURL=direct.decorator.js.map

package/dist/direct/direct.decorator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct.decorator.js","sourceRoot":"","sources":["../../src/direct/direct.decorator.ts"],"names":[],"mappings":";;;AAAA,2CAAwC;AACxC,6CAGwB;AAGxB,yDAA+D;AAC/D,uEAAkE;AAE3D,MAAM,oBAAoB,GAAG,CAClC,SAAoB,EACA,EAAE;IACtB,OAAO,CAAC,MAAM,EAAE,WAAW,EAAE,cAAc,EAAE,EAAE;QAC7C,MAAM,mBAAmB,GAAG,IAAA,4CAAyB,EAAC,SAAS,CAAC,CAAC;QAEjE,IAAA,eAAM,EAAC,mBAAmB,CAAC,CAAC,MAAM,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;QACjE,uBAAY,CAAC,YAAY,CACvB,MAAM,EACN,6CAAkC,EAClC,mBAAmB,EACnB,+CAAqB,CAAC,YAAY,CAAC,SAAS,CAAC,EAC7C,IAAI,CACL,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC,CAAC;AAfW,QAAA,oBAAoB,wBAe/B"}

package/dist/direct/direct.event-factory.d.ts

@@ -0,0 +1,10 @@
+import type { EventPayload, EventType } from '@rsdk/nats.common';
+import { EventsFactory } from '@rsdk/nats.producer';
+import type { MsgHdrs, Payload } from 'nats';
+export declare class EventsFactoryDirect<E extends EventType> extends EventsFactory<E, Payload> {
+    create<P extends EventPayload<E>>(payload: P): {
+        headers: MsgHdrs;
+        payload: Payload;
+    };
+    private buildHeaders;
+}

package/dist/direct/direct.event-factory.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EventsFactoryDirect = void 0;
+const nats_producer_1 = require("@rsdk/nats.producer");
+const nats_1 = require("nats");
+class EventsFactoryDirect extends nats_producer_1.EventsFactory {
+    create(payload) {
+        const metadata = this.getMetadata(this.eventType.$type, null);
+        return {
+            headers: this.buildHeaders(metadata),
+            payload: this.serializePayload(payload),
+        };
+    }
+    buildHeaders(metadata) {
+        const headers = new nats_1.MsgHdrsImpl();
+        for (const [key, value] of Object.entries(metadata)) {
+            headers.set(key, value);
+        }
+        return headers;
+    }
+}
+exports.EventsFactoryDirect = EventsFactoryDirect;
+//# sourceMappingURL=direct.event-factory.js.map

package/dist/direct/direct.event-factory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct.event-factory.js","sourceRoot":"","sources":["../../src/direct/direct.event-factory.ts"],"names":[],"mappings":";;;AACA,uDAAoD;AAEpD,+BAAmC;AAEnC,MAAa,mBAAyC,SAAQ,6BAG7D;IACC,MAAM,CACJ,OAAU;QAEV,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAE9D,OAAO;YACL,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC;YACpC,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC;SACxC,CAAC;IACJ,CAAC;IAEO,YAAY,CAAC,QAAgC;QACnD,MAAM,OAAO,GAAG,IAAI,kBAAW,EAAE,CAAC;QAElC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpD,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAC1B,CAAC;QAED,OAAO,OAAO,CAAC;IACjB,CAAC;CACF;AAxBD,kDAwBC"}

package/dist/direct/direct.module-generator.d.ts

@@ -0,0 +1,5 @@
+import type { DynamicModule } from '@nestjs/common';
+import type { EventType } from '@rsdk/nats.common';
+export declare class DirectModuleGenerator {
+    static createModule(eventType: EventType): DynamicModule;
+}

package/dist/direct/direct.module-generator.js

@@ -0,0 +1,51 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DirectModuleGenerator = void 0;
+const autodoc_protocol_1 = require("@rsdk/autodoc.protocol");
+const core_1 = require("@rsdk/core");
+const metadata_1 = require("@rsdk/metadata");
+const nats_common_1 = require("@rsdk/nats.common");
+const nats_producer_1 = require("@rsdk/nats.producer");
+const omit_1 = __importDefault(require("lodash/omit"));
+const direct_autodoc_1 = require("./direct.autodoc");
+const direct_constants_1 = require("./direct.constants");
+const direct_module_1 = require("./direct.module");
+const direct_producer_1 = require("./direct.producer");
+class DirectModuleGenerator {
+    static createModule(eventType) {
+        const providers = [
+            {
+                inject: [nats_common_1.NATS_PRODUCER_INJECTION_TOKEN, nats_producer_1.ProducerConfig],
+                provide: (0, direct_constants_1.createDirectProducerToken)(eventType),
+                useFactory: (client, config) => {
+                    return new direct_producer_1.DirectProducer(client, eventType, config);
+                },
+            },
+        ];
+        const moduleDef = {
+            imports: [
+                nats_common_1.NatsClientModule,
+                direct_module_1.DirectNatsProducerModule,
+                core_1.PlatformConfigModule.forFeature(nats_producer_1.ProducerConfig),
+            ],
+            providers,
+            module: DirectModuleGenerator,
+            exports: providers,
+        };
+        const rsdkNatsMetadata = new metadata_1.RsdkMetadata(moduleDef, nats_common_1.NATS_TOPIC_RSDK_METADATA_SCOPE);
+        rsdkNatsMetadata.add({
+            type: nats_common_1.TopicType.PRODUCE,
+            topicName: (0, nats_common_1.getSubjectName)(eventType),
+            group: eventType.$group,
+            partitionKeyField: eventType.$partitionKeyField,
+            eventType: (0, omit_1.default)(eventType, 'toJSON', 'encode', 'decode'),
+        });
+        autodoc_protocol_1.AutodocMetadata.defineResolver(moduleDef, nats_common_1.NATS_TOPIC_RSDK_METADATA_SCOPE, direct_autodoc_1.natsProducerDirectAutodocResolver);
+        return moduleDef;
+    }
+}
+exports.DirectModuleGenerator = DirectModuleGenerator;
+//# sourceMappingURL=direct.module-generator.js.map

package/dist/direct/direct.module-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct.module-generator.js","sourceRoot":"","sources":["../../src/direct/direct.module-generator.ts"],"names":[],"mappings":";;;;;;AACA,6DAAyD;AACzD,qCAAkD;AAClD,6CAA8C;AAE9C,mDAM2B;AAC3B,uDAAqD;AACrD,uDAA+B;AAE/B,qDAAqE;AACrE,yDAA+D;AAC/D,mDAA2D;AAC3D,uDAAmD;AAEnD,MAAa,qBAAqB;IAChC,MAAM,CAAC,YAAY,CAAC,SAAoB;QACtC,MAAM,SAAS,GAAsB;YACnC;gBACE,MAAM,EAAE,CAAC,2CAA6B,EAAE,8BAAc,CAAC;gBACvD,OAAO,EAAE,IAAA,4CAAyB,EAAC,SAAS,CAAC;gBAC7C,UAAU,EAAE,CACV,MAAgB,EAChB,MAAsB,EACK,EAAE;oBAC7B,OAAO,IAAI,gCAAc,CAAC,MAAM,EAAE,SAAS,EAAE,MAAM,CAAC,CAAC;gBACvD,CAAC;aACF;SACF,CAAC;QAEF,MAAM,SAAS,GAAG;YAChB,OAAO,EAAE;gBACP,8BAAgB;gBAChB,wCAAwB;gBACxB,2BAAoB,CAAC,UAAU,CAAC,8BAAc,CAAC;aAChD;YACD,SAAS;YACT,MAAM,EAAE,qBAAqB;YAC7B,OAAO,EAAE,SAAS;SACnB,CAAC;QAEF,MAAM,gBAAgB,GAAG,IAAI,uBAAY,CACvC,SAAS,EACT,4CAA8B,CAC/B,CAAC;QAEF,gBAAgB,CAAC,GAAG,CAAC;YACnB,IAAI,EAAE,uBAAS,CAAC,OAAO;YACvB,SAAS,EAAE,IAAA,4BAAc,EAAC,SAAS,CAAC;YACpC,KAAK,EAAE,SAAS,CAAC,MAAM;YACvB,iBAAiB,EAAE,SAAS,CAAC,kBAAkB;YAC/C,SAAS,EAAE,IAAA,cAAI,EAAC,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC;SACzD,CAAC,CAAC;QAEH,kCAAe,CAAC,cAAc,CAC5B,SAAS,EACT,4CAA8B,EAC9B,kDAAiC,CAClC,CAAC;QAEF,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AA/CD,sDA+CC"}

package/dist/direct/direct.module.d.ts

@@ -0,0 +1,2 @@
+export declare class DirectNatsProducerModule {
+}

package/dist/direct/direct.module.js

@@ -0,0 +1,22 @@
+"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;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DirectNatsProducerModule = void 0;
+const common_1 = require("@nestjs/common");
+const nats_common_1 = require("@rsdk/nats.common");
+// import { DirectProducerIndicator } from './kafka-producer-direct.health-indicator';
+let DirectNatsProducerModule = class DirectNatsProducerModule {
+};
+exports.DirectNatsProducerModule = DirectNatsProducerModule;
+exports.DirectNatsProducerModule = DirectNatsProducerModule = __decorate([
+    (0, common_1.Module)({
+        imports: [nats_common_1.NatsClientModule],
+        // providers: [DirectProducerIndicator],
+    })
+], DirectNatsProducerModule);
+//# sourceMappingURL=direct.module.js.map

package/dist/direct/direct.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct.module.js","sourceRoot":"","sources":["../../src/direct/direct.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAwC;AACxC,mDAAqD;AAErD,sFAAsF;AAM/E,IAAM,wBAAwB,GAA9B,MAAM,wBAAwB;CAAG,CAAA;AAA3B,4DAAwB;mCAAxB,wBAAwB;IAJpC,IAAA,eAAM,EAAC;QACN,OAAO,EAAE,CAAC,8BAAgB,CAAC;QAC3B,wCAAwC;KACzC,CAAC;GACW,wBAAwB,CAAG"}

package/dist/direct/direct.producer.d.ts

@@ -0,0 +1,13 @@
+import type { EventPayload, EventType, Producer } from '@rsdk/nats.common';
+import type { NatsProducer, ProducerOptions, PublishOptions } from '@rsdk/nats.producer';
+/**
+ * Producer, специализированный под конкретный ивент.
+ */
+export declare class DirectProducer<T extends EventType> implements NatsProducer<T> {
+    private readonly client;
+    private readonly eventType;
+    private readonly factory;
+    private readonly logger;
+    constructor(client: Producer, eventType: T, config: ProducerOptions);
+    publish<P extends EventPayload<T>>(payload: P, options?: PublishOptions): Promise<void>;
+}

package/dist/direct/direct.producer.js

@@ -0,0 +1,40 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DirectProducer = void 0;
+const logging_1 = require("@rsdk/logging");
+const nats_common_1 = require("@rsdk/nats.common");
+const direct_event_factory_1 = require("./direct.event-factory");
+/**
+ * Producer, специализированный под конкретный ивент.
+ */
+class DirectProducer {
+    client;
+    eventType;
+    factory;
+    logger = logging_1.LoggerFactory.create(DirectProducer.name);
+    constructor(client, eventType, config) {
+        this.client = client;
+        this.eventType = eventType;
+        this.client = client;
+        this.eventType = eventType;
+        this.factory = new direct_event_factory_1.EventsFactoryDirect(eventType, config);
+    }
+    async publish(payload, options) {
+        const message = this.factory.create(payload);
+        const partitionKey = this.factory.getPartitionKey(payload, options);
+        if (!partitionKey) {
+            this.logger.warn('Partition key field is not found in the payload. It may lead to performance issues', { options });
+        }
+        const subject = (0, nats_common_1.getSubjectName)(this.eventType, partitionKey);
+        try {
+            await this.client.publish(subject, message.payload, {
+                headers: message.headers,
+            });
+        }
+        catch (error) {
+            (0, nats_common_1.handleNatsException)(error, subject, options);
+        }
+    }
+}
+exports.DirectProducer = DirectProducer;
+//# sourceMappingURL=direct.producer.js.map

package/dist/direct/direct.producer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct.producer.js","sourceRoot":"","sources":["../../src/direct/direct.producer.ts"],"names":[],"mappings":";;;AACA,2CAA8C;AAE9C,mDAAwE;AAOxE,iEAA6D;AAE7D;;GAEG;AACH,MAAa,cAAc;IAKN;IACA;IALF,OAAO,CAAyB;IAChC,MAAM,GAAY,uBAAa,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IAE7E,YACmB,MAAgB,EAChB,SAAY,EAC7B,MAAuB;QAFN,WAAM,GAAN,MAAM,CAAU;QAChB,cAAS,GAAT,SAAS,CAAG;QAG7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,IAAI,0CAAmB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,OAAO,CACX,OAAU,EACV,OAAwB;QAExB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC7C,MAAM,YAAY,GAAG,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACpE,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,oFAAoF,EACpF,EAAE,OAAO,EAAE,CACZ,CAAC;QACJ,CAAC;QACD,MAAM,OAAO,GAAG,IAAA,4BAAc,EAAC,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;QAE7D,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE;gBAClD,OAAO,EAAE,OAAO,CAAC,OAAO;aACzB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAA,iCAAmB,EAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;CACF;AApCD,wCAoCC"}

package/dist/direct/index.d.ts

@@ -0,0 +1,2 @@
+export { DirectProducer } from './direct.producer';
+export { InjectDirectProducer } from './direct.decorator';

package/dist/direct/index.js

@@ -0,0 +1,8 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InjectDirectProducer = exports.DirectProducer = void 0;
+var direct_producer_1 = require("./direct.producer");
+Object.defineProperty(exports, "DirectProducer", { enumerable: true, get: function () { return direct_producer_1.DirectProducer; } });
+var direct_decorator_1 = require("./direct.decorator");
+Object.defineProperty(exports, "InjectDirectProducer", { enumerable: true, get: function () { return direct_decorator_1.InjectDirectProducer; } });
+//# sourceMappingURL=index.js.map

package/dist/direct/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/direct/index.ts"],"names":[],"mappings":";;;AAAA,qDAAmD;AAA1C,iHAAA,cAAc,OAAA;AACvB,uDAA0D;AAAjD,wHAAA,oBAAoB,OAAA"}

package/dist/index.d.ts

@@ -0,0 +1 @@
+export { DirectProducer, InjectDirectProducer } from './direct';

package/dist/index.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InjectDirectProducer = exports.DirectProducer = void 0;
+var direct_1 = require("./direct");
+Object.defineProperty(exports, "DirectProducer", { enumerable: true, get: function () { return direct_1.DirectProducer; } });
+Object.defineProperty(exports, "InjectDirectProducer", { enumerable: true, get: function () { return direct_1.InjectDirectProducer; } });
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;AAAA,mCAAgE;AAAvD,wGAAA,cAAc,OAAA;AAAE,8GAAA,oBAAoB,OAAA"}

package/jest.config.js

@@ -0,0 +1 @@
+module.exports = require('@rsdk/jest/jest.config');

package/jest.config.unit.js

@@ -0,0 +1 @@
+module.exports = require('@rsdk/jest/jest.config.unit');

package/package.json

@@ -0,0 +1,33 @@
+{
+  "name": "@rsdk/nats.producer.direct",
+  "version": "5.3.0-next.4",
+  "description": "Publishing messages directly to nats jetstream",
+  "license": "Apache License 2.0",
+  "publishConfig": {
+    "access": "public"
+  },
+  "repository": {
+    "url": "https://github.com/R-Vision/rsdk"
+  },
+  "main": "dist/index.js",
+  "peerDependencies": {
+    "@grpc/grpc-js": "^1.10.9",
+    "@nestjs/common": "^10.0.0",
+    "@nestjs/core": "^10.0.0",
+    "@nestjs/microservices": "^10.0.0",
+    "@rsdk/autodoc.protocol": "*",
+    "@rsdk/common": "*",
+    "@rsdk/core": "*",
+    "@rsdk/logging": "*",
+    "@rsdk/metadata": "*",
+    "@rsdk/nats.common": "*",
+    "@rsdk/nats.producer": "*",
+    "nats": "^2.28.2",
+    "reflect-metadata": "^0.1.12 || ^0.2.0",
+    "rxjs": "^7.8.1"
+  },
+  "dependencies": {
+    "lodash": "^4.17.21"
+  },
+  "gitHead": "6d80aacd87b9cc36cfcff8c0c050e4fe31810276"
+}

package/src/direct/direct.autodoc.ts

@@ -0,0 +1,48 @@
+import type { DocumentResolver } from '@rsdk/autodoc.protocol';
+import { Composite, Table } from '@rsdk/autodoc.protocol';
+import type { RsdkMetadataProvider } from '@rsdk/metadata';
+import type {
+  NatsTopicMetadata,
+  NatsTopicMetadataByEventType,
+} from '@rsdk/nats.common';
+import { TopicType } from '@rsdk/nats.common';
+
+import { NATS_PRODUCED_RSDK_METADATA_SCOPE } from './direct.constants';
+
+export class NatsProducerDirectAutodocResolver implements DocumentResolver {
+  getNode(rsdkMetadataProvider: RsdkMetadataProvider): Composite | null {
+    const resources = rsdkMetadataProvider.get<NatsTopicMetadata>(
+      NATS_PRODUCED_RSDK_METADATA_SCOPE,
+    );
+
+    if (!resources) {
+      return null;
+    }
+
+    return new Composite(
+      new Table(
+        resources
+          .filter((resource) => resource.value.type === TopicType.PRODUCE)
+          .map((resource) => {
+            const {
+              topicName,
+              group: group,
+              partitionKeyField,
+              eventType,
+            } = <NatsTopicMetadataByEventType>resource.value;
+
+            return {
+              TopicName: topicName,
+              Group: group ?? '-',
+              PartitionKeyField: partitionKeyField ?? '-',
+              EventType: eventType?.$type ?? '-',
+            };
+          }),
+      ),
+      'Nats JetStream producer direct',
+    );
+  }
+}
+
+export const natsProducerDirectAutodocResolver =
+  new NatsProducerDirectAutodocResolver();

package/src/direct/direct.constants.ts

@@ -0,0 +1,6 @@
+import type { EventType } from '@rsdk/nats.common';
+
+export const NATS_PRODUCED_RSDK_METADATA_SCOPE = 'nats-producer-direct';
+
+export const createDirectProducerToken = (eventType: EventType<any>): string =>
+  `DIRECT_NATS_PRODUCER_${eventType.$type}`;

package/src/direct/direct.decorator.ts

@@ -0,0 +1,26 @@
+import { Inject } from '@nestjs/common';
+import {
+  PLATFORM_RAW_GLOBAL_METADATA_SCOPE,
+  RsdkMetadata,
+} from '@rsdk/metadata';
+import type { EventType } from '@rsdk/nats.common';
+
+import { createDirectProducerToken } from './direct.constants';
+import { DirectModuleGenerator } from './direct.module-generator';
+
+export const InjectDirectProducer = (
+  eventType: EventType,
+): ParameterDecorator => {
+  return (target, propertyKey, parameterIndex) => {
+    const directProducerToken = createDirectProducerToken(eventType);
+
+    Inject(directProducerToken)(target, propertyKey, parameterIndex);
+    RsdkMetadata.setWithScope(
+      target,
+      PLATFORM_RAW_GLOBAL_METADATA_SCOPE,
+      directProducerToken,
+      DirectModuleGenerator.createModule(eventType),
+      true,
+    );
+  };
+};

package/src/direct/direct.event-factory.ts

@@ -0,0 +1,30 @@
+import type { EventPayload, EventType } from '@rsdk/nats.common';
+import { EventsFactory } from '@rsdk/nats.producer';
+import type { MsgHdrs, Payload } from 'nats';
+import { MsgHdrsImpl } from 'nats';
+
+export class EventsFactoryDirect<E extends EventType> extends EventsFactory<
+  E,
+  Payload
+> {
+  create<P extends EventPayload<E>>(
+    payload: P,
+  ): { headers: MsgHdrs; payload: Payload } {
+    const metadata = this.getMetadata(this.eventType.$type, null);
+
+    return {
+      headers: this.buildHeaders(metadata),
+      payload: this.serializePayload(payload),
+    };
+  }
+
+  private buildHeaders(metadata: Record<string, string>): MsgHdrs {
+    const headers = new MsgHdrsImpl();
+
+    for (const [key, value] of Object.entries(metadata)) {
+      headers.set(key, value);
+    }
+
+    return headers;
+  }
+}

package/src/direct/direct.module-generator.ts

@@ -0,0 +1,68 @@
+import type { DynamicModule, FactoryProvider } from '@nestjs/common';
+import { AutodocMetadata } from '@rsdk/autodoc.protocol';
+import { PlatformConfigModule } from '@rsdk/core';
+import { RsdkMetadata } from '@rsdk/metadata';
+import type { EventType, NatsTopicMetadata, Producer } from '@rsdk/nats.common';
+import {
+  getSubjectName,
+  NATS_PRODUCER_INJECTION_TOKEN,
+  NATS_TOPIC_RSDK_METADATA_SCOPE,
+  NatsClientModule,
+  TopicType,
+} from '@rsdk/nats.common';
+import { ProducerConfig } from '@rsdk/nats.producer';
+import omit from 'lodash/omit';
+
+import { natsProducerDirectAutodocResolver } from './direct.autodoc';
+import { createDirectProducerToken } from './direct.constants';
+import { DirectNatsProducerModule } from './direct.module';
+import { DirectProducer } from './direct.producer';
+
+export class DirectModuleGenerator {
+  static createModule(eventType: EventType): DynamicModule {
+    const providers: FactoryProvider[] = [
+      {
+        inject: [NATS_PRODUCER_INJECTION_TOKEN, ProducerConfig],
+        provide: createDirectProducerToken(eventType),
+        useFactory: (
+          client: Producer,
+          config: ProducerConfig,
+        ): DirectProducer<EventType> => {
+          return new DirectProducer(client, eventType, config);
+        },
+      },
+    ];
+
+    const moduleDef = {
+      imports: [
+        NatsClientModule,
+        DirectNatsProducerModule,
+        PlatformConfigModule.forFeature(ProducerConfig),
+      ],
+      providers,
+      module: DirectModuleGenerator,
+      exports: providers,
+    };
+
+    const rsdkNatsMetadata = new RsdkMetadata<NatsTopicMetadata>(
+      moduleDef,
+      NATS_TOPIC_RSDK_METADATA_SCOPE,
+    );
+
+    rsdkNatsMetadata.add({
+      type: TopicType.PRODUCE,
+      topicName: getSubjectName(eventType),
+      group: eventType.$group,
+      partitionKeyField: eventType.$partitionKeyField,
+      eventType: omit(eventType, 'toJSON', 'encode', 'decode'),
+    });
+
+    AutodocMetadata.defineResolver(
+      moduleDef,
+      NATS_TOPIC_RSDK_METADATA_SCOPE,
+      natsProducerDirectAutodocResolver,
+    );
+
+    return moduleDef;
+  }
+}

package/src/direct/direct.module.ts

@@ -0,0 +1,10 @@
+import { Module } from '@nestjs/common';
+import { NatsClientModule } from '@rsdk/nats.common';
+
+// import { DirectProducerIndicator } from './kafka-producer-direct.health-indicator';
+
+@Module({
+  imports: [NatsClientModule],
+  // providers: [DirectProducerIndicator],
+})
+export class DirectNatsProducerModule {}

package/src/direct/direct.producer.ts

@@ -0,0 +1,52 @@
+import type { ILogger } from '@rsdk/logging';
+import { LoggerFactory } from '@rsdk/logging';
+import type { EventPayload, EventType, Producer } from '@rsdk/nats.common';
+import { getSubjectName, handleNatsException } from '@rsdk/nats.common';
+import type {
+  NatsProducer,
+  ProducerOptions,
+  PublishOptions,
+} from '@rsdk/nats.producer';
+
+import { EventsFactoryDirect } from './direct.event-factory';
+
+/**
+ * Producer, специализированный под конкретный ивент.
+ */
+export class DirectProducer<T extends EventType> implements NatsProducer<T> {
+  private readonly factory: EventsFactoryDirect<T>;
+  private readonly logger: ILogger = LoggerFactory.create(DirectProducer.name);
+
+  constructor(
+    private readonly client: Producer,
+    private readonly eventType: T,
+    config: ProducerOptions,
+  ) {
+    this.client = client;
+    this.eventType = eventType;
+    this.factory = new EventsFactoryDirect(eventType, config);
+  }
+
+  async publish<P extends EventPayload<T>>(
+    payload: P,
+    options?: PublishOptions,
+  ): Promise<void> {
+    const message = this.factory.create(payload);
+    const partitionKey = this.factory.getPartitionKey(payload, options);
+    if (!partitionKey) {
+      this.logger.warn(
+        'Partition key field is not found in the payload. It may lead to performance issues',
+        { options },
+      );
+    }
+    const subject = getSubjectName(this.eventType, partitionKey);
+
+    try {
+      await this.client.publish(subject, message.payload, {
+        headers: message.headers,
+      });
+    } catch (error) {
+      handleNatsException(error, subject, options);
+    }
+  }
+}

package/src/direct/index.ts

@@ -0,0 +1,2 @@
+export { DirectProducer } from './direct.producer';
+export { InjectDirectProducer } from './direct.decorator';

package/src/direct.spec.ts

@@ -0,0 +1,200 @@
+import { Injectable, Module } from '@nestjs/common';
+import type { EventType } from '@rsdk/nats.common';
+import { X_FORMAT_HEADER, X_TYPE_HEADER } from '@rsdk/nats.common';
+import { NatsProducer } from '@rsdk/nats.producer';
+import type { PlatformTestingModule } from '@rsdk/testing';
+import { PlatformTest } from '@rsdk/testing';
+import { connect } from 'nats';
+import { Reader, Writer } from 'protobufjs';
+import type { StartedTestContainer } from 'testcontainers';
+import { GenericContainer, Wait } from 'testcontainers';
+
+import { InjectDirectProducer } from './direct';
+
+jest.setTimeout(60_000);
+
+type TestMessage = { content: string };
+
+class TestEventContract implements EventType {
+  $type = TestEventContract.name;
+
+  constructor(
+    public readonly $group: string,
+    public readonly $partitionKeyField: string | null = null,
+  ) {}
+
+  static encode(message: TestMessage): Writer {
+    return new Writer().string(message.content);
+  }
+
+  static decode(data: Uint8Array | Reader): TestMessage {
+    return data instanceof Uint8Array
+      ? { content: new Reader(data).string() }
+      : { content: data.string() };
+  }
+
+  encode(message: TestMessage): Writer {
+    return TestEventContract.encode(message);
+  }
+
+  decode(data: Uint8Array | Reader): TestMessage {
+    return TestEventContract.decode(data);
+  }
+
+  toJSON(message: TestMessage): string {
+    return JSON.stringify(message);
+  }
+}
+
+const TestEvent = new TestEventContract('kek', null);
+
+@Injectable()
+class TestService {
+  constructor(
+    @InjectDirectProducer(TestEvent)
+    public readonly producer: NatsProducer<TestEventContract>,
+  ) {}
+}
+
+@Module({
+  providers: [TestService],
+})
+class AppModule {}
+
+const buildModule = async (): Promise<PlatformTestingModule> => {
+  const Test = PlatformTest.create({});
+
+  return await Test.createTestingModule({
+    imports: [AppModule],
+  }).compile();
+};
+
+describe('nats direct producer', () => {
+  let natsContainer: StartedTestContainer;
+  let natsServers: string;
+
+  beforeAll(async () => {
+    const container = new GenericContainer('nats:latest')
+      .withExposedPorts({ container: 4222, host: 4222 })
+      .withCommand(['--jetstream'])
+      .withWaitStrategy(Wait.forLogMessage('Server is ready'));
+
+    natsContainer = await container.start();
+    natsServers = `nats://${natsContainer.getHost()}:${natsContainer.getMappedPort(4222)}`;
+    process.env.NATS_SERVERS = natsServers;
+  });
+
+  afterAll(async () => {
+    await natsContainer.stop();
+  });
+
+  it('injects producer to service', async () => {
+    const app = await buildModule();
+    const service = app.get(TestService);
+
+    expect(service).toBeDefined();
+    expect(service.producer).toBeDefined();
+  });
+
+  it('allows to send message with json codec', async () => {
+    // Arrange
+    process.env.NATS_PAYLOAD_FORMAT = 'protobuf';
+    const connection = await connect({ servers: natsServers });
+    const jsm = await connection.jetstreamManager();
+    const streamName = TestEvent.$group;
+
+    await jsm.streams.add({ name: streamName, subjects: ['events.>'] });
+
+    const app = await buildModule();
+    const service = app.get(TestService);
+    const testMessage: TestMessage = { content: 'kek' };
+
+    // Ack
+    await service.producer.publish(testMessage);
+
+    // Assert
+    const message = await jsm.streams.getMessage(streamName, { seq: 1 });
+
+    expect(TestEvent.decode(message.data)).toStrictEqual(testMessage);
+
+    await jsm.streams.delete(streamName);
+    await connection.drain();
+  });
+
+  it('allows to send message with json codec', async () => {
+    // Arrange
+    process.env.NATS_PAYLOAD_FORMAT = 'json';
+    const connection = await connect({ servers: natsServers });
+    const jsm = await connection.jetstreamManager();
+    const streamName = TestEvent.$group;
+
+    await jsm.streams.add({ name: streamName, subjects: ['events.>'] });
+
+    const app = await buildModule();
+    const service = app.get(TestService);
+    const testMessage: TestMessage = { content: 'kek' };
+
+    // Ack
+    await service.producer.publish(testMessage);
+
+    // Assert
+    const message = await jsm.streams.getMessage(streamName, { seq: 1 });
+
+    expect(message.json()).toStrictEqual(testMessage);
+
+    await jsm.streams.delete(streamName);
+    await connection.drain();
+  });
+
+  it('uses partition key as part of the subject', async () => {
+    // Arrange
+    process.env.NATS_PAYLOAD_FORMAT = 'json';
+    const connection = await connect({ servers: natsServers });
+    const jsm = await connection.jetstreamManager();
+    const streamName = TestEvent.$group;
+
+    await jsm.streams.add({ name: streamName, subjects: ['events.>'] });
+
+    const app = await buildModule();
+    const service = app.get(TestService);
+    const testMessage: TestMessage = { content: 'kek' };
+    const options = { partitionKey: 'test-partition-key' };
+
+    // Ack
+    await service.producer.publish(testMessage, options);
+
+    // Assert
+    const message = await jsm.streams.getMessage(streamName, { seq: 1 });
+
+    expect(message.subject).toContain(options.partitionKey);
+
+    await jsm.streams.delete(streamName);
+    await connection.drain();
+  });
+
+  it('provide type and used format if not default', async () => {
+    // Arrange
+    process.env.NATS_PAYLOAD_FORMAT = 'json';
+    const connection = await connect({ servers: natsServers });
+    const jsm = await connection.jetstreamManager();
+    const streamName = TestEvent.$group;
+
+    await jsm.streams.add({ name: streamName, subjects: ['events.>'] });
+
+    const app = await buildModule();
+    const service = app.get(TestService);
+    const testMessage: TestMessage = { content: 'kek' };
+
+    // Ack
+    await service.producer.publish(testMessage);
+
+    // Assert
+    const message = await jsm.streams.getMessage(streamName, { seq: 1 });
+
+    expect(message.header.get(X_FORMAT_HEADER)).toBe('json');
+    expect(message.header.get(X_TYPE_HEADER)).toBe(TestEvent.$type);
+
+    await jsm.streams.delete(streamName);
+    await connection.drain();
+  });
+});

package/src/index.ts

@@ -0,0 +1 @@
+export { DirectProducer, InjectDirectProducer } from './direct';

package/tsconfig.build.json

@@ -0,0 +1,12 @@
+{
+  "extends": "./tsconfig.json",
+  "exclude": [
+    "node_modules",
+    "dist",
+    "test",
+    "**/*.spec.ts",
+    "**/*.test.ts",
+    "**/*.test.e2e.ts",
+    "**/*.test.manual-e2e.ts",
+  ]
+}

package/tsconfig.json

@@ -0,0 +1,7 @@
+{
+  "extends": "@rsdk/tsconfig/base.json",
+  "compilerOptions": {
+    "declaration": true,
+    "outDir": "dist"
+  }
+}