mirror of
https://github.com/whyour/qinglong.git
synced 2026-09-20 08:05:22 +08:00
79 lines
2.6 KiB
TypeScript
79 lines
2.6 KiB
TypeScript
import { definePostgresSqlMigration } from './sqlMigration';
|
|
|
|
export const POSTGRESQL_RUN_RETRY_POLICY_TABLE = 'run_retry_policies';
|
|
|
|
export const pg0003RunRetryPolicyMigration = definePostgresSqlMigration({
|
|
id: 'pg-0003-run-retry-policy',
|
|
statements: [
|
|
`
|
|
ALTER TABLE "ql3"."runs"
|
|
ADD COLUMN legacy_cron_id integer
|
|
CONSTRAINT ql3_runs_legacy_cron_id_check
|
|
CHECK (legacy_cron_id IS NULL OR legacy_cron_id >= 1)
|
|
`.trim(),
|
|
`
|
|
CREATE TABLE "ql3"."${POSTGRESQL_RUN_RETRY_POLICY_TABLE}" (
|
|
run_id varchar(36) PRIMARY KEY,
|
|
max_attempts integer NOT NULL
|
|
CONSTRAINT ql3_run_retry_policies_max_attempts_check
|
|
CHECK (max_attempts BETWEEN 1 AND 16),
|
|
retry_on_lost boolean NOT NULL,
|
|
safety varchar(16) NOT NULL
|
|
CONSTRAINT ql3_run_retry_policies_safety_check
|
|
CHECK (safety IN ('unknown', 'idempotent', 'deduplicated')),
|
|
backoff_base_ms bigint NOT NULL
|
|
CONSTRAINT ql3_run_retry_policies_backoff_base_check
|
|
CHECK (backoff_base_ms BETWEEN 0 AND 86400000),
|
|
backoff_max_ms bigint NOT NULL
|
|
CONSTRAINT ql3_run_retry_policies_backoff_max_check
|
|
CHECK (backoff_max_ms BETWEEN backoff_base_ms AND 86400000),
|
|
next_attempt_at_ms bigint
|
|
CONSTRAINT ql3_run_retry_policies_next_attempt_check
|
|
CHECK (next_attempt_at_ms IS NULL OR next_attempt_at_ms >= 0),
|
|
version integer NOT NULL DEFAULT 0
|
|
CONSTRAINT ql3_run_retry_policies_version_check
|
|
CHECK (version >= 0),
|
|
created_at_ms bigint NOT NULL
|
|
CONSTRAINT ql3_run_retry_policies_created_at_check
|
|
CHECK (created_at_ms >= 0),
|
|
updated_at_ms bigint NOT NULL
|
|
CONSTRAINT ql3_run_retry_policies_updated_at_check
|
|
CHECK (updated_at_ms >= created_at_ms),
|
|
CONSTRAINT ql3_run_retry_policies_run_fk
|
|
FOREIGN KEY (run_id) REFERENCES "ql3"."runs" (id) ON DELETE CASCADE
|
|
)
|
|
`.trim(),
|
|
`
|
|
CREATE INDEX ql3_run_retry_policies_due_idx
|
|
ON "ql3"."${POSTGRESQL_RUN_RETRY_POLICY_TABLE}" (next_attempt_at_ms, run_id)
|
|
WHERE next_attempt_at_ms IS NOT NULL
|
|
`.trim(),
|
|
`
|
|
CREATE INDEX ql3_runs_lost_retry_idx
|
|
ON "ql3"."runs" (execution_owner, status, id)
|
|
`.trim(),
|
|
`
|
|
DO $ql3$
|
|
BEGIN
|
|
UPDATE "ql3"."schema_capabilities"
|
|
SET contract_version = 2,
|
|
migration_id = 'pg-0003-run-retry-policy',
|
|
capabilities = '{"run_core":1,"run_retry_policy":1}'::jsonb,
|
|
updated_at_ms = floor(
|
|
extract(epoch FROM transaction_timestamp()) * 1000
|
|
)::bigint
|
|
WHERE contract_name = 'control-core'
|
|
AND contract_version = 1
|
|
AND migration_id = 'pg-0002-run-core'
|
|
AND capabilities = '{"run_core":1}'::jsonb;
|
|
|
|
IF NOT FOUND THEN
|
|
RAISE EXCEPTION 'control-core capability is not at version 1'
|
|
USING ERRCODE = 'check_violation';
|
|
END IF;
|
|
END
|
|
$ql3$
|
|
`.trim(),
|
|
],
|
|
});
|