170 lines
5.8 KiB
Go
170 lines
5.8 KiB
Go
package logic
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"fusenapi/constants"
|
|
"fusenapi/utils/auth"
|
|
"fusenapi/utils/basic"
|
|
"fusenapi/utils/format"
|
|
"strings"
|
|
|
|
"context"
|
|
|
|
"fusenapi/server/product/internal/svc"
|
|
"fusenapi/server/product/internal/types"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type GetProductModelsLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewGetProductModelsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetProductModelsLogic {
|
|
return &GetProductModelsLogic{
|
|
Logger: logx.WithContext(ctx),
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
}
|
|
}
|
|
|
|
// 处理进入前逻辑w,r
|
|
// func (l *GetProductModelsLogic) BeforeLogic(w http.ResponseWriter, r *http.Request) {
|
|
// }
|
|
|
|
func (l *GetProductModelsLogic) GetProductModels(req *types.GetProductModelsReq, userinfo *auth.UserInfo) (resp *basic.Response) {
|
|
req.ProductIds = strings.Trim(req.ProductIds, " ")
|
|
productIds, err := format.StrSlicToInt64Slice(strings.Split(req.ProductIds, ","))
|
|
if err != nil {
|
|
logx.Error(err)
|
|
return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, "err param(format err)")
|
|
}
|
|
if len(productIds) > 10 {
|
|
return resp.SetStatusWithMessage(basic.CodeRequestParamsErr, "the count of product can`t greater than 10")
|
|
}
|
|
if len(productIds) == 0 {
|
|
return resp.SetStatusWithMessage(basic.CodeOK, "success", nil)
|
|
}
|
|
//获取产品信息
|
|
productList, err := l.svcCtx.AllModels.FsProduct.GetProductListByIds(l.ctx, productIds, "")
|
|
if err != nil {
|
|
logx.Error(err)
|
|
return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get product list")
|
|
}
|
|
if len(productList) != len(productIds) {
|
|
return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "some one product is invalid")
|
|
}
|
|
//获取产品所有的模型以及配件
|
|
modelList, err := l.svcCtx.AllModels.FsProductModel3d.GetAllByProductIdsTags(l.ctx, productIds, []int{constants.TAG_MODEL, constants.TAG_PARTS})
|
|
if err != nil {
|
|
logx.Error(err)
|
|
return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get model list")
|
|
}
|
|
mapProductModels := make(map[int64][]types.ModelItem)
|
|
lightIds := make([]int64, 0, len(modelList))
|
|
for _, v := range modelList {
|
|
lightIds = append(lightIds, *v.Light)
|
|
}
|
|
//获取光源列表
|
|
lightList, err := l.svcCtx.AllModels.FsProductModel3dLight.GetAllByIds(l.ctx, lightIds)
|
|
if err != nil {
|
|
logx.Error(err)
|
|
return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get light list")
|
|
}
|
|
mapLight := make(map[int64]int)
|
|
for k, v := range lightList {
|
|
mapLight[v.Id] = k
|
|
}
|
|
//获取配件默认贴图
|
|
fittingPublicTemplateIds := make([]int64, 0, len(modelList)) //绑定了公共模板
|
|
fittingNotPublicIds := make([]int64, 0, len(modelList)) //没绑定公共模板
|
|
for _, v := range modelList {
|
|
if *v.Tag != constants.TAG_PARTS {
|
|
continue
|
|
}
|
|
//配件绑定公共模板
|
|
if *v.OptionTemplate > 0 {
|
|
fittingPublicTemplateIds = append(fittingPublicTemplateIds, *v.OptionTemplate)
|
|
} else {
|
|
fittingNotPublicIds = append(fittingNotPublicIds, v.Id)
|
|
}
|
|
}
|
|
//公共模板配件的模板列表
|
|
fittingPublicTemplateList, err := l.svcCtx.AllModels.FsProductTemplateV2.FindAllByIds(l.ctx, fittingPublicTemplateIds, "id,model_id,material_img")
|
|
if err != nil {
|
|
logx.Error(err)
|
|
return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get fitting templates!")
|
|
}
|
|
mapFittingPublicTemplate := make(map[int64]int)
|
|
for k, v := range fittingPublicTemplateList {
|
|
mapFittingPublicTemplate[v.Id] = k
|
|
}
|
|
//非公共模板配件的模板列表
|
|
fittingNotPublicTemplateList, err := l.svcCtx.AllModels.FsProductTemplateV2.FindAllByFittingIds(l.ctx, fittingNotPublicIds)
|
|
if err != nil {
|
|
logx.Error(err)
|
|
return resp.SetStatusWithMessage(basic.CodeDbSqlErr, "failed to get fitting templates!!")
|
|
}
|
|
mapFittingNotPublicTemplate := make(map[int64]int)
|
|
for k, v := range fittingNotPublicTemplateList {
|
|
mapFittingNotPublicTemplate[*v.ModelId] = k
|
|
}
|
|
for _, v := range modelList {
|
|
//模型设计信息
|
|
var modelDesignInfo interface{}
|
|
if v.ModelInfo != nil && *v.ModelInfo != "" {
|
|
if err = json.Unmarshal([]byte(*v.ModelInfo), &modelDesignInfo); err != nil {
|
|
logx.Error(err)
|
|
return resp.SetStatusWithMessage(basic.CodeJsonErr, fmt.Sprintf("failed to parse model design info[:%d]", v.Id))
|
|
}
|
|
}
|
|
//光源信息
|
|
var lightDesignInfo interface{}
|
|
if lightIndex, ok := mapLight[*v.Light]; ok {
|
|
lightInfo := lightList[lightIndex]
|
|
if lightInfo.Info != nil && *lightInfo.Info != "" {
|
|
if err = json.Unmarshal([]byte(*lightInfo.Info), &lightDesignInfo); err != nil {
|
|
logx.Error(err)
|
|
return resp.SetStatusWithMessage(basic.CodeJsonErr, fmt.Sprintf("failed to parse light design info[:%d]", lightInfo.Id))
|
|
}
|
|
}
|
|
}
|
|
fittingMaterialImage := ""
|
|
//配件有默认贴图
|
|
if *v.Tag == constants.TAG_PARTS {
|
|
switch *v.OptionTemplate {
|
|
case 0: //没有绑定公共模板的配件
|
|
if index, ok := mapFittingNotPublicTemplate[v.Id]; ok {
|
|
fittingMaterialImage = *fittingNotPublicTemplateList[index].MaterialImg
|
|
}
|
|
default:
|
|
if index, ok := mapFittingPublicTemplate[*v.OptionTemplate]; ok {
|
|
fittingMaterialImage = *fittingPublicTemplateList[index].MaterialImg
|
|
}
|
|
}
|
|
}
|
|
mapProductModels[*v.ProductId] = append(mapProductModels[*v.ProductId], types.ModelItem{
|
|
Id: v.Id,
|
|
Type: *v.Tag,
|
|
Title: *v.Title,
|
|
Name: *v.Name,
|
|
MaterialImage: fittingMaterialImage,
|
|
DesignInfo: modelDesignInfo,
|
|
LightInfo: lightDesignInfo,
|
|
})
|
|
}
|
|
mapProductInfo := make(map[int64][]types.ModelItem)
|
|
for _, v := range productList {
|
|
mapProductInfo[v.Id] = mapProductModels[v.Id]
|
|
}
|
|
return resp.SetStatusWithMessage(basic.CodeOK, "success", mapProductInfo)
|
|
}
|
|
|
|
// 处理逻辑后 w,r 如:重定向, resp 必须重新处理
|
|
// func (l *GetProductModelsLogic) AfterLogic(w http.ResponseWriter, r *http.Request, resp *basic.Response) {
|
|
// // httpx.OkJsonCtx(r.Context(), w, resp)
|
|
// }
|