avoid nil

This commit is contained in:
Chris Lu 2021-02-16 04:16:46 -08:00
parent cb9cc29518
commit 38bbef7ec1

View file

@ -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)
}
}