Fix RateLimiter.

FStarLang · May 9, 2024 · 29afd2d · 29afd2d
1 parent f37e526
commit 29afd2d
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/server/src/signals.ts b/server/src/signals.ts
@@ -63,15 +63,20 @@ export class RateLimiter implements SignalProc {
 		if (this.timeout) {
 			this.missedFire = true;
 		} else {
-			this.timeout = setTimeout(() => {
-				this.timeout = undefined;
-				if (this.missedFire) {
-					this.missedFire = false;
-					this.handler();
-				}
-			}, this.millis);
+			this.setTimeout();
 			// Run async
 			setTimeout(() => this.handler());
 		}
 	}
+
+	private setTimeout() {
+		this.timeout = setTimeout(() => {
+			this.timeout = undefined;
+			if (this.missedFire) {
+				this.missedFire = false;
+				this.setTimeout();
+				this.handler();
+			}
+		}, this.millis);
+	}
 }