Skip to content

Commit 26c76be

Browse files
committed
support fetch log and fetch job result seperately
1 parent e5e7419 commit 26c76be

File tree

2 files changed

+19
-13
lines changed

2 files changed

+19
-13
lines changed

python/ppc_model/model_result/task_result_handler.py

Lines changed: 14 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,13 @@
1111

1212

1313
class TaskResultRequest:
14-
def __init__(self, job_id, task_type, only_fetch_log, user):
14+
def __init__(self, job_id, task_type,
15+
fetch_log: bool,
16+
fetch_result: bool, user):
1517
self.job_id = job_id
1618
self.task_type = task_type
17-
self.only_fetch_log = only_fetch_log
19+
self.fetch_log = fetch_log
20+
self.fetch_result = fetch_result
1821
self.user = user
1922

2023

@@ -346,7 +349,7 @@ def __init__(self, task_result_request: TaskResultRequest, components: Initializ
346349

347350
def get_response(self):
348351
response = dict()
349-
if not self.task_result_request.only_fetch_log:
352+
if self.task_result_request.fetch_result is True:
350353
merged_result = dict()
351354
for result in self.result_list:
352355
merged_result.update(result.to_dict())
@@ -356,13 +359,14 @@ def get_response(self):
356359
response = {"jobPlanetResult": merged_result,
357360
"modelData": self.model_data}
358361
# record the log
359-
log_size, log_path, log_content = self.components.log_retriever.retrieve_log(
360-
self.task_result_request.job_id, self.task_result_request.user)
361-
log_result = {}
362-
log_result.update({"logSize": log_size})
363-
log_result.update({"logPath": log_path})
364-
log_result.update({"logContent": log_content})
365-
response.update({"logDetail": log_result})
362+
if self.task_result_request.fetch_log is True:
363+
log_size, log_path, log_content = self.components.log_retriever.retrieve_log(
364+
self.task_result_request.job_id, self.task_result_request.user)
365+
log_result = {}
366+
log_result.update({"logSize": log_size})
367+
log_result.update({"logPath": log_path})
368+
log_result.update({"logContent": log_content})
369+
response.update({"logDetail": log_result})
366370
return utils.make_response(PpcErrorCode.SUCCESS.get_code(), PpcErrorCode.SUCCESS.get_msg(), response)
367371

368372
def _get_evaluation_result(self):

python/ppc_model/network/http/model_controller.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -105,12 +105,14 @@ def post(self, task_id):
105105
f"get task result, task_id: {task_id}, args: {args}")
106106
user_name = args['user']
107107
task_type = args['jobType']
108-
only_fetch_log = {'True': True, 'False': False}.get(
109-
args['onlyFetchLog'])
108+
fetch_log = {'True': True, 'False': False}.get(
109+
args['fetchLog'])
110+
fetch_job_result = {'True': True, 'False': False}.get(
111+
args['fetchJobResult'])
110112
components.logger().info(
111113
f"get_job_direct_result_response, job: {task_id}")
112114
task_result_request = TaskResultRequest(
113-
task_id, task_type, only_fetch_log, user_name)
115+
task_id, task_type, fetch_log, fetch_job_result, user_name)
114116
job_result_handler = TaskResultHandler(
115117
task_result_request=task_result_request, components=components)
116118
response = job_result_handler.get_response()

0 commit comments

Comments
 (0)