fix: prevent reboot loops (#345)

apify · Dec 19, 2024 · 271bc99 · 271bc99
1 parent deda03b
commit 271bc99
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/packages/apify/src/actor.ts b/packages/apify/src/actor.ts
@@ -81,6 +81,11 @@ export class Actor<Data extends Dictionary = Dictionary> {
      */
     private warnedAboutMissingInitCall = false;
 
+    /**
+     * Set if the Actor is currently rebooting.
+     */
+    private isRebooting = false;
+
     constructor(options: ConfigurationOptions = {}) {
         // use default configuration object if nothing overridden (it fallbacks to env vars)
         this.config = Object.keys(options).length === 0 ? Configuration.getGlobalConfig() : new Configuration(options);
@@ -459,6 +464,13 @@ export class Actor<Data extends Dictionary = Dictionary> {
             return;
         }
 
+        if (this.isRebooting) {
+            log.debug('Actor is already rebooting, skipping the additional reboot call.');
+            return;
+        }
+
+        this.isRebooting = true;
+
         // Waiting for all the listeners to finish, as `.reboot()` kills the container.
         await Promise.all([
             // `persistState` for individual RequestLists, RequestQueue... instances to be persisted