414 lines
13 KiB
Go
414 lines
13 KiB
Go
package reward
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"github.com/gogf/gf/v2/os/gtime"
|
||
"server/internal/consts"
|
||
"server/internal/dao"
|
||
"server/internal/model"
|
||
"server/internal/model/do"
|
||
"server/internal/service"
|
||
"server/utility/ecode"
|
||
|
||
"github.com/gogf/gf/v2/database/gdb"
|
||
"github.com/gogf/gf/v2/errors/gerror"
|
||
)
|
||
|
||
type sReward struct{}
|
||
|
||
func New() service.IReward {
|
||
return &sReward{}
|
||
}
|
||
|
||
func init() {
|
||
service.RegisterReward(New())
|
||
}
|
||
|
||
// CreateSystemReward 创建系统奖励
|
||
func (s *sReward) CreateSystemReward(ctx context.Context, in *model.RewardCreateIn) (out *model.CreateOut, err error) {
|
||
if in.Name == "" {
|
||
return nil, gerror.New("奖励名称不能为空")
|
||
}
|
||
exist, err := dao.RewardTypes.Ctx(ctx).WherePri(in.RewardTypeID).Exist()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("奖励类型不存在")
|
||
}
|
||
if !exist {
|
||
return nil, ecode.Params.Sub("奖励类型不存在")
|
||
}
|
||
|
||
// TODO 生成奖励唯一 code
|
||
|
||
// 插入数据
|
||
id, err := dao.Rewards.Ctx(ctx).Data(do.Rewards{
|
||
RewardTypeId: in.RewardTypeID,
|
||
RewardScope: consts.SystemReward, // 系统奖励
|
||
Name: in.Name,
|
||
Code: "xmax",
|
||
Description: in.Description,
|
||
Status: in.Status,
|
||
Stock: in.Stock,
|
||
StartAt: in.StartAt,
|
||
ExpireAt: in.ExpireAt,
|
||
}).OmitEmptyData().InsertAndGetId()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("创建奖励失败")
|
||
}
|
||
return &model.CreateOut{
|
||
Id: id,
|
||
}, nil
|
||
}
|
||
|
||
// CreateStoreReward 创建门店奖励
|
||
func (s *sReward) CreateStoreReward(ctx context.Context, in *model.RewardCreateIn) (out *model.CreateOut, err error) {
|
||
if in.OperatorRole != consts.MerchantRoleCode && in.OperatorRole != consts.StoreRoleCode {
|
||
return nil, ecode.Params.Sub("只有管理员和商户才能创建门店奖励")
|
||
}
|
||
// 验证 OperatorId 对 门店 id 是否存在权限
|
||
if in.OperatorRole == consts.MerchantRoleCode {
|
||
count, err := dao.MerchantAdmins.Ctx(ctx).LeftJoin(
|
||
dao.Stores.Table(),
|
||
fmt.Sprintf("%s.%s=%s.%s", dao.Stores.Table(), dao.Stores.Columns().MerchantId, dao.MerchantAdmins.Table(), dao.MerchantAdmins.Columns().MerchantId),
|
||
).Where(dao.Stores.Columns().Id, in.StoreIDs).WherePri(in.OperatorId).Count()
|
||
if err != nil {
|
||
return nil, gerror.Wrap(err, "检查商户权限失败")
|
||
}
|
||
if count != len(in.StoreIDs) {
|
||
return nil, ecode.Params.Sub("商户权限不足")
|
||
}
|
||
} else {
|
||
count, err := dao.StoreAdmins.Ctx(ctx).
|
||
LeftJoin(
|
||
dao.Stores.Table(),
|
||
fmt.Sprintf("%s.%s=%s.%s", dao.Stores.Table(), dao.Stores.Columns().Id, dao.StoreAdmins.Table(), dao.StoreAdmins.Columns().StoreId),
|
||
).WherePri(in.OperatorId).WhereIn(dao.Stores.Columns().Id, in.StoreIDs).Count()
|
||
if err != nil {
|
||
return nil, gerror.Wrap(err, "检查门店权限失败")
|
||
}
|
||
if count != len(in.StoreIDs) {
|
||
return nil, ecode.Params.Sub("门店权限不足")
|
||
}
|
||
|
||
}
|
||
if in.Name == "" {
|
||
return nil, gerror.New("奖励名称不能为空")
|
||
}
|
||
exist, err := dao.RewardTypes.Ctx(ctx).WherePri(in.RewardTypeID).Exist()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("奖励类型不存在")
|
||
}
|
||
if !exist {
|
||
return nil, ecode.Params.Sub("奖励类型不存在")
|
||
}
|
||
if err = dao.Rewards.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||
// TODO 生成奖励唯一 code
|
||
|
||
// 首先插入奖励
|
||
id, err := dao.Rewards.Ctx(ctx).Data(do.Rewards{
|
||
RewardTypeId: in.RewardTypeID,
|
||
RewardScope: consts.StoreReward, // 门店奖励
|
||
Name: in.Name,
|
||
Code: "",
|
||
Description: in.Description,
|
||
Status: in.Status,
|
||
Stock: in.Stock,
|
||
StartAt: in.StartAt,
|
||
ExpireAt: in.ExpireAt,
|
||
}).OmitEmptyData().InsertAndGetId() // 忽略空值插入
|
||
if err != nil {
|
||
return ecode.Fail.Sub("创建门店奖励失败")
|
||
}
|
||
insertDatas := make([]do.StoreRewards, 0, len(in.StoreIDs))
|
||
// 插入门店奖励关系表
|
||
for _, d := range in.StoreIDs {
|
||
insertDatas = append(insertDatas, do.StoreRewards{
|
||
RewardId: id,
|
||
StoreId: d,
|
||
})
|
||
}
|
||
if _, err := dao.StoreRewards.Ctx(ctx).Data(insertDatas).Insert(); err != nil {
|
||
return ecode.Fail.Sub("创建门店奖励关联失败")
|
||
}
|
||
return nil
|
||
}); err != nil {
|
||
return nil, err
|
||
}
|
||
return
|
||
}
|
||
|
||
// ListSystemReward 查询系统奖励列表
|
||
func (s *sReward) ListSystemReward(ctx context.Context, in *model.RewardListIn) (out *model.RewardListOut, err error) {
|
||
m := dao.Rewards.Ctx(ctx)
|
||
|
||
// 构建查询条件
|
||
if in.Name != "" {
|
||
m = m.WhereLike(dao.Rewards.Columns().Name, "%"+in.Name+"%")
|
||
}
|
||
|
||
if in.RewardTypeID != 0 {
|
||
m = m.Where(do.Rewards{RewardTypeId: in.RewardTypeID})
|
||
}
|
||
|
||
list := make([]model.Reward, 0)
|
||
var total int
|
||
err = m.Page(in.Page, in.Size).Where(do.Rewards{RewardScope: 1, Status: in.Status}).OrderDesc(dao.Rewards.Columns().CreatedAt).OrderDesc(dao.Rewards.Columns().Id).ScanAndCount(&list, &total, false)
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("查询系统奖励失败")
|
||
}
|
||
|
||
return &model.RewardListOut{
|
||
List: list,
|
||
Total: total,
|
||
}, nil
|
||
}
|
||
|
||
// ListStoreReward 查询门店奖励列表
|
||
func (s *sReward) ListStoreReward(ctx context.Context, in *model.RewardListIn) (out *model.RewardListOut, err error) {
|
||
// 商户可以查询下属门店的所有商品信息，商户只能查询自己的门店商品信息
|
||
if in.OperatorRole != consts.MerchantRoleCode && in.OperatorRole != consts.StoreRoleCode {
|
||
return nil, ecode.Params.Sub("只有管理员和商户才能查询门店奖励")
|
||
}
|
||
if in.OperatorRole == consts.MerchantRoleCode {
|
||
// 获取商户管理员所属商户 id，根据 商户 id 和门店 id 查询记录是否
|
||
exist, err := dao.MerchantAdmins.Ctx(ctx).LeftJoin(
|
||
dao.Stores.Table(),
|
||
fmt.Sprintf("%s.%s=%s.%s", dao.Stores.Table(), dao.Stores.Columns().MerchantId, dao.MerchantAdmins.Table(), dao.MerchantAdmins.Columns().MerchantId),
|
||
).WherePri(in.StoreID).Exist()
|
||
if err != nil {
|
||
return nil, gerror.Wrap(err, "检查商户权限失败")
|
||
}
|
||
if !exist {
|
||
return nil, ecode.Params.Sub("商户权限不足")
|
||
}
|
||
}
|
||
|
||
// 构建查询
|
||
query := dao.Rewards.Ctx(ctx).LeftJoin(
|
||
dao.StoreRewards.Table(),
|
||
fmt.Sprintf("%s.%s=%s.%s", dao.Rewards.Table(), dao.Rewards.Columns().Id, dao.StoreRewards.Table(), dao.StoreRewards.Columns().RewardId),
|
||
).Where(dao.Rewards.Columns().RewardScope, consts.StoreReward)
|
||
|
||
// 应用过滤条件
|
||
if in.StoreID != nil && *in.StoreID > 0 {
|
||
query = query.Where(dao.StoreRewards.Columns().StoreId, *in.StoreID)
|
||
}
|
||
if in.RewardTypeID > 0 {
|
||
query = query.Where(dao.Rewards.Columns().RewardTypeId, in.RewardTypeID)
|
||
}
|
||
if in.Name != "" {
|
||
query = query.WhereLike(dao.Rewards.Columns().Name, "%"+in.Name+"%")
|
||
}
|
||
if in.Status > 0 {
|
||
query = query.Where(dao.Rewards.Columns().Status, in.Status)
|
||
}
|
||
if in.RewardScope > 0 {
|
||
query = query.Where(dao.Rewards.Columns().RewardScope, in.RewardScope)
|
||
}
|
||
|
||
// 查询奖励列表
|
||
list := make([]model.Reward, 0)
|
||
var total int
|
||
err = query.Fields(dao.Rewards.Columns()).Page(in.Page, in.Size).ScanAndCount(&list, &total, false)
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("查询奖励列表失败")
|
||
}
|
||
|
||
return &model.RewardListOut{
|
||
List: list,
|
||
Total: total,
|
||
}, nil
|
||
}
|
||
|
||
// UpdateSystemReward 更新系统奖励
|
||
func (s *sReward) UpdateSystemReward(ctx context.Context, in *model.RewardUpdateIn) (out *model.UpdateOut, err error) {
|
||
exist, err := dao.Rewards.Ctx(ctx).Where(do.Rewards{Id: in.ID}).Exist()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("查询该奖励失败")
|
||
}
|
||
if !exist {
|
||
return nil, ecode.Params.Sub("该奖励不存在")
|
||
}
|
||
|
||
_, err = dao.Rewards.Ctx(ctx).Where(do.Rewards{Id: in.ID, RewardScope: 1}).Update(do.Rewards{
|
||
RewardTypeId: in.RewardTypeID,
|
||
Name: in.Name,
|
||
Description: in.Description,
|
||
Status: in.Status,
|
||
Stock: in.Stock,
|
||
StartAt: in.StartAt,
|
||
ExpireAt: in.ExpireAt,
|
||
})
|
||
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("更新奖励失败")
|
||
}
|
||
return &model.UpdateOut{Success: true}, nil
|
||
}
|
||
|
||
// UpdateStoreReward 更新门店奖励
|
||
func (s *sReward) UpdateStoreReward(ctx context.Context, in *model.RewardUpdateIn) (out *model.UpdateOut, err error) {
|
||
if in.OperatorRole != consts.MerchantRoleCode && in.OperatorRole != consts.StoreRoleCode {
|
||
return nil, ecode.Params.Sub("只有管理员和商户才能更新门店奖励")
|
||
}
|
||
// 1. 如果是门店管理员，只需要更新奖励信息即可
|
||
// 2. 如果是商户管理员，更新奖励信息，同时将 StoreIds 中id 不在门店关系表的数据删除
|
||
|
||
// 验证奖励是否存在且为门店奖励
|
||
exist, err := dao.Rewards.Ctx(ctx).Where("id = ? AND reward_scope = ? AND deleted_at IS NULL", in.ID, consts.StoreReward).Exist()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("检查奖励是否存在失败")
|
||
}
|
||
if !exist {
|
||
return nil, ecode.Params.Sub("门店奖励不存在或已删除")
|
||
}
|
||
|
||
if in.Name == "" {
|
||
return nil, ecode.Params.Sub("奖励名称不能为空")
|
||
}
|
||
exist, err = dao.RewardTypes.Ctx(ctx).WherePri(in.RewardTypeID).Exist()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("检查奖励类型失败")
|
||
}
|
||
if !exist {
|
||
return nil, ecode.Params.Sub("奖励类型不存在")
|
||
}
|
||
|
||
// 使用事务确保数据一致性
|
||
err = dao.Rewards.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||
// 更新 rewards 表
|
||
_, err := tx.Model(dao.Rewards).Data(do.Rewards{
|
||
RewardTypeId: in.RewardTypeID,
|
||
Name: in.Name,
|
||
Description: in.Description,
|
||
Status: in.Status,
|
||
Stock: in.Stock,
|
||
StartAt: in.StartAt,
|
||
ExpireAt: in.ExpireAt,
|
||
UpdatedAt: gtime.Now(),
|
||
}).Where("id = ?", in.ID).OmitEmptyData().Update()
|
||
if err != nil {
|
||
return ecode.Fail.Sub("更新奖励信息失败")
|
||
}
|
||
|
||
// 门店管理员不修改门店关联
|
||
if in.OperatorRole == consts.MerchantRoleCode {
|
||
return nil
|
||
}
|
||
|
||
// 验证 StoreIDs 是否存在
|
||
count, err := dao.Stores.Ctx(ctx).WhereIn("id", in.StoreIDs).Count()
|
||
if err != nil {
|
||
return ecode.Fail.Sub("检查门店ID失败")
|
||
}
|
||
if count != len(in.StoreIDs) {
|
||
return ecode.Params.Sub("存在无效的门店ID")
|
||
}
|
||
|
||
// 删除旧的门店关联
|
||
_, err = tx.Model(dao.StoreRewards).Where(do.StoreRewards{RewardId: in.ID}).Delete()
|
||
if err != nil {
|
||
return ecode.Fail.Sub("删除旧门店关联失败")
|
||
}
|
||
|
||
// 插入新的门店关联
|
||
storeRewardData := make([]do.StoreRewards, len(in.StoreIDs))
|
||
for i, storeID := range in.StoreIDs {
|
||
storeRewardData[i] = do.StoreRewards{
|
||
StoreId: storeID,
|
||
RewardId: in.ID,
|
||
}
|
||
}
|
||
_, err = tx.Model(dao.StoreRewards).Data(storeRewardData).Insert()
|
||
if err != nil {
|
||
return ecode.Fail.Sub("关联门店失败")
|
||
}
|
||
|
||
return nil
|
||
})
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return &model.UpdateOut{
|
||
Success: true,
|
||
}, nil
|
||
}
|
||
|
||
// DeleteSystemReward 删除系统奖励
|
||
func (s *sReward) DeleteSystemReward(ctx context.Context, in *model.RewardDeleteIn) (out *model.DeleteOut, err error) {
|
||
|
||
exist, err := dao.Rewards.Ctx(ctx).Where(do.Rewards{Id: in.ID}).Exist()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("查询该奖励失败")
|
||
}
|
||
|
||
if !exist {
|
||
return nil, ecode.Params.Sub("该奖励不存在")
|
||
}
|
||
|
||
_, err = dao.Rewards.Ctx(ctx).Where(do.Rewards{Id: in.ID}).Delete()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("删除奖励失败")
|
||
}
|
||
|
||
return &model.DeleteOut{Success: true}, nil
|
||
}
|
||
|
||
// DeleteStoreReward 删除门店奖励
|
||
func (s *sReward) DeleteStoreReward(ctx context.Context, in *model.RewardDeleteIn) (out *model.DeleteOut, err error) {
|
||
|
||
// 验证操作者角色
|
||
if in.OperatorRole != consts.MerchantRoleCode && in.OperatorRole != consts.StoreRoleCode {
|
||
return nil, ecode.Params.Sub("只有管理员和商户才能删除门店奖励")
|
||
}
|
||
|
||
// 验证奖励是否存在且为门店奖励
|
||
exist, err := dao.Rewards.Ctx(ctx).WherePri(in.ID).Where(do.Rewards{RewardScope: consts.StoreReward}).Exist()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("检查奖励是否存在失败")
|
||
}
|
||
if !exist {
|
||
return nil, ecode.Params.Sub("门店奖励不存在或已删除")
|
||
}
|
||
|
||
storeIds := make([]int64, 0)
|
||
if in.OperatorRole == consts.MerchantRoleCode {
|
||
array, err := dao.MerchantAdmins.Ctx(ctx).LeftJoin(
|
||
dao.Stores.Table(),
|
||
fmt.Sprintf("%s.%s=%s.%s", dao.MerchantAdmins.Table(), dao.MerchantAdmins.Columns().MerchantId, dao.Stores.Table(), dao.Stores.Columns().MerchantId),
|
||
).Where(do.MerchantAdmins{Id: in.OperatorId}).Fields(dao.Stores.Columns().Id).Array()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("查询商户门店失败")
|
||
}
|
||
for _, value := range array {
|
||
storeIds = append(storeIds, value.Int64())
|
||
}
|
||
} else {
|
||
value, err := dao.StoreAdmins.Ctx(ctx).LeftJoin().WherePri(in.OperatorId).Fields(dao.StoreRewards.Columns().StoreId).Value()
|
||
if err != nil {
|
||
return nil, ecode.Fail.Sub("查询门店失败")
|
||
}
|
||
storeIds = append(storeIds, value.Int64())
|
||
}
|
||
if err = dao.Rewards.Transaction(ctx, func(ctx context.Context, tx gdb.TX) error {
|
||
if _, err := dao.Rewards.Ctx(ctx).WherePri(in.ID).Delete(); err != nil {
|
||
return ecode.Fail.Sub("删除奖励失败")
|
||
}
|
||
if _, err := dao.StoreRewards.Ctx(ctx).Where(do.StoreRewards{RewardId: in.ID}).WhereIn(dao.StoreRewards.Columns().StoreId, storeIds).Delete(); err != nil {
|
||
return ecode.Fail.Sub("删除门店奖励关联失败")
|
||
}
|
||
return nil
|
||
}); err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
return &model.DeleteOut{
|
||
Success: true,
|
||
}, nil
|
||
}
|
||
|
||
func (s *sReward) Detail(ctx context.Context, in *model.RewardDetailIn) (out *model.RewardDetailOut, err error) {
|
||
return
|
||
}
|