wechat-api/internal/logic/credit_balance/operate_credit_balance_logic.go

package credit_balance

import (
	"context"
	"github.com/suyuan32/simple-admin-common/msg/errormsg"
	"github.com/suyuan32/simple-admin-core/rpc/types/core"
	"github.com/zeromicro/go-zero/core/errorx"
	"wechat-api/ent/creditbalance"

	"wechat-api/internal/svc"
	"wechat-api/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
)

type OperateCreditBalanceLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewOperateCreditBalanceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *OperateCreditBalanceLogic {
	return &OperateCreditBalanceLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx}
}

func (l *OperateCreditBalanceLogic) OperateCreditBalance(req *types.CreditBalanceOperateReq) (*types.BaseMsgResp, error) {
	isAdmin := l.ctx.Value("isAdmin").(bool)
	organizationId := l.ctx.Value("organizationId").(uint64)

	if !isAdmin {
		return nil, errorx.NewInvalidArgumentError("权限不足")
	}

	if req.Number == nil || *req.Number == 0 {
		return nil, errorx.NewInvalidArgumentError("Number参数非法")
	}

	_, err := l.svcCtx.CoreRpc.GetDepartmentById(l.ctx, &core.IDReq{Id: *req.OrganizationId})
	if err != nil {
		return nil, errorx.NewAlreadyExistsError("租户不存在")
	}

	creditBalance, err := l.svcCtx.DB.CreditBalance.Query().Where(creditbalance.OrganizationID(*req.OrganizationId)).Only(l.ctx)
	if err != nil {
		l.Logger.Errorf("query credit_balance error:%v\n", err)
		return nil, errorx.NewInvalidArgumentError(err.Error())
	}

	var ntype int
	if *req.Number > 0 {
		ntype = 2
	} else {
		ntype = 1
	}

	userId := l.ctx.Value("userId").(string)
	tx, err := l.svcCtx.DB.Tx(context.Background())

	var beforeNumber, afterNumber float64
	if creditBalance == nil {
		beforeNumber = 0
		afterNumber = *req.Number
	} else {
		beforeNumber = creditBalance.Balance
		afterNumber = beforeNumber + *req.Number
	}

	_, err = tx.CreditUsage.Create().
		SetNumber(*req.Number).
		SetNotNilPayMethod(req.PayMethod). //支付方式
		SetNotNilPayNumber(req.PayNumber). //支付金额
		SetNtype(ntype).
		SetNid(0).
		SetTable("").
		SetNotNilReason(req.Reason).
		SetOrganizationID(organizationId).
		SetOperator(userId).
		SetBeforeNumber(beforeNumber).
		SetAfterNumber(afterNumber).
		Save(l.ctx)
	if err != nil {
		l.Logger.Errorf("create credit_usage error:%v\n", err)
		_ = tx.Rollback()
		return nil, errorx.NewInvalidArgumentError("创建租户积分明细失败，请重试")
	}

	if *req.Number > 0 {
		if creditBalance == nil {
			_, err = l.svcCtx.DB.CreditBalance.Create().
				SetOrganizationID(*req.OrganizationId).
				SetBalance(*req.Number).
				Save(l.ctx)
			if err != nil {
				l.Logger.Errorf("create credit_balance error:%v\n", err)
				_ = tx.Rollback()
				return nil, errorx.NewInvalidArgumentError("创建租户积分余额失败，请重试")
			}
		} else {
			_, err = l.svcCtx.DB.CreditBalance.Update().Where(creditbalance.OrganizationID(*req.OrganizationId)).SetBalance(creditBalance.Balance + *req.Number).Save(l.ctx)
			if err != nil {
				l.Logger.Errorf("update credit_balance error:%v\n", err)
				_ = tx.Rollback()
				return nil, errorx.NewInvalidArgumentError("修改租户积分余额失败，请重试.")
			}
		}
	} else {
		_, err := l.svcCtx.DB.CreditBalance.Update().Where(creditbalance.OrganizationID(*req.OrganizationId)).SetBalance(creditBalance.Balance + *req.Number).Save(l.ctx)
		if err != nil {
			l.Logger.Errorf("update credit_balance error:%v\n", err)
			_ = tx.Rollback()
			return nil, errorx.NewInvalidArgumentError("修改租户积分余额失败，请重试...")
		}
	}
	_ = tx.Commit()

	return &types.BaseMsgResp{
		Code: 0,
		Msg:  errormsg.Success,
	}, nil
}