coder
diff --git a/‎coderd/database/dump.sql
Lines changed: 58 additions & 0 deletions b/‎coderd/database/dump.sql
Lines changed: 58 additions & 0 deletions
diff --git a/‎coderd/database/migrations/000260_agent_id_name_pair.down.sql
Lines changed: 0 additions & 1 deletion b/‎coderd/database/migrations/000260_agent_id_name_pair.down.sql
Lines changed: 0 additions & 1 deletion
diff --git a/‎coderd/database/migrations/000260_agent_id_name_pair.up.sql
Lines changed: 0 additions & 4 deletions b/‎coderd/database/migrations/000260_agent_id_name_pair.up.sql
Lines changed: 0 additions & 4 deletions
diff --git a/‎coderd/database/migrations/000260_workspace_updates.down.sql
Lines changed: 7 additions & 0 deletions b/‎coderd/database/migrations/000260_workspace_updates.down.sql
Lines changed: 7 additions & 0 deletions
diff --git a/‎coderd/database/migrations/000260_workspace_updates.up.sql
Lines changed: 71 additions & 0 deletions b/‎coderd/database/migrations/000260_workspace_updates.up.sql
Lines changed: 71 additions & 0 deletions
@@ -0,0 +1,7 @@
+DROP TYPE agent_id_name_pair;
+
+DROP TRIGGER IF EXISTS new_workspace_notify ON workspaces;
+DROP FUNCTION IF EXISTS new_workspace_notify;
+
+DROP TRIGGER IF EXISTS new_agent_notify ON workspace_agents;
+DROP FUNCTION new_agent_notify;
@@ -0,0 +1,71 @@
+CREATE TYPE agent_id_name_pair AS (
+	id uuid,
+	name text
+);
+
+CREATE FUNCTION new_workspace_notify() RETURNS trigger
+	LANGUAGE plpgsql
+AS $$
+DECLARE
+BEGIN
+	-- Write to the notification channel `new_workspace:owner_id`
+	-- with the workspace id as the payload.
+	PERFORM pg_notify('new_workspace:' || NEW.owner_id, NEW.id);
+	RETURN NEW;
+END;
+$$;
+
+CREATE TRIGGER new_workspace_notify
+	AFTER INSERT ON workspaces
+	FOR EACH ROW
+EXECUTE FUNCTION new_workspace_notify();
+
+
+CREATE FUNCTION new_agent_notify() RETURNS trigger
+	LANGUAGE plpgsql
+AS $$
+DECLARE
+	workspace_owner_id uuid;
+BEGIN
+	SELECT workspaces.owner_id
+	INTO workspace_owner_id
+	FROM
+		workspaces
+	WHERE
+		workspaces.id = (
+			SELECT
+				workspace_id
+			FROM
+				workspace_builds
+			WHERE
+				workspace_builds.job_id = (
+					SELECT
+						job_id
+					FROM
+						workspace_resources
+					WHERE
+						workspace_resources.id = (
+							SELECT
+								resource_id
+							FROM
+								workspace_agents
+							WHERE
+								workspace_agents.id = NEW.id
+						)
+				)
+		);
+	-- Agents might not belong to a workspace (template imports)
+	IF workspace_owner_id IS NOT NULL THEN
+		-- Write to the notification channel `new_agent:workspace_owner_id`
+		PERFORM pg_notify('new_agent:' || workspace_owner_id, '');
+	END IF;
+	RETURN NEW;
+END;
+$$;
+
+
+CREATE TRIGGER new_agent_notify
+	AFTER INSERT ON workspace_agents
+	FOR EACH ROW
+EXECUTE FUNCTION new_agent_notify();
+