@hg-ts/async-context 0.1.0

package/dist/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+tslib_1.__exportStar(require("./async-context.provider"), exports);
+tslib_1.__exportStar(require("./async-context.not-found.exception"), exports);
+//# 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,mEAAyC;AACzC,8EAAoD"}

package/dist-esm/async-context.not-found.exception.d.ts

@@ -0,0 +1,5 @@
+import { BaseException } from '@hg-ts/exception';
+export declare class AsyncContextNotFoundException extends BaseException {
+    constructor();
+}
+//# sourceMappingURL=async-context.not-found.exception.d.ts.map

package/dist-esm/async-context.not-found.exception.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"async-context.not-found.exception.d.ts","sourceRoot":"","sources":["../src/async-context.not-found.exception.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEjD,qBAAa,6BAA8B,SAAQ,aAAa;;CAI/D"}

package/dist-esm/async-context.not-found.exception.js

@@ -0,0 +1,7 @@
+import { BaseException } from '@hg-ts/exception';
+export class AsyncContextNotFoundException extends BaseException {
+    constructor() {
+        super('Async context not provided');
+    }
+}
+//# sourceMappingURL=async-context.not-found.exception.js.map

package/dist-esm/async-context.not-found.exception.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"async-context.not-found.exception.js","sourceRoot":"","sources":["../src/async-context.not-found.exception.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEjD,MAAM,OAAO,6BAA8B,SAAQ,aAAa;IAC/D;QACC,KAAK,CAAC,4BAA4B,CAAC,CAAC;IACrC,CAAC;CACD"}

package/dist-esm/async-context.provider.d.ts

@@ -0,0 +1,18 @@
+/// <reference types="node" />
+/// <reference types="@hg-ts/types" />
+import { AsyncLocalStorage } from 'node:async_hooks';
+import { MonoTypeOperatorFunction } from 'rxjs';
+type AsyncContextProviderCallback<T> = () => T;
+type AsyncContextFactory<InputType, Context> = (input: InputType) => Context;
+export declare class AsyncContextProvider<Context> {
+    protected readonly storage: AsyncLocalStorage<Context>;
+    get(): Nullable<Context>;
+    getOrFail(): NonNullable<Context>;
+    run<ReturnType>(context: Context, callback: AsyncContextProviderCallback<ReturnType>): ReturnType;
+    exit<ReturnType>(callback: AsyncContextProviderCallback<ReturnType>): ReturnType;
+    runOperator<T>(factory?: AsyncContextFactory<T, Context>): MonoTypeOperatorFunction<T>;
+    exitOperator<T>(): MonoTypeOperatorFunction<T>;
+    private createOperator;
+}
+export {};
+//# sourceMappingURL=async-context.provider.d.ts.map

package/dist-esm/async-context.provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"async-context.provider.d.ts","sourceRoot":"","sources":["../src/async-context.provider.ts"],"names":[],"mappings":";;AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EACN,wBAAwB,EAExB,MAAM,MAAM,CAAC;AAGd,KAAK,4BAA4B,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC;AAC/C,KAAK,mBAAmB,CAAC,SAAS,EAAE,OAAO,IAAI,CAAC,KAAK,EAAE,SAAS,KAAK,OAAO,CAAC;AAI7E,qBAAa,oBAAoB,CAAC,OAAO;IACxC,SAAS,CAAC,QAAQ,CAAC,OAAO,6BAAoC;IAEvD,GAAG,IAAI,QAAQ,CAAC,OAAO,CAAC;IAIxB,SAAS,IAAI,WAAW,CAAC,OAAO,CAAC;IAUjC,GAAG,CAAC,UAAU,EACpB,OAAO,EAAE,OAAO,EAChB,QAAQ,EAAE,4BAA4B,CAAC,UAAU,CAAC,GAChD,UAAU;IAIN,IAAI,CAAC,UAAU,EACrB,QAAQ,EAAE,4BAA4B,CAAC,UAAU,CAAC,GAChD,UAAU;IAIN,WAAW,CAAC,CAAC,EACnB,OAAO,CAAC,EAAE,mBAAmB,CAAC,CAAC,EAAE,OAAO,CAAC,GACvC,wBAAwB,CAAC,CAAC,CAAC;IAOvB,YAAY,CAAC,CAAC,KAAK,wBAAwB,CAAC,CAAC,CAAC;IAIrD,OAAO,CAAC,cAAc;CA0BtB"}

