feat: load empty return false in persistentcache (#3697)

Signed-off-by: Gaius <[email protected]>
dragonflyoss · Dec 10, 2024 · 79a845e · 79a845e
1 parent 50cbd6f
commit 79a845e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/scheduler/resource/persistentcache/host_manager.go b/scheduler/resource/persistentcache/host_manager.go
@@ -89,6 +89,10 @@ func (h *hostManager) Load(ctx context.Context, hostID string) (*Host, bool) {
 		return nil, false
 	}
 
+	if len(rawHost) == 0 {
+		return nil, false
+	}
+
 	// Set integer fields from raw host.
 	port, err := strconv.ParseInt(rawHost["port"], 10, 32)
 	if err != nil {

diff --git a/scheduler/resource/persistentcache/peer_manager.go b/scheduler/resource/persistentcache/peer_manager.go
@@ -98,6 +98,10 @@ func (p *peerManager) Load(ctx context.Context, peerID string) (*Peer, bool) {
 		return nil, false
 	}
 
+	if len(rawPeer) == 0 {
+		return nil, false
+	}
+
 	persistent, err := strconv.ParseBool(rawPeer["persistent"])
 	if err != nil {
 		log.Errorf("parsing persistent failed: %v", err)

diff --git a/scheduler/resource/persistentcache/task_manager.go b/scheduler/resource/persistentcache/task_manager.go
@@ -75,6 +75,10 @@ func (t *taskManager) Load(ctx context.Context, taskID string) (*Task, bool) {
 		return nil, false
 	}
 
+	if len(rawTask) == 0 {
+		return nil, false
+	}
+
 	// Set integer fields from raw task.
 	persistentReplicaCount, err := strconv.ParseUint(rawTask["persistent_replica_count"], 10, 64)
 	if err != nil {