84e778e033
- 创建完整的迁移框架 (pkg/migration/) - 版本管理系统，时间戳版本号 (YYYYMMDD_HHMMSS) - 事务安全的上下迁移机制 (Up/Down) - 迁移状态跟踪和记录 (migration_records 表) - 命令行迁移工具 (cmd/migrate/main.go) - 生产环境迁移脚本 (scripts/production-migrate.sh) - 生产环境初始化脚本 (scripts/init-production-db.sh) - 迁移测试脚本 (scripts/test-migration.sh) - Makefile 集成 (migrate-up, migrate-down, migrate-status) - 5个预定义迁移 (基础表、默认数据、元数据、收藏、用户资料) - 自动备份机制、预览模式、详细日志 - 完整文档 (docs/DATABASE_MIGRATION.md) 任务13完成，项目完成率达到42.5%
321 lines
11 KiB
Go
321 lines
11 KiB
Go
package migration
|
||
|
||
import (
|
||
"time"
|
||
)
|
||
|
||
// GetAllMigrations 返回所有迁移定义
|
||
func GetAllMigrations() []Migration {
|
||
return []Migration{
|
||
{
|
||
Version: "20250101_000001",
|
||
Description: "Create initial tables - users, categories, photos",
|
||
Timestamp: time.Date(2025, 1, 1, 0, 0, 1, 0, time.UTC),
|
||
UpSQL: `
|
||
-- 用户表
|
||
CREATE TABLE IF NOT EXISTS user (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
username VARCHAR(50) UNIQUE NOT NULL,
|
||
password VARCHAR(255) NOT NULL,
|
||
email VARCHAR(100) UNIQUE NOT NULL,
|
||
avatar VARCHAR(255) DEFAULT '',
|
||
status INTEGER DEFAULT 1, -- 1:启用 0:禁用
|
||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||
);
|
||
|
||
-- 分类表
|
||
CREATE TABLE IF NOT EXISTS category (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
name VARCHAR(100) NOT NULL,
|
||
description TEXT,
|
||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||
);
|
||
|
||
-- 照片表
|
||
CREATE TABLE IF NOT EXISTS photo (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
title VARCHAR(255) NOT NULL,
|
||
description TEXT,
|
||
file_path VARCHAR(500) NOT NULL,
|
||
thumbnail_path VARCHAR(500),
|
||
user_id INTEGER NOT NULL,
|
||
category_id INTEGER,
|
||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (user_id) REFERENCES user(id),
|
||
FOREIGN KEY (category_id) REFERENCES category(id)
|
||
);
|
||
|
||
-- 创建索引
|
||
CREATE INDEX IF NOT EXISTS idx_user_username ON user(username);
|
||
CREATE INDEX IF NOT EXISTS idx_user_email ON user(email);
|
||
CREATE INDEX IF NOT EXISTS idx_photo_user_id ON photo(user_id);
|
||
CREATE INDEX IF NOT EXISTS idx_photo_category_id ON photo(category_id);
|
||
CREATE INDEX IF NOT EXISTS idx_photo_created_at ON photo(created_at);
|
||
`,
|
||
DownSQL: `
|
||
-- 删除索引
|
||
DROP INDEX IF EXISTS idx_photo_created_at;
|
||
DROP INDEX IF EXISTS idx_photo_category_id;
|
||
DROP INDEX IF EXISTS idx_photo_user_id;
|
||
DROP INDEX IF EXISTS idx_user_email;
|
||
DROP INDEX IF EXISTS idx_user_username;
|
||
|
||
-- 删除表（注意外键约束，先删除依赖表）
|
||
DROP TABLE IF EXISTS photo;
|
||
DROP TABLE IF EXISTS category;
|
||
DROP TABLE IF EXISTS user;
|
||
`,
|
||
},
|
||
{
|
||
Version: "20250101_000002",
|
||
Description: "Insert default admin user and categories",
|
||
Timestamp: time.Date(2025, 1, 1, 0, 0, 2, 0, time.UTC),
|
||
UpSQL: `
|
||
-- 插入默认管理员用户
|
||
INSERT OR IGNORE INTO user (username, password, email, avatar, status)
|
||
VALUES ('admin', '$2a$10$K8H7YjN5hOcE0zWTz1YuAuYqFyQ9cqUdFHJgJdKxA5wGv3LUQHgKq', 'admin@example.com', '', 1);
|
||
-- 密码是 admin123 的 bcrypt 哈希
|
||
|
||
-- 插入默认分类
|
||
INSERT OR IGNORE INTO category (name, description)
|
||
VALUES
|
||
('风景', '自然风景摄影作品'),
|
||
('人像', '人物肖像摄影作品'),
|
||
('建筑', '建筑摄影作品'),
|
||
('街拍', '街头摄影作品');
|
||
`,
|
||
DownSQL: `
|
||
-- 删除默认数据（保留用户数据的完整性）
|
||
DELETE FROM photo WHERE category_id IN (SELECT id FROM category WHERE name IN ('风景', '人像', '建筑', '街拍'));
|
||
DELETE FROM category WHERE name IN ('风景', '人像', '建筑', '街拍');
|
||
DELETE FROM user WHERE username = 'admin' AND email = 'admin@example.com';
|
||
`,
|
||
},
|
||
{
|
||
Version: "20250111_000001",
|
||
Description: "Add photo metadata fields - EXIF data, tags, location",
|
||
Timestamp: time.Date(2025, 1, 11, 0, 0, 1, 0, time.UTC),
|
||
UpSQL: `
|
||
-- 为照片表添加元数据字段
|
||
ALTER TABLE photo ADD COLUMN exif_data TEXT DEFAULT '{}';
|
||
ALTER TABLE photo ADD COLUMN tags VARCHAR(500) DEFAULT '';
|
||
ALTER TABLE photo ADD COLUMN location VARCHAR(200) DEFAULT '';
|
||
ALTER TABLE photo ADD COLUMN camera_model VARCHAR(100) DEFAULT '';
|
||
ALTER TABLE photo ADD COLUMN lens_model VARCHAR(100) DEFAULT '';
|
||
ALTER TABLE photo ADD COLUMN focal_length INTEGER DEFAULT 0;
|
||
ALTER TABLE photo ADD COLUMN aperture VARCHAR(10) DEFAULT '';
|
||
ALTER TABLE photo ADD COLUMN shutter_speed VARCHAR(20) DEFAULT '';
|
||
ALTER TABLE photo ADD COLUMN iso INTEGER DEFAULT 0;
|
||
ALTER TABLE photo ADD COLUMN flash_used BOOLEAN DEFAULT FALSE;
|
||
ALTER TABLE photo ADD COLUMN orientation INTEGER DEFAULT 1;
|
||
|
||
-- 创建标签索引（用于搜索）
|
||
CREATE INDEX IF NOT EXISTS idx_photo_tags ON photo(tags);
|
||
CREATE INDEX IF NOT EXISTS idx_photo_location ON photo(location);
|
||
`,
|
||
DownSQL: `
|
||
-- 删除索引
|
||
DROP INDEX IF EXISTS idx_photo_location;
|
||
DROP INDEX IF EXISTS idx_photo_tags;
|
||
|
||
-- 注意：SQLite 不支持 DROP COLUMN，需要重建表
|
||
-- 创建临时表（不包含新添加的列）
|
||
CREATE TABLE photo_temp (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
title VARCHAR(255) NOT NULL,
|
||
description TEXT,
|
||
file_path VARCHAR(500) NOT NULL,
|
||
thumbnail_path VARCHAR(500),
|
||
user_id INTEGER NOT NULL,
|
||
category_id INTEGER,
|
||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (user_id) REFERENCES user(id),
|
||
FOREIGN KEY (category_id) REFERENCES category(id)
|
||
);
|
||
|
||
-- 复制数据
|
||
INSERT INTO photo_temp (id, title, description, file_path, thumbnail_path, user_id, category_id, created_at, updated_at)
|
||
SELECT id, title, description, file_path, thumbnail_path, user_id, category_id, created_at, updated_at FROM photo;
|
||
|
||
-- 删除原表
|
||
DROP TABLE photo;
|
||
|
||
-- 重命名临时表
|
||
ALTER TABLE photo_temp RENAME TO photo;
|
||
|
||
-- 重新创建索引
|
||
CREATE INDEX IF NOT EXISTS idx_photo_user_id ON photo(user_id);
|
||
CREATE INDEX IF NOT EXISTS idx_photo_category_id ON photo(category_id);
|
||
CREATE INDEX IF NOT EXISTS idx_photo_created_at ON photo(created_at);
|
||
`,
|
||
},
|
||
{
|
||
Version: "20250111_000002",
|
||
Description: "Create photo collections and favorites system",
|
||
Timestamp: time.Date(2025, 1, 11, 0, 0, 2, 0, time.UTC),
|
||
UpSQL: `
|
||
-- 照片集合表
|
||
CREATE TABLE IF NOT EXISTS collection (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
name VARCHAR(100) NOT NULL,
|
||
description TEXT,
|
||
user_id INTEGER NOT NULL,
|
||
is_public BOOLEAN DEFAULT FALSE,
|
||
cover_photo_id INTEGER,
|
||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (user_id) REFERENCES user(id),
|
||
FOREIGN KEY (cover_photo_id) REFERENCES photo(id)
|
||
);
|
||
|
||
-- 照片集合关联表（多对多）
|
||
CREATE TABLE IF NOT EXISTS collection_photo (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
collection_id INTEGER NOT NULL,
|
||
photo_id INTEGER NOT NULL,
|
||
sort_order INTEGER DEFAULT 0,
|
||
added_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (collection_id) REFERENCES collection(id) ON DELETE CASCADE,
|
||
FOREIGN KEY (photo_id) REFERENCES photo(id) ON DELETE CASCADE,
|
||
UNIQUE(collection_id, photo_id)
|
||
);
|
||
|
||
-- 用户收藏表
|
||
CREATE TABLE IF NOT EXISTS user_favorite (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
user_id INTEGER NOT NULL,
|
||
photo_id INTEGER NOT NULL,
|
||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (user_id) REFERENCES user(id) ON DELETE CASCADE,
|
||
FOREIGN KEY (photo_id) REFERENCES photo(id) ON DELETE CASCADE,
|
||
UNIQUE(user_id, photo_id)
|
||
);
|
||
|
||
-- 创建索引
|
||
CREATE INDEX IF NOT EXISTS idx_collection_user_id ON collection(user_id);
|
||
CREATE INDEX IF NOT EXISTS idx_collection_public ON collection(is_public);
|
||
CREATE INDEX IF NOT EXISTS idx_collection_photo_collection ON collection_photo(collection_id);
|
||
CREATE INDEX IF NOT EXISTS idx_collection_photo_photo ON collection_photo(photo_id);
|
||
CREATE INDEX IF NOT EXISTS idx_user_favorite_user ON user_favorite(user_id);
|
||
CREATE INDEX IF NOT EXISTS idx_user_favorite_photo ON user_favorite(photo_id);
|
||
`,
|
||
DownSQL: `
|
||
-- 删除索引
|
||
DROP INDEX IF EXISTS idx_user_favorite_photo;
|
||
DROP INDEX IF EXISTS idx_user_favorite_user;
|
||
DROP INDEX IF EXISTS idx_collection_photo_photo;
|
||
DROP INDEX IF EXISTS idx_collection_photo_collection;
|
||
DROP INDEX IF EXISTS idx_collection_public;
|
||
DROP INDEX IF EXISTS idx_collection_user_id;
|
||
|
||
-- 删除表
|
||
DROP TABLE IF EXISTS user_favorite;
|
||
DROP TABLE IF EXISTS collection_photo;
|
||
DROP TABLE IF EXISTS collection;
|
||
`,
|
||
},
|
||
{
|
||
Version: "20250111_000003",
|
||
Description: "Add user profile and preferences",
|
||
Timestamp: time.Date(2025, 1, 11, 0, 0, 3, 0, time.UTC),
|
||
UpSQL: `
|
||
-- 为用户表添加更多字段
|
||
ALTER TABLE user ADD COLUMN bio TEXT DEFAULT '';
|
||
ALTER TABLE user ADD COLUMN website VARCHAR(255) DEFAULT '';
|
||
ALTER TABLE user ADD COLUMN location VARCHAR(100) DEFAULT '';
|
||
ALTER TABLE user ADD COLUMN birth_date DATE;
|
||
ALTER TABLE user ADD COLUMN phone VARCHAR(20) DEFAULT '';
|
||
ALTER TABLE user ADD COLUMN is_verified BOOLEAN DEFAULT FALSE;
|
||
ALTER TABLE user ADD COLUMN last_login_at DATETIME;
|
||
|
||
-- 用户设置表
|
||
CREATE TABLE IF NOT EXISTS user_setting (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
user_id INTEGER UNIQUE NOT NULL,
|
||
theme VARCHAR(20) DEFAULT 'light', -- light, dark, auto
|
||
language VARCHAR(10) DEFAULT 'zh-CN',
|
||
timezone VARCHAR(50) DEFAULT 'Asia/Shanghai',
|
||
email_notifications BOOLEAN DEFAULT TRUE,
|
||
public_profile BOOLEAN DEFAULT TRUE,
|
||
show_exif BOOLEAN DEFAULT TRUE,
|
||
watermark_enabled BOOLEAN DEFAULT FALSE,
|
||
watermark_text VARCHAR(100) DEFAULT '',
|
||
watermark_position VARCHAR(20) DEFAULT 'bottom-right',
|
||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
FOREIGN KEY (user_id) REFERENCES user(id) ON DELETE CASCADE
|
||
);
|
||
|
||
-- 为新字段创建索引
|
||
CREATE INDEX IF NOT EXISTS idx_user_verified ON user(is_verified);
|
||
CREATE INDEX IF NOT EXISTS idx_user_last_login ON user(last_login_at);
|
||
`,
|
||
DownSQL: `
|
||
-- 删除索引
|
||
DROP INDEX IF EXISTS idx_user_last_login;
|
||
DROP INDEX IF EXISTS idx_user_verified;
|
||
|
||
-- 删除用户设置表
|
||
DROP TABLE IF EXISTS user_setting;
|
||
|
||
-- SQLite 不支持 DROP COLUMN，需要重建用户表
|
||
CREATE TABLE user_temp (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
username VARCHAR(50) UNIQUE NOT NULL,
|
||
password VARCHAR(255) NOT NULL,
|
||
email VARCHAR(100) UNIQUE NOT NULL,
|
||
avatar VARCHAR(255) DEFAULT '',
|
||
status INTEGER DEFAULT 1,
|
||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||
);
|
||
|
||
-- 复制原始数据
|
||
INSERT INTO user_temp (id, username, password, email, avatar, status, created_at, updated_at)
|
||
SELECT id, username, password, email, avatar, status, created_at, updated_at FROM user;
|
||
|
||
-- 删除原表
|
||
DROP TABLE user;
|
||
|
||
-- 重命名临时表
|
||
ALTER TABLE user_temp RENAME TO user;
|
||
|
||
-- 重新创建索引
|
||
CREATE INDEX IF NOT EXISTS idx_user_username ON user(username);
|
||
CREATE INDEX IF NOT EXISTS idx_user_email ON user(email);
|
||
`,
|
||
},
|
||
}
|
||
}
|
||
|
||
// GetMigrationByVersion 根据版本号获取迁移
|
||
func GetMigrationByVersion(version string) *Migration {
|
||
migrations := GetAllMigrations()
|
||
for _, migration := range migrations {
|
||
if migration.Version == version {
|
||
return &migration
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// GetLatestMigrationVersion 获取最新的迁移版本
|
||
func GetLatestMigrationVersion() string {
|
||
migrations := GetAllMigrations()
|
||
if len(migrations) == 0 {
|
||
return ""
|
||
}
|
||
|
||
latest := migrations[0]
|
||
for _, migration := range migrations {
|
||
if migration.Version > latest.Version {
|
||
latest = migration
|
||
}
|
||
}
|
||
|
||
return latest.Version
|
||
} |