mirror of
https://github.com/QingdaoU/OnlineJudge.git
synced 2024-09-21 16:33:22 +00:00
45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
# coding=utf-8
|
||
from django.db import models
|
||
|
||
|
||
class JudgeServer(models.Model):
|
||
ip = models.GenericIPAddressField()
|
||
port = models.IntegerField()
|
||
# 这个服务器最大可能运行的判题实例数量
|
||
max_instance_number = models.IntegerField()
|
||
left_instance_number = models.IntegerField()
|
||
workload = models.IntegerField(default=0)
|
||
token = models.CharField(max_length=30)
|
||
# 进行测试用例同步的时候加锁
|
||
lock = models.BooleanField(default=False)
|
||
# status 为 false 的时候代表不使用这个服务器
|
||
status = models.BooleanField(default=True)
|
||
|
||
def use_judge_instance(self):
|
||
# 因为use 和 release 中间是判题时间,可能这个 model 的数据已经被修改了,所以不能直接使用self.xxx,否则取到的是旧数据
|
||
server = JudgeServer.objects.select_for_update().get(id=self.id)
|
||
server.left_instance_number -= 1
|
||
server.workload = 100 - int(float(server.left_instance_number) / server.max_instance_number * 100)
|
||
server.save()
|
||
|
||
def release_judge_instance(self):
|
||
# 使用原子操作
|
||
server = JudgeServer.objects.select_for_update().get(id=self.id)
|
||
server.left_instance_number += 1
|
||
server.workload = 100 - int(float(server.left_instance_number) / server.max_instance_number * 100)
|
||
server.save()
|
||
|
||
class Meta:
|
||
db_table = "judge_server"
|
||
|
||
|
||
class JudgeWaitingQueue(models.Model):
|
||
submission_id = models.CharField(max_length=40)
|
||
time_limit = models.IntegerField()
|
||
memory_limit = models.IntegerField()
|
||
test_case_id = models.CharField(max_length=40)
|
||
create_time = models.DateTimeField(auto_now_add=True)
|
||
|
||
class Meta:
|
||
db_table = "judge_waiting_queue"
|