feat: 实现 RBAC 权限系统和数据权限控制 (004-rbac-data-permission)
主要功能: - 实现完整的 RBAC 权限系统（账号、角色、权限的多对多关联） - 基于 owner_id + shop_id 的自动数据权限过滤 - 使用 PostgreSQL WITH RECURSIVE 查询下级账号 - Redis 缓存优化下级账号查询性能（30分钟过期） - 支持多租户数据隔离和层级权限管理 技术实现: - 新增 Account、Role、Permission 模型及关联关系表 - 实现 GORM Scopes 自动应用数据权限过滤 - 添加数据库迁移脚本（000002_rbac_data_permission、000003_add_owner_id_shop_id） - 完善错误码定义（1010-1027 为 RBAC 相关错误） - 重构 main.go 采用函数拆分提高可读性 测试覆盖: - 添加 Account、Role、Permission 的集成测试 - 添加数据权限过滤的单元测试和集成测试 - 添加下级账号查询和缓存的单元测试 - 添加 API 回归测试确保向后兼容 文档更新: - 更新 README.md 添加 RBAC 功能说明 - 更新 CLAUDE.md 添加技术栈和开发原则 - 添加 docs/004-rbac-data-permission/ 功能总结和使用指南 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
234
tests/integration/migration_test.go
Normal file
234
tests/integration/migration_test.go
Normal file
@@ -0,0 +1,234 @@
|
||||
package integration
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/golang-migrate/migrate/v4"
|
||||
_ "github.com/golang-migrate/migrate/v4/database/postgres"
|
||||
_ "github.com/golang-migrate/migrate/v4/source/file"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
"github.com/testcontainers/testcontainers-go"
|
||||
testcontainers_postgres "github.com/testcontainers/testcontainers-go/modules/postgres"
|
||||
"github.com/testcontainers/testcontainers-go/wait"
|
||||
postgresDriver "gorm.io/driver/postgres"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/logger"
|
||||
)
|
||||
|
||||
// TestMigration_UpAndDown 测试迁移脚本的向上和向下迁移
|
||||
func TestMigration_UpAndDown(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
// 启动 PostgreSQL 容器
|
||||
postgresContainer, err := testcontainers_postgres.RunContainer(ctx,
|
||||
testcontainers.WithImage("postgres:14-alpine"),
|
||||
testcontainers_postgres.WithDatabase("testdb"),
|
||||
testcontainers_postgres.WithUsername("postgres"),
|
||||
testcontainers_postgres.WithPassword("password"),
|
||||
testcontainers.WithWaitStrategy(
|
||||
wait.ForLog("database system is ready to accept connections").
|
||||
WithOccurrence(2).
|
||||
WithStartupTimeout(30*time.Second),
|
||||
),
|
||||
)
|
||||
require.NoError(t, err, "启动 PostgreSQL 容器失败")
|
||||
defer func() {
|
||||
if err := postgresContainer.Terminate(ctx); err != nil {
|
||||
t.Logf("终止容器失败: %v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
// 获取连接字符串
|
||||
connStr, err := postgresContainer.ConnectionString(ctx, "sslmode=disable")
|
||||
require.NoError(t, err, "获取数据库连接字符串失败")
|
||||
|
||||
// 应用数据库迁移
|
||||
migrationsPath := getMigrationsPath(t)
|
||||
m, err := migrate.New(
|
||||
fmt.Sprintf("file://%s", migrationsPath),
|
||||
connStr,
|
||||
)
|
||||
require.NoError(t, err, "创建迁移实例失败")
|
||||
defer func() { _, _ = m.Close() }()
|
||||
|
||||
t.Run("向上迁移", func(t *testing.T) {
|
||||
err := m.Up()
|
||||
require.NoError(t, err, "执行向上迁移失败")
|
||||
|
||||
// 验证表已创建
|
||||
db, err := gorm.Open(postgresDriver.Open(connStr), &gorm.Config{
|
||||
Logger: logger.Default.LogMode(logger.Silent),
|
||||
})
|
||||
require.NoError(t, err, "连接数据库失败")
|
||||
|
||||
// 检查 RBAC 表存在
|
||||
tables := []string{
|
||||
"tb_account",
|
||||
"tb_role",
|
||||
"tb_permission",
|
||||
"tb_account_role",
|
||||
"tb_role_permission",
|
||||
}
|
||||
|
||||
for _, table := range tables {
|
||||
var exists bool
|
||||
err := db.Raw("SELECT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = ?)", table).Scan(&exists).Error
|
||||
assert.NoError(t, err)
|
||||
assert.True(t, exists, "表 %s 应该存在", table)
|
||||
}
|
||||
|
||||
// 检查索引
|
||||
var indexCount int64
|
||||
err = db.Raw(`
|
||||
SELECT COUNT(*) FROM pg_indexes
|
||||
WHERE tablename = 'tb_account'
|
||||
AND indexname LIKE 'idx_account_%'
|
||||
`).Scan(&indexCount).Error
|
||||
assert.NoError(t, err)
|
||||
assert.Greater(t, indexCount, int64(0), "tb_account 表应该有索引")
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
if sqlDB != nil {
|
||||
_ = sqlDB.Close()
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("向下迁移", func(t *testing.T) {
|
||||
err := m.Down()
|
||||
require.NoError(t, err, "执行向下迁移失败")
|
||||
|
||||
// 验证表已删除
|
||||
db, err := gorm.Open(postgresDriver.Open(connStr), &gorm.Config{
|
||||
Logger: logger.Default.LogMode(logger.Silent),
|
||||
})
|
||||
require.NoError(t, err, "连接数据库失败")
|
||||
|
||||
// 检查 RBAC 表已删除
|
||||
tables := []string{
|
||||
"tb_account",
|
||||
"tb_role",
|
||||
"tb_permission",
|
||||
"tb_account_role",
|
||||
"tb_role_permission",
|
||||
}
|
||||
|
||||
for _, table := range tables {
|
||||
var exists bool
|
||||
err := db.Raw("SELECT EXISTS (SELECT 1 FROM information_schema.tables WHERE table_name = ?)", table).Scan(&exists).Error
|
||||
assert.NoError(t, err)
|
||||
assert.False(t, exists, "表 %s 应该已删除", table)
|
||||
}
|
||||
|
||||
sqlDB, _ := db.DB()
|
||||
if sqlDB != nil {
|
||||
_ = sqlDB.Close()
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// TestMigration_NoForeignKeys 验证迁移脚本不包含外键约束
|
||||
func TestMigration_NoForeignKeys(t *testing.T) {
|
||||
// 获取迁移目录
|
||||
migrationsPath := getMigrationsPath(t)
|
||||
|
||||
// 读取所有迁移文件
|
||||
files, err := filepath.Glob(filepath.Join(migrationsPath, "*.up.sql"))
|
||||
require.NoError(t, err)
|
||||
|
||||
forbiddenKeywords := []string{
|
||||
"FOREIGN KEY",
|
||||
"REFERENCES",
|
||||
"ON DELETE CASCADE",
|
||||
"ON UPDATE CASCADE",
|
||||
}
|
||||
|
||||
for _, file := range files {
|
||||
content, err := os.ReadFile(file)
|
||||
require.NoError(t, err)
|
||||
|
||||
for _, keyword := range forbiddenKeywords {
|
||||
assert.NotContains(t, string(content), keyword,
|
||||
"迁移文件 %s 不应包含外键约束关键字: %s", filepath.Base(file), keyword)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TestMigration_SoftDeleteSupport 验证表支持软删除
|
||||
func TestMigration_SoftDeleteSupport(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
|
||||
// 启动 PostgreSQL 容器
|
||||
postgresContainer, err := testcontainers_postgres.RunContainer(ctx,
|
||||
testcontainers.WithImage("postgres:14-alpine"),
|
||||
testcontainers_postgres.WithDatabase("testdb"),
|
||||
testcontainers_postgres.WithUsername("postgres"),
|
||||
testcontainers_postgres.WithPassword("password"),
|
||||
testcontainers.WithWaitStrategy(
|
||||
wait.ForLog("database system is ready to accept connections").
|
||||
WithOccurrence(2).
|
||||
WithStartupTimeout(30*time.Second),
|
||||
),
|
||||
)
|
||||
require.NoError(t, err, "启动 PostgreSQL 容器失败")
|
||||
defer func() {
|
||||
if err := postgresContainer.Terminate(ctx); err != nil {
|
||||
t.Logf("终止容器失败: %v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
// 获取连接字符串
|
||||
connStr, err := postgresContainer.ConnectionString(ctx, "sslmode=disable")
|
||||
require.NoError(t, err, "获取数据库连接字符串失败")
|
||||
|
||||
// 应用迁移
|
||||
migrationsPath := getMigrationsPath(t)
|
||||
m, err := migrate.New(
|
||||
fmt.Sprintf("file://%s", migrationsPath),
|
||||
connStr,
|
||||
)
|
||||
require.NoError(t, err, "创建迁移实例失败")
|
||||
defer func() { _, _ = m.Close() }()
|
||||
|
||||
err = m.Up()
|
||||
require.NoError(t, err, "执行向上迁移失败")
|
||||
|
||||
// 连接数据库验证
|
||||
db, err := gorm.Open(postgresDriver.Open(connStr), &gorm.Config{
|
||||
Logger: logger.Default.LogMode(logger.Silent),
|
||||
})
|
||||
require.NoError(t, err, "连接数据库失败")
|
||||
defer func() {
|
||||
sqlDB, _ := db.DB()
|
||||
if sqlDB != nil {
|
||||
_ = sqlDB.Close()
|
||||
}
|
||||
}()
|
||||
|
||||
// 检查每个表都有 deleted_at 列和索引
|
||||
tables := []string{
|
||||
"tb_account",
|
||||
"tb_role",
|
||||
"tb_permission",
|
||||
"tb_account_role",
|
||||
"tb_role_permission",
|
||||
}
|
||||
|
||||
for _, table := range tables {
|
||||
// 检查 deleted_at 列存在
|
||||
var columnExists bool
|
||||
err := db.Raw(`
|
||||
SELECT EXISTS (
|
||||
SELECT 1 FROM information_schema.columns
|
||||
WHERE table_name = ? AND column_name = 'deleted_at'
|
||||
)
|
||||
`, table).Scan(&columnExists).Error
|
||||
assert.NoError(t, err)
|
||||
assert.True(t, columnExists, "表 %s 应该有 deleted_at 列", table)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user