Cloned DMOJ

judge/bridge/djangohandler.py

@@ -0,0 +1,67 @@
+import json
+import logging
+import struct
+
+from event_socket_server import ZlibPacketHandler
+
+logger = logging.getLogger('judge.bridge')
+size_pack = struct.Struct('!I')
+
+
+class DjangoHandler(ZlibPacketHandler):
+    def __init__(self, server, socket):
+        super(DjangoHandler, self).__init__(server, socket)
+
+        self.handlers = {
+            'submission-request': self.on_submission,
+            'terminate-submission': self.on_termination,
+            'disconnect-judge': self.on_disconnect,
+        }
+        self._to_kill = True
+        # self.server.schedule(5, self._kill_if_no_request)
+
+    def _kill_if_no_request(self):
+        if self._to_kill:
+            logger.info('Killed inactive connection: %s', self._socket.getpeername())
+            self.close()
+
+    def _format_send(self, data):
+        return super(DjangoHandler, self)._format_send(json.dumps(data, separators=(',', ':')))
+
+    def packet(self, packet):
+        self._to_kill = False
+        packet = json.loads(packet)
+        try:
+            result = self.handlers.get(packet.get('name', None), self.on_malformed)(packet)
+        except Exception:
+            logger.exception('Error in packet handling (Django-facing)')
+            result = {'name': 'bad-request'}
+        self.send(result, self._schedule_close)
+
+    def _schedule_close(self):
+        self.server.schedule(0, self.close)
+
+    def on_submission(self, data):
+        id = data['submission-id']
+        problem = data['problem-id']
+        language = data['language']
+        source = data['source']
+        priority = data['priority']
+        if not self.server.judges.check_priority(priority):
+            return {'name': 'bad-request'}
+        self.server.judges.judge(id, problem, language, source, priority)
+        return {'name': 'submission-received', 'submission-id': id}
+
+    def on_termination(self, data):
+        return {'name': 'submission-received', 'judge-aborted': self.server.judges.abort(data['submission-id'])}
+
+    def on_disconnect(self, data):
+        judge_id = data['judge-id']
+        force = data['force']
+        self.server.judges.disconnect(judge_id, force=force)
+
+    def on_malformed(self, packet):
+        logger.error('Malformed packet: %s', packet)
+
+    def on_close(self):
+        self._to_kill = False