diff options
Diffstat (limited to 'weed/shell/command_volume_balance.go')
| -rw-r--r-- | weed/shell/command_volume_balance.go | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/weed/shell/command_volume_balance.go b/weed/shell/command_volume_balance.go index 9fa915ea4..f5fdf2216 100644 --- 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) } } |