mirror of
https://github.com/coder/coder.git
synced 2026-09-22 05:05:20 +08:00
George K
6af0f4d698
This models restart as durable orchestration of existing stop and start workspace builds instead of adding a new restart transition. Keeping restart as two existing transitions preserves the current build/provisioner model. The child start build is created only after the parent stop build succeeds, rather than being inserted immediately in a pending state. That keeps `workspace_builds` aligned with actual provisioner-ready work and avoids introducing a second pending-build lifecycle that the provisioner and build acquisition paths would need to understand. Refs: https://linear.app/codercom/issue/PLAT-143
134 lines
3.2 KiB
SQL
134 lines
3.2 KiB
SQL
-- name: InsertWorkspaceBuildOrchestration :one
|
|
INSERT INTO workspace_build_orchestrations (
|
|
id,
|
|
created_at,
|
|
updated_at,
|
|
parent_build_id,
|
|
workspace_id,
|
|
child_transition,
|
|
child_template_version_id,
|
|
child_template_version_preset_id,
|
|
child_rich_parameter_values,
|
|
child_log_level,
|
|
child_reason,
|
|
status,
|
|
error
|
|
)
|
|
VALUES (
|
|
@id,
|
|
@created_at,
|
|
@updated_at,
|
|
@parent_build_id,
|
|
(SELECT workspace_id FROM workspace_builds WHERE id = @parent_build_id),
|
|
@child_transition,
|
|
@child_template_version_id,
|
|
@child_template_version_preset_id,
|
|
@child_rich_parameter_values,
|
|
@child_log_level,
|
|
@child_reason,
|
|
'pending',
|
|
NULL
|
|
)
|
|
RETURNING *;
|
|
|
|
-- name: GetNextPendingWorkspaceBuildOrchestrationForUpdate :one
|
|
-- Must be called from within a transaction. The row lock is released
|
|
-- when the transaction ends.
|
|
SELECT
|
|
wbo.*
|
|
FROM
|
|
workspace_build_orchestrations wbo
|
|
JOIN workspace_builds wb ON wbo.parent_build_id = wb.id
|
|
JOIN provisioner_jobs pj ON wb.job_id = pj.id
|
|
WHERE
|
|
wbo.status = 'pending'
|
|
AND (
|
|
wbo.next_retry_after IS NULL
|
|
OR wbo.next_retry_after <= NOW()
|
|
)
|
|
-- Include all terminal parent states so pending orchestration
|
|
-- rows are processed and resolved even when no child build should
|
|
-- be created.
|
|
AND pj.job_status IN ('succeeded', 'failed', 'canceled')
|
|
ORDER BY
|
|
wbo.created_at ASC
|
|
LIMIT 1
|
|
FOR UPDATE OF wbo SKIP LOCKED;
|
|
|
|
-- name: UpdateWorkspaceBuildOrchestrationCompletedByID :one
|
|
UPDATE
|
|
workspace_build_orchestrations
|
|
SET
|
|
child_build_id = @child_build_id,
|
|
status = 'completed',
|
|
next_retry_after = NULL,
|
|
error = NULL,
|
|
updated_at = @updated_at
|
|
WHERE
|
|
id = @id
|
|
AND status = 'pending'
|
|
RETURNING *;
|
|
|
|
-- name: UpdateWorkspaceBuildOrchestrationFailedByID :one
|
|
UPDATE
|
|
workspace_build_orchestrations
|
|
SET
|
|
status = 'failed',
|
|
next_retry_after = NULL,
|
|
error = @error,
|
|
updated_at = @updated_at
|
|
WHERE
|
|
id = @id
|
|
AND status = 'pending'
|
|
RETURNING *;
|
|
|
|
-- name: UpdateWorkspaceBuildOrchestrationRetryByID :one
|
|
UPDATE
|
|
workspace_build_orchestrations
|
|
SET
|
|
attempt_count = attempt_count + 1,
|
|
next_retry_after = CASE
|
|
WHEN attempt_count + 1 >= @max_attempt_count::int THEN NULL
|
|
ELSE @next_retry_after::timestamptz
|
|
END,
|
|
status = CASE
|
|
WHEN attempt_count + 1 >= @max_attempt_count::int THEN 'failed'
|
|
ELSE status
|
|
END,
|
|
error = @error,
|
|
updated_at = @updated_at
|
|
WHERE
|
|
id = @id
|
|
AND status = 'pending'
|
|
RETURNING *;
|
|
|
|
-- name: UpdateWorkspaceBuildOrchestrationCanceledByID :one
|
|
UPDATE
|
|
workspace_build_orchestrations
|
|
SET
|
|
status = 'canceled',
|
|
next_retry_after = NULL,
|
|
error = NULL,
|
|
updated_at = @updated_at
|
|
WHERE
|
|
id = @id
|
|
AND status = 'pending'
|
|
RETURNING *;
|
|
|
|
-- name: DeleteOldWorkspaceBuildOrchestrations :execrows
|
|
WITH deletable AS (
|
|
SELECT
|
|
id
|
|
FROM
|
|
workspace_build_orchestrations
|
|
WHERE
|
|
status IN ('completed', 'failed', 'canceled')
|
|
AND updated_at < @before_time::timestamptz
|
|
ORDER BY
|
|
updated_at ASC
|
|
LIMIT @limit_count::int
|
|
)
|
|
DELETE FROM workspace_build_orchestrations
|
|
USING deletable
|
|
WHERE workspace_build_orchestrations.id = deletable.id;
|