评测一个代码的主要流程代码
def run(problem_id, solution_id, language, data_count, user_id, dblock):
low_level()
'''获取程序执行时间和内存'''
#dblock.acquire()
#time_limit, mem_limit = get_problem_limit(problem_id)
#dblock.release()
time_limit = 5 * 1000 #in MS
mem_limit = 64 * 1024 #in KB
program_info = {
"solution_id": solution_id,
"problem_id": problem_id,
"take_time": 0,
"take_memory": 0,
"user_id": user_id,
"result": 12,
"compile_info": "null",
}
if check_dangerous_code(solution_id, language) == False:
program_info['result'] = result_code["Runtime Error"]
return program_info
compile_result = compile(solution_id, language, program_info, dblock)
if compile_result is False: # 编译错误
program_info['result'] = result_code["Compile Error"]
return program_info
if data_count == 0: # 没有测试数据
program_info['result'] = result_code["System Error"]
return program_info
result = judge(
solution_id,
problem_id,
data_count,
time_limit,
mem_limit,
program_info,
result_code,
language)
logging.info("Running program over. \n\tThe result is : " + str(result))
logging.debug(result)
return result
清理临时文件,删除数据缓存
def clean_work_dir(solution_id):
'''清理word目录,删除临时文件'''
dir_name = os.path.join(config.work_dir, str(solution_id))
print(dir_name)
shutil.rmtree(dir_name)
开启进程
def start_work_thread():
'''开启工作线程'''
for i in range(config.count_thread):
t = threading.Thread(target=worker)
t.deamon = True
t.start()
def start_get_task():
'''开启获取任务线程'''
t = threading.Thread(target=put_task_into_queue, name="get_task")
t.deamon = True
t.start()
配置文件
#!/usr/bin/env python
#coding=utf-8
#开启评测线程数目
count_thread = 4
#评测程序队列容量
queue_size = 4
#数据库地址
db_host = "localhost"
#数据库用户名
db_user = "user"
#数据库密码
db_password = "password"
#数据库名字
db_name = "judge_ol"
#数据库编码
db_charset = "utf8"
#work评判目录
work_dir = "/tmp/OJ_Working/working"
#data测试数据目录
data_dir = "/tmp/OJ_Working/testdata"
#自动清理评work目录
auto_clean = True