avoid nil

4 years ago · 38bbef7ec1
1 changed files with 4 additions and 1 deletions
--- a/weed/shell/command_volume_balance.go
+++ b/weed/shell/command_volume_balance.go
@ -205,7 +205,10 @@ type CapacityFunc func(*master_pb.DataNodeInfo) int

 func capacityByMaxVolumeCount(diskType types.DiskType) CapacityFunc {
 	return func(info *master_pb.DataNodeInfo) int {
-		diskInfo := info.DiskInfos[string(diskType)]
+		diskInfo, found := info.DiskInfos[string(diskType)]
+		if !found {
+			return 0
+		}
 		return int(diskInfo.MaxVolumeCount)
 	}
 }