package/dist-esm/async-context.provider.js

@@ -0,0 +1,57 @@
+import { AsyncLocalStorage } from 'node:async_hooks';
+import { Observable, } from 'rxjs';
+import { AsyncContextNotFoundException } from './async-context.not-found.exception';
+export class AsyncContextProvider {
+    storage = new AsyncLocalStorage();
+    get() {
+        return this.storage.getStore() ?? null;
+    }
+    getOrFail() {
+        const context = this.get();
+        if (typeof context === 'undefined' || context === null) {
+            throw new AsyncContextNotFoundException();
+        }
+        return context;
+    }
+    run(context, callback) {
+        return this.storage.run(context, callback);
+    }
+    exit(callback) {
+        return this.storage.exit(callback);
+    }
+    runOperator(factory) {
+        return this.createOperator((next, value) => {
+            const context = factory ? factory(value) : value;
+            this.storage.run(context, next);
+        });
+    }
+    exitOperator() {
+        return this.createOperator(next => this.storage.exit(next));
+    }
+    createOperator(onNext) {
+        return (source) => new Observable(subscriber => {
+            const callNext = input => {
+                const next = () => {
+                    if (!subscriber.closed) {
+                        subscriber.next(input);
+                    }
+                };
+                onNext(next, input);
+            };
+            const callError = input => {
+                const next = () => {
+                    if (!subscriber.closed) {
+                        subscriber.error(input);
+                    }
+                };
+                onNext(next, input);
+            };
+            source.subscribe({
+                next: callNext,
+                error: callError,
+                complete: () => subscriber.complete(),
+            });
+        });
+    }
+}
+//# sourceMappingURL=async-context.provider.js.map

package/dist-esm/async-context.provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"async-context.provider.js","sourceRoot":"","sources":["../src/async-context.provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAEN,UAAU,GACV,MAAM,MAAM,CAAC;AACd,OAAO,EAAE,6BAA6B,EAAE,MAAM,qCAAqC,CAAC;AAOpF,MAAM,OAAO,oBAAoB;IACb,OAAO,GAAG,IAAI,iBAAiB,EAAW,CAAC;IAEvD,GAAG;QACT,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,IAAI,CAAC;IACxC,CAAC;IAEM,SAAS;QACf,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE3B,IAAI,OAAO,OAAO,KAAK,WAAW,IAAI,OAAO,KAAK,IAAI,EAAE;YACvD,MAAM,IAAI,6BAA6B,EAAE,CAAC;SAC1C;QAED,OAAO,OAA+B,CAAC;IACxC,CAAC;IAEM,GAAG,CACT,OAAgB,EAChB,QAAkD;QAElD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IAC5C,CAAC;IAEM,IAAI,CACV,QAAkD;QAElD,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IAEM,WAAW,CACjB,OAAyC;QAEzC,OAAO,IAAI,CAAC,cAAc,CAAI,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YAC7C,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAY,CAAC;YACxD,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;IACJ,CAAC;IAEM,YAAY;QAClB,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IAC7D,CAAC;IAEO,cAAc,CAAI,MAA4C;QACrE,OAAO,CAAC,MAAM,EAAiB,EAAE,CAAC,IAAI,UAAU,CAAI,UAAU,CAAC,EAAE;YAChE,MAAM,QAAQ,GAAoB,KAAK,CAAC,EAAE;gBACzC,MAAM,IAAI,GAAG,GAAS,EAAE;oBACvB,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE;wBACvB,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;qBACvB;gBACF,CAAC,CAAC;gBACF,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YACF,MAAM,SAAS,GAAoB,KAAK,CAAC,EAAE;gBAC1C,MAAM,IAAI,GAAG,GAAS,EAAE;oBACvB,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE;wBACvB,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;qBACxB;gBACF,CAAC,CAAC;gBACF,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;YACrB,CAAC,CAAC;YAEF,MAAM,CAAC,SAAS,CAAC;gBAChB,IAAI,EAAE,QAAQ;gBACd,KAAK,EAAE,SAAS;gBAChB,QAAQ,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE;aACrC,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;IACJ,CAAC;CACD"}

package/dist-esm/async-context.test-suite.d.ts

