From 49ac8d52bde038cda3897ef072c44e7802dd69d7 Mon Sep 17 00:00:00 2001
From: Qi <37133148+aobt@users.noreply.github.com>
Date: Mon, 25 Mar 2024 21:07:57 +0800
Subject: [PATCH] fix index map in storage_manager (#3145)

Signed-off-by: Qi <37133148+aobt@users.noreply.github.com>
---
 client/daemon/storage/storage_manager.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/client/daemon/storage/storage_manager.go b/client/daemon/storage/storage_manager.go
index 48a90c7e7..ea12a3696 100644
--- a/client/daemon/storage/storage_manager.go
+++ b/client/daemon/storage/storage_manager.go
@@ -625,7 +625,11 @@ func (s *storageManager) cleanIndex(taskID, peerID string) {
 		}
 		remain = append(remain, t)
 	}
-	s.indexTask2PeerTask[taskID] = remain
+	if len(remain) > 0 {
+		s.indexTask2PeerTask[taskID] = remain
+	} else {
+		delete(s.indexTask2PeerTask, taskID)
+	}
 }
 
 func (s *storageManager) cleanSubIndex(taskID, peerID string) {
@@ -644,7 +648,11 @@ func (s *storageManager) cleanSubIndex(taskID, peerID string) {
 		}
 		remain = append(remain, t)
 	}
-	s.subIndexTask2PeerTask[taskID] = remain
+	if len(remain) > 0 {
+		s.subIndexTask2PeerTask[taskID] = remain
+	} else {
+		delete(s.subIndexTask2PeerTask, taskID)
+	}
 }
 
 func (s *storageManager) ValidateDigest(req *PeerTaskMetadata) error {