mirror of
https://github.com/coder/coder.git
synced 2025-07-06 15:41:45 +00:00
8e29ee50a3
Adds database migrations required for the Tasks feature. There's a slight difference between the migrations in this PR and the RFC: this PR adds `NOT NULL` constraints to the `has_ai_task` columns. It was an oversight on my part when I wrote the RFC - I assumed the `DEFAULT FALSE` value would make the columns implicitly NOT NULL, but that's not the case with Postgres. We have no use for the NULL value. The `DEFAULT FALSE` statement ensures that the migration will pass even when there are existing rows in the template version and workspace builds tables, so there's no danger in adding the `NOT NULL` constraints.
104 lines
3.4 KiB
SQL
104 lines
3.4 KiB
SQL
-- Determines if a coder_ai_task resource was included in a
|
|
-- workspace build.
|
|
ALTER TABLE
|
|
workspace_builds
|
|
ADD
|
|
COLUMN has_ai_task BOOLEAN NOT NULL DEFAULT FALSE;
|
|
|
|
-- The app that is displayed in the ai tasks sidebar.
|
|
ALTER TABLE
|
|
workspace_builds
|
|
ADD
|
|
COLUMN ai_tasks_sidebar_app_id UUID DEFAULT NULL;
|
|
|
|
ALTER TABLE
|
|
workspace_builds
|
|
ADD
|
|
CONSTRAINT workspace_builds_ai_tasks_sidebar_app_id_fkey FOREIGN KEY (ai_tasks_sidebar_app_id) REFERENCES workspace_apps(id);
|
|
|
|
-- Determines if a coder_ai_task resource is defined in a template version.
|
|
ALTER TABLE
|
|
template_versions
|
|
ADD
|
|
COLUMN has_ai_task BOOLEAN NOT NULL DEFAULT FALSE;
|
|
|
|
-- The Tasks tab will be rendered in the UI only if there's at least one template version with has_ai_task set to true.
|
|
-- The query to determine this will be run on every UI render, and this index speeds it up.
|
|
-- SELECT EXISTS (SELECT 1 FROM template_versions WHERE has_ai_task = TRUE);
|
|
CREATE INDEX idx_template_versions_has_ai_task ON template_versions USING btree (has_ai_task);
|
|
|
|
DROP VIEW workspace_build_with_user;
|
|
|
|
-- We're adding the has_ai_task and ai_tasks_sidebar_app_id columns.
|
|
CREATE VIEW workspace_build_with_user AS
|
|
SELECT
|
|
workspace_builds.id,
|
|
workspace_builds.created_at,
|
|
workspace_builds.updated_at,
|
|
workspace_builds.workspace_id,
|
|
workspace_builds.template_version_id,
|
|
workspace_builds.build_number,
|
|
workspace_builds.transition,
|
|
workspace_builds.initiator_id,
|
|
workspace_builds.provisioner_state,
|
|
workspace_builds.job_id,
|
|
workspace_builds.deadline,
|
|
workspace_builds.reason,
|
|
workspace_builds.daily_cost,
|
|
workspace_builds.max_deadline,
|
|
workspace_builds.template_version_preset_id,
|
|
workspace_builds.has_ai_task,
|
|
workspace_builds.ai_tasks_sidebar_app_id,
|
|
COALESCE(
|
|
visible_users.avatar_url,
|
|
'' :: text
|
|
) AS initiator_by_avatar_url,
|
|
COALESCE(
|
|
visible_users.username,
|
|
'' :: text
|
|
) AS initiator_by_username,
|
|
COALESCE(visible_users.name, '' :: text) AS initiator_by_name
|
|
FROM
|
|
(
|
|
workspace_builds
|
|
LEFT JOIN visible_users ON (
|
|
(
|
|
workspace_builds.initiator_id = visible_users.id
|
|
)
|
|
)
|
|
);
|
|
|
|
COMMENT ON VIEW workspace_build_with_user IS 'Joins in the username + avatar url of the initiated by user.';
|
|
|
|
DROP VIEW template_version_with_user;
|
|
|
|
-- We're adding the has_ai_task column.
|
|
CREATE VIEW template_version_with_user AS
|
|
SELECT
|
|
template_versions.id,
|
|
template_versions.template_id,
|
|
template_versions.organization_id,
|
|
template_versions.created_at,
|
|
template_versions.updated_at,
|
|
template_versions.name,
|
|
template_versions.readme,
|
|
template_versions.job_id,
|
|
template_versions.created_by,
|
|
template_versions.external_auth_providers,
|
|
template_versions.message,
|
|
template_versions.archived,
|
|
template_versions.source_example_id,
|
|
template_versions.has_ai_task,
|
|
COALESCE(visible_users.avatar_url, '' :: text) AS created_by_avatar_url,
|
|
COALESCE(visible_users.username, '' :: text) AS created_by_username,
|
|
COALESCE(visible_users.name, '' :: text) AS created_by_name
|
|
FROM
|
|
(
|
|
template_versions
|
|
LEFT JOIN visible_users ON (
|
|
(template_versions.created_by = visible_users.id)
|
|
)
|
|
);
|
|
|
|
COMMENT ON VIEW template_version_with_user IS 'Joins in the username + avatar url of the created by user.';
|