feat: supporting nosync and nofailover node tags

k-web-s · Mar 18, 2024 · 51bc574 · 51bc574
1 parent 83661e3
commit 51bc574
Show file tree

Hide file tree

Showing 4 changed files with 70 additions and 0 deletions.
diff --git a/api/v1alpha1/patronipostgres_types.go b/api/v1alpha1/patronipostgres_types.go
@@ -31,13 +31,28 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 )
 
+// NodeTags holds a subset of Patroni tags
+// https://patroni.readthedocs.io/en/latest/yaml_configuration.html#tags
+type NodeTags struct {
+	// NoSync If set to true the node will never be selected as a synchronous replica.
+	NoSync bool `json:"nosync,omitempty"`
+
+	// NoFailover controls whether this node is allowed to participate in the leader
+	// race and become a leader. Defaults to false, meaning this node _can_
+	// participate in leader races.
+	NoFailover bool `json:"nofailover,omitempty"`
+}
+
 // Node represents a PatroniPostgres node's configuration
 type Node struct {
 	// StorageClassName references a storage class to allocate volume from
 	StorageClassName string `json:"storageClassName"`
 
 	// AccessMode allows for overriding implicit ReadWriteOnce accessmode
 	AccessMode corev1.PersistentVolumeAccessMode `json:"accessMode,omitempty"`
+
+	// Tags for Node
+	Tags NodeTags `json:"tags,omitempty"`
 }
 
 type VolumeStatus struct {

diff --git a/config/crd/bases/kwebs.cloud_patronipostgres.yaml b/config/crd/bases/kwebs.cloud_patronipostgres.yaml
@@ -2191,6 +2191,20 @@ spec:
                       description: StorageClassName references a storage class to
                         allocate volume from
                       type: string
+                    tags:
+                      description: Tags for Node
+                      properties:
+                        nofailover:
+                          description: NoFailover controls whether this node is allowed
+                            to participate in the leader race and become a leader.
+                            Defaults to false, meaning this node _can_ participate
+                            in leader races.
+                          type: boolean
+                        nosync:
+                          description: NoSync If set to true the node will never be
+                            selected as a synchronous replica.
+                          type: boolean
+                      type: object
                   required:
                   - storageClassName
                   type: object

diff --git a/config/deploy.yaml b/config/deploy.yaml
@@ -2474,6 +2474,20 @@ spec:
                       description: StorageClassName references a storage class to
                         allocate volume from
                       type: string
+                    tags:
+                      description: Tags for Node
+                      properties:
+                        nofailover:
+                          description: NoFailover controls whether this node is allowed
+                            to participate in the leader race and become a leader.
+                            Defaults to false, meaning this node _can_ participate
+                            in leader races.
+                          type: boolean
+                        nosync:
+                          description: NoSync If set to true the node will never be
+                            selected as a synchronous replica.
+                          type: boolean
+                      type: object
                   required:
                   - storageClassName
                   type: object

diff --git a/private/controllers/statefulset/statefulset.go b/private/controllers/statefulset/statefulset.go
@@ -28,6 +28,7 @@ package statefulset
 import (
 	"encoding/json"
 	"fmt"
+	"strings"
 
 	appsv1 "k8s.io/api/apps/v1"
 	corev1 "k8s.io/api/core/v1"
@@ -281,6 +282,8 @@ func ReconcileSts(ctx context.Context, p *v1alpha1.PatroniPostgres) (sts *appsv1
 		},
 	}
 
+	sts.Spec.Template.Spec.Containers[0].Env = append(sts.Spec.Template.Spec.Containers[0].Env, genNodeTagsEnvs(p)...)
+
 	sts.Spec.Template.Spec.Containers = append(sts.Spec.Template.Spec.Containers, p.Spec.ExtraContainers...)
 
 	if create {
@@ -314,3 +317,27 @@ func GetK8SStatefulSet(ctx context.Context, p *v1alpha1.PatroniPostgres) (sts *a
 	err = ctx.Get(ctx, types.NamespacedName{Namespace: p.Namespace, Name: p.Name}, sts)
 	return
 }
+
+func genNodeTagsEnvs(p *v1alpha1.PatroniPostgres) (envs []corev1.EnvVar) {
+	podPrefix := strings.ReplaceAll(p.Name, "-", "_")
+
+	for idx := range p.Spec.Nodes {
+		node := &p.Spec.Nodes[idx]
+
+		if node.Tags.NoSync {
+			envs = append(envs, corev1.EnvVar{
+				Name:  fmt.Sprintf("PATRONI_NODE_%s_%d_TAG_%s", podPrefix, idx, "nosync"),
+				Value: "true",
+			})
+		}
+
+		if node.Tags.NoFailover {
+			envs = append(envs, corev1.EnvVar{
+				Name:  fmt.Sprintf("PATRONI_NODE_%s_%d_TAG_%s", podPrefix, idx, "nofailover"),
+				Value: "true",
+			})
+		}
+	}
+
+	return
+}