fix: 用 YAML 配置 HWLAB runtime store pool

2026-06-20 00:02:08 +00:00
parent c9a5e2bede
commit 3533868a41
3 changed files with 31 additions and 0 deletions
@@ -79,6 +79,14 @@ export interface HwlabRuntimeExternalPostgresSpec {
  readonly openfga: HwlabRuntimeExternalPostgresComponentSpec;
 }

+export interface HwlabRuntimePostgresStoreSpec {
+  readonly poolMax: number;
+}
+
+export interface HwlabRuntimeStoreSpec {
+  readonly postgres?: HwlabRuntimePostgresStoreSpec;
+}
+
 export interface HwlabRuntimeBuildkitSpec {
  readonly sidecarImage: string;
 }
@@ -216,6 +224,7 @@ export interface HwlabRuntimeLaneSpec {
  readonly buildkit?: HwlabRuntimeBuildkitSpec;
  readonly bootstrapAdmin?: HwlabRuntimeBootstrapAdminSpec;
  readonly externalPostgres?: HwlabRuntimeExternalPostgresSpec;
+  readonly runtimeStore?: HwlabRuntimeStoreSpec;
  readonly publicExposure: HwlabRuntimePublicExposureSpec | null;
  readonly observability: HwlabRuntimeObservabilitySpec;
  readonly runtimeImageRewrites: readonly HwlabRuntimeImageRewriteSpec[];
@@ -257,6 +266,7 @@ interface HwlabLaneConfig {
  readonly buildkit?: HwlabRuntimeBuildkitSpec;
  readonly bootstrapAdmin?: HwlabRuntimeBootstrapAdminSpec;
  readonly externalPostgres?: HwlabRuntimeExternalPostgresSpec;
+  readonly runtimeStore?: HwlabRuntimeStoreSpec;
  readonly publicExposure: HwlabRuntimePublicExposureSpec | null;
  readonly observability: HwlabRuntimeObservabilitySpec;
  readonly runtimeImageRewrites: readonly HwlabRuntimeImageRewriteSpec[];
@@ -474,6 +484,7 @@ function laneConfig(id: HwlabRuntimeLane, raw: Record<string, unknown>): HwlabLa
    buildkit: buildkitConfig(raw.buildkit, `lanes.${id}.buildkit`),
    bootstrapAdmin: bootstrapAdminConfig(raw.bootstrapAdmin, `lanes.${id}.bootstrapAdmin`),
    externalPostgres: externalPostgresConfig(raw.externalPostgres, `lanes.${id}.externalPostgres`),
+    runtimeStore: runtimeStoreConfig(raw.runtimeStore, `lanes.${id}.runtimeStore`),
    publicExposure: publicExposureConfig(raw.publicExposure, `lanes.${id}.publicExposure`),
    observability: observabilityConfig(raw.observability, `lanes.${id}.observability`),
    runtimeImageRewrites: runtimeImageRewritesConfig(raw.runtimeImageRewrites, `lanes.${id}.runtimeImageRewrites`),
@@ -493,6 +504,7 @@ function laneTargetConfig(id: HwlabRuntimeLane, nodeId: string, baseRaw: Record<
    buildkit: mergeOptionalRecord(baseRaw.buildkit, targetRaw.buildkit),
    bootstrapAdmin: mergeOptionalRecord(baseRaw.bootstrapAdmin, targetRaw.bootstrapAdmin),
    externalPostgres: mergeOptionalRecord(baseRaw.externalPostgres, targetRaw.externalPostgres),
+    runtimeStore: mergeOptionalRecord(baseRaw.runtimeStore, targetRaw.runtimeStore),
    publicExposure: mergeOptionalRecord(baseRaw.publicExposure, targetRaw.publicExposure),
    observability: mergeOptionalRecord(baseRaw.observability, targetRaw.observability),
  };
@@ -569,6 +581,15 @@ function externalPostgresConfig(value: unknown, path: string): HwlabRuntimeExter
  };
 }

+function runtimeStoreConfig(value: unknown, path: string): HwlabRuntimeStoreSpec | undefined {
+  if (value === undefined) return undefined;
+  const raw = asRecord(value, path);
+  const postgres = raw.postgres === undefined ? undefined : asRecord(raw.postgres, `${path}.postgres`);
+  return {
+    ...(postgres === undefined ? {} : { postgres: { poolMax: numberField(postgres, "poolMax", `${path}.postgres`) } }),
+  };
+}
+
 function publicExposureProxyConfig(value: unknown, path: string): HwlabRuntimePublicExposureFrpcProxySpec {
  const raw = asRecord(value, path);
  return {
@@ -823,6 +844,7 @@ function buildRuntimeLaneSpec(config: HwlabLaneConfig): HwlabRuntimeLaneSpec {
    ...(config.buildkit === undefined ? {} : { buildkit: config.buildkit }),
    ...(config.bootstrapAdmin === undefined ? {} : { bootstrapAdmin: config.bootstrapAdmin }),
    ...(config.externalPostgres === undefined ? {} : { externalPostgres: config.externalPostgres }),
+    ...(config.runtimeStore === undefined ? {} : { runtimeStore: config.runtimeStore }),
    publicExposure: config.publicExposure,
    observability: config.observability,
    runtimeImageRewrites: config.runtimeImageRewrites,