@@ -0,0 +1,16 @@
+import { TestSuite } from '@hg-ts/tests';
+export declare class AsyncContextTestSuite extends TestSuite {
+    private context;
+    emptyContext(): Promise<void>;
+    noPromiseContext(): Promise<void>;
+    promiseContext(): Promise<void>;
+    operatorContext(): Promise<void>;
+    operatorWithFactory(): Promise<void>;
+    operatorWithErrorContext(): Promise<void>;
+    operatorAfterUnsubscribe(): Promise<void>;
+    operatorWithErrorAfterUnsubscribe(): Promise<void>;
+    contextNotFound(): Promise<void>;
+    beforeEach(): Promise<void>;
+    private createOperatorTestSubscription;
+}
+//# sourceMappingURL=async-context.test-suite.d.ts.map

package/dist-esm/async-context.test-suite.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"async-context.test-suite.d.ts","sourceRoot":"","sources":["../src/async-context.test-suite.ts"],"names":[],"mappings":"AAAA,OAAO,EAKN,SAAS,EACT,MAAM,cAAc,CAAC;AAiBtB,qBACa,qBAAsB,SAAQ,SAAS;IACnD,OAAO,CAAC,OAAO,CAA+B;IAGjC,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;IAK7B,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;IAiBjC,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAiB/B,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAUhC,mBAAmB,IAAI,OAAO,CAAC,IAAI,CAAC;IAUpC,wBAAwB,IAAI,OAAO,CAAC,IAAI,CAAC;IAUzC,wBAAwB,IAAI,OAAO,CAAC,IAAI,CAAC;IAWzC,iCAAiC,IAAI,OAAO,CAAC,IAAI,CAAC;IAYlD,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC;IAIvB,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAIjD,OAAO,CAAC,8BAA8B;CAgCtC"}

package/dist-esm/async-context.test-suite.js

