From a82cceada0df66cf48d646a3cd6a55556ebf9962 Mon Sep 17 00:00:00 2001
From: rabbitali <wenxin32@foxmail.com>
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

