feat(delivery): integrate PostgreSQL delivery

2026-07-31 10:30:03 +08:00
parent 048ac6507e
commit 66fde73792
25 changed files with 2110 additions and 139 deletions
@@ -0,0 +1,104 @@
+package service
+
+import (
+	"context"
+	"testing"
+)
+
+func TestValidatePostgreSQLDeliveryInput(t *testing.T) {
+	valid := PostgreSQLDeliveryInput{BusinessLineID: 1, TargetID: 2, Namespace: "team-a", ClusterName: "orders-pg", VersionMajor: "16", Topology: "standalone", CPUMilli: 500, MemoryMi: 2048, StorageGi: 20}
+	if err := validatePostgreSQLDeliveryInput(valid, []string{"/data"}); err != nil {
+		t.Fatalf("valid standalone input rejected: %v", err)
+	}
+	aliasVersion := valid
+	aliasVersion.VersionMajor = ""
+	aliasVersion.PostgreSQLVersion = "15"
+	if err := validatePostgreSQLDeliveryInput(aliasVersion, []string{"/data"}); err != nil {
+		t.Fatalf("postgresql_version alias rejected: %v", err)
+	}
+	replicated := valid
+	replicated.VersionMajor = "15"
+	replicated.Topology = "primary_replica"
+	replicated.ReplicaCount = 2
+	replicated.TargetHosts = []string{"pg-a", "pg-b", "pg-c"}
+	replicated.DataRoot = "/data/postgresql"
+	replicated.MaxConnections = 500
+	if err := validatePostgreSQLDeliveryInput(replicated, []string{"/data"}); err != nil {
+		t.Fatalf("valid primary_replica input rejected: %v", err)
+	}
+	for name, mutate := range map[string]func(*PostgreSQLDeliveryInput){
+		"unsupported version": func(in *PostgreSQLDeliveryInput) { in.VersionMajor = "14" },
+		"bad topology":        func(in *PostgreSQLDeliveryInput) { in.Topology = "patroni" },
+		"standalone replicas": func(in *PostgreSQLDeliveryInput) { in.ReplicaCount = 1 },
+		"missing replicas":    func(in *PostgreSQLDeliveryInput) { in.Topology = "primary_replica" },
+		"too many replicas":   func(in *PostgreSQLDeliveryInput) { in.Topology = "primary_replica"; in.ReplicaCount = 8 },
+		"duplicate hosts": func(in *PostgreSQLDeliveryInput) {
+			in.Topology = "primary_replica"
+			in.ReplicaCount = 1
+			in.TargetHosts = []string{"pg-a", "pg-a"}
+		},
+		"wrong host count": func(in *PostgreSQLDeliveryInput) {
+			in.Topology = "primary_replica"
+			in.ReplicaCount = 2
+			in.TargetHosts = []string{"pg-a", "pg-b"}
+		},
+		"bad root":              func(in *PostgreSQLDeliveryInput) { in.DataRoot = "/tmp/postgresql" },
+		"unsupported root disk": func(in *PostgreSQLDeliveryInput) { in.DataRoot = "/disk1/postgresql" },
+	} {
+		input := valid
+		mutate(&input)
+		if err := validatePostgreSQLDeliveryInput(input, []string{"/data"}); err == nil {
+			t.Errorf("%s was accepted", name)
+		}
+	}
+}
+
+func TestAllocatePostgreSQLPort(t *testing.T) {
+	if port, err := allocatePostgreSQLPort(0, nil); err != nil || port != postgresqlPortPoolStart {
+		t.Fatalf("expected pool start %d, got %d err=%v", postgresqlPortPoolStart, port, err)
+	}
+	if port, err := allocatePostgreSQLPort(0, []int{15432, 15433}); err != nil || port != 15434 {
+		t.Fatalf("expected 15434, got %d err=%v", port, err)
+	}
+	if _, err := allocatePostgreSQLPort(15432, []int{15432}); err == nil {
+		t.Fatal("occupied requested port was accepted")
+	}
+	if _, err := allocatePostgreSQLPort(5432, nil); err == nil {
+		t.Fatal("port outside the PostgreSQL pool was accepted")
+	}
+}
+
+func TestAllocateReachablePostgreSQLPortSkipsListeningPorts(t *testing.T) {
+	probed := []int{}
+	port, err := allocateReachablePostgreSQLPort(context.Background(), "pg.example", nil, func(_ context.Context, host string, port int) bool {
+		if host != "pg.example" {
+			t.Fatalf("unexpected probe host %q", host)
+		}
+		probed = append(probed, port)
+		return port == postgresqlPortPoolStart
+	})
+	if err != nil {
+		t.Fatalf("allocate reachable port: %v", err)
+	}
+	if port != postgresqlPortPoolStart+1 {
+		t.Fatalf("port = %d, want %d", port, postgresqlPortPoolStart+1)
+	}
+	if len(probed) != 2 || probed[0] != postgresqlPortPoolStart || probed[1] != postgresqlPortPoolStart+1 {
+		t.Fatalf("unexpected probes: %v", probed)
+	}
+}
+
+func TestSelectPostgreSQLHosts(t *testing.T) {
+	hosts := []targetHost{{Name: "pg-a"}, {Name: "pg-b"}, {Name: "pg-c"}}
+	selected, err := selectPostgreSQLHosts(hosts, nil, 2)
+	if err != nil || len(selected) != 2 || selected[0].Name != "pg-a" || selected[1].Name != "pg-b" {
+		t.Fatalf("unexpected automatic selection: %+v err=%v", selected, err)
+	}
+	selected, err = selectPostgreSQLHosts(hosts, []string{"pg-c", "pg-a"}, 2)
+	if err != nil || selected[0].Name != "pg-c" || selected[1].Name != "pg-a" {
+		t.Fatalf("unexpected pinned selection: %+v err=%v", selected, err)
+	}
+	if _, err := selectPostgreSQLHosts(hosts, []string{"missing"}, 1); err == nil {
+		t.Fatal("host outside the pool was accepted")
+	}
+}