66 lines
1.7 KiB
Go
66 lines
1.7 KiB
Go
package logic
|
|
|
|
import (
|
|
"context"
|
|
|
|
"lone-services/pkg/utils"
|
|
"lone-services/pkg/validate"
|
|
product "lone-services/rpc/product/pb"
|
|
"lone-services/services/product/internal/dao"
|
|
"lone-services/services/product/internal/model"
|
|
"lone-services/services/product/internal/svc"
|
|
"lone-services/services/product/validator"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
type NumberAddLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
}
|
|
|
|
func NewNumberAddLogic(ctx context.Context, svcCtx *svc.ServiceContext) *NumberAddLogic {
|
|
return &NumberAddLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
func (l *NumberAddLogic) NumberAdd(in *product.NumberAddReq) (*product.Response, error) {
|
|
var req validator.ProductNumberAddValidator
|
|
if msg := validate.ValidateFromProto(in, &req); msg != utils.StringEmpty {
|
|
return outResponse(utils.ErrorParams, msg), nil
|
|
}
|
|
|
|
adminInfo := utils.GetUserFromCtx(l.ctx)
|
|
if adminInfo.ID < utils.NumberOne {
|
|
return failResponse(utils.ErrorNoLoginInfo), nil
|
|
}
|
|
|
|
if err := changeProductNumber(l.Logger, model.ProductModel{}.Init(), req.Id, dao.NumberTypeAdd, req.Number); err != nil {
|
|
return mapNumberErr(err), nil
|
|
}
|
|
return okResponse(nil), nil
|
|
}
|
|
|
|
func mapNumberErr(err error) *product.Response {
|
|
st, ok := status.FromError(err)
|
|
if !ok {
|
|
return failResponse(utils.Fail)
|
|
}
|
|
switch st.Code() {
|
|
case codes.InvalidArgument:
|
|
return outResponse(utils.ErrorParams, st.Message())
|
|
case codes.NotFound:
|
|
return failResponse(utils.ErrorNotFund)
|
|
case codes.FailedPrecondition:
|
|
return failResponse(utils.ErrorStockNotEnough)
|
|
default:
|
|
return failResponse(utils.Fail)
|
|
}
|
|
}
|