csxj2026/junhong_cmp_fiber
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects 1 Releases Wiki Activity

refactor: align framework cleanup with new bootstrap flow

Browse Source 
Co-authored-by: factory-droid[bot] <138933559+factory-droid[bot]@users.noreply.github.com>
This commit is contained in:
huang
2025-11-19 12:47:25 +08:00
parent 39d14ec093
commit d66323487b
67 changed files with 3020 additions and 3992 deletions
Download Patch File Download Diff File Expand all files Collapse all files

104
internal/store/postgres/order_store.go
View File

@@ -1,104 +0,0 @@
package postgres
import (
"context"
"github.com/break/junhong_cmp_fiber/internal/model"
"gorm.io/gorm"
)
// OrderStore 订单数据访问层
type OrderStore struct {
db *gorm.DB
}
// NewOrderStore 创建订单 Store
func NewOrderStore(db *gorm.DB) *OrderStore {
return &OrderStore{db: db}
}
// Create 创建订单
func (s *OrderStore) Create(ctx context.Context, order *model.Order) error {
return s.db.WithContext(ctx).Create(order).Error
}
// GetByID 根据 ID 获取订单
func (s *OrderStore) GetByID(ctx context.Context, id uint) (*model.Order, error) {
var order model.Order
err := s.db.WithContext(ctx).First(&order, id).Error
if err != nil {
return nil, err
}
return &order, nil
}
// GetByOrderID 根据订单号获取订单
func (s *OrderStore) GetByOrderID(ctx context.Context, orderID string) (*model.Order, error) {
var order model.Order
err := s.db.WithContext(ctx).Where("order_id = ?", orderID).First(&order).Error
if err != nil {
return nil, err
}
return &order, nil
}
// ListByUserID 根据用户 ID 分页获取订单列表
func (s *OrderStore) ListByUserID(ctx context.Context, userID uint, page, pageSize int) ([]model.Order, int64, error) {
var orders []model.Order
var total int64
// 计算总数
if err := s.db.WithContext(ctx).Model(&model.Order{}).Where("user_id = ?", userID).Count(&total).Error; err != nil {
return nil, 0, err
}
// 分页查询
offset := (page - 1) * pageSize
err := s.db.WithContext(ctx).
Where("user_id = ?", userID).
Offset(offset).
Limit(pageSize).
Order("created_at DESC").
Find(&orders).Error
if err != nil {
return nil, 0, err
}
return orders, total, nil
}
// List 分页获取订单列表（全部订单）
func (s *OrderStore) List(ctx context.Context, page, pageSize int) ([]model.Order, int64, error) {
var orders []model.Order
var total int64
// 计算总数
if err := s.db.WithContext(ctx).Model(&model.Order{}).Count(&total).Error; err != nil {
return nil, 0, err
}
// 分页查询
offset := (page - 1) * pageSize
err := s.db.WithContext(ctx).
Offset(offset).
Limit(pageSize).
Order("created_at DESC").
Find(&orders).Error
if err != nil {
return nil, 0, err
}
return orders, total, nil
}
// Update 更新订单
func (s *OrderStore) Update(ctx context.Context, order *model.Order) error {
return s.db.WithContext(ctx).Save(order).Error
}
// Delete 软删除订单
func (s *OrderStore) Delete(ctx context.Context, id uint) error {
return s.db.WithContext(ctx).Delete(&model.Order{}, id).Error
}

86
internal/store/postgres/scopes.go
View File

