Merge branch 'develop' of https://gitee.com/fusenpack/fusenapi into develop

server/inventory/etc/inventory.yaml

@@ -0,0 +1,8 @@
+Name: inventory
+Host: 0.0.0.0
+Port: 8898
+SourceMysql: fusentest:XErSYmLELKMnf3Dh@tcp(110.41.19.98:3306)/fusentest
+Auth:
+    AccessSecret: fusen2023
+    AccessExpire: 604800
+    RefreshAfter: 345600

server/inventory/internal/config/config.go

@@ -0,0 +1,12 @@
+package config
+
+import (
+	"fusenapi/server/inventory/internal/types"
+	"github.com/zeromicro/go-zero/rest"
+)
+
+type Config struct {
+	rest.RestConf
+	SourceMysql string
+	Auth        types.Auth
+}

server/inventory/internal/handler/getcloudlisthandler.go

@@ -0,0 +1,78 @@
+package handler
+
+import (
+	"errors"
+	"net/http"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"fusenapi/utils/auth"
+	"fusenapi/utils/basic"
+
+	"fusenapi/server/inventory/internal/logic"
+	"fusenapi/server/inventory/internal/svc"
+	"fusenapi/server/inventory/internal/types"
+)
+
+func GetCloudListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+
+		var (
+			// 定义错误变量
+			err error
+			// 定义用户信息变量
+			userinfo *auth.UserInfo
+		)
+		// 解析JWT token,并对空用户进行判断
+		claims, err := svcCtx.ParseJwtToken(r)
+		// 如果解析JWT token出错,则返回未授权的JSON响应并记录错误消息
+		if err != nil {
+			httpx.OkJsonCtx(r.Context(), w, &basic.Response{
+				Code:    401,            // 返回401状态码,表示未授权
+				Message: "unauthorized", // 返回未授权信息
+			})
+			logx.Info("unauthorized:", err.Error()) // 记录错误日志
+			return
+		}
+
+		if claims != nil {
+			// 从token中获取对应的用户信息
+			userinfo, err = auth.GetUserInfoFormMapClaims(claims)
+			// 如果获取用户信息出错,则返回未授权的JSON响应并记录错误消息
+			if err != nil {
+				httpx.OkJsonCtx(r.Context(), w, &basic.Response{
+					Code:    401,
+					Message: "unauthorized",
+				})
+				logx.Info("unauthorized:", err.Error())
+				return
+			}
+		} else {
+			// 如果claims为nil,则认为用户身份为白板用户
+			userinfo = &auth.UserInfo{UserId: 0, GuestId: 0}
+		}
+
+		var req types.GetCloudListReq
+		// 如果端点有请求结构体，则使用httpx.Parse方法从HTTP请求体中解析请求数据
+		if err := httpx.Parse(r, &req); err != nil {
+			httpx.OkJsonCtx(r.Context(), w, &basic.Response{
+				Code:    510,
+				Message: "parameter error",
+			})
+			logx.Info(err)
+			return
+		}
+		// 创建一个业务逻辑层实例
+		l := logic.NewGetCloudListLogic(r.Context(), svcCtx)
+		resp := l.GetCloudList(&req, userinfo)
+		// 如果响应不为nil，则使用httpx.OkJsonCtx方法返回JSON响应;
+		if resp != nil {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		} else {
+			err := errors.New("server logic is error, resp must not be nil")
+			httpx.ErrorCtx(r.Context(), w, err)
+			logx.Error(err)
+		}
+	}
+}

server/inventory/internal/handler/routes.go

@@ -0,0 +1,27 @@
+// Code generated by goctl. DO NOT EDIT.
+package handler
+
+import (
+	"net/http"
+
+	"fusenapi/server/inventory/internal/svc"
+
+	"github.com/zeromicro/go-zero/rest"
+)
+
+func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
+	server.AddRoutes(
+		[]rest.Route{
+			{
+				Method:  http.MethodPost,
+				Path:    "/inventory/take",
+				Handler: TakeHandler(serverCtx),
+			},
+			{
+				Method:  http.MethodGet,
+				Path:    "/inventory/list",
+				Handler: GetCloudListHandler(serverCtx),
+			},
+		},
+	)
+}

