refactor: 重构后端架构,采用 Go 风格四层设计模式
Some checks failed
部署后端服务 / 🧪 测试后端 (push) Failing after 1m37s
部署后端服务 / 🚀 构建并部署 (push) Has been skipped
部署后端服务 / 🔄 回滚部署 (push) Has been skipped

## 主要变更

### 🏗️ 架构重构
- 采用简洁的四层架构:API → Service → Repository → Model
- 遵循 Go 语言最佳实践和命名规范
- 实现依赖注入和接口导向设计
- 统一错误处理和响应格式

### 📁 目录结构优化
- 删除重复模块 (application/, domain/, infrastructure/ 等)
- 规范化命名 (使用 Go 风格的 snake_case)
- 清理无关文件 (package.json, node_modules/ 等)
- 新增规范化的测试目录结构

### 📚 文档系统
- 为每个模块创建详细的 CLAUDE.md 指导文件
- 包含开发规范、最佳实践和使用示例
- 支持模块化开发,缩短上下文长度

### 🔧 开发规范
- 统一接口命名规范 (UserServicer, PhotoRepositoryr)
- 标准化错误处理机制
- 完善的测试策略 (单元测试、集成测试、性能测试)
- 规范化的配置管理

### 🗂️ 新增文件
- cmd/server/ - 服务启动入口和配置
- internal/model/ - 数据模型层 (entity, dto, request)
- pkg/ - 共享工具包 (logger, response, validator)
- tests/ - 完整测试结构
- docs/ - API 文档和架构设计
- .gitignore - Git 忽略文件配置

### 🗑️ 清理内容
- 删除 Node.js 相关文件 (package.json, node_modules/)
- 移除重复的架构目录
- 清理临时文件和构建产物
- 删除重复的文档文件

## 影响
- 提高代码可维护性和可扩展性
- 统一开发规范,提升团队协作效率
- 优化项目结构,符合 Go 语言生态标准
- 完善文档体系,降低上手难度
This commit is contained in:
xujiang
2025-07-10 11:20:59 +08:00
parent 540593f1dc
commit a2f2f66f88
40 changed files with 9682 additions and 1798 deletions

View File