@@ -1,86 +0,0 @@
package postgres
import (
"context"
"github.com/break/junhong_cmp_fiber/pkg/constants"
"github.com/break/junhong_cmp_fiber/pkg/logger"
"github.com/break/junhong_cmp_fiber/pkg/middleware"
"go.uber.org/zap"
"gorm.io/gorm"
)
// DataPermissionScope 数据权限过滤 Scope
// 根据 context 中的用户信息自动过滤数据
// - root 用户跳过过滤
// - 普通用户只能查看自己和下级的数据
// - 同时限制 shop_id 相同
func DataPermissionScope(ctx context.Context, accountStore *AccountStore) func(db *gorm.DB) *gorm.DB {
return func(db *gorm.DB) *gorm.DB {
// 1. 检查是否为 root 用户，root 用户跳过数据权限过滤
if middleware.IsRootUser(ctx) {
return db
}
// 2. 获取当前用户 ID
userID := middleware.GetUserIDFromContext(ctx)
if userID == 0 {
// 未登录用户返回空结果
logger.GetAppLogger().Warn("数据权限过滤：未获取到用户 ID")
return db.Where("1 = 0")
}
// 3. 获取当前用户的 shop_id
shopID := middleware.GetShopIDFromContext(ctx)
// 4. 获取当前用户及所有下级的 ID
subordinateIDs, err := accountStore.GetSubordinateIDs(ctx, userID)
if err != nil {
// 查询失败时，降级为只能看自己的数据
logger.GetAppLogger().Error("数据权限过滤：获取下级 ID 失败",
zap.Uint("user_id", userID),
zap.Error(err))
subordinateIDs = []uint{userID}
}
// 5. 应用数据权限过滤条件
// owner_id IN (用户自己及所有下级) AND shop_id = 当前用户 shop_id
if len(subordinateIDs) == 0 {
subordinateIDs = []uint{userID}
}
// 根据是否有 shop_id 过滤条件决定 SQL
if shopID != 0 {
return db.Where("owner_id IN ? AND shop_id = ?", subordinateIDs, shopID)
}
// 如果 shop_id 为 0，只根据 owner_id 过滤
return db.Where("owner_id IN ?", subordinateIDs)
}
}
// WithoutDataPermission 跳过数据权限过滤的 Scope
// 用于需要查询所有数据的场景（如管理后台统计、系统任务等）
func WithoutDataPermission() func(db *gorm.DB) *gorm.DB {
return func(db *gorm.DB) *gorm.DB {
// 什么都不做，直接返回原 db
return db
}
}
// SoftDeleteScope 软删除过滤 Scope（GORM 默认已支持，此处作为示例）
// 只查询未软删除的记录
func SoftDeleteScope() func(db *gorm.DB) *gorm.DB {
return func(db *gorm.DB) *gorm.DB {
return db.Where("deleted_at IS NULL")
}
}
// StatusEnabledScope 状态启用过滤 Scope
// 只查询状态为启用的记录
func StatusEnabledScope() func(db *gorm.DB) *gorm.DB {
return func(db *gorm.DB) *gorm.DB {
return db.Where("status = ?", constants.StatusEnabled)
}
}

53
internal/store/postgres/store.go
View File

@@ -1,53 +0,0 @@
package postgres
import (
"context"
"go.uber.org/zap"
"gorm.io/gorm"
)
// Store PostgreSQL 数据访问层整合结构
type Store struct {
db *gorm.DB
logger *zap.Logger
User *UserStore
Order *OrderStore
}
// NewStore 创建新的 PostgreSQL Store 实例
func NewStore(db *gorm.DB, logger *zap.Logger) *Store {
return &Store{
db: db,
logger: logger,
User: NewUserStore(db),
Order: NewOrderStore(db),
}
}
// DB 获取数据库连接
func (s *Store) DB() *gorm.DB {
return s.db
}
// Transaction 执行事务
// 提供统一的事务管理接口，自动处理提交和回滚
// 在事务内部，所有 Store 操作都会使用事务连接
func (s *Store) Transaction(ctx context.Context, fn func(*Store) error) error {
return s.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
// 创建事务内的 Store 实例
txStore := &Store{
db: tx,
logger: s.logger,
User: NewUserStore(tx),
Order: NewOrderStore(tx),
}
return fn(txStore)
})
}
// WithContext 返回带上下文的数据库实例
func (s *Store) WithContext(ctx context.Context) *gorm.DB {
return s.db.WithContext(ctx)
}

78
internal/store/postgres/user_store.go
View File

@@ -1,78 +0,0 @@
package postgres
import (
"context"
"github.com/break/junhong_cmp_fiber/internal/model"
"gorm.io/gorm"
)
// UserStore 用户数据访问层
type UserStore struct {
db *gorm.DB
}
// NewUserStore 创建用户 Store
func NewUserStore(db *gorm.DB) *UserStore {
return &UserStore{db: db}
}
// Create 创建用户
func (s *UserStore) Create(ctx context.Context, user *model.User) error {
return s.db.WithContext(ctx).Create(user).Error
}
// GetByID 根据 ID 获取用户
func (s *UserStore) GetByID(ctx context.Context, id uint) (*model.User, error) {
var user model.User
err := s.db.WithContext(ctx).First(&user, id).Error
if err != nil {
return nil, err
}
return &user, nil
}
// GetByUsername 根据用户名获取用户
func (s *UserStore) GetByUsername(ctx context.Context, username string) (*model.User, error) {
var user model.User
err := s.db.WithContext(ctx).Where("username = ?", username).First(&user).Error
if err != nil {
return nil, err
}
return &user, nil
}
// List 分页获取用户列表
func (s *UserStore) List(ctx context.Context, page, pageSize int) ([]model.User, int64, error) {
var users []model.User
var total int64
// 计算总数
if err := s.db.WithContext(ctx).Model(&model.User{}).Count(&total).Error; err != nil {
return nil, 0, err
}
// 分页查询
offset := (page - 1) * pageSize
err := s.db.WithContext(ctx).
Offset(offset).
Limit(pageSize).
Order("created_at DESC").
Find(&users).Error
if err != nil {
return nil, 0, err
}
return users, total, nil
}
// Update 更新用户
func (s *UserStore) Update(ctx context.Context, user *model.User) error {
return s.db.WithContext(ctx).Save(user).Error
}
// Delete 软删除用户
func (s *UserStore) Delete(ctx context.Context, id uint) error {
return s.db.WithContext(ctx).Delete(&model.User{}, id).Error
}