typekro 0.22.0 → 0.23.0

package/dist/factories/clickhouse/utils/zone-layout.js

@@ -0,0 +1,103 @@
+/**
+ * Zone-pinned CHI layout compilation.
+ *
+ * WHY THIS EXISTS: the clickhouse-operator's own `podDistribution` supports
+ * ONLY the `kubernetes.io/hostname` topologyKey (Altinity/clickhouse-operator
+ * issue #772), so per-ZONE spreading cannot be expressed operator-natively.
+ * On EKS this matters operationally: EBS volumes are ZONAL, so a replica pod
+ * rescheduled into another AZ strands its PVC and the pod wedges Pending
+ * (we hit exactly this in production). The fix the operator DOES support is a
+ * per-replica `layout.replicas:` list where each replica's pod template pins
+ * one zone via nodeAffinity on `topology.kubernetes.io/zone` — this module
+ * compiles that layout deterministically.
+ *
+ * These helpers are pure (no resource construction) and unit-tested directly.
+ */
+/** Standard Kubernetes zone topology label. */
+export const ZONE_TOPOLOGY_KEY = 'topology.kubernetes.io/zone';
+/**
+ * Assign one zone per replica index, round-robin.
+ *
+ * When `replicaCount > zones.length`, zones repeat in order:
+ * replicas=5, zones=[a,b,c] → [a, b, c, a, b].
+ *
+ * @param replicaCount - Number of replicas per shard
+ * @param zones - Non-empty list of availability zones
+ * @returns Zone for each replica index (length === replicaCount)
+ */
+export function assignZonesRoundRobin(replicaCount, zones) {
+    if (zones.length === 0) {
+        throw new Error('assignZonesRoundRobin requires at least one zone');
+    }
+    if (!Number.isInteger(replicaCount) || replicaCount < 1) {
+        throw new Error(`assignZonesRoundRobin requires a positive integer replica count, got ${replicaCount}`);
+    }
+    return Array.from({ length: replicaCount }, (_, index) => {
+        const zone = zones[index % zones.length];
+        if (zone === undefined) {
+            throw new Error(`Zone assignment failed for replica ${index}`);
+        }
+        return zone;
+    });
+}
+/**
+ * Deterministic pod template name for a zone-pinned template.
+ *
+ * Zone names are DNS-safe label values, so embedding them keeps the template
+ * name valid and self-describing (e.g. `clickhouse-us-east-2a`).
+ */
+export function zonePodTemplateName(baseName, zone) {
+    return `${baseName}-${zone}`;
+}
+/**
+ * Explicit required nodeAffinity pinning a pod to one zone.
+ *
+ * Emitted as a plain PodSpec affinity (rather than the operator's
+ * `podTemplate.zone` sugar) so the pinning is visible verbatim in the
+ * rendered CHI and doesn't depend on operator-side template expansion.
+ */
+export function zoneNodeAffinity(zone) {
+    return {
+        nodeAffinity: {
+            requiredDuringSchedulingIgnoredDuringExecution: {
+                nodeSelectorTerms: [
+                    {
+                        matchExpressions: [
+                            {
+                                key: ZONE_TOPOLOGY_KEY,
+                                operator: 'In',
+                                values: [zone],
+                            },
+                        ],
+                    },
+                ],
+            },
+        },
+    };
+}
+/**
+ * Compile the per-replica zone-pinned layout for a CHI cluster.
+ *
+ * @param options.replicaCount - Replicas per shard
+ * @param options.zones - Availability zones to round-robin replicas across
+ * @param options.podTemplateBaseName - Base name for generated pod templates
+ * @param options.podSpec - Shared PodSpec (containers etc.); the zone
+ *   nodeAffinity is merged in per template
+ */
+export function compileZonePinnedLayout(options) {
+    const { replicaCount, zones, podTemplateBaseName, podSpec } = options;
+    const assignments = assignZonesRoundRobin(replicaCount, zones);
+    const distinctZones = [...new Set(assignments)];
+    const podTemplates = distinctZones.map((zone) => ({
+        name: zonePodTemplateName(podTemplateBaseName, zone),
+        spec: {
+            ...podSpec,
+            affinity: zoneNodeAffinity(zone),
+        },
+    }));
+    const replicas = assignments.map((zone) => ({
+        templates: { podTemplate: zonePodTemplateName(podTemplateBaseName, zone) },
+    }));
+    return { replicas, podTemplates };
+}
+//# sourceMappingURL=zone-layout.js.map

package/dist/factories/clickhouse/utils/zone-layout.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zone-layout.js","sourceRoot":"","sources":["../../../../src/factories/clickhouse/utils/zone-layout.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,+CAA+C;AAC/C,MAAM,CAAC,MAAM,iBAAiB,GAAG,6BAA6B,CAAC;AAE/D;;;;;;;;;GASG;AACH,MAAM,UAAU,qBAAqB,CACnC,YAAoB,EACpB,KAAwB;IAExB,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;IACtE,CAAC;IACD,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,IAAI,YAAY,GAAG,CAAC,EAAE,CAAC;QACxD,MAAM,IAAI,KAAK,CACb,wEAAwE,YAAY,EAAE,CACvF,CAAC;IACJ,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE;QACvD,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC;QACzC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,sCAAsC,KAAK,EAAE,CAAC,CAAC;QACjE,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,mBAAmB,CAAC,QAAgB,EAAE,IAAY;IAChE,OAAO,GAAG,QAAQ,IAAI,IAAI,EAAE,CAAC;AAC/B,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,gBAAgB,CAAC,IAAY;IAC3C,OAAO;QACL,YAAY,EAAE;YACZ,8CAA8C,EAAE;gBAC9C,iBAAiB,EAAE;oBACjB;wBACE,gBAAgB,EAAE;4BAChB;gCACE,GAAG,EAAE,iBAAiB;gCACtB,QAAQ,EAAE,IAAI;gCACd,MAAM,EAAE,CAAC,IAAI,CAAC;6BACf;yBACF;qBACF;iBACF;aACF;SACF;KACF,CAAC;AACJ,CAAC;AAiBD;;;;;;;;GAQG;AACH,MAAM,UAAU,uBAAuB,CAAC,OAKvC;IACC,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,mBAAmB,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC;IACtE,MAAM,WAAW,GAAG,qBAAqB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC;IAC/D,MAAM,aAAa,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC;IAEhD,MAAM,YAAY,GAAqB,aAAa,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAClE,IAAI,EAAE,mBAAmB,CAAC,mBAAmB,EAAE,IAAI,CAAC;QACpD,IAAI,EAAE;YACJ,GAAG,OAAO;YACV,QAAQ,EAAE,gBAAgB,CAAC,IAAI,CAAC;SACjC;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,QAAQ,GAA8B,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACrE,SAAS,EAAE,EAAE,WAAW,EAAE,mBAAmB,CAAC,mBAAmB,EAAE,IAAI,CAAC,EAAE;KAC3E,CAAC,CAAC,CAAC;IAEJ,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;AACpC,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "typekro",
-  "version": "0.22.0",
+  "version": "0.23.0",
   "description": "A control plane aware framework for orchestrating kubernetes resources like a programmer.",
   "type": "module",
   "repository": {
@@ -46,6 +46,10 @@
       "import": "./dist/factories/cnpg/index.js",
       "types": "./dist/factories/cnpg/index.d.ts"
     },
+    "./clickhouse": {
+      "import": "./dist/factories/clickhouse/index.js",
+      "types": "./dist/factories/clickhouse/index.d.ts"
+    },
     "./flux": {
       "import": "./dist/factories/flux/index.js",
       "types": "./dist/factories/flux/index.d.ts"