From a82cceada0df66cf48d646a3cd6a55556ebf9962 Mon Sep 17 00:00:00 2001 From: rabbitali Date: Tue, 23 Jul 2024 18:55:50 +0800 Subject: [PATCH 1/1] fix bug with host count in cve fix task --- apollo/database/proxy/task/cve_fix.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/apollo/database/proxy/task/cve_fix.py b/apollo/database/proxy/task/cve_fix.py index fa3f2a9..1aa5546 100644 --- a/apollo/database/proxy/task/cve_fix.py +++ b/apollo/database/proxy/task/cve_fix.py @@ -121,7 +121,8 @@ class CveFixTaskProxy(TaskProxy): wait_fix_rpms = dict() for task_info in fix_host_rpm_info: - wait_fix_rpms[task_info["cve_id"]] = dict(rpms=task_info.get("rpms", []), hosts=list(host_dict.keys())) + host_list = [host_info["host_id"] for host_info in task_info["host_info"]] + wait_fix_rpms[task_info["cve_id"]] = dict(rpms=task_info.get("rpms", []), hosts=host_list) hotpatch_fix_rpms, coldpatch_fix_rpms = self._get_cold_and_hotpatch_fix_rpm(wait_fix_rpms, data["takeover"]) fix_tasks = [] -- 2.33.0