# coding=utf-8 from __future__ import unicode_literals import _judger import psutil import socket import logging import fcntl import hashlib import os from config import COUNTER_FILE_PATH, TOKEN_FILE_PATH from exception import JudgeClientError logger = logging.getLogger(__name__) handler = logging.FileHandler("/log/judge_server.log") formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s') handler.setFormatter(formatter) logger.addHandler(handler) logger.setLevel(logging.WARNING) class TaskCounter(object): def __init__(self, file_path=COUNTER_FILE_PATH): self.f = open(file_path, "r+") self.fd = self.f.fileno() def update(self, action): # lock file fcntl.lockf(self.fd, fcntl.LOCK_EX) try: value = self.f.read() self.f.seek(0) self.f.write(str(int(value) + action)) finally: # release lock fcntl.lockf(self.fd, fcntl.LOCK_UN) def get(self): # lock file fcntl.lockf(self.fd, fcntl.LOCK_EX) try: value = self.f.read() return int(value) finally: # release lock fcntl.lockf(self.fd, fcntl.LOCK_UN) def server_info(): ver = _judger.VERSION return {"hostname": socket.gethostname(), "cpu": psutil.cpu_percent(), "cpu_core": psutil.cpu_count(), "memory": psutil.virtual_memory().percent, "judger_version": ".".join([str((ver >> 16) & 0xff), str((ver >> 8) & 0xff), str(ver & 0xff)]), "running_task_number": TaskCounter().get()} def get_token(): try: with open(TOKEN_FILE_PATH, "r") as f: return f.read() except IOError: raise JudgeClientError("token.txt not found") token = hashlib.sha256(get_token()).hexdigest()