bugfix: avoid nil point

projecteru2 · Nov 25, 2020 · 2c80812 · 2c80812
1 parent c306654
commit 2c80812
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/rpc/transform.go b/rpc/transform.go
@@ -531,6 +531,9 @@ func toCoreExecuteWorkloadOptions(b *pb.ExecuteWorkloadOptions) (opts *types.Exe
 }
 
 func toRPCCapacityMessage(msg *types.CapacityMessage) *pb.CapacityMessage {
+	if msg == nil {
+		return nil
+	}
 	caps := map[string]int64{}
 	for nodename, capacity := range msg.NodeCapacities {
 		caps[nodename] = int64(capacity)