fix

2023-06-25 17:11:42 +08:00
parent 002ec9d292
commit d876884397
14 changed files with 602 additions and 5 deletions
--- a/server/product-model/internal/config/config.go
+++ b/server/product-model/internal/config/config.go
@@ -0,0 +1,12 @@
+package config
+
+import (
+	"fusenapi/server/product-model/internal/types"
+	"github.com/zeromicro/go-zero/rest"
+)
+
+type Config struct {
+	rest.RestConf
+	SourceMysql string
+	Auth        types.Auth
+}
--- a/server/product-model/internal/handler/getmodeldetailhandler.go
+++ b/server/product-model/internal/handler/getmodeldetailhandler.go
@@ -0,0 +1,41 @@
+package handler
+
+import (
+	"errors"
+	"net/http"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"fusenapi/utils/basic"
+
+	"fusenapi/server/product-model/internal/logic"
+	"fusenapi/server/product-model/internal/svc"
+	"fusenapi/server/product-model/internal/types"
+)
+
+func GetModelDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.GetModelDetailReq
+		// 如果端点有请求结构体，则使用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.NewGetModelDetailLogic(r.Context(), svcCtx)
+		resp := l.GetModelDetail(&req, r)
+		// 如果响应不为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)
+		}
+	}
+}
--- a/server/product-model/internal/handler/getmodelotherinfohandler.go
+++ b/server/product-model/internal/handler/getmodelotherinfohandler.go
@@ -0,0 +1,41 @@
+package handler
+
+import (
+	"errors"
+	"net/http"
+
+	"github.com/zeromicro/go-zero/core/logx"
+	"github.com/zeromicro/go-zero/rest/httpx"
+
+	"fusenapi/utils/basic"
+
+	"fusenapi/server/product-model/internal/logic"
+	"fusenapi/server/product-model/internal/svc"
+	"fusenapi/server/product-model/internal/types"
+)
+
+func GetModelOtherInfoHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.GetModelOtherInfoReq
+		// 如果端点有请求结构体，则使用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.NewGetModelOtherInfoLogic(r.Context(), svcCtx)
+		resp := l.GetModelOtherInfo(&req, r)
+		// 如果响应不为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)
+		}
+	}
+}
--- a/server/product-model/internal/handler/routes.go
+++ b/server/product-model/internal/handler/routes.go
@@ -0,0 +1,27 @@
+// Code generated by goctl. DO NOT EDIT.
+package handler
+
+import (
+	"net/http"
+
+	"fusenapi/server/product-model/internal/svc"
+
+	"github.com/zeromicro/go-zero/rest"
+)
+
+func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
+	server.AddRoutes(
+		[]rest.Route{
+			{
+				Method:  http.MethodGet,
+				Path:    "/product-model/detail",
+				Handler: GetModelDetailHandler(serverCtx),
+			},
+			{
+				Method:  http.MethodGet,
+				Path:    "/product-model/other-info",
+				Handler: GetModelOtherInfoHandler(serverCtx),
+			},
+		},
+	)
+}
--- a/server/product-model/internal/logic/getmodeldetaillogic.go
+++ b/server/product-model/internal/logic/getmodeldetaillogic.go
@@ -0,0 +1,68 @@
+package logic
+
+import (
+	"encoding/json"
+	"errors"
+	"fusenapi/model/gmodel"
+	"fusenapi/utils/basic"
+	"gorm.io/gorm"
+	"net/http"
+
+	"context"
+
+	"fusenapi/server/product-model/internal/svc"
+	"fusenapi/server/product-model/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type GetModelDetailLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewGetModelDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetModelDetailLogic {
+	return &GetModelDetailLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *GetModelDetailLogic) GetModelDetail(req *types.GetModelDetailReq, r *http.Request) (resp *basic.Response) {
+	authKey := r.Header.Get("Auth-Key")
+	genentModel := gmodel.NewFsGerentModel(l.svcCtx.MysqlConn)
+	_, err := genentModel.Find(l.ctx, authKey)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return resp.SetStatusWithMessage(basic.CodeUnAuth, "please login first..")
+		}
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeUnAuth, "failed to get user info")
+	}
+	if req.ModelId <= 0 {
+		return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, "param model_id is required")
+	}
+	//获取产品模型信息
+	productModel3dInfo, err := l.svcCtx.AllModels.FsProductModel3d.FindOne(l.ctx, req.ModelId)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return resp.SetStatusWithMessage(basic.CodeDbRecordNotFoundErr, "product model info is not exists")
+		}
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get product model info")
+	}
+	//解析模型json数据
+	var modelInfoRsp interface{}
+	if productModel3dInfo.ModelInfo != nil && *productModel3dInfo.ModelInfo != "" {
+		if err = json.Unmarshal([]byte(*productModel3dInfo.ModelInfo), &modelInfoRsp); err != nil {
+			logx.Error(err)
+			return resp.SetStatusWithMessage(basic.CodeServiceErr, "failed to parse model info")
+		}
+	}
+	return resp.SetStatusWithMessage(basic.CodeOK, "success", types.GetModelDetailRsp{
+		Tag:              *productModel3dInfo.Tag,
+		ProductModelInfo: modelInfoRsp,
+	})
+}
--- a/server/product-model/internal/logic/getmodelotherinfologic.go
+++ b/server/product-model/internal/logic/getmodelotherinfologic.go
@@ -0,0 +1,120 @@
+package logic
+
+import (
+	"encoding/json"
+	"errors"
+	"fusenapi/constants"
+	"fusenapi/model/gmodel"
+	"fusenapi/utils/basic"
+	"gorm.io/gorm"
+	"net/http"
+
+	"context"
+
+	"fusenapi/server/product-model/internal/svc"
+	"fusenapi/server/product-model/internal/types"
+
+	"github.com/zeromicro/go-zero/core/logx"
+)
+
+type GetModelOtherInfoLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewGetModelOtherInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetModelOtherInfoLogic {
+	return &GetModelOtherInfoLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *GetModelOtherInfoLogic) GetModelOtherInfo(req *types.GetModelOtherInfoReq, r *http.Request) (resp *basic.Response) {
+	authKey := r.Header.Get("Auth-Key")
+	genentModel := gmodel.NewFsGerentModel(l.svcCtx.MysqlConn)
+	_, err := genentModel.Find(l.ctx, authKey)
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return resp.SetStatusWithMessage(basic.CodeUnAuth, "please login first..")
+		}
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeUnAuth, "failed to get user info")
+	}
+	//获取所有灯光列表
+	modelLightList, err := l.svcCtx.AllModels.FsProductModel3dLight.GetAll(l.ctx)
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get model light list")
+	}
+	//获取配件列表
+	Get3dModelsByParamReq := gmodel.Get3dModelsByParamReq{
+		Tag: constants.TAG_PARTS,
+	}
+	if req.ProductId > 0 {
+		Get3dModelsByParamReq.ProductId = req.ProductId
+	}
+	model3dList, err := l.svcCtx.AllModels.FsProductModel3d.Get3dModelsByParam(l.ctx, Get3dModelsByParamReq)
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get 3d model list")
+	}
+	//组装灯光列表
+	lightListRsp := make([]types.LightListItem, 0, len(modelLightList))
+	for _, v := range modelLightList {
+		var info interface{}
+		if v.Info != nil && *v.Info != "" {
+			if err = json.Unmarshal([]byte(*v.Info), &info); err != nil {
+				logx.Error(err)
+				return resp.SetStatusWithMessage(basic.CodeServiceErr, "failed to parse light info")
+			}
+		}
+		lightListRsp = append(lightListRsp, types.LightListItem{
+			Id:   v.Id,
+			Name: *v.Name,
+			Info: info,
+		})
+	}
+	modelIds := make([]int64, 0, len(model3dList))
+	map3dModel := make(map[int64]int)
+	for k, v := range model3dList {
+		modelIds = append(modelIds, v.Id)
+		map3dModel[v.Id] = k
+	}
+	//根据模型ids获取产品模板
+	productTemplateV2List, err := l.svcCtx.AllModels.FsProductTemplateV2.FindAllByModelIds(l.ctx, modelIds)
+	if err != nil {
+		logx.Error(err)
+		return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get product v2 template list")
+	}
+	mapProductTemplate := make(map[int64]int)
+	for k, v := range productTemplateV2List {
+		mapProductTemplate[v.Id] = k
+	}
+	//组装配件数据
+	partListRsp := make([]types.PartListItem, 0, len(model3dList))
+	for _, v := range model3dList {
+		materialImg := ""
+		if templateIndex, ok := mapProductTemplate[*v.OptionTemplate]; ok {
+			materialImg = *productTemplateV2List[templateIndex].MaterialImg
+		}
+		var info interface{}
+		if v.ModelInfo != nil && *v.ModelInfo != "" {
+			if err = json.Unmarshal([]byte(*v.ModelInfo), &info); err != nil {
+				logx.Error(err)
+				return resp.SetStatusWithMessage(basic.CodeServiceErr, "failed to parse model info")
+			}
+		}
+		partListRsp = append(partListRsp, types.PartListItem{
+			Id:          v.Id,
+			Name:        *v.Name,
+			MaterialImg: materialImg,
+			ModelInfo:   info,
+		})
+	}
+	return resp.SetStatusWithMessage(basic.CodeOK, "success", types.GetModelOtherInfoRsp{
+		LightList: lightListRsp,
+		PartList:  partListRsp,
+	})
+}
--- a/server/product-model/internal/svc/servicecontext.go
+++ b/server/product-model/internal/svc/servicecontext.go
@@ -0,0 +1,60 @@
+package svc
+
+import (
+	"errors"
+	"fmt"
+	"fusenapi/server/product-model/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))
+}
--- a/server/product-model/internal/types/types.go
+++ b/server/product-model/internal/types/types.go
@@ -0,0 +1,92 @@
+// Code generated by goctl. DO NOT EDIT.
+package types
+
+import (
+	"fusenapi/utils/basic"
+)
+
+type GetModelDetailReq struct {
+	ModelId int64 `form:"model_id"`
+}
+
+type GetModelDetailRsp struct {
+	Tag              int64       `json:"tag"`
+	ProductModelInfo interface{} `json:"product_model_info"`
+}
+
+type GetModelOtherInfoReq struct {
+	ProductId int64 `form:"product_id,optional"`
+}
+
+type GetModelOtherInfoRsp struct {
+	LightList []LightListItem `json:"light_list"`
+	PartList  []PartListItem  `json:"part_list"`
+}
+
+type LightListItem struct {
+	Id   int64       `json:"id"`
+	Name string      `json:"name"`
+	Info interface{} `json:"info"`
+}
+
+type PartListItem struct {
+	Id          int64       `json:"id"`
+	Name        string      `json:"name"`
+	MaterialImg string      `json:"material_img"`
+	ModelInfo   interface{} `json:"model_info"`
+}
+
+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"`
+}
+
+// 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
+}