@ -7,9 +7,15 @@ CREATE TABLE users (
password VARCHAR(255) NOT NULL,
name VARCHAR(100),
avatar VARCHAR(500),
role VARCHAR(20) DEFAULT 'user',
bio TEXT,
website VARCHAR(200),
location VARCHAR(100),
role VARCHAR(20) DEFAULT 'user' CHECK (role IN ('user', 'admin', 'photographer')),
is_active BOOLEAN DEFAULT true,
is_public BOOLEAN DEFAULT true,
email_verified BOOLEAN DEFAULT false,
last_login TIMESTAMP,
login_count INTEGER DEFAULT 0,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
deleted_at TIMESTAMP
@ -19,12 +25,30 @@ CREATE TABLE users (
CREATE INDEX idx_users_username ON users(username);
CREATE INDEX idx_users_email ON users(email);
CREATE INDEX idx_users_role ON users(role);
CREATE INDEX idx_users_deleted_at ON users(deleted_at);
CREATE INDEX idx_users_is_active ON users(is_active);
CREATE INDEX idx_users_is_public ON users(is_public);
CREATE INDEX idx_users_email_verified ON users(email_verified);
CREATE INDEX idx_users_created_at ON users(created_at);
CREATE INDEX idx_users_deleted_at ON users(deleted_at) WHERE deleted_at IS NOT NULL;
-- 添加更新时间触发器
CREATE OR REPLACE FUNCTION update_updated_at_column()
RETURNS TRIGGER AS $$
BEGIN
NEW.updated_at = CURRENT_TIMESTAMP;
RETURN NEW;
END;
$$ language 'plpgsql';
CREATE TRIGGER update_users_updated_at BEFORE UPDATE ON users
FOR EACH ROW EXECUTE FUNCTION update_updated_at_column();
-- 插入默认管理员用户 (密码: admin123)
INSERT INTO users (username, email, password, name, role) VALUES
('admin', 'admin@photography.com', '$2a$10$D4Zz6m3j1YJzp8Y7zW4l2OXcQ5f6g7h8i9j0k1l2m3n4o5p6q7r8s9t0', '管理员', 'admin');
INSERT INTO users (username, email, password, name, role, email_verified) VALUES
('admin', 'admin@photography.com', '$2a$10$D4Zz6m3j1YJzp8Y7zW4l2OXcQ5f6g7h8i9j0k1l2m3n4o5p6q7r8s9t0', '管理员', 'admin', true);
-- +migrate Down
DROP TRIGGER IF EXISTS update_users_updated_at ON users;
DROP FUNCTION IF EXISTS update_updated_at_column();
DROP TABLE IF EXISTS users;

View File

@ -0,0 +1,64 @@
-- +migrate Up
CREATE TABLE photos (
id SERIAL PRIMARY KEY,
title VARCHAR(200) NOT NULL,
description TEXT,
filename VARCHAR(255) NOT NULL,
original_url VARCHAR(500) NOT NULL,
thumbnail_url VARCHAR(500),
medium_url VARCHAR(500),
large_url VARCHAR(500),
file_size BIGINT,
mime_type VARCHAR(100),
width INTEGER,
height INTEGER,
camera_make VARCHAR(100),
camera_model VARCHAR(100),
lens_model VARCHAR(100),
focal_length DECIMAL(5,2),
aperture DECIMAL(3,1),
shutter_speed VARCHAR(20),
iso INTEGER,
taken_at TIMESTAMP,
location_name VARCHAR(200),
latitude DECIMAL(10,8),
longitude DECIMAL(11,8),
user_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
album_id INTEGER,
category_id INTEGER,
is_public BOOLEAN DEFAULT true,
is_featured BOOLEAN DEFAULT false,
view_count INTEGER DEFAULT 0,
like_count INTEGER DEFAULT 0,
download_count INTEGER DEFAULT 0,
sort_order INTEGER DEFAULT 0,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
deleted_at TIMESTAMP
);
-- 创建索引
CREATE INDEX idx_photos_user_id ON photos(user_id);
CREATE INDEX idx_photos_album_id ON photos(album_id);
CREATE INDEX idx_photos_category_id ON photos(category_id);
CREATE INDEX idx_photos_is_public ON photos(is_public);
CREATE INDEX idx_photos_is_featured ON photos(is_featured);
CREATE INDEX idx_photos_taken_at ON photos(taken_at);
CREATE INDEX idx_photos_created_at ON photos(created_at);
CREATE INDEX idx_photos_view_count ON photos(view_count);
CREATE INDEX idx_photos_like_count ON photos(like_count);
CREATE INDEX idx_photos_sort_order ON photos(sort_order);
CREATE INDEX idx_photos_deleted_at ON photos(deleted_at) WHERE deleted_at IS NOT NULL;
-- 地理位置索引
CREATE INDEX idx_photos_location ON photos(latitude, longitude) WHERE latitude IS NOT NULL AND longitude IS NOT NULL;
-- 添加更新时间触发器
CREATE TRIGGER update_photos_updated_at BEFORE UPDATE ON photos
FOR EACH ROW EXECUTE FUNCTION update_updated_at_column();
-- +migrate Down
DROP TRIGGER IF EXISTS update_photos_updated_at ON photos;
DROP TABLE IF EXISTS photos;

View File

@ -0,0 +1,73 @@
-- +migrate Up
CREATE TABLE albums (
id SERIAL PRIMARY KEY,
title VARCHAR(200) NOT NULL,
description TEXT,
slug VARCHAR(255) UNIQUE,
cover_photo_id INTEGER REFERENCES photos(id) ON DELETE SET NULL,
user_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
category_id INTEGER REFERENCES categories(id) ON DELETE SET NULL,
is_public BOOLEAN DEFAULT true,
is_featured BOOLEAN DEFAULT false,
password VARCHAR(255),
view_count INTEGER DEFAULT 0,
like_count INTEGER DEFAULT 0,
photo_count INTEGER DEFAULT 0,
sort_order INTEGER DEFAULT 0,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
deleted_at TIMESTAMP
);
-- 创建索引
CREATE INDEX idx_albums_user_id ON albums(user_id);
CREATE INDEX idx_albums_category_id ON albums(category_id);
CREATE INDEX idx_albums_cover_photo_id ON albums(cover_photo_id);
CREATE INDEX idx_albums_slug ON albums(slug);
CREATE INDEX idx_albums_is_public ON albums(is_public);
CREATE INDEX idx_albums_is_featured ON albums(is_featured);
CREATE INDEX idx_albums_created_at ON albums(created_at);
CREATE INDEX idx_albums_view_count ON albums(view_count);
CREATE INDEX idx_albums_like_count ON albums(like_count);
CREATE INDEX idx_albums_photo_count ON albums(photo_count);
CREATE INDEX idx_albums_sort_order ON albums(sort_order);
CREATE INDEX idx_albums_deleted_at ON albums(deleted_at) WHERE deleted_at IS NOT NULL;
-- 添加更新时间触发器
CREATE TRIGGER update_albums_updated_at BEFORE UPDATE ON albums
FOR EACH ROW EXECUTE FUNCTION update_updated_at_column();
-- 添加 slug 自动生成触发器
CREATE OR REPLACE FUNCTION generate_album_slug()
RETURNS TRIGGER AS $$
BEGIN
IF NEW.slug IS NULL OR NEW.slug = '' THEN
NEW.slug = lower(regexp_replace(NEW.title, '[^a-zA-Z0-9]+', '-', 'g'));
NEW.slug = trim(both '-' from NEW.slug);
-- 确保 slug 唯一
DECLARE
counter INTEGER := 0;
base_slug VARCHAR(255);
BEGIN
base_slug := NEW.slug;
WHILE EXISTS (SELECT 1 FROM albums WHERE slug = NEW.slug AND id != COALESCE(NEW.id, 0)) LOOP
counter := counter + 1;
NEW.slug := base_slug || '-' || counter;
END LOOP;
END;
END IF;
RETURN NEW;
END;
$$ language 'plpgsql';
CREATE TRIGGER generate_albums_slug BEFORE INSERT OR UPDATE ON albums
FOR EACH ROW EXECUTE FUNCTION generate_album_slug();
-- +migrate Down
DROP TRIGGER IF EXISTS generate_albums_slug ON albums;
DROP FUNCTION IF EXISTS generate_album_slug();
DROP TRIGGER IF EXISTS update_albums_updated_at ON albums;
DROP TABLE IF EXISTS albums;

View File

@ -0,0 +1,113 @@
-- +migrate Up
CREATE TABLE categories (
id SERIAL PRIMARY KEY,
name VARCHAR(100) NOT NULL,
description TEXT,
slug VARCHAR(255) UNIQUE,
parent_id INTEGER REFERENCES categories(id) ON DELETE CASCADE,
user_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
cover_photo_id INTEGER REFERENCES photos(id) ON DELETE SET NULL,
color VARCHAR(7) DEFAULT '#3B82F6',
is_public BOOLEAN DEFAULT true,
photo_count INTEGER DEFAULT 0,
album_count INTEGER DEFAULT 0,
sort_order INTEGER DEFAULT 0,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
deleted_at TIMESTAMP
);
-- 创建索引
CREATE INDEX idx_categories_user_id ON categories(user_id);
CREATE INDEX idx_categories_parent_id ON categories(parent_id);
CREATE INDEX idx_categories_cover_photo_id ON categories(cover_photo_id);
CREATE INDEX idx_categories_slug ON categories(slug);
CREATE INDEX idx_categories_is_public ON categories(is_public);
CREATE INDEX idx_categories_created_at ON categories(created_at);
CREATE INDEX idx_categories_photo_count ON categories(photo_count);
CREATE INDEX idx_categories_album_count ON categories(album_count);
CREATE INDEX idx_categories_sort_order ON categories(sort_order);
CREATE INDEX idx_categories_deleted_at ON categories(deleted_at) WHERE deleted_at IS NOT NULL;
-- 树形结构查询优化
CREATE INDEX idx_categories_parent_sort ON categories(parent_id, sort_order);
-- 添加更新时间触发器
CREATE TRIGGER update_categories_updated_at BEFORE UPDATE ON categories
FOR EACH ROW EXECUTE FUNCTION update_updated_at_column();
-- 添加 slug 自动生成触发器
CREATE OR REPLACE FUNCTION generate_category_slug()
RETURNS TRIGGER AS $$
BEGIN
IF NEW.slug IS NULL OR NEW.slug = '' THEN
NEW.slug = lower(regexp_replace(NEW.name, '[^a-zA-Z0-9]+', '-', 'g'));
NEW.slug = trim(both '-' from NEW.slug);
-- 确保 slug 唯一
DECLARE
counter INTEGER := 0;
base_slug VARCHAR(255);
BEGIN
base_slug := NEW.slug;
WHILE EXISTS (SELECT 1 FROM categories WHERE slug = NEW.slug AND id != COALESCE(NEW.id, 0)) LOOP
counter := counter + 1;
NEW.slug := base_slug || '-' || counter;
END LOOP;
END;
END IF;
RETURN NEW;
END;
$$ language 'plpgsql';
CREATE TRIGGER generate_categories_slug BEFORE INSERT OR UPDATE ON categories
FOR EACH ROW EXECUTE FUNCTION generate_category_slug();
-- 插入默认分类
INSERT INTO categories (name, description, slug, user_id, color, is_public)
SELECT
'风景摄影' as name,
'自然风光、城市景观等风景类摄影作品' as description,
'landscape' as slug,
1 as user_id,
'#10B981' as color,
true as is_public
WHERE EXISTS (SELECT 1 FROM users WHERE id = 1);
INSERT INTO categories (name, description, slug, user_id, color, is_public)
SELECT
'人像摄影' as name,
'肖像、写真、人物摄影作品' as description,
'portrait' as slug,
1 as user_id,
'#F59E0B' as color,
true as is_public
WHERE EXISTS (SELECT 1 FROM users WHERE id = 1);
INSERT INTO categories (name, description, slug, user_id, color, is_public)
SELECT
'街拍摄影' as name,
'街头摄影、日常生活记录' as description,
'street' as slug,
1 as user_id,
'#EF4444' as color,
true as is_public
WHERE EXISTS (SELECT 1 FROM users WHERE id = 1);
INSERT INTO categories (name, description, slug, user_id, color, is_public)
SELECT
'建筑摄影' as name,
'建筑、室内设计等建筑类摄影作品' as description,
'architecture' as slug,
1 as user_id,
'#8B5CF6' as color,
true as is_public
WHERE EXISTS (SELECT 1 FROM users WHERE id = 1);
-- +migrate Down
DROP TRIGGER IF EXISTS generate_categories_slug ON categories;
DROP FUNCTION IF EXISTS generate_category_slug();
DROP TRIGGER IF EXISTS update_categories_updated_at ON categories;
DROP TABLE IF EXISTS categories;

View File

@ -0,0 +1,164 @@
-- +migrate Up
-- 创建标签表
CREATE TABLE tags (
id SERIAL PRIMARY KEY,
name VARCHAR(100) NOT NULL,
slug VARCHAR(255) UNIQUE,
color VARCHAR(7) DEFAULT '#6B7280',
description TEXT,
user_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
is_public BOOLEAN DEFAULT true,
photo_count INTEGER DEFAULT 0,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
deleted_at TIMESTAMP
);
-- 创建照片标签关联表(多对多)
CREATE TABLE photo_tags (
id SERIAL PRIMARY KEY,
photo_id INTEGER NOT NULL REFERENCES photos(id) ON DELETE CASCADE,
tag_id INTEGER NOT NULL REFERENCES tags(id) ON DELETE CASCADE,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
UNIQUE(photo_id, tag_id)
);
-- 创建相册标签关联表(多对多)
CREATE TABLE album_tags (
id SERIAL PRIMARY KEY,
album_id INTEGER NOT NULL REFERENCES albums(id) ON DELETE CASCADE,
tag_id INTEGER NOT NULL REFERENCES tags(id) ON DELETE CASCADE,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
UNIQUE(album_id, tag_id)
);
-- 创建标签表索引
CREATE INDEX idx_tags_user_id ON tags(user_id);
CREATE INDEX idx_tags_slug ON tags(slug);
CREATE INDEX idx_tags_is_public ON tags(is_public);
CREATE INDEX idx_tags_created_at ON tags(created_at);
CREATE INDEX idx_tags_photo_count ON tags(photo_count);
CREATE INDEX idx_tags_deleted_at ON tags(deleted_at) WHERE deleted_at IS NOT NULL;
-- 创建照片标签关联表索引
CREATE INDEX idx_photo_tags_photo_id ON photo_tags(photo_id);
CREATE INDEX idx_photo_tags_tag_id ON photo_tags(tag_id);
CREATE INDEX idx_photo_tags_created_at ON photo_tags(created_at);
-- 创建相册标签关联表索引
CREATE INDEX idx_album_tags_album_id ON album_tags(album_id);
CREATE INDEX idx_album_tags_tag_id ON album_tags(tag_id);
CREATE INDEX idx_album_tags_created_at ON album_tags(created_at);
-- 添加更新时间触发器
CREATE TRIGGER update_tags_updated_at BEFORE UPDATE ON tags
FOR EACH ROW EXECUTE FUNCTION update_updated_at_column();
-- 添加 slug 自动生成触发器
CREATE OR REPLACE FUNCTION generate_tag_slug()
RETURNS TRIGGER AS $$
BEGIN
IF NEW.slug IS NULL OR NEW.slug = '' THEN
NEW.slug = lower(regexp_replace(NEW.name, '[^a-zA-Z0-9]+', '-', 'g'));
NEW.slug = trim(both '-' from NEW.slug);
-- 确保 slug 唯一
DECLARE
counter INTEGER := 0;
base_slug VARCHAR(255);
BEGIN
base_slug := NEW.slug;
WHILE EXISTS (SELECT 1 FROM tags WHERE slug = NEW.slug AND id != COALESCE(NEW.id, 0)) LOOP
counter := counter + 1;
NEW.slug := base_slug || '-' || counter;
END LOOP;
END;
END IF;
RETURN NEW;
END;
$$ language 'plpgsql';
CREATE TRIGGER generate_tags_slug BEFORE INSERT OR UPDATE ON tags
FOR EACH ROW EXECUTE FUNCTION generate_tag_slug();
-- 创建标签计数更新触发器
CREATE OR REPLACE FUNCTION update_tag_photo_count()
RETURNS TRIGGER AS $$
BEGIN
IF TG_OP = 'INSERT' THEN
UPDATE tags SET photo_count = photo_count + 1 WHERE id = NEW.tag_id;
RETURN NEW;
ELSIF TG_OP = 'DELETE' THEN
UPDATE tags SET photo_count = photo_count - 1 WHERE id = OLD.tag_id;
RETURN OLD;
END IF;
RETURN NULL;
END;
$$ language 'plpgsql';
CREATE TRIGGER trigger_update_tag_photo_count
AFTER INSERT OR DELETE ON photo_tags
FOR EACH ROW EXECUTE FUNCTION update_tag_photo_count();
-- 插入默认标签
INSERT INTO tags (name, description, slug, user_id, color, is_public)
SELECT
'自然' as name,
'自然风光、植物、动物等自然主题' as description,
'nature' as slug,
1 as user_id,
'#10B981' as color,
true as is_public
WHERE EXISTS (SELECT 1 FROM users WHERE id = 1);
INSERT INTO tags (name, description, slug, user_id, color, is_public)
SELECT
'城市' as name,
'城市景观、建筑、街道等城市主题' as description,
'city' as slug,
1 as user_id,
'#3B82F6' as color,
true as is_public
WHERE EXISTS (SELECT 1 FROM users WHERE id = 1);
INSERT INTO tags (name, description, slug, user_id, color, is_public)
SELECT
'黑白' as name,
'黑白摄影作品' as description,
'black-white' as slug,
1 as user_id,
'#6B7280' as color,
true as is_public
WHERE EXISTS (SELECT 1 FROM users WHERE id = 1);
INSERT INTO tags (name, description, slug, user_id, color, is_public)
SELECT
'夜景' as name,
'夜晚拍摄的照片' as description,
'night' as slug,
1 as user_id,
'#1F2937' as color,
true as is_public
WHERE EXISTS (SELECT 1 FROM users WHERE id = 1);
INSERT INTO tags (name, description, slug, user_id, color, is_public)
SELECT
'日出日落' as name,
'日出、日落、黄金时段拍摄' as description,
'sunrise-sunset' as slug,
1 as user_id,
'#F59E0B' as color,
true as is_public
WHERE EXISTS (SELECT 1 FROM users WHERE id = 1);
-- +migrate Down
DROP TRIGGER IF EXISTS trigger_update_tag_photo_count ON photo_tags;
DROP FUNCTION IF EXISTS update_tag_photo_count();
DROP TRIGGER IF EXISTS generate_tags_slug ON tags;
DROP FUNCTION IF EXISTS generate_tag_slug();
DROP TRIGGER IF EXISTS update_tags_updated_at ON tags;
DROP TABLE IF EXISTS album_tags;
DROP TABLE IF EXISTS photo_tags;
DROP TABLE IF EXISTS tags;

View File

@ -0,0 +1,111 @@
-- +migrate Up
-- 添加照片表的外键约束
ALTER TABLE photos
ADD CONSTRAINT fk_photos_album_id
FOREIGN KEY (album_id) REFERENCES albums(id) ON DELETE SET NULL;
ALTER TABLE photos
ADD CONSTRAINT fk_photos_category_id
FOREIGN KEY (category_id) REFERENCES categories(id) ON DELETE SET NULL;
-- 添加相册表的外键约束(如果还没有的话)
-- 这些约束在创建相册表时可能已经存在,这里做一个保险
-- 添加一些有用的触发器和函数
-- 创建照片计数更新触发器(用于相册和分类)
CREATE OR REPLACE FUNCTION update_photo_counts()
RETURNS TRIGGER AS $$
BEGIN
IF TG_OP = 'INSERT' THEN
-- 更新相册照片数量
IF NEW.album_id IS NOT NULL THEN
UPDATE albums SET photo_count = photo_count + 1 WHERE id = NEW.album_id;
END IF;
-- 更新分类照片数量
IF NEW.category_id IS NOT NULL THEN
UPDATE categories SET photo_count = photo_count + 1 WHERE id = NEW.category_id;
END IF;
RETURN NEW;
ELSIF TG_OP = 'DELETE' THEN
-- 更新相册照片数量
IF OLD.album_id IS NOT NULL THEN
UPDATE albums SET photo_count = photo_count - 1 WHERE id = OLD.album_id;
END IF;
-- 更新分类照片数量
IF OLD.category_id IS NOT NULL THEN
UPDATE categories SET photo_count = photo_count - 1 WHERE id = OLD.category_id;
END IF;
RETURN OLD;
ELSIF TG_OP = 'UPDATE' THEN
-- 处理相册变更
IF OLD.album_id IS DISTINCT FROM NEW.album_id THEN
IF OLD.album_id IS NOT NULL THEN
UPDATE albums SET photo_count = photo_count - 1 WHERE id = OLD.album_id;
END IF;
IF NEW.album_id IS NOT NULL THEN
UPDATE albums SET photo_count = photo_count + 1 WHERE id = NEW.album_id;
END IF;
END IF;
-- 处理分类变更
IF OLD.category_id IS DISTINCT FROM NEW.category_id THEN
IF OLD.category_id IS NOT NULL THEN
UPDATE categories SET photo_count = photo_count - 1 WHERE id = OLD.category_id;
END IF;
IF NEW.category_id IS NOT NULL THEN
UPDATE categories SET photo_count = photo_count + 1 WHERE id = NEW.category_id;
END IF;
END IF;
RETURN NEW;
END IF;
RETURN NULL;
END;
$$ language 'plpgsql';
-- 创建触发器
CREATE TRIGGER trigger_update_photo_counts
AFTER INSERT OR UPDATE OR DELETE ON photos
FOR EACH ROW EXECUTE FUNCTION update_photo_counts();
-- 创建相册计数更新触发器(用于分类)
CREATE OR REPLACE FUNCTION update_album_counts()
RETURNS TRIGGER AS $$
BEGIN
IF TG_OP = 'INSERT' THEN
IF NEW.category_id IS NOT NULL THEN
UPDATE categories SET album_count = album_count + 1 WHERE id = NEW.category_id;
END IF;
RETURN NEW;
ELSIF TG_OP = 'DELETE' THEN
IF OLD.category_id IS NOT NULL THEN
UPDATE categories SET album_count = album_count - 1 WHERE id = OLD.category_id;
END IF;
RETURN OLD;
ELSIF TG_OP = 'UPDATE' THEN
IF OLD.category_id IS DISTINCT FROM NEW.category_id THEN
IF OLD.category_id IS NOT NULL THEN
UPDATE categories SET album_count = album_count - 1 WHERE id = OLD.category_id;
END IF;
IF NEW.category_id IS NOT NULL THEN
UPDATE categories SET album_count = album_count + 1 WHERE id = NEW.category_id;
END IF;
END IF;
RETURN NEW;
END IF;
RETURN NULL;
END;
$$ language 'plpgsql';
CREATE TRIGGER trigger_update_album_counts
AFTER INSERT OR UPDATE OR DELETE ON albums
FOR EACH ROW EXECUTE FUNCTION update_album_counts();
-- +migrate Down
DROP TRIGGER IF EXISTS trigger_update_album_counts ON albums;
DROP FUNCTION IF EXISTS update_album_counts();
DROP TRIGGER IF EXISTS trigger_update_photo_counts ON photos;
DROP FUNCTION IF EXISTS update_photo_counts();
ALTER TABLE photos DROP CONSTRAINT IF EXISTS fk_photos_category_id;
ALTER TABLE photos DROP CONSTRAINT IF EXISTS fk_photos_album_id;