From dffeb775d77e1c608eed8dc9553fcc46ebbfc450 Mon Sep 17 00:00:00 2001
From: root <root@localhost.localdomain>
Date: Wed, 24 Jul 2024 16:46:56 +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 efdc91e..468c03a 100644
--- a/apollo/database/proxy/task/cve_fix.py
+++ b/apollo/database/proxy/task/cve_fix.py
@@ -133,7 +133,8 @@ class CveFixTaskProxy(TaskProxy):
             return PARAM_ERROR, 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
