@travetto/context 5.1.0 → 6.0.0-rc.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@travetto/context",
-  "version": "5.1.0",
+  "version": "6.0.0-rc.1",
   "description": "Async-aware state management, maintaining context across asynchronous calls.",
   "keywords": [
     "async-hooks",
@@ -26,10 +26,10 @@
     "directory": "module/context"
   },
   "dependencies": {
-    "@travetto/di": "^5.1.0"
+    "@travetto/di": "^6.0.0-rc.1"
   },
   "peerDependencies": {
-    "@travetto/test": "^5.1.0"
+    "@travetto/test": "^6.0.0-rc.1"
   },
   "peerDependenciesMeta": {
     "@travetto/test": {

package/src/value.ts

@@ -6,9 +6,9 @@ type Storage<T = unknown> = AsyncLocalStorage<Payload<T>>;
 type Key = string | symbol;
 type StorageSource = Storage | (() => Storage) | { storage: Storage } | { context: { storage: Storage } };
 
-type ContextConfig = {
-  failIfUnbound?: boolean;
-};
+type ReadWriteConfig = { read?: boolean, write?: boolean };
+
+type ContextConfig = { failIfUnbound?: ReadWriteConfig };
 
 /**
  * Async Context Value
@@ -18,7 +18,7 @@ export class AsyncContextValue<T = unknown> {
   #source: () => Storage<T>;
   #storage?: Storage<T>;
   #key: Key = Symbol();
-  #failIfUnbound: boolean;
+  #failIfUnbound: ReadWriteConfig;
 
   constructor(source: StorageSource, config?: ContextConfig) {
     this.#source = castTo(typeof source === 'function' ?
@@ -30,12 +30,12 @@ export class AsyncContextValue<T = unknown> {
           source.context.storage
         ))
     );
-    this.#failIfUnbound = config?.failIfUnbound ?? true;
+    this.#failIfUnbound = { read: true, write: true, ...config?.failIfUnbound };
   }
 
-  get #store(): Payload<T> | undefined {
+  #store(mode: keyof ReadWriteConfig): Payload<T> | undefined {
     const store = (this.#storage ??= this.#source()).getStore();
-    if (!store && this.#failIfUnbound) {
+    if (!store && this.#failIfUnbound[mode]) {
       throw new AppError('Context not initialized');
     }
     return store;
@@ -45,14 +45,17 @@ export class AsyncContextValue<T = unknown> {
    * Get value
    */
   get(): T | undefined {
-    return this.#store?.[this.#key];
+    const store = this.#store('read');
+    if (store) {
+      return store[this.#key];
+    }
   }
 
   /**
    * Set value
    */
   set(value: T | undefined): void {
-    const store = this.#store;
+    const store = this.#store('write');
     if (store) {
       store[this.#key] = value;
     }