142 lines
6.7 KiB
SQL
142 lines
6.7 KiB
SQL
-- 数据库结构同步迁移脚本（与 api/database/models.py 保持一致）
|
|
-- 幂等：可重复执行，已存在的表/列会跳过。
|
|
-- 执行方式: psql -U <user> -d <database> -f api/migrations/sync_schema_to_models.sql
|
|
-- 执行时间: 2026-02
|
|
|
|
-- ========== 1. users 表缺失列 ==========
|
|
DO $$
|
|
BEGIN
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'users' AND column_name = 'subscription_type') THEN
|
|
ALTER TABLE users ADD COLUMN subscription_type VARCHAR DEFAULT 'free';
|
|
RAISE NOTICE '已添加 users.subscription_type';
|
|
END IF;
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'users' AND column_name = 'subscription_expires_at') THEN
|
|
ALTER TABLE users ADD COLUMN subscription_expires_at TIMESTAMP WITH TIME ZONE DEFAULT NULL;
|
|
RAISE NOTICE '已添加 users.subscription_expires_at';
|
|
END IF;
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'users' AND column_name = 'openid') THEN
|
|
ALTER TABLE users ADD COLUMN openid VARCHAR UNIQUE;
|
|
RAISE NOTICE '已添加 users.openid';
|
|
END IF;
|
|
END $$;
|
|
|
|
-- ========== 2. refresh_tokens 表缺失列 ==========
|
|
DO $$
|
|
BEGIN
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'refresh_tokens' AND column_name = 'device_info') THEN
|
|
ALTER TABLE refresh_tokens ADD COLUMN device_info VARCHAR;
|
|
RAISE NOTICE '已添加 refresh_tokens.device_info';
|
|
END IF;
|
|
END $$;
|
|
|
|
-- ========== 3. conversations 表缺失列 ==========
|
|
DO $$
|
|
BEGIN
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'conversations' AND column_name = 'current_topic') THEN
|
|
ALTER TABLE conversations ADD COLUMN current_topic VARCHAR;
|
|
RAISE NOTICE '已添加 conversations.current_topic';
|
|
END IF;
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'conversations' AND column_name = 'conversation_stage') THEN
|
|
ALTER TABLE conversations ADD COLUMN conversation_stage VARCHAR;
|
|
RAISE NOTICE '已添加 conversations.conversation_stage';
|
|
END IF;
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'conversations' AND column_name = 'last_message_at') THEN
|
|
ALTER TABLE conversations ADD COLUMN last_message_at TIMESTAMP WITH TIME ZONE;
|
|
RAISE NOTICE '已添加 conversations.last_message_at';
|
|
END IF;
|
|
END $$;
|
|
|
|
UPDATE conversations
|
|
SET last_message_at = started_at
|
|
WHERE last_message_at IS NULL
|
|
AND started_at IS NOT NULL;
|
|
|
|
CREATE INDEX IF NOT EXISTS ix_conversations_last_message_at ON conversations(last_message_at);
|
|
|
|
-- ========== 4. chapters 表缺失列 ==========
|
|
DO $$
|
|
BEGIN
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'chapters' AND column_name = 'category') THEN
|
|
ALTER TABLE chapters ADD COLUMN category VARCHAR;
|
|
RAISE NOTICE '已添加 chapters.category';
|
|
END IF;
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'chapters' AND column_name = 'is_new') THEN
|
|
ALTER TABLE chapters ADD COLUMN is_new BOOLEAN DEFAULT TRUE;
|
|
RAISE NOTICE '已添加 chapters.is_new';
|
|
END IF;
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'chapters' AND column_name = 'source_segments') THEN
|
|
ALTER TABLE chapters ADD COLUMN source_segments JSONB;
|
|
RAISE NOTICE '已添加 chapters.source_segments';
|
|
END IF;
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'chapters' AND column_name = 'images') THEN
|
|
ALTER TABLE chapters ADD COLUMN images JSONB;
|
|
RAISE NOTICE '已添加 chapters.images';
|
|
END IF;
|
|
END $$;
|
|
|
|
-- ========== 5. books 表缺失列 ==========
|
|
DO $$
|
|
BEGIN
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'books' AND column_name = 'has_update') THEN
|
|
ALTER TABLE books ADD COLUMN has_update BOOLEAN DEFAULT FALSE;
|
|
RAISE NOTICE '已添加 books.has_update';
|
|
END IF;
|
|
IF NOT EXISTS (SELECT 1 FROM information_schema.columns WHERE table_schema = 'public' AND table_name = 'books' AND column_name = 'last_update_chapter_id') THEN
|
|
ALTER TABLE books ADD COLUMN last_update_chapter_id VARCHAR;
|
|
RAISE NOTICE '已添加 books.last_update_chapter_id';
|
|
END IF;
|
|
END $$;
|
|
|
|
-- ========== 6. orders 表（若无则创建） ==========
|
|
CREATE TABLE IF NOT EXISTS orders (
|
|
id VARCHAR NOT NULL PRIMARY KEY,
|
|
user_id VARCHAR NOT NULL REFERENCES users(id),
|
|
plan_id VARCHAR NOT NULL,
|
|
plan_name VARCHAR NOT NULL,
|
|
amount INTEGER NOT NULL,
|
|
currency VARCHAR DEFAULT 'CNY',
|
|
payment_method VARCHAR NOT NULL,
|
|
status VARCHAR DEFAULT 'pending',
|
|
trade_no VARCHAR,
|
|
paid_at TIMESTAMP WITH TIME ZONE,
|
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
|
expired_at TIMESTAMP WITH TIME ZONE
|
|
);
|
|
CREATE INDEX IF NOT EXISTS ix_orders_user_id ON orders(user_id);
|
|
CREATE INDEX IF NOT EXISTS ix_orders_trade_no ON orders(trade_no);
|
|
CREATE INDEX IF NOT EXISTS ix_orders_status ON orders(status);
|
|
|
|
-- ========== 7. sms_verification_codes 表（若无则创建） ==========
|
|
CREATE TABLE IF NOT EXISTS sms_verification_codes (
|
|
id VARCHAR PRIMARY KEY,
|
|
phone VARCHAR NOT NULL,
|
|
code VARCHAR NOT NULL,
|
|
purpose VARCHAR NOT NULL,
|
|
is_used BOOLEAN DEFAULT FALSE,
|
|
is_expired BOOLEAN DEFAULT FALSE,
|
|
expires_at TIMESTAMP WITH TIME ZONE NOT NULL,
|
|
created_at TIMESTAMP WITH TIME ZONE DEFAULT NOW(),
|
|
verified_at TIMESTAMP WITH TIME ZONE,
|
|
ip_address VARCHAR
|
|
);
|
|
CREATE INDEX IF NOT EXISTS idx_sms_phone ON sms_verification_codes(phone);
|
|
CREATE INDEX IF NOT EXISTS idx_sms_created_at ON sms_verification_codes(created_at);
|
|
CREATE INDEX IF NOT EXISTS idx_sms_purpose ON sms_verification_codes(purpose);
|
|
CREATE INDEX IF NOT EXISTS idx_sms_phone_purpose ON sms_verification_codes(phone, purpose);
|
|
|
|
-- ========== 8. memoir_states 表（若无则创建，供 create_all 未执行环境使用） ==========
|
|
CREATE TABLE IF NOT EXISTS memoir_states (
|
|
id VARCHAR NOT NULL PRIMARY KEY,
|
|
user_id VARCHAR NOT NULL UNIQUE REFERENCES users(id),
|
|
stage_order JSONB DEFAULT '[]'::jsonb,
|
|
current_stage VARCHAR DEFAULT 'childhood',
|
|
covered_stages JSONB DEFAULT '[]'::jsonb,
|
|
slots JSONB NOT NULL,
|
|
updated_at TIMESTAMP WITH TIME ZONE DEFAULT NOW()
|
|
);
|
|
|
|
DO $$
|
|
BEGIN
|
|
RAISE NOTICE 'sync_schema_to_models 迁移执行完成';
|
|
END $$;
|