Test GetUserStatusCountsByDay

coder · SasSwart · Jan 13, 2025 · Dec 14, 2024 · Dec 14, 2024 · Dec 14, 2024
commit eb6e2498855f34875003abc67c587f7024bf16e8
diff --git a/coderd/database/dump.sql b/coderd/database/dump.sql
diff --git a/coderd/database/migrations/000279_user_status_changes.up.sql b/coderd/database/migrations/000279_user_status_changes.up.sql
@@ -7,7 +7,6 @@ CREATE TABLE user_status_changes (
 
 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 (
@@ -22,22 +21,24 @@ SELECT
 FROM users
 WHERE NOT deleted;
 
-CREATE FUNCTION record_user_status_change() RETURNS trigger AS $$
+CREATE OR REPLACE FUNCTION record_user_status_change() RETURNS trigger AS $$
 BEGIN
-    IF OLD.status IS DISTINCT FROM NEW.status THEN
+    IF TG_OP = 'INSERT' OR OLD.status IS DISTINCT FROM NEW.status THEN
         INSERT INTO user_status_changes (
             user_id,
-            new_status
+            new_status,
+            changed_at
         ) VALUES (
             NEW.id,
-            NEW.status
+            NEW.status,
+            NEW.updated_at
         );
     END IF;
     RETURN NEW;
 END;
 $$ LANGUAGE plpgsql;
 
 CREATE TRIGGER user_status_change_trigger
-    BEFORE UPDATE ON users
+    AFTER INSERT OR UPDATE ON users
     FOR EACH ROW
     EXECUTE FUNCTION record_user_status_change();