fix missing piece

slackhq · Jun 28, 2024 · cd5adbb · cd5adbb
1 parent 31bb813
commit cd5adbb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/go/vt/topo/topoproto/srvkeyspace.go b/go/vt/topo/topoproto/srvkeyspace.go
@@ -51,6 +51,9 @@ func (sra ShardReferenceArray) Sort() { sort.Sort(sra) }
 // SrvKeyspaceGetPartition returns a Partition for the given tablet type,
 // or nil if it's not there.
 func SrvKeyspaceGetPartition(sk *topodatapb.SrvKeyspace, tabletType topodatapb.TabletType) *topodatapb.SrvKeyspace_KeyspacePartition {
+	if sk == nil {
+		return nil
+	}
 	for _, p := range sk.Partitions {
 		if p.ServedType == tabletType {
 			return p