reset config stores in the config requestor when set on the client

Author: typotter

src/client/eppo-client.ts

@@ -209,13 +209,31 @@ export default class EppoClient {
   // noinspection JSUnusedGlobalSymbols
   setFlagConfigurationStore(flagConfigurationStore: IConfigurationStore<Flag | ObfuscatedFlag>) {
     this.flagConfigurationStore = flagConfigurationStore;
+
+    // Update the ConfigurationRequestor if it exists
+    if (this.configurationRequestor) {
+      this.configurationRequestor.setConfigurationStores(
+        this.flagConfigurationStore,
+        this.banditVariationConfigurationStore || null,
+        this.banditModelConfigurationStore || null,
+      );
+    }
   }
 
   // noinspection JSUnusedGlobalSymbols
   setBanditVariationConfigurationStore(
     banditVariationConfigurationStore: IConfigurationStore<BanditVariation[]>,
   ) {
     this.banditVariationConfigurationStore = banditVariationConfigurationStore;
+
+    // Update the ConfigurationRequestor if it exists
+    if (this.configurationRequestor) {
+      this.configurationRequestor.setConfigurationStores(
+        this.flagConfigurationStore,
+        this.banditVariationConfigurationStore,
+        this.banditModelConfigurationStore || null,
+      );
+    }
   }
 
   /** Sets the EventDispatcher instance to use when tracking events with {@link track}. */
@@ -244,6 +262,15 @@ export default class EppoClient {
     banditModelConfigurationStore: IConfigurationStore<BanditParameters>,
   ) {
     this.banditModelConfigurationStore = banditModelConfigurationStore;
+
+    // Update the ConfigurationRequestor if it exists
+    if (this.configurationRequestor) {
+      this.configurationRequestor.setConfigurationStores(
+        this.flagConfigurationStore,
+        this.banditVariationConfigurationStore || null,
+        this.banditModelConfigurationStore,
+      );
+    }
   }
 
   // noinspection JSUnusedGlobalSymbols

src/configuration-requestor.ts

@@ -10,15 +10,13 @@ import { BanditVariation, BanditParameters, Flag, BanditReference } from './inte
 // Requests AND stores flag configurations
 export default class ConfigurationRequestor {
   private banditModelVersions: string[] = [];
-  private readonly configuration: StoreBackedConfiguration;
+  private configuration: StoreBackedConfiguration;
 
   constructor(
     private readonly httpClient: IHttpClient,
-    private readonly flagConfigurationStore: IConfigurationStore<Flag>,
-    private readonly banditVariationConfigurationStore: IConfigurationStore<
-      BanditVariation[]
-    > | null,
-    private readonly banditModelConfigurationStore: IConfigurationStore<BanditParameters> | null,
+    private flagConfigurationStore: IConfigurationStore<Flag>,
+    private banditVariationConfigurationStore: IConfigurationStore<BanditVariation[]> | null,
+    private banditModelConfigurationStore: IConfigurationStore<BanditParameters> | null,
   ) {
     this.configuration = new StoreBackedConfiguration(
       this.flagConfigurationStore,
@@ -27,6 +25,26 @@ export default class ConfigurationRequestor {
     );
   }
 
+  /**
+   * Updates the configuration stores and recreates the StoreBackedConfiguration
+   */
+  public setConfigurationStores(
+    flagConfigurationStore: IConfigurationStore<Flag>,
+    banditVariationConfigurationStore: IConfigurationStore<BanditVariation[]> | null,
+    banditModelConfigurationStore: IConfigurationStore<BanditParameters> | null,
+  ): void {
+    this.flagConfigurationStore = flagConfigurationStore;
+    this.banditVariationConfigurationStore = banditVariationConfigurationStore;
+    this.banditModelConfigurationStore = banditModelConfigurationStore;
+
+    // Recreate the configuration with the new stores
+    this.configuration = new StoreBackedConfiguration(
+      this.flagConfigurationStore,
+      this.banditVariationConfigurationStore,
+      this.banditModelConfigurationStore,
+    );
+  }
+
   public isFlagConfigExpired(): Promise<boolean> {
     return this.flagConfigurationStore.isExpired();
   }