@@ -0,0 +1,156 @@
+import { __decorate, __metadata } from "tslib";
+import { Describe, expect, ExpectException, Test, TestSuite, } from '@hg-ts/tests';
+import { Subject, tap, } from 'rxjs';
+import { AsyncContextNotFoundException } from './async-context.not-found.exception';
+import { AsyncContextProvider } from './async-context.provider';
+let AsyncContextTestSuite = class AsyncContextTestSuite extends TestSuite {
+    context;
+    async emptyContext() {
+        expect(this.context.get()).toBe(null);
+    }
+    async noPromiseContext() {
+        const contextValue = Math.random();
+        const resultValue = Math.random();
+        const result = this.context.run(contextValue, () => {
+            expect(this.context.getOrFail()).toBe(contextValue);
+            this.context.exit(() => {
+                expect(this.context.get()).toBe(null);
+            });
+            return resultValue;
+        });
+        expect(result).toBe(resultValue);
+    }
+    async promiseContext() {
+        const contextValue = Math.random();
+        const resultValue = Math.random();
+        const result = await this.context.run(contextValue, async () => {
+            expect(this.context.getOrFail()).toBe(contextValue);
+            await this.context.exit(async () => {
+                expect(this.context.get()).toBe(null);
+            });
+            return resultValue;
+        });
+        expect(result).toBe(resultValue);
+    }
+    async operatorContext() {
+        const { subject, input } = this.createOperatorTestSubscription();
+        subject.next(input);
+        subject.complete();
+        expect.assertions(4);
+    }
+    async operatorWithFactory() {
+        const { subject, input } = this.createOperatorTestSubscription(value => -value);
+        subject.next(input);
+        subject.complete();
+        expect.assertions(4);
+    }
+    async operatorWithErrorContext() {
+        const { subject, input } = this.createOperatorTestSubscription();
+        subject.error(input);
+        subject.complete();
+        expect.assertions(4);
+    }
+    async operatorAfterUnsubscribe() {
+        const { subject, input, subscription } = this.createOperatorTestSubscription();
+        subscription.unsubscribe();
+        subject.next(input);
+        subject.complete();
+        expect.assertions(2);
+    }
+    async operatorWithErrorAfterUnsubscribe() {
+        const { subject, input, subscription } = this.createOperatorTestSubscription();
+        subscription.unsubscribe();
+        subject.error(input);
+        subject.complete();
+        expect.assertions(2);
+    }
+    async contextNotFound() {
+        this.context.getOrFail();
+    }
+    async beforeEach() {
+        this.context = new AsyncContextProvider();
+    }
+    createOperatorTestSubscription(factory) {
+        const subject = new Subject();
+        const input = Math.random();
+        const context = factory ? factory(input) : input;
+        const withContext = (value) => {
+            expect(value).toBe(input);
+            expect(this.context.getOrFail()).toBe(context);
+        };
+        const withoutContext = (value) => {
+            expect(value).toBe(input);
+            expect(this.context.get()).toBe(null);
+        };
+        const subscription = subject.asObservable()
+            .pipe(this.context.runOperator(factory), tap({
+            next: withContext,
+            error: withContext,
+        }), this.context.exitOperator(), tap({
+            next: withoutContext,
+            error: withoutContext,
+        }))
+            .subscribe({ next() { }, error() { } });
+        return { subject, subscription, input, context };
+    }
+};
+__decorate([
+    Test(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AsyncContextTestSuite.prototype, "emptyContext", null);
+__decorate([
+    Test(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AsyncContextTestSuite.prototype, "noPromiseContext", null);
+__decorate([
+    Test(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AsyncContextTestSuite.prototype, "promiseContext", null);
+__decorate([
+    Test(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AsyncContextTestSuite.prototype, "operatorContext", null);
+__decorate([
+    Test(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AsyncContextTestSuite.prototype, "operatorWithFactory", null);
+__decorate([
+    Test(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AsyncContextTestSuite.prototype, "operatorWithErrorContext", null);
+__decorate([
+    Test(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AsyncContextTestSuite.prototype, "operatorAfterUnsubscribe", null);
+__decorate([
+    Test(),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AsyncContextTestSuite.prototype, "operatorWithErrorAfterUnsubscribe", null);
+__decorate([
+    Test(),
+    ExpectException(AsyncContextNotFoundException),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", []),
+    __metadata("design:returntype", Promise)
+], AsyncContextTestSuite.prototype, "contextNotFound", null);
+AsyncContextTestSuite = __decorate([
+    Describe()
+], AsyncContextTestSuite);
+export { AsyncContextTestSuite };
+//# sourceMappingURL=async-context.test-suite.js.map

package/dist-esm/async-context.test-suite.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"async-context.test-suite.js","sourceRoot":"","sources":["../src/async-context.test-suite.ts"],"names":[],"mappings":";AAAA,OAAO,EACN,QAAQ,EACR,MAAM,EACN,eAAe,EACf,IAAI,EACJ,SAAS,GACT,MAAM,cAAc,CAAC;AACtB,OAAO,EACN,OAAO,EAEP,GAAG,GACH,MAAM,MAAM,CAAC;AAEd,OAAO,EAAE,6BAA6B,EAAE,MAAM,qCAAqC,CAAC;AACpF,OAAO,EAAE,oBAAoB,EAAE,MAAM,0BAA0B,CAAC;AAUzD,IAAM,qBAAqB,GAA3B,MAAM,qBAAsB,SAAQ,SAAS;IAC3C,OAAO,CAA+B;IAGjC,AAAN,KAAK,CAAC,YAAY;QACxB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC;IAGY,AAAN,KAAK,CAAC,gBAAgB;QAC5B,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACnC,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAElC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,GAAG,EAAE;YAClD,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAEpD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE;gBACtB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvC,CAAC,CAAC,CAAC;YACH,OAAO,WAAW,CAAC;QACpB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAClC,CAAC;IAGY,AAAN,KAAK,CAAC,cAAc;QAC1B,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACnC,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAElC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,KAAK,IAAG,EAAE;YAC7D,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAEpD,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,IAAG,EAAE;gBACjC,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvC,CAAC,CAAC,CAAC;YACH,OAAO,WAAW,CAAC;QACpB,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAClC,CAAC;IAGY,AAAN,KAAK,CAAC,eAAe;QAC3B,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,8BAA8B,EAAE,CAAC;QAEjE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpB,OAAO,CAAC,QAAQ,EAAE,CAAC;QAEnB,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC;IAGY,AAAN,KAAK,CAAC,mBAAmB;QAC/B,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC;QAEhF,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpB,OAAO,CAAC,QAAQ,EAAE,CAAC;QAEnB,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC;IAGY,AAAN,KAAK,CAAC,wBAAwB;QACpC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,8BAA8B,EAAE,CAAC;QAEjE,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO,CAAC,QAAQ,EAAE,CAAC;QAEnB,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC;IAGY,AAAN,KAAK,CAAC,wBAAwB;QACpC,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,8BAA8B,EAAE,CAAC;QAE/E,YAAY,CAAC,WAAW,EAAE,CAAC;QAC3B,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpB,OAAO,CAAC,QAAQ,EAAE,CAAC;QAEnB,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC;IAGY,AAAN,KAAK,CAAC,iCAAiC;QAC7C,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,8BAA8B,EAAE,CAAC;QAE/E,YAAY,CAAC,WAAW,EAAE,CAAC;QAC3B,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACrB,OAAO,CAAC,QAAQ,EAAE,CAAC;QAEnB,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC;IAIY,AAAN,KAAK,CAAC,eAAe;QAC3B,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;IAC1B,CAAC;IAEe,KAAK,CAAC,UAAU;QAC/B,IAAI,CAAC,OAAO,GAAG,IAAI,oBAAoB,EAAE,CAAC;IAC3C,CAAC;IAEO,8BAA8B,CAAC,OAAmC;QACzE,MAAM,OAAO,GAAG,IAAI,OAAO,EAAU,CAAC;QACtC,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAEjD,MAAM,WAAW,GAAG,CAAC,KAAa,EAAQ,EAAE;YAC3C,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC,CAAC;QAEF,MAAM,cAAc,GAAG,CAAC,KAAa,EAAQ,EAAE;YAC9C,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,CAAC,CAAC;QAEF,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,EAAE;aACzC,IAAI,CACJ,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,EACjC,GAAG,CAAC;YACH,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,WAAW;SAClB,CAAC,EACF,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,EAC3B,GAAG,CAAC;YACH,IAAI,EAAE,cAAc;YACpB,KAAK,EAAE,cAAc;SACrB,CAAC,CACF;aACA,SAAS,CAAC,EAAE,IAAI,KAAI,CAAC,EAAE,KAAK,KAAI,CAAC,EAAE,CAAC,CAAC;QAEvC,OAAO,EAAE,OAAO,EAAE,YAAY,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;IAClD,CAAC;CACD,CAAA;AApIa;IADZ,IAAI,EAAE;;;;yDAGN;AAGY;IADZ,IAAI,EAAE;;;;6DAeN;AAGY;IADZ,IAAI,EAAE;;;;2DAeN;AAGY;IADZ,IAAI,EAAE;;;;4DAQN;AAGY;IADZ,IAAI,EAAE;;;;gEAQN;AAGY;IADZ,IAAI,EAAE;;;;qEAQN;AAGY;IADZ,IAAI,EAAE;;;;qEASN;AAGY;IADZ,IAAI,EAAE;;;;8EASN;AAIY;IAFZ,IAAI,EAAE;IACN,eAAe,CAAC,6BAA6B,CAAC;;;;4DAG9C;AAlGW,qBAAqB;IADjC,QAAQ,EAAE;GACE,qBAAqB,CAwIjC"}

package/dist-esm/index.d.ts

@@ -0,0 +1,3 @@
+export * from './async-context.provider';
+export * from './async-context.not-found.exception';
+//# sourceMappingURL=index.d.ts.map

package/dist-esm/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,0BAA0B,CAAC;AACzC,cAAc,qCAAqC,CAAC"}

package/dist-esm/index.js

@@ -0,0 +1,3 @@
+export * from './async-context.provider';
+export * from './async-context.not-found.exception';
+//# sourceMappingURL=index.js.map

package/dist-esm/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,0BAA0B,CAAC;AACzC,cAAc,qCAAqC,CAAC"}

package/package.json

@@ -0,0 +1,44 @@
+{
+  "name": "@hg-ts/async-context",
+  "version": "0.1.0",
+  "main": "dist/index.js",
+  "module": "dist-esm/index.js",
+  "repository": "git@gitlab.com:hyper-graph/framework.git",
+  "scripts": {
+    "prepack": "yarn build && yarn lint && yarn test",
+    "build": "hg build",
+    "build:dev": "hg build:dev",
+    "lint": "hg lint",
+    "lint:fix": "hg lint:fix",
+    "test": "hg test",
+    "test:dev": "hg test:dev",
+    "test:coverage": "hg test:coverage"
+  },
+  "devDependencies": {
+    "@hg-ts-config/eslint-config": "0.1.0",
+    "@hg-ts-config/nyc": "0.1.0",
+    "@hg-ts-config/typescript": "0.1.0",
+    "@hg-ts/cli": "0.1.0",
+    "@hg-ts/exception": "0.1.0",
+    "@hg-ts/tests": "0.1.0",
+    "@hg-ts/types": "0.1.0",
+    "@types/node": "20.8.7",
+    "@typescript-eslint/eslint-plugin": "6.8.0",
+    "@typescript-eslint/parser": "6.8.0",
+    "eslint": "8.52.0",
+    "mocha": "10.2.0",
+    "mocha-junit-reporter": "2.2.1",
+    "nyc": "15.1.0",
+    "reflect-metadata": "0.1.13",
+    "rxjs": "7.8.1",
+    "tsc-watch": "6.0.4",
+    "tslib": "2.6.2",
+    "typescript": "5.2.2"
+  },
+  "peerDependencies": {
+    "@hg-ts/exception": "0.1.0",
+    "reflect-metadata": "*",
+    "rxjs": "*",
+    "tslib": "*"
+  }
+}

package/src/async-context.not-found.exception.ts

@@ -0,0 +1,7 @@
+import { BaseException } from '@hg-ts/exception';
+
+export class AsyncContextNotFoundException extends BaseException {
+	public constructor() {
+		super('Async context not provided');
+	}
+}

package/src/async-context.provider.ts

@@ -0,0 +1,82 @@
+import { AsyncLocalStorage } from 'node:async_hooks';
+import {
+	MonoTypeOperatorFunction,
+	Observable,
+} from 'rxjs';
+import { AsyncContextNotFoundException } from './async-context.not-found.exception';
+
+type AsyncContextProviderCallback<T> = () => T;
+type AsyncContextFactory<InputType, Context> = (input: InputType) => Context;
+
+type NextCallback<T> = (value: T) => void;
+
+export class AsyncContextProvider<Context> {
+	protected readonly storage = new AsyncLocalStorage<Context>();
+
+	public get(): Nullable<Context> {
+		return this.storage.getStore() ?? null;
+	}
+
+	public getOrFail(): NonNullable<Context> {
+		const context = this.get();
+
+		if (typeof context === 'undefined' || context === null) {
+			throw new AsyncContextNotFoundException();
+		}
+
+		return context as NonNullable<Context>;
+	}
+
+	public run<ReturnType>(
+		context: Context,
+		callback: AsyncContextProviderCallback<ReturnType>,
+	): ReturnType {
+		return this.storage.run(context, callback);
+	}
+
+	public exit<ReturnType>(
+		callback: AsyncContextProviderCallback<ReturnType>,
+	): ReturnType {
+		return this.storage.exit(callback);
+	}
+
+	public runOperator<T>(
+		factory?: AsyncContextFactory<T, Context>,
+	): MonoTypeOperatorFunction<T> {
+		return this.createOperator<T>((next, value) => {
+			const context = factory ? factory(value) : value as any;
+			this.storage.run(context, next);
+		});
+	}
+
+	public exitOperator<T>(): MonoTypeOperatorFunction<T> {
+		return this.createOperator(next => this.storage.exit(next));
+	}
+
+	private createOperator<T>(onNext: (next: () => void, value: T) => void): MonoTypeOperatorFunction<any> {
+		return (source): Observable<T> => new Observable<T>(subscriber => {
+			const callNext: NextCallback<T> = input => {
+				const next = (): void => {
+					if (!subscriber.closed) {
+						subscriber.next(input);
+					}
+				};
+				onNext(next, input);
+			};
+			const callError: NextCallback<T> = input => {
+				const next = (): void => {
+					if (!subscriber.closed) {
+						subscriber.error(input);
+					}
+				};
+				onNext(next, input);
+			};
+
+			source.subscribe({
+				next: callNext,
+				error: callError,
+				complete: () => subscriber.complete(),
+			});
+		});
+	}
+}

package/src/async-context.test-suite.ts

@@ -0,0 +1,161 @@
+import {
+	Describe,
+	expect,
+	ExpectException,
+	Test,
+	TestSuite,
+} from '@hg-ts/tests';
+import {
+	Subject,
+	Subscription,
+	tap,
+} from 'rxjs';
+
+import { AsyncContextNotFoundException } from './async-context.not-found.exception';
+import { AsyncContextProvider } from './async-context.provider';
+
+type OperatorTestSubscription = {
+	subscription: Subscription;
+	subject: Subject<number>;
+	input: number;
+	context: number;
+};
+
+@Describe()
+export class AsyncContextTestSuite extends TestSuite {
+	private context: AsyncContextProvider<number>;
+
+	@Test()
+	public async emptyContext(): Promise<void> {
+		expect(this.context.get()).toBe(null);
+	}
+
+	@Test()
+	public async noPromiseContext(): Promise<void> {
+		const contextValue = Math.random();
+		const resultValue = Math.random();
+
+		const result = this.context.run(contextValue, () => {
+			expect(this.context.getOrFail()).toBe(contextValue);
+
+			this.context.exit(() => {
+				expect(this.context.get()).toBe(null);
+			});
+			return resultValue;
+		});
+
+		expect(result).toBe(resultValue);
+	}
+
+	@Test()
+	public async promiseContext(): Promise<void> {
+		const contextValue = Math.random();
+		const resultValue = Math.random();
+
+		const result = await this.context.run(contextValue, async() => {
+			expect(this.context.getOrFail()).toBe(contextValue);
+
+			await this.context.exit(async() => {
+				expect(this.context.get()).toBe(null);
+			});
+			return resultValue;
+		});
+
+		expect(result).toBe(resultValue);
+	}
+
+	@Test()
+	public async operatorContext(): Promise<void> {
+		const { subject, input } = this.createOperatorTestSubscription();
+
+		subject.next(input);
+		subject.complete();
+
+		expect.assertions(4);
+	}
+
+	@Test()
+	public async operatorWithFactory(): Promise<void> {
+		const { subject, input } = this.createOperatorTestSubscription(value => -value);
+
+		subject.next(input);
+		subject.complete();
+
+		expect.assertions(4);
+	}
+
+	@Test()
+	public async operatorWithErrorContext(): Promise<void> {
+		const { subject, input } = this.createOperatorTestSubscription();
+
+		subject.error(input);
+		subject.complete();
+
+		expect.assertions(4);
+	}
+
+	@Test()
+	public async operatorAfterUnsubscribe(): Promise<void> {
+		const { subject, input, subscription } = this.createOperatorTestSubscription();
+
+		subscription.unsubscribe();
+		subject.next(input);
+		subject.complete();
+
+		expect.assertions(2);
+	}
+
+	@Test()
+	public async operatorWithErrorAfterUnsubscribe(): Promise<void> {
+		const { subject, input, subscription } = this.createOperatorTestSubscription();
+
+		subscription.unsubscribe();
+		subject.error(input);
+		subject.complete();
+
+		expect.assertions(2);
+	}
+
+	@Test()
+	@ExpectException(AsyncContextNotFoundException)
+	public async contextNotFound(): Promise<void> {
+		this.context.getOrFail();
+	}
+
+	public override async beforeEach(): Promise<void> {
+		this.context = new AsyncContextProvider();
+	}
+
+	private createOperatorTestSubscription(factory?: (input: number) => number): OperatorTestSubscription {
+		const subject = new Subject<number>();
+		const input = Math.random();
+		const context = factory ? factory(input) : input;
+
+		const withContext = (value: number): void => {
+			expect(value).toBe(input);
+			expect(this.context.getOrFail()).toBe(context);
+		};
+
+		const withoutContext = (value: number): void => {
+			expect(value).toBe(input);
+			expect(this.context.get()).toBe(null);
+		};
+
+		const subscription = subject.asObservable()
+			.pipe(
+				this.context.runOperator(factory),
+				tap({
+					next: withContext,
+					error: withContext,
+				}),
+				this.context.exitOperator(),
+				tap({
+					next: withoutContext,
+					error: withoutContext,
+				}),
+			)
+			.subscribe({ next() {}, error() {} });
+
+		return { subject, subscription, input, context };
+	}
+}

package/src/index.ts

@@ -0,0 +1,2 @@
+export * from './async-context.provider';
+export * from './async-context.not-found.exception';

package/tsconfig.esm.json

@@ -0,0 +1,9 @@
+{
+	"extends": "@hg-ts-config/typescript/esm",
+	"compilerOptions": {
+		"baseUrl": "src",
+		"rootDir": "src",
+		"outDir": "dist-esm"
+	},
+	"exclude": ["dist-esm", "dist"]
+}

package/tsconfig.json

@@ -0,0 +1,9 @@
+{
+	"extends": "@hg-ts-config/typescript/commonjs",
+	"compilerOptions": {
+		"baseUrl": "src",
+		"rootDir": "src",
+		"outDir": "dist"
+	},
+	"exclude": ["dist-esm", "dist"]
+}