Fix bridge (https://github.com/DMOJ/online-judge/pull/2080)

2022-12-27 13:19:02 -06:00 · 2022-12-27 13:19:02 -06:00 · 7d5efa60c7
commit 7d5efa60c7
parent 304fa0e0f3
3 changed files with 13 additions and 6 deletions
--- a/judge/bridge/base_handler.py
+++ b/judge/bridge/base_handler.py
@ -145,6 +145,9 @@ class ZlibPacketHandler(metaclass=RequestHandlerMeta):
    def on_timeout(self):
        pass

+    def on_cleanup(self):
+        pass
+
    def handle(self):
        try:
            tag = self.read_size()
@ -204,6 +207,8 @@ class ZlibPacketHandler(metaclass=RequestHandlerMeta):
            if e.__class__.__name__ == "cancel_wait_ex":
                return
            raise
+        finally:
+            self.on_cleanup()

    def send(self, data):
        compressed = zlib.compress(data.encode("utf-8"))