feat: product number

services/product/internal/logic/numberAddLogic.go

@@ -2,6 +2,7 @@ package logic
 
 import (
 	"context"
+
 	"lone-services/pkg/utils"
 	"lone-services/pkg/validate"
 	product "lone-services/rpc/product/pb"
@@ -10,6 +11,8 @@ import (
 	"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 {
@@ -37,5 +40,25 @@ func (l *NumberAddLogic) NumberAdd(in *product.NumberAddReq) (*product.Response,
 		return failResponse(utils.ErrorNoLoginInfo), nil
 	}
 
-	return changeProductNumber(l.Logger, req.Id, dao.NumberTypeAdd, req.Number)
+	if err := changeProductNumber(l.Logger, req.Id, dao.NumberTypeAdd, req.Number); err != nil {
+		return mapNumberErr(err), nil
+	}
+	return okResponse(utils.StringEmpty), 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)
+	}
 }