package logic import ( "context" "lone-services/pkg/modelbase" "lone-services/pkg/utils" order "lone-services/rpc/order/pb" sale "lone-services/rpc/sale/pb" "lone-services/services/store/internal/dao" "lone-services/services/store/internal/model" "lone-services/services/store/validator" "strconv" "lone-services/rpc/store/pb" "lone-services/services/store/internal/svc" "github.com/zeromicro/go-zero/core/logx" ) type ToSaleLogic struct { ctx context.Context svcCtx *svc.ServiceContext logx.Logger BaseLogic } func NewToSaleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ToSaleLogic { return &ToSaleLogic{ ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx), } } func (l *ToSaleLogic) ToSale(in *store.ToSaleRequest) (*store.Response, error) { var v validator.StoreToSaleValidator if fail := l.checkParams(in, &v); fail != nil { return fail, nil } adminInfo := utils.GetUserFromCtx(l.ctx) if adminInfo.ID < utils.NumberOne { return l.fail(utils.ErrorNoLoginInfo) } var info dao.StoreToSale w := modelbase.Params{ Eq: map[string]string{"id": strconv.Itoa(int(in.Id))}} modelObj := model.StoreModel{}.Init() err := modelObj.GetOne(w, &info) if err != nil { l.Logger.Error(err) return l.fail(utils.Fail) } if info.Id < utils.NumberOne { return l.fail(utils.ErrorNotFund) } if info.SaleId == in.SaleId { return l.ok(utils.NumberOne) } info.SaleId = in.SaleId info.AdminId = adminInfo.ID info.AdminName = adminInfo.Name groupId, sErr := l.getSaleInfo(info.SaleId) if sErr != utils.Ok { return l.fail(sErr) } if groupId < utils.NumberOne { return l.fail(utils.ErrorNotFund) } info.GroupId = groupId modelObj.Begin() _, err = modelObj.Edit(w, &info) if err != nil { modelObj.Rollback() l.Logger.Error(err) return l.fail(utils.Fail) } history := dao.SaleChangeHistoryCreate{ StoreId: info.Id, SaleId: info.SaleId, AdminId: adminInfo.ID, } err = model.SaleChangeHistoryModel{}.Init().Create(&history) if err != nil { modelObj.Rollback() l.Logger.Error(err) return l.fail(utils.Fail) } sErr = l.setAddressSaleInfo(info) if sErr != utils.Ok { modelObj.Rollback() return l.fail(sErr) } modelObj.Commit() return l.ok(utils.NumberOne) } func (l *ToSaleLogic) getSaleInfo(id int64) (int64, utils.Status) { cli, err := svc.GetRpcClient(l.svcCtx.SaleName) if err != nil { logx.Errorf("get order err: %v", err) return utils.NumberZero, utils.ErrorInternalServer } client := sale.NewSaleClient(cli.Conn()) w := sale.InfoReq{Id: id} info, err := client.InfoInternal(context.Background(), &w) if err != nil { logx.Errorf("get order err: %v", err) return utils.NumberZero, utils.ErrorInternalServer } return info.GroupId, utils.Ok } func (l *ToSaleLogic) setAddressSaleInfo(info dao.StoreToSale) utils.Status { cli, err := svc.GetRpcClient(l.svcCtx.OrderName) if err != nil { logx.Errorf("get order err: %v", err) return utils.ErrorInternalServer } client := order.NewOrderClient(cli.Conn()) data := order.UpdateAddressSaleInfoRequest{ Id: info.Id, SaleId: info.SaleId, GroupId: info.GroupId, } _, err = client.UpdateAddressSaleInfo(context.Background(), &data) if err != nil { logx.Errorf("get order err: %v", err) return utils.ErrorInternalServer } return utils.Ok }