add user_status_changes table

coder · SasSwart · Dec 14, 2024 · Dec 14, 2024 · Dec 14, 2024 · Dec 17, 2024
commit 4cd66af5fbff7c29bdbefae36026305d785cbe8e
diff --git a/coderd/database/dump.sql b/coderd/database/dump.sql
diff --git a/coderd/database/foreign_key_constraint.go b/coderd/database/foreign_key_constraint.go
diff --git a/coderd/database/migrations/000279_user_status_changes.down.sql b/coderd/database/migrations/000279_user_status_changes.down.sql
@@ -0,0 +1,12 @@
+-- Drop the trigger first
+DROP TRIGGER IF EXISTS user_status_change_trigger ON users;
+
+-- Drop the trigger function
+DROP FUNCTION IF EXISTS record_user_status_change();
+
+-- Drop the indexes
+DROP INDEX IF EXISTS idx_user_status_changes_changed_at;
+DROP INDEX IF EXISTS idx_user_status_changes_user_id;
+
+-- Drop the table
+DROP TABLE IF EXISTS user_status_changes;
diff --git a/coderd/database/migrations/000279_user_status_changes.up.sql b/coderd/database/migrations/000279_user_status_changes.up.sql
@@ -0,0 +1,43 @@
+CREATE TABLE user_status_changes (
+    id uuid PRIMARY KEY DEFAULT gen_random_uuid(),
+    user_id uuid NOT NULL REFERENCES users(id),
+    new_status user_status NOT NULL,
+    changed_at timestamptz NOT NULL DEFAULT CURRENT_TIMESTAMP
+);
+
+COMMENT ON TABLE user_status_changes IS 'Tracks the history of user status changes';
+
+CREATE INDEX idx_user_status_changes_user_id ON user_status_changes(user_id);
+CREATE INDEX idx_user_status_changes_changed_at ON user_status_changes(changed_at);
+
+INSERT INTO user_status_changes (
+    user_id,
+    new_status,
+    changed_at
+)
+SELECT
+    id,
+    status,
+    created_at
+FROM users
+WHERE NOT deleted;
+
+CREATE FUNCTION record_user_status_change() RETURNS trigger AS $$
+BEGIN
+    IF OLD.status IS DISTINCT FROM NEW.status THEN
+        INSERT INTO user_status_changes (
+            user_id,
+            new_status
+        ) VALUES (
+            NEW.id,
+            NEW.status
+        );
+    END IF;
+    RETURN NEW;
+END;
+$$ LANGUAGE plpgsql;
+
+CREATE TRIGGER user_status_change_trigger
+    BEFORE UPDATE ON users
+    FOR EACH ROW
+    EXECUTE FUNCTION record_user_status_change();
diff --git a/coderd/database/models.go b/coderd/database/models.go
diff --git a/coderd/database/unique_constraint.go b/coderd/database/unique_constraint.go