mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
feat: add ai tasks migrations (#18359)
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.
This commit is contained in:
Hugo Dutka
@@ -0,0 +1,77 @@
|
||||
DROP VIEW workspace_build_with_user;
|
||||
|
||||
DROP VIEW template_version_with_user;
|
||||
|
||||
DROP INDEX idx_template_versions_has_ai_task;
|
||||
|
||||
ALTER TABLE
|
||||
template_versions DROP COLUMN has_ai_task;
|
||||
|
||||
ALTER TABLE
|
||||
workspace_builds DROP CONSTRAINT workspace_builds_ai_tasks_sidebar_app_id_fkey;
|
||||
|
||||
ALTER TABLE
|
||||
workspace_builds DROP COLUMN ai_tasks_sidebar_app_id;
|
||||
|
||||
ALTER TABLE
|
||||
workspace_builds DROP COLUMN has_ai_task;
|
||||
|
||||
-- Recreate `workspace_build_with_user` as defined in dump.sql
|
||||
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,
|
||||
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.';
|
||||
|
||||
-- Recreate `template_version_with_user` as defined in dump.sql
|
||||
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,
|
||||
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.';
|
||||
@@ -0,0 +1,103 @@
|
||||
-- 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.';
|
||||
Reference in New Issue
Block a user