def _handler(sid): status = Submission.get_status(sid) if status is None: return print_log('Analysing submission %s' % (sid, )) test_status = pg_session.query(pg_models.TestDataStatus).filter_by(submission_id=sid).first() info = test_status.status finished = True max_time = -1 max_memory = -1 max_status = 4 for test_data_id, solution_id in status.items(): solution = mysql_session.query(mysql_models.Solution).filter_by(solution_id=solution_id).first() result = solution.result info[str(test_data_id)]['status'] = status_map[result] info[str(test_data_id)]['time'] = solution.time info[str(test_data_id)]['memory'] = solution.memory if solution.time > max_time: max_time = solution.time if solution.memory > max_memory: max_memory = solution.memory if priority[result] > priority[max_status]: max_status = result print('\tGot solution %s, result is %s' % (solution_id, result)) if result in final_status: Submission.remove_status(sid, test_data_id) print('\t\tSolution is judged, removed from mark.') else: finished = False table = pg_models.TestDataStatus.__table__ pg_session.execute( table.update().where(table.c.submission_id == sid), {'status': info} ) pg_session.commit() submission = pg_session.query(pg_models.Submission).filter_by(id=sid).first() submission.judge_id = judger_id if not finished: Submission.push(sid) submission.status = status_map[max_status] print('\tNot finished, pushed.') else: submission.time = max_time submission.memory = max_memory submission.status = status_map[max_status] submission.finished = True print('\tFinished, unmarked.') pg_session.commit() print('\tSubmission %s updated.' % (sid, ))
def handler(sid): status = Submission.get_status(sid) # 获得该提交的数据。 oj = status['oj'] if status is not None and 'oj' in status else None if oj is not None: print("Report Submission: sid=%s, oj=%s" % (sid, oj)) is_ok, new_status, update = reporter_functions[oj](sid, status) # 回执的ret为(isOk, status, update)。包括轮询完成标志和新的status,以及做出的修改。 # update包括如下内容: # status, score, time, memory, finished if is_ok and status is not None: Submission.unmark(sid) else: Submission.push(sid) Submission.mark(sid, new_status) if update is not None: # 该项不为None时,需要对sdustoj的提交做出修改。 submission = pg_session.query( pg_models.Submission).filter_by(id=sid).first() AlchemyUtils.update_submission(submission, update) pg_session.commit() else: Submission.unmark(sid)
def _handler(sid): # 处理某个提交。 global mysql_session, pg_session status = Submission.get_status(sid) # sdustoj上的提交ID -> 提交的内容字典(sdustoj上的测试数据ID,hustoj上的提交ID) if status is None: # 如果字典不存在,也就是这个提交没有被标记处理,那么就离开。 return print_log('Analysing submission %s' % (sid, )) test_status = pg_session.query(pg_models.TestDataStatus).filter_by(submission_id=sid).first() # 查询在sdustoj上的该提交的提交状态。 info = test_status.status # 获得这个提交状态。这个提交状态是一个json结构表的字典表。包含如下信息: # status: 提交状态 # time:耗时 # memory: 空间 finished = True # 这一部分用来从所有的测试数据中得出最长用时/最大内存/最高优先级的结果状态。 max_time = -1 max_memory = -1 max_status = 4 sum_score = 0 sum_weight = 0 for test_data_id, solution_id in status.items(): # 遍历查询所有测试数据的提交的测试数据ID和hustoj提交的ID。 while True: try: solution = mysql_session.query(mysql_models.Solution).filter_by(solution_id=solution_id).first() except OperationalError: # MySQL数据库因为长连接断开。重连一次。 mysql_session = MysqlSession() solution = None if solution is not None: break # 从hustoj查询这个提交。 result = solution.result # 获得提交的结果状态 map_status, map_score = status_map[result] # 这个map将hustoj的结果编号映射到sdustoj的缩写上去。 weight = info[str(test_data_id)]['weight'] if 'weight' in info[str(test_data_id)] else 1 info[str(test_data_id)]['status'] = map_status info[str(test_data_id)]['score'] = map_score info[str(test_data_id)]['time'] = solution.time info[str(test_data_id)]['memory'] = solution.memory # 上三行是将正在查看的测试数据的评测结果写到结果信息表中。 if solution.time > max_time: max_time = solution.time if solution.memory > max_memory: max_memory = solution.memory if priority[result] > priority[max_status]: max_status = result # 上三行获得更高优先级的结果数据。 priority用来规定结果状态的优先级。 # 下面计算权重数值。 sum_score += map_score sum_weight += weight print('\tGot solution %s, result is %s' % (solution_id, result)) if result in final_status: # 终止状态代号。 Submission.remove_status(sid, test_data_id) # 既然已经终止了就将此条测试数据删掉防止下次还用。 print('\t\tSolution is judged, removed from mark.') else: finished = False # 这个标记用来标记还没有完成全部的测试数据的评测。 table = pg_models.TestDataStatus.__table__ # 测试数据状态的表本身? pg_session.execute( table.update().where(table.c.submission_id == sid), {'status': info} ) # 看起来这句话的意思是更新该条提交在sdustoj数据库中的信息…… pg_session.commit() # 提交更改 submission = pg_session.query(pg_models.Submission).filter_by(id=sid).first() # 查询该条提交。 submission.judge_id = judger_id # 更新评测机的ID为本机ID。 if not finished: # 还没有完成全部测试数据评测。 Submission.push(sid) # 将该提交放回分析队列等待下一次分析 map_status, _ = status_map[max_status] submission.status = map_status submission.score_info = sum_score / sum_weight if sum_weight > 0 else 0 print('\tNot finished, pushed.') else: submission.time = max_time submission.memory = max_memory # 现在可以更新最大用时和最大空间占用了。 map_status, _ = status_map[max_status] submission.status = map_status submission.score_info = sum_score / sum_weight if sum_weight > 0 else 0 submission.finished = True # 标记该提交为已完成。 print('\tFinished, unmarked.') submission.update_time = datetime.now() pg_session.commit() print('\tSubmission %s updated.' % (sid, ))