feat(ql3): add strong cluster run management

2026-09-23 03:18:09 +08:00 · 2026-08-12 07:28:43 +08:00
parent e38b143dbb
commit c0ab62e64a
58 changed files with 3087 additions and 105 deletions
@@ -0,0 +1,106 @@
+import {
+  RUN_MANUAL_RETRY_SCHEMA,
+  normalizeRunManualRetryResult,
+} from '@qinglong/runtime-core/run-manual-retry';
+import {
+  ClusterPluginPackageManagementClientRequestError,
+  executeClusterAuthenticatedManagementClient,
+  type ClusterAuthenticatedManagementClientResult,
+  type ClusterPluginPackageManagementClientConnectionOptions,
+  type ClusterPluginPackageManagementClientPaths,
+} from '../management-support/pluginPackageManagementClient';
+import {
+  normalizeClusterRunManagementCommand,
+  type ClusterRunManagementCommand,
+  type ClusterRunManagementTransportResult,
+} from './runManagementTransport';
+
+const MANAGEMENT_PATH = '/api/v3/runs/management';
+
+export type ClusterRunManagementClientPaths =
+  ClusterPluginPackageManagementClientPaths;
+export type ClusterRunManagementClientConnectionOptions =
+  ClusterPluginPackageManagementClientConnectionOptions;
+export type ClusterRunManagementClientResult =
+  ClusterAuthenticatedManagementClientResult<ClusterRunManagementTransportResult>;
+
+function invalid(): never {
+  throw new ClusterPluginPackageManagementClientRequestError();
+}
+
+function exact(value: unknown, keys: readonly string[]): Record<string, unknown> {
+  if (!value || typeof value !== 'object' || Array.isArray(value)) invalid();
+  const actual = Object.keys(value as object).sort();
+  const expected = [...keys].sort();
+  if (
+    actual.length !== expected.length ||
+    actual.some((key, index) => key !== expected[index])
+  ) {
+    invalid();
+  }
+  return value as Record<string, unknown>;
+}
+
+export function validateClusterRunManagementClientResult(
+  value: unknown,
+  command: Readonly<ClusterRunManagementCommand>,
+): Readonly<ClusterRunManagementTransportResult> {
+  const envelope = exact(value, ['schemaVersion', 'operation', 'retry']);
+  if (envelope.schemaVersion !== 1 || envelope.operation !== 'run.retry') invalid();
+  const retry = exact(envelope.retry, [
+    'schema',
+    'status',
+    'projectId',
+    'sourceRunId',
+    'sourceRunStatus',
+    'sourceRunVersion',
+    'runId',
+    'retryOfRunId',
+    'taskId',
+    'taskRevision',
+    'attemptId',
+    'runStatus',
+    'runVersion',
+    'eventSequence',
+    'executorType',
+    'executionRevisionDigest',
+    'createdAtMs',
+  ]);
+  if (retry.schema !== RUN_MANUAL_RETRY_SCHEMA) invalid();
+  try {
+    const { schema: _schema, ...result } = retry;
+    const normalized = normalizeRunManualRetryResult(result as never);
+    if (
+      normalized.projectId !== command.request.projectId ||
+      normalized.sourceRunId !== command.request.sourceRunId ||
+      normalized.sourceRunVersion !== command.request.body.expectedRunVersion ||
+      normalized.sourceRunStatus !== command.request.body.expectedRunStatus ||
+      normalized.executorType !== 'remote_worker'
+    ) {
+      invalid();
+    }
+  } catch {
+    invalid();
+  }
+  return Object.freeze(
+    envelope as unknown as ClusterRunManagementTransportResult,
+  );
+}
+
+const PROTOCOL = Object.freeze({
+  managementPath: MANAGEMENT_PATH,
+  clientCertificate: 'required' as const,
+  normalizeCommand: normalizeClusterRunManagementCommand,
+  validateResult: validateClusterRunManagementClientResult,
+});
+
+export function executeClusterRunManagementClient(
+  paths: ClusterRunManagementClientPaths,
+  connectionOptions?: ClusterRunManagementClientConnectionOptions,
+): Promise<Readonly<ClusterRunManagementClientResult>> {
+  return executeClusterAuthenticatedManagementClient(
+    paths,
+    PROTOCOL,
+    connectionOptions,
+  );
+}