fix warmup deadlock in cherry

bananaml · Oct 25, 2023 · 0580ba9 · 0580ba9
1 parent b6962c3
commit 0580ba9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/potassium/potassium.py b/potassium/potassium.py
@@ -225,6 +225,7 @@ def _create_flask_app(self):
         def handle(path):
             with self._sequence_number_lock:
                 self._sequence_number += 1
+
             route = "/" + path
             if route not in self._endpoints:
                 abort(404)
@@ -234,6 +235,8 @@ def handle(path):
 
         @flask_app.route('/_k/warmup', methods=["POST"])
         def warm():
+            with self._sequence_number_lock:
+                self._sequence_number += 1
             res = make_response({
                 "warm": True,
             })