server/inventory/internal/handler/takehandler.go

@@ -0,0 +1,78 @@
+package handler
+
+import (
+	"errors"
+	"net/http"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"fusenapi/utils/auth"
+	"fusenapi/utils/basic"
+
+	"fusenapi/server/inventory/internal/logic"
+	"fusenapi/server/inventory/internal/svc"
+	"fusenapi/server/inventory/internal/types"
+)
+
+func TakeHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+
+		var (
+			// 定义错误变量
+			err error
+			// 定义用户信息变量
+			userinfo *auth.UserInfo
+		)
+		// 解析JWT token,并对空用户进行判断
+		claims, err := svcCtx.ParseJwtToken(r)
+		// 如果解析JWT token出错,则返回未授权的JSON响应并记录错误消息
+		if err != nil {
+			httpx.OkJsonCtx(r.Context(), w, &basic.Response{
+				Code:    401,            // 返回401状态码,表示未授权
+				Message: "unauthorized", // 返回未授权信息
+			})
+			logx.Info("unauthorized:", err.Error()) // 记录错误日志
+			return
+		}
+
+		if claims != nil {
+			// 从token中获取对应的用户信息
+			userinfo, err = auth.GetUserInfoFormMapClaims(claims)
+			// 如果获取用户信息出错,则返回未授权的JSON响应并记录错误消息
+			if err != nil {
+				httpx.OkJsonCtx(r.Context(), w, &basic.Response{
+					Code:    401,
+					Message: "unauthorized",
+				})
+				logx.Info("unauthorized:", err.Error())
+				return
+			}
+		} else {
+			// 如果claims为nil,则认为用户身份为白板用户
+			userinfo = &auth.UserInfo{UserId: 0, GuestId: 0}
+		}
+
+		var req types.TakeReq
+		// 如果端点有请求结构体，则使用httpx.Parse方法从HTTP请求体中解析请求数据
+		if err := httpx.Parse(r, &req); err != nil {
+			httpx.OkJsonCtx(r.Context(), w, &basic.Response{
+				Code:    510,
+				Message: "parameter error",
+			})
+			logx.Info(err)
+			return
+		}
+		// 创建一个业务逻辑层实例
+		l := logic.NewTakeLogic(r.Context(), svcCtx)
+		resp := l.Take(&req, userinfo)
+		// 如果响应不为nil，则使用httpx.OkJsonCtx方法返回JSON响应;
+		if resp != nil {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		} else {
+			err := errors.New("server logic is error, resp must not be nil")
+			httpx.ErrorCtx(r.Context(), w, err)
+			logx.Error(err)
+		}
+	}
+}

server/inventory/internal/logic/getcloudlistlogic.go

@@ -0,0 +1,154 @@
+package logic
+
+import (
+	"fusenapi/constants"
+	"fusenapi/model/gmodel"
+	"fusenapi/utils/auth"
+	"fusenapi/utils/basic"
+	"fusenapi/utils/format"
+	"strings"
+
+	"context"
+
+	"fusenapi/server/inventory/internal/svc"
+	"fusenapi/server/inventory/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type GetCloudListLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewGetCloudListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetCloudListLogic {
+	return &GetCloudListLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *GetCloudListLogic) GetCloudList(req *types.GetCloudListReq, userinfo *auth.UserInfo) (resp *basic.Response) {
+	if userinfo.GetIdType() != auth.IDTYPE_User {
+		return resp.SetStatusWithMessage(basic.CodeServiceErr, "please login first")
+	}
+	if req.Page <= 0 {
+		req.Page = constants.DEFAULT_PAGE
+	}
+	if req.PageSize <= 0 || req.PageSize > 200 {
+		req.Page = constants.DEFAULT_PAGE_SIZE
+	}
+	//获取个人云仓列表
+	stockList, total, err := l.svcCtx.AllModels.FsUserStock.GetStockList(l.ctx, gmodel.GetStockListReq{
+		UserId: userinfo.UserId,
+		Page:   int(req.Page),
+		Limit:  int(req.PageSize),
+	})
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get user stock list")
+	}
+	if len(stockList) == 0 {
+		return resp.SetStatusWithMessage(basic.CodeOK, "success", types.GetCloudListRsp{
+			ListData: []types.ListDataItem{},
+		})
+	}
+	designIds := make([]int64, 0, len(stockList))
+	for _, v := range stockList {
+		designIds = append(designIds, *v.DesignId)
+	}
+	//获取设计数据
+	productDesignList, err := l.svcCtx.AllModels.FsProductDesign.GetAllByIds(l.ctx, designIds)
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get product design list")
+	}
+	//尺寸ids
+	sizeIds := make([]int64, 0, len(productDesignList))
+	//产品ids
+	productIds := make([]int64, 0, len(productDesignList))
+	//模板ids
+	templateIds := make([]int64, 0, len(productDesignList))
+	//配件ids
+	optionalIds := make([]int64, 0, len(productDesignList))
+	mapProductDesign := make(map[int64]int)
+	for k, v := range productDesignList {
+		sizeIds = append(sizeIds, *v.SizeId)
+		productIds = append(productIds, *v.ProductId)
+		templateIds = append(templateIds, *v.TemplateId)
+		optionalIds = append(optionalIds, *v.OptionalId)
+		mapProductDesign[v.Id] = k
+	}
+	//获取尺寸信息
+	sizeList, err := l.svcCtx.AllModels.FsProductSize.GetAllByProductIdsWithoutStatus(l.ctx, sizeIds, "")
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get product size list")
+	}
+	mapSize := make(map[int64]int)
+	for k, v := range sizeList {
+		mapSize[v.Id] = k
+	}
+	//获取产品信息
+	productList, err := l.svcCtx.AllModels.FsProduct.GetProductListByIdsWithoutStatus(l.ctx, productIds, "")
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get product list")
+	}
+	mapProduct := make(map[int64]int)
+	for k, v := range productList {
+		mapProduct[v.Id] = k
+	}
+	//获取模板信息
+	productTemplateList, err := l.svcCtx.AllModels.FsProductTemplateV2.FindAllByIdsWithoutStatus(l.ctx, templateIds)
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get product template list")
+	}
+	mapTemplate := make(map[int64]int)
+	for k, v := range productTemplateList {
+		mapTemplate[v.Id] = k
+	}
+	//获取配件列表
+	productModel3dList, err := l.svcCtx.AllModels.FsProductModel3d.GetAllByIdsWithoutStatus(l.ctx, optionalIds)
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get product 3d model list")
+	}
+	mapProductModel := make(map[int64]int)
+	for k, v := range productModel3dList {
+		mapProductModel[v.Id] = k
+	}
+	//根据产品ids获取产品价格
+	priceList, err := l.svcCtx.AllModels.FsProductPrice.GetPriceListByProductIds(l.ctx, productIds)
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get product price list")
+	}
+	for _, v := range priceList {
+		if *v.StepNum == "" || *v.StepPrice == "" {
+			return resp.SetStatusWithMessage(basic.CodeServiceErr, "price data`s step num or step price is empty")
+		}
+		stepNum, err := format.StrSlicToIntSlice(strings.Split(*v.StepNum, ","))
+		if err != nil {
+			logx.Error(err)
+			return resp.SetStatusWithMessage(basic.CodeServiceErr, "parse step num err")
+		}
+		lenStepNum := len(stepNum)
+		stepPrice, err := format.StrSlicToIntSlice(strings.Split(*v.StepPrice, ","))
+		if err != nil {
+			logx.Error(err)
+			return resp.SetStatusWithMessage(basic.CodeServiceErr, "parse step price err")
+		}
+		lenStepPrice := len(stepPrice)
+		for *v.MinBuyNum < int64(stepPrice[lenStepPrice-1]+5) {
+			//根据材质，尺寸，价格计算阶梯价
+
+			*v.MinBuyNum++
+		}
+
+	}
+	return resp.SetStatus(basic.CodeOK)
+}

server/inventory/internal/logic/takelogic.go

@@ -0,0 +1,129 @@
+package logic
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"fusenapi/constants"
+	"fusenapi/model/gmodel"
+	"fusenapi/server/inventory/internal/types"
+	"fusenapi/utils/auth"
+	"fusenapi/utils/basic"
+	"fusenapi/utils/id_generator"
+	"gorm.io/gorm"
+	"time"
+
+	"fusenapi/server/inventory/internal/svc"
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type TakeLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewTakeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *TakeLogic {
+	return &TakeLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *TakeLogic) Take(req *types.TakeReq, userinfo *auth.UserInfo) (resp *basic.Response) {
+	if userinfo.GetIdType() != auth.IDTYPE_User {
+		return resp.SetStatusWithMessage(basic.CodeServiceErr, "please login first")
+	}
+	if len(req.Form) == 0 {
+		return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, "param err :form can`t be empty array")
+	}
+	if req.AddressId <= 0 {
+		return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, "param err :address_id is required")
+	}
+	//获取地址信息
+	addressInfo, err := l.svcCtx.AllModels.FsAddress.GetOne(l.ctx, req.AddressId, userinfo.UserId)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return resp.SetStatusWithMessage(basic.CodeDbRecordNotFoundErr, "your address is not exists")
+		}
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get address info")
+	}
+	stockIds := make([]int64, 0, len(req.Form))
+	for _, v := range req.Form {
+		stockIds = append(stockIds, v.Id)
+	}
+	//提货单总单
+	addressInfoBytes, _ := json.Marshal(addressInfo)
+	addressInfoJson := string(addressInfoBytes)
+	trackNum := id_generator.GenPickUpTrackNum()
+	status := int64(constants.STATUS_ORDERD)
+	now := time.Now().Unix()
+	pickUpData := gmodel.FsCloudPickUp{
+		UserId:      &userinfo.UserId,
+		TrackNum:    &trackNum,
+		AddressId:   &req.AddressId,
+		AddressInfo: &addressInfoJson,
+		Status:      &status,
+		Ctime:       &now,
+	}
+	//箱数验证
+	boxes := int64(0)
+	//需要更新的库存信息
+	stockUpdateList := make([]gmodel.FsUserStock, 0, len(req.Form))
+	//需要新增的提货详情单
+	pickUpDetailAddList := make([]gmodel.FsCloudPickUpDetail, 0, len(req.Form))
+	for k, val := range req.Form {
+		formItem := val
+		//验证提取数量
+		if formItem.Num <= 0 {
+			return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, fmt.Sprintf("row %d inventory data`s take num can`be less than 0", k+1))
+		}
+		//获取库存信息（枷锁）
+		stockInfo, err := l.svcCtx.AllModels.FsUserStock.FindOne(l.ctx, formItem.Id, userinfo.UserId, true)
+		if err != nil {
+			if errors.Is(err, gorm.ErrRecordNotFound) {
+				return resp.SetStatusWithMessage(basic.CodeDbRecordNotFoundErr, fmt.Sprintf("row %d inventory data is not availabled for you", k+1))
+			}
+			logx.Error(err)
+			return resp.SetStatusWithMessage(basic.CodeDbSqlErr, fmt.Sprintf("failed to get row %d`s stock info ", k+1))
+		}
+		//校验取货数量
+		if *stockInfo.Stick < formItem.Num {
+			return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, fmt.Sprintf("row %d inventory data is shortage", k+1))
+		}
+		if *stockInfo.EachBoxNum <= 0 {
+			return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, fmt.Sprintf("row %d inventory data each box num can`t be less than 0", k+1))
+		}
+		boxes += (formItem.Num / *stockInfo.EachBoxNum)
+		//库存变更
+		newStick := *stockInfo.Stick - formItem.Num
+		newTransNum := *stockInfo.TransNum + formItem.Num
+		stockUpdateList = append(stockUpdateList, gmodel.FsUserStock{
+			Id:       stockInfo.Id,
+			Stick:    &newStick,
+			TransNum: &newTransNum,
+		})
+		//提货详情单
+		detailBoxes := formItem.Num / *stockInfo.EachBoxNum
+		pickUpDetailAddList = append(pickUpDetailAddList, gmodel.FsCloudPickUpDetail{
+			PickId:  nil, //到model里方法会给他赋值pick总单id
+			StockId: &formItem.Id,
+			Num:     &formItem.Num,
+			Boxes:   &detailBoxes,
+			Ctime:   &now,
+		})
+	}
+	if boxes < 3 {
+		return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, "Take out more than three boxes")
+	}
+	//事务处理数据
+	err = l.svcCtx.AllModels.FsCloudPickUp.SavePickUpWithTransaction(l.ctx, &pickUpData, stockUpdateList, pickUpDetailAddList)
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to take your goods")
+	}
+	return resp.SetStatusWithMessage(basic.CodeOK, "success", []int64{})
+}

server/inventory/internal/svc/servicecontext.go

@@ -0,0 +1,60 @@
+package svc
+
+import (
+	"errors"
+	"fmt"
+	"fusenapi/server/inventory/internal/config"
+	"net/http"
+
+	"fusenapi/initalize"
+	"fusenapi/model/gmodel"
+
+	"github.com/golang-jwt/jwt"
+	"gorm.io/gorm"
+)
+
+type ServiceContext struct {
+	Config config.Config
+
+	MysqlConn *gorm.DB
+	AllModels *gmodel.AllModelsGen
+}
+
+func NewServiceContext(c config.Config) *ServiceContext {
+
+	return &ServiceContext{
+		Config:    c,
+		MysqlConn: initalize.InitMysql(c.SourceMysql),
+		AllModels: gmodel.NewAllModels(initalize.InitMysql(c.SourceMysql)),
+	}
+}
+
+func (svcCtx *ServiceContext) ParseJwtToken(r *http.Request) (jwt.MapClaims, error) {
+	AuthKey := r.Header.Get("Authorization")
+	if AuthKey == "" {
+		return nil, nil
+	}
+
+	if len(AuthKey) <= 50 {
+		return nil, errors.New(fmt.Sprint("Error parsing token, len:", len(AuthKey)))
+	}
+
+	token, err := jwt.Parse(AuthKey, func(token *jwt.Token) (interface{}, error) {
+		// 检查签名方法是否为 HS256
+		if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
+			return nil, fmt.Errorf("unexpected signing method: %v", token.Header["alg"])
+		}
+		// 返回用于验证签名的密钥
+		return []byte(svcCtx.Config.Auth.AccessSecret), nil
+	})
+	if err != nil {
+		return nil, errors.New(fmt.Sprint("Error parsing token:", err))
+	}
+
+	// 验证成功返回
+	if claims, ok := token.Claims.(jwt.MapClaims); ok && token.Valid {
+		return claims, nil
+	}
+
+	return nil, errors.New(fmt.Sprint("Invalid token", err))
+}

server/inventory/internal/types/types.go

@@ -0,0 +1,117 @@
+// Code generated by goctl. DO NOT EDIT.
+package types
+
+import (
+	"fusenapi/utils/basic"
+)
+
+type TakeReq struct {
+	Form      []TakeForm `json:"form"`
+	AddressId int64      `json:"address_id"`
+}
+
+type TakeForm struct {
+	Id  int64 `json:"id"`
+	Num int64 `json:"num"`
+}
+
+type GetCloudListReq struct {
+	Page     int64 `json:"page"`
+	PageSize int64 `json:"page_size"`
+	Size     int64 `json:"size"`
+}
+
+type GetCloudListRsp struct {
+	WarehouseBoxes int64          `json:"warehouse_boxes"`
+	TransitBoxes   int64          `json:"transit_boxes"`
+	MinTakeNum     int64          `json:"minTakeNum"`
+	ListData       []ListDataItem `json:"listData"`
+	Pagnation      Pagnation      `json:"pagnation"`
+}
+
+type ListDataItem struct {
+	Id            int64       `json:"id"`
+	Sn            string      `json:"sn"`
+	Cover         string      `json:"cover"`
+	Name          string      `json:"name"`
+	DesignSn      string      `json:"design_sn"`
+	Fitting       string      `json:"fitting"`
+	Production    int64       `json:"production"`
+	ProductionBox int64       `json:"production_box"`
+	EachBoxNum    int64       `json:"each_box_num"`
+	Stick         int64       `json:"stick"`
+	StickBox      int64       `json:"stick_box"`
+	Type          int64       `json:"type"`
+	TakeNum       int64       `json:"takeNum"`
+	Size          string      `json:"size"`
+	IsStop        int64       `json:"is_stop"`
+	PriceList     []PriceItem `json:"price"`
+}
+
+type PriceItem struct {
+	Num      int `json:"num"`
+	TotalNum int `json:"total_num"`
+	Price    int `json:"price"`
+}
+
+type Request struct {
+}
+
+type Response struct {
+	Code    int         `json:"code"`
+	Message string      `json:"msg"`
+	Data    interface{} `json:"data"`
+}
+
+type Auth struct {
+	AccessSecret string `json:"accessSecret"`
+	AccessExpire int64  `json:"accessExpire"`
+	RefreshAfter int64  `json:"refreshAfter"`
+}
+
+type Pagnation struct {
+	TotalCount int64 `json:"total_count"`
+	TotalPage  int64 `json:"total_page"`
+	CurPage    int64 `json:"cur_page"`
+	PageSize   int64 `json:"page_size"`
+}
+
+// Set 设置Response的Code和Message值
+func (resp *Response) Set(Code int, Message string) *Response {
+	return &Response{
+		Code:    Code,
+		Message: Message,
+	}
+}
+
+// Set 设置整个Response
+func (resp *Response) SetWithData(Code int, Message string, Data interface{}) *Response {
+	return &Response{
+		Code:    Code,
+		Message: Message,
+		Data:    Data,
+	}
+}
+
+// SetStatus 设置默认StatusResponse(内部自定义) 默认msg, 可以带data, data只使用一个参数
+func (resp *Response) SetStatus(sr *basic.StatusResponse, data ...interface{}) *Response {
+	newResp := &Response{
+		Code: sr.Code,
+	}
+	if len(data) == 1 {
+		newResp.Data = data[0]
+	}
+	return newResp
+}
+
+// SetStatusWithMessage 设置默认StatusResponse(内部自定义) 非默认msg, 可以带data, data只使用一个参数
+func (resp *Response) SetStatusWithMessage(sr *basic.StatusResponse, msg string, data ...interface{}) *Response {
+	newResp := &Response{
+		Code:    sr.Code,
+		Message: msg,
+	}
+	if len(data) == 1 {
+		newResp.Data = data[0]
+	}
+	return newResp
+}

server/inventory/inventory.go

@@ -0,0 +1,49 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+
+	"fusenapi/server/inventory/internal/config"
+	"fusenapi/server/inventory/internal/handler"
+	"fusenapi/server/inventory/internal/svc"
+
+	"github.com/zeromicro/go-zero/core/conf"
+	"github.com/zeromicro/go-zero/rest"
+)
+
+var configFile = flag.String("f", "etc/inventory.yaml", "the config file")
+
+func main() {
+	flag.Parse()
+
+	var c config.Config
+	conf.MustLoad(*configFile, &c)
+
+	server := rest.MustNewServer(c.RestConf)
+	defer server.Stop()
+
+	ctx := svc.NewServiceContext(c)
+	handler.RegisterHandlers(server, ctx)
+
+	fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port)
+	server.Start()
+}
+
+// var testConfigFile = flag.String("f", "../etc/inventory.yaml", "the config file")
+// var cnf config.Config
+
+// func GetTestServer() *rest.Server {
+// 	flag.Parse()
+
+// 	conf.MustLoad(*testConfigFile, &cnf)
+
+// 	server := rest.MustNewServer(cnf.RestConf)
+// 	defer server.Stop()
+
+// 	ctx := svc.NewServiceContext(cnf)
+// 	handler.RegisterHandlers(server, ctx)
+
+// 	fmt.Printf("Starting server at %s:%d...\n", cnf.Host, cnf.Port)
+// 	return server
+// }