K8SPSMDB-1061: Refactor reconcileStatefulSet

pkg/apis/psmdb/v1/psmdb_types.go

@@ -178,6 +178,7 @@ type MultiCluster struct {
 type AppState string
 
 const (
+	AppStateNone     AppState = ""
 	AppStateInit     AppState = "initializing"
 	AppStateStopping AppState = "stopping"
 	AppStatePaused   AppState = "paused"
@@ -551,6 +552,29 @@ func (r ReplsetSpec) CustomReplsetName() (string, error) {
 	return cfg.Replication.ReplSetName, nil
 }
 
+func (r *ReplsetSpec) MongodLabels(cr *PerconaServerMongoDB) map[string]string {
+	return map[string]string{
+		"app.kubernetes.io/name":       "percona-server-mongodb",
+		"app.kubernetes.io/instance":   cr.Name,
+		"app.kubernetes.io/replset":    r.Name,
+		"app.kubernetes.io/managed-by": "percona-server-mongodb-operator",
+		"app.kubernetes.io/part-of":    "percona-server-mongodb",
+		"app.kubernetes.io/component":  "mongod",
+	}
+}
+
+func (r *ReplsetSpec) ArbiterLabels(cr *PerconaServerMongoDB) map[string]string {
+	ls := r.MongodLabels(cr)
+	ls["app.kubernetes.io/component"] = "arbiter"
+	return ls
+}
+
+func (r *ReplsetSpec) NonVotingLabels(cr *PerconaServerMongoDB) map[string]string {
+	ls := r.MongodLabels(cr)
+	ls["app.kubernetes.io/component"] = "nonVoting"
+	return ls
+}
+
 type LivenessProbeExtended struct {
 	corev1.Probe        `json:",inline"`
 	StartupDelaySeconds int `json:"startupDelaySeconds,omitempty"`

pkg/controller/perconaservermongodb/helpers_test.go

@@ -0,0 +1,31 @@
+package perconaservermongodb
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+
+	"k8s.io/apimachinery/pkg/util/yaml"
+
+	api "github.com/percona/percona-server-mongodb-operator/pkg/apis/psmdb/v1"
+)
+
+func readDefaultCR(t *testing.T, name, namespace string) *api.PerconaServerMongoDB {
+	t.Helper()
+
+	data, err := os.ReadFile(filepath.Join("..", "..", "..", "deploy", "cr.yaml"))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	cr := new(api.PerconaServerMongoDB)
+
+	if err := yaml.Unmarshal(data, cr); err != nil {
+		t.Fatal(err)
+	}
+
+	cr.Name = name
+	cr.Namespace = namespace
+
+	return cr
+}