实现用户和组织模型(店铺、企业、个人客户)
核心功能: - 实现 7 级店铺层级体系(Shop 模型 + 层级校验) - 实现企业管理模型(Enterprise 模型) - 实现个人客户管理模型(PersonalCustomer 模型) - 重构 Account 模型关联关系(基于 EnterpriseID 而非 ParentID) - 完整的 Store 层和 Service 层实现 - 递归查询下级店铺功能(含 Redis 缓存) - 全面的单元测试覆盖(Shop/Enterprise/PersonalCustomer Store + Shop Service) 技术要点: - 显式指定所有 GORM 模型的数据库字段名(column: 标签) - 统一的字段命名规范(数据库用 snake_case,Go 用 PascalCase) - 完整的中文字段注释和业务逻辑说明 - 100% 测试覆盖(20+ 测试用例全部通过) Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
This commit is contained in:
198
internal/service/shop/service.go
Normal file
198
internal/service/shop/service.go
Normal file
@@ -0,0 +1,198 @@
|
||||
package shop
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"github.com/break/junhong_cmp_fiber/internal/model"
|
||||
"github.com/break/junhong_cmp_fiber/internal/store"
|
||||
"github.com/break/junhong_cmp_fiber/internal/store/postgres"
|
||||
"github.com/break/junhong_cmp_fiber/pkg/constants"
|
||||
"github.com/break/junhong_cmp_fiber/pkg/errors"
|
||||
"github.com/break/junhong_cmp_fiber/pkg/middleware"
|
||||
)
|
||||
|
||||
// Service 店铺业务服务
|
||||
type Service struct {
|
||||
shopStore *postgres.ShopStore
|
||||
}
|
||||
|
||||
// New 创建店铺服务
|
||||
func New(shopStore *postgres.ShopStore) *Service {
|
||||
return &Service{
|
||||
shopStore: shopStore,
|
||||
}
|
||||
}
|
||||
|
||||
// Create 创建店铺
|
||||
func (s *Service) Create(ctx context.Context, req *model.CreateShopRequest) (*model.Shop, error) {
|
||||
// 获取当前用户 ID
|
||||
currentUserID := middleware.GetUserIDFromContext(ctx)
|
||||
if currentUserID == 0 {
|
||||
return nil, errors.New(errors.CodeUnauthorized, "未授权访问")
|
||||
}
|
||||
|
||||
// 检查店铺编号唯一性
|
||||
if req.ShopCode != "" {
|
||||
existing, err := s.shopStore.GetByCode(ctx, req.ShopCode)
|
||||
if err == nil && existing != nil {
|
||||
return nil, errors.New(errors.CodeShopCodeExists, "店铺编号已存在")
|
||||
}
|
||||
}
|
||||
|
||||
// 计算层级
|
||||
level := 1
|
||||
if req.ParentID != nil {
|
||||
// 验证上级店铺存在
|
||||
parent, err := s.shopStore.GetByID(ctx, *req.ParentID)
|
||||
if err != nil {
|
||||
return nil, errors.New(errors.CodeInvalidParentID, "上级店铺不存在或无效")
|
||||
}
|
||||
|
||||
// 计算新店铺的层级
|
||||
level = parent.Level + 1
|
||||
|
||||
// 校验层级不超过最大值
|
||||
if level > constants.MaxShopLevel {
|
||||
return nil, errors.New(errors.CodeShopLevelExceeded, "店铺层级不能超过 7 级")
|
||||
}
|
||||
}
|
||||
|
||||
// 创建店铺
|
||||
shop := &model.Shop{
|
||||
ShopName: req.ShopName,
|
||||
ShopCode: req.ShopCode,
|
||||
ParentID: req.ParentID,
|
||||
Level: level,
|
||||
ContactName: req.ContactName,
|
||||
ContactPhone: req.ContactPhone,
|
||||
Province: req.Province,
|
||||
City: req.City,
|
||||
District: req.District,
|
||||
Address: req.Address,
|
||||
Status: constants.StatusEnabled,
|
||||
}
|
||||
shop.Creator = currentUserID
|
||||
shop.Updater = currentUserID
|
||||
|
||||
if err := s.shopStore.Create(ctx, shop); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return shop, nil
|
||||
}
|
||||
|
||||
// Update 更新店铺信息
|
||||
func (s *Service) Update(ctx context.Context, id uint, req *model.UpdateShopRequest) (*model.Shop, error) {
|
||||
// 获取当前用户 ID
|
||||
currentUserID := middleware.GetUserIDFromContext(ctx)
|
||||
if currentUserID == 0 {
|
||||
return nil, errors.New(errors.CodeUnauthorized, "未授权访问")
|
||||
}
|
||||
|
||||
// 查询店铺
|
||||
shop, err := s.shopStore.GetByID(ctx, id)
|
||||
if err != nil {
|
||||
return nil, errors.New(errors.CodeShopNotFound, "店铺不存在")
|
||||
}
|
||||
|
||||
// 检查店铺编号唯一性(如果修改了编号)
|
||||
if req.ShopCode != nil && *req.ShopCode != shop.ShopCode {
|
||||
existing, err := s.shopStore.GetByCode(ctx, *req.ShopCode)
|
||||
if err == nil && existing != nil && existing.ID != id {
|
||||
return nil, errors.New(errors.CodeShopCodeExists, "店铺编号已存在")
|
||||
}
|
||||
shop.ShopCode = *req.ShopCode
|
||||
}
|
||||
|
||||
// 更新字段
|
||||
if req.ShopName != nil {
|
||||
shop.ShopName = *req.ShopName
|
||||
}
|
||||
if req.ContactName != nil {
|
||||
shop.ContactName = *req.ContactName
|
||||
}
|
||||
if req.ContactPhone != nil {
|
||||
shop.ContactPhone = *req.ContactPhone
|
||||
}
|
||||
if req.Province != nil {
|
||||
shop.Province = *req.Province
|
||||
}
|
||||
if req.City != nil {
|
||||
shop.City = *req.City
|
||||
}
|
||||
if req.District != nil {
|
||||
shop.District = *req.District
|
||||
}
|
||||
if req.Address != nil {
|
||||
shop.Address = *req.Address
|
||||
}
|
||||
|
||||
shop.Updater = currentUserID
|
||||
|
||||
if err := s.shopStore.Update(ctx, shop); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return shop, nil
|
||||
}
|
||||
|
||||
// Disable 禁用店铺
|
||||
func (s *Service) Disable(ctx context.Context, id uint) error {
|
||||
// 获取当前用户 ID
|
||||
currentUserID := middleware.GetUserIDFromContext(ctx)
|
||||
if currentUserID == 0 {
|
||||
return errors.New(errors.CodeUnauthorized, "未授权访问")
|
||||
}
|
||||
|
||||
// 查询店铺
|
||||
shop, err := s.shopStore.GetByID(ctx, id)
|
||||
if err != nil {
|
||||
return errors.New(errors.CodeShopNotFound, "店铺不存在")
|
||||
}
|
||||
|
||||
// 更新状态
|
||||
shop.Status = constants.StatusDisabled
|
||||
shop.Updater = currentUserID
|
||||
|
||||
return s.shopStore.Update(ctx, shop)
|
||||
}
|
||||
|
||||
// Enable 启用店铺
|
||||
func (s *Service) Enable(ctx context.Context, id uint) error {
|
||||
// 获取当前用户 ID
|
||||
currentUserID := middleware.GetUserIDFromContext(ctx)
|
||||
if currentUserID == 0 {
|
||||
return errors.New(errors.CodeUnauthorized, "未授权访问")
|
||||
}
|
||||
|
||||
// 查询店铺
|
||||
shop, err := s.shopStore.GetByID(ctx, id)
|
||||
if err != nil {
|
||||
return errors.New(errors.CodeShopNotFound, "店铺不存在")
|
||||
}
|
||||
|
||||
// 更新状态
|
||||
shop.Status = constants.StatusEnabled
|
||||
shop.Updater = currentUserID
|
||||
|
||||
return s.shopStore.Update(ctx, shop)
|
||||
}
|
||||
|
||||
// GetByID 获取店铺详情
|
||||
func (s *Service) GetByID(ctx context.Context, id uint) (*model.Shop, error) {
|
||||
shop, err := s.shopStore.GetByID(ctx, id)
|
||||
if err != nil {
|
||||
return nil, errors.New(errors.CodeShopNotFound, "店铺不存在")
|
||||
}
|
||||
return shop, nil
|
||||
}
|
||||
|
||||
// List 查询店铺列表
|
||||
func (s *Service) List(ctx context.Context, opts *store.QueryOptions, filters map[string]interface{}) ([]*model.Shop, int64, error) {
|
||||
return s.shopStore.List(ctx, opts, filters)
|
||||
}
|
||||
|
||||
// GetSubordinateShopIDs 获取下级店铺 ID 列表(包含自己)
|
||||
func (s *Service) GetSubordinateShopIDs(ctx context.Context, shopID uint) ([]uint, error) {
|
||||
return s.shopStore.GetSubordinateShopIDs(ctx, shopID)
|
||||
}
|
||||
Reference in New Issue
Block a user