QingdaoU/JudgeServer
1
0
Fork 0
mirror of https://github.com/QingdaoU/JudgeServer.git synced 2024-12-27 12:21:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Pool 资源泄漏 (#59)

Browse Source
This commit is contained in:
Helson 2024-04-05 23:36:46 +08:00 committed by GitHub
parent 205ae66a45
commit b28aa56d60
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
server/judge_client.py
View File

@ -32,7 +32,6 @@ class JudgeClient(object):
self._test_case_dir = test_case_dir
self._submission_dir = submission_dir
self._pool = Pool(processes=psutil.cpu_count())
self._test_case_info = self._load_test_case_info()
self._spj_version = spj_version
@ -179,18 +178,17 @@ class JudgeClient(object):
def run(self):
tmp_result = []
result = []
for test_case_file_id, _ in self._test_case_info["test_cases"].items():
tmp_result.append(self._pool.apply_async(_run, (self, test_case_file_id)))
self._pool.close()
self._pool.join()
pool = Pool(processes=psutil.cpu_count())
try:
for test_case_file_id, _ in self._test_case_info["test_cases"].items():
tmp_result.append(pool.apply_async(_run, (self, test_case_file_id)))
except Exception as e:
raise e
finally:
pool.close()
pool.join()
for item in tmp_result:
# exception will be raised, when get() is called
# # http://stackoverflow.com/questions/22094852/how-to-catch-exceptions-in-workers-in-multiprocessing
# http://stackoverflow.com/questions/22094852/how-to-catch-exceptions-in-workers-in-multiprocessing
result.append(item.get())
return result
def __getstate__(self):
# http://stackoverflow.com/questions/25382455/python-notimplementederror-pool-objects-cannot-be-passed-between-processes
self_dict = self.__dict__.copy()
del self_dict["_pool"]
return self_dict