From 8a8717a81f9619c9a616c36c27ac1703afb7dbc9 Mon Sep 17 00:00:00 2001
From: TIANHE <TIANHE@GMAIL.COM>
Date: Sat, 31 Jan 2026 22:40:41 +0800
Subject: [PATCH] chore: merge user_id migration into init.sql

---
 .../add_user_id_to_analysis_memory.sql        | 21 -------------------
 backend_api_python/migrations/init.sql        | 15 +++++++++++++
 2 files changed, 15 insertions(+), 21 deletions(-)
 delete mode 100644 backend_api_python/migrations/add_user_id_to_analysis_memory.sql

diff --git a/backend_api_python/migrations/add_user_id_to_analysis_memory.sql b/backend_api_python/migrations/add_user_id_to_analysis_memory.sql
deleted file mode 100644
index 7f648ed..0000000
--- a/backend_api_python/migrations/add_user_id_to_analysis_memory.sql
+++ /dev/null
@@ -1,21 +0,0 @@
--- Migration: Add user_id column to qd_analysis_memory table
--- This allows filtering analysis history by user
--- Run this migration to update existing databases
-
--- Add user_id column if it doesn't exist
-DO $$
-BEGIN
-    IF NOT EXISTS (
-        SELECT 1 FROM information_schema.columns 
-        WHERE table_name = 'qd_analysis_memory' AND column_name = 'user_id'
-    ) THEN
-        ALTER TABLE qd_analysis_memory ADD COLUMN user_id INT;
-        
-        -- Create index for efficient user-based queries
-        CREATE INDEX IF NOT EXISTS idx_analysis_memory_user ON qd_analysis_memory(user_id);
-        
-        RAISE NOTICE 'Added user_id column to qd_analysis_memory';
-    ELSE
-        RAISE NOTICE 'user_id column already exists in qd_analysis_memory';
-    END IF;
-END $$;
diff --git a/backend_api_python/migrations/init.sql b/backend_api_python/migrations/init.sql
index 1d69e79..7562e93 100644
--- a/backend_api_python/migrations/init.sql
+++ b/backend_api_python/migrations/init.sql
@@ -636,6 +636,7 @@ CREATE INDEX IF NOT EXISTS idx_reflection_market ON qd_reflection_records(market
 
 CREATE TABLE IF NOT EXISTS qd_analysis_memory (
     id SERIAL PRIMARY KEY,
+    user_id INT,                                -- User who created this analysis (for filtering)
     market VARCHAR(50) NOT NULL,
     symbol VARCHAR(50) NOT NULL,
     decision VARCHAR(10) NOT NULL,
@@ -662,6 +663,20 @@ CREATE TABLE IF NOT EXISTS qd_analysis_memory (
 CREATE INDEX IF NOT EXISTS idx_analysis_memory_symbol ON qd_analysis_memory(market, symbol);
 CREATE INDEX IF NOT EXISTS idx_analysis_memory_created ON qd_analysis_memory(created_at DESC);
 CREATE INDEX IF NOT EXISTS idx_analysis_memory_validated ON qd_analysis_memory(validated_at) WHERE validated_at IS NOT NULL;
+CREATE INDEX IF NOT EXISTS idx_analysis_memory_user ON qd_analysis_memory(user_id);
+
+-- Migration: Add user_id column to existing qd_analysis_memory table
+DO $$
+BEGIN
+    IF NOT EXISTS (
+        SELECT 1 FROM information_schema.columns 
+        WHERE table_name = 'qd_analysis_memory' AND column_name = 'user_id'
+    ) THEN
+        ALTER TABLE qd_analysis_memory ADD COLUMN user_id INT;
+        CREATE INDEX IF NOT EXISTS idx_analysis_memory_user ON qd_analysis_memory(user_id);
+        RAISE NOTICE 'Added user_id column to qd_analysis_memory';
+    END IF;
+END $$;
 
 -- =============================================================================
 -- 20. Migration: Add token_version for single-client login