chore: cluster connect to replica whose pod role label is set as prim…

…ary (#65)
apecloud · Nov 22, 2023 · e4302b6 · e4302b6
1 parent 756333e
commit e4302b6
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/pkg/cluster/helper.go b/pkg/cluster/helper.go
@@ -136,7 +136,17 @@ func getInstanceInfoByList(dynamic dynamic.Interface, name, componentName, names
 	}
 
 	for _, o := range objs.Items {
-		infos = append(infos, &InstanceInfo{Name: o.GetName()})
+		info := &InstanceInfo{Name: o.GetName()}
+		podLabels := o.GetLabels()
+		role, ok := podLabels[constant.RoleLabelKey]
+		if ok {
+			info.Role = role
+		}
+		if role == constant.Primary || role == constant.Leader {
+			infos = append([]*InstanceInfo{info}, infos...)
+		} else {
+			infos = append(infos, info)
+		}
 	}
 	return infos
 }