feat: add locking to state generate

paraswap · Sep 14, 2023 · 6714ef8 · 6714ef8
1 parent 4704f8c
commit 6714ef8
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/stateful-event-subscriber.ts b/src/stateful-event-subscriber.ts
@@ -51,6 +51,8 @@ export abstract class StatefulEventSubscriber<State>
 
   private _lastPublishedTimeMs: number = 0;
 
+  private isForcedGenerateStateIsRunning = false;
+
   constructor(
     public readonly parentName: string,
     _name: string,
@@ -95,8 +97,16 @@ export abstract class StatefulEventSubscriber<State>
       this.logger.debug(
         `${this.parentName}: ${this.name}: forced to regenerate state`,
       );
-      const state = await this.generateState(blockNumber);
-      this.setState(state, blockNumber);
+      if (this.isForcedGenerateStateIsRunning) {
+        return;
+      }
+      this.isForcedGenerateStateIsRunning = true;
+      try {
+        const state = await this.generateState(blockNumber);
+        this.setState(state, blockNumber);
+      } finally {
+        this.isForcedGenerateStateIsRunning = false;
+      }
     } else {
       if (this.dexHelper.config.isSlave && this.masterPoolNeeded) {
         let stateAsString = await this.dexHelper.cache.hget(