mirror of
https://github.com/QingdaoU/JudgeServer.git
synced 2025-11-04 14:50:01 +08:00
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
# coding=utf-8
|
|
from __future__ import unicode_literals
|
|
import os
|
|
import json
|
|
import requests
|
|
import hashlib
|
|
|
|
from exception import JudgeServiceError
|
|
from utils import server_info, logger, token
|
|
|
|
|
|
class JudgeService(object):
|
|
def __init__(self):
|
|
self.service_url = os.environ["SERVICE_URL"]
|
|
self.backend_url = os.environ["BACKEND_URL"]
|
|
|
|
def _request(self, data):
|
|
try:
|
|
r = requests.post(self.backend_url, json=data,
|
|
headers={"X-JUDGE-SERVER-TOKEN": token}, timeout=5).json()
|
|
except Exception as e:
|
|
logger.exception(e)
|
|
raise JudgeServiceError(e.message)
|
|
if r["error"]:
|
|
raise JudgeServiceError(r["data"])
|
|
|
|
def heartbeat(self):
|
|
data = server_info()
|
|
data["action"] = "heartbeat"
|
|
data["service_url"] = self.service_url
|
|
self._request(data)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
service = JudgeService()
|
|
service.heartbeat()
|
|
exit(0)
|
|
except Exception as e:
|
|
logger.exception(e)
|
|
exit(1)
|