@azure-net/kit 1.2.7 → 1.3.0

package/dist/core/shared/boundaryProvider/Provider.d.ts

@@ -11,9 +11,11 @@ export interface ProviderWithType<T extends ServiceMap> {
 type ProviderFactory<T extends ServiceMap, D extends Record<string, ProviderWithType<ServiceMap>>> = (context: {
     [K in keyof D]: InferProviderType<D[K]>;
 }) => T;
-interface LayerProviderOptions<D> {
+export interface ProviderSettings<T extends ServiceMap, D extends Record<string, ProviderWithType<ServiceMap>>> {
     dependsOn?: D;
+    boot?: (services: ResolvedServices<T>) => void | Promise<void>;
+    register: ProviderFactory<T, D>;
 }
-export declare const createBoundaryProvider: <T extends ServiceMap, D extends Record<string, ProviderWithType<ServiceMap>>>(name: string, services: ProviderFactory<T, D>, options?: LayerProviderOptions<D>) => ProviderWithType<T>;
+export declare const createBoundaryProvider: <T extends ServiceMap, D extends Record<string, ProviderWithType<ServiceMap>>>(name: string, settings: ProviderSettings<T, D>) => ProviderWithType<T>;
 export declare function cleanupProvider(name: string): void;
 export {};

package/dist/core/shared/boundaryProvider/Provider.js

@@ -3,6 +3,7 @@ import { EnvironmentUtil } from 'azure-net-tools';
 const clientCache = new Map();
 const factoriesCache = new WeakMap();
 const providerProxyCache = new Map();
+const clientBootFlags = new Map();
 const getProviderCache = (providerName) => {
     if (EnvironmentUtil.isBrowser) {
         if (!clientCache.has(providerName)) {
@@ -26,8 +27,34 @@ const getProviderCache = (providerName) => {
         return providers.get(providerName);
     }
 };
-export const createBoundaryProvider = (name, services, options) => {
-    const { dependsOn = {} } = options ?? {};
+const getBootFlag = (providerName) => {
+    if (EnvironmentUtil.isBrowser) {
+        return clientBootFlags.get(providerName) ?? false;
+    }
+    else {
+        const context = RequestContext.current();
+        if (!context.data.bootFlags) {
+            context.data.bootFlags = new Map();
+        }
+        const bootFlags = context.data.bootFlags;
+        return bootFlags.get(providerName) ?? false;
+    }
+};
+const setBootFlag = (providerName, value) => {
+    if (EnvironmentUtil.isBrowser) {
+        clientBootFlags.set(providerName, value);
+    }
+    else {
+        const context = RequestContext.current();
+        if (!context.data.bootFlags) {
+            context.data.bootFlags = new Map();
+        }
+        const bootFlags = context.data.bootFlags;
+        bootFlags.set(providerName, value);
+    }
+};
+export const createBoundaryProvider = (name, settings) => {
+    const { dependsOn = {}, boot, register } = settings;
     const providerFn = () => {
         if (EnvironmentUtil.isBrowser && providerProxyCache.has(name)) {
             return providerProxyCache.get(name);
@@ -37,8 +64,8 @@ export const createBoundaryProvider = (name, services, options) => {
         const getFactories = () => {
             if (factories)
                 return factories;
-            if (factoriesCache.has(services)) {
-                factories = factoriesCache.get(services);
+            if (factoriesCache.has(register)) {
+                factories = factoriesCache.get(register);
                 return factories;
             }
             const deps = {};
@@ -54,9 +81,9 @@ export const createBoundaryProvider = (name, services, options) => {
                     }
                 });
             }
-            factories = services(deps);
+            factories = register(deps);
             if (EnvironmentUtil.isBrowser) {
-                factoriesCache.set(services, factories);
+                factoriesCache.set(register, factories);
             }
             return factories;
         };
@@ -90,6 +117,14 @@ export const createBoundaryProvider = (name, services, options) => {
         if (EnvironmentUtil.isBrowser) {
             providerProxyCache.set(name, providerProxy);
         }
+        if (boot && !getBootFlag(name)) {
+            setBootFlag(name, true);
+            getFactories();
+            const bootResult = boot(providerProxy);
+            if (bootResult instanceof Promise) {
+                bootResult.catch((err) => console.error(`Error in boot for provider '${name}':`, err));
+            }
+        }
         return providerProxy;
     };
     return providerFn;
@@ -114,6 +149,7 @@ export function cleanupProvider(name) {
             void cleanupCache(cache);
         }
         providerProxyCache.delete(name);
+        clientBootFlags.delete(name);
     }
     else {
         const context = RequestContext.current();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@azure-net/kit",
-  "version": "1.2.7",
+  "version": "1.3.0",
   "files": [
     "dist",
     "!dist/**/*.test.*",