2023-08-14 09:56:06 +00:00
|
|
|
|
package repositories
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"context"
|
|
|
|
|
"encoding/json"
|
|
|
|
|
"errors"
|
|
|
|
|
"fusenapi/constants"
|
|
|
|
|
"fusenapi/model/gmodel"
|
|
|
|
|
"fusenapi/utils/curl"
|
|
|
|
|
"fusenapi/utils/file"
|
|
|
|
|
"fusenapi/utils/hash"
|
2023-09-21 03:35:00 +00:00
|
|
|
|
"strings"
|
2023-08-24 07:35:31 +00:00
|
|
|
|
"time"
|
2023-08-14 09:56:06 +00:00
|
|
|
|
|
|
|
|
|
"github.com/aws/aws-sdk-go/aws/session"
|
2023-08-24 07:35:31 +00:00
|
|
|
|
"github.com/zeromicro/go-zero/core/logc"
|
2023-08-14 09:56:06 +00:00
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
|
|
|
"gorm.io/gorm"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func NewImageHandle(gormDB *gorm.DB, bLMServiceUrl *string, awsSession *session.Session) ImageHandle {
|
|
|
|
|
return &defaultImageHandle{
|
|
|
|
|
MysqlConn: gormDB,
|
|
|
|
|
BLMServiceUrl: bLMServiceUrl,
|
|
|
|
|
AwsSession: awsSession,
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
type (
|
|
|
|
|
defaultImageHandle struct {
|
|
|
|
|
MysqlConn *gorm.DB
|
|
|
|
|
BLMServiceUrl *string
|
|
|
|
|
AwsSession *session.Session
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ImageHandle = interface {
|
2023-08-29 10:00:37 +00:00
|
|
|
|
// 获取logo最新信息
|
|
|
|
|
LogoInfo(ctx context.Context, in *LogoInfoReq) (*LogoInfoRes, error)
|
|
|
|
|
|
2023-08-23 03:09:14 +00:00
|
|
|
|
// logo信息
|
|
|
|
|
LogoInfoSet(ctx context.Context, in *LogoInfoSetReq) (*LogoInfoSetRes, error)
|
2023-08-14 09:56:06 +00:00
|
|
|
|
|
|
|
|
|
// logo合图
|
|
|
|
|
LogoCombine(ctx context.Context, in *LogoCombineReq) (*LogoCombineRes, error)
|
2023-08-18 08:47:22 +00:00
|
|
|
|
|
|
|
|
|
// logo裁剪
|
|
|
|
|
LogoStandard(ctx context.Context, in *LogoStandardReq) (*LogoStandardRes, error)
|
2023-08-14 09:56:06 +00:00
|
|
|
|
}
|
|
|
|
|
)
|
|
|
|
|
|
2023-08-29 10:00:37 +00:00
|
|
|
|
/* 获取logo最新信息 */
|
|
|
|
|
type (
|
|
|
|
|
LogoInfoReq struct {
|
|
|
|
|
UserId int64 `json:"user_id"`
|
|
|
|
|
GuestId int64 `json:"guest_id"`
|
|
|
|
|
}
|
|
|
|
|
LogoInfoRes struct {
|
2023-09-18 09:53:05 +00:00
|
|
|
|
Metadata *string `json:"metadata"`
|
|
|
|
|
LogoUrl *string `json:"logo_url"`
|
|
|
|
|
UserInfoMetadata *string `json:"user_info_metadata"`
|
2023-08-29 10:00:37 +00:00
|
|
|
|
}
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func (l *defaultImageHandle) LogoInfo(ctx context.Context, in *LogoInfoReq) (*LogoInfoRes, error) {
|
|
|
|
|
var metadata *string
|
|
|
|
|
var logoUrl *string
|
2023-09-18 09:53:05 +00:00
|
|
|
|
var userInfoMetadata *string
|
2023-09-07 08:25:29 +00:00
|
|
|
|
// 更新用户信息
|
|
|
|
|
var module = "profile"
|
|
|
|
|
userInfoGorm := l.MysqlConn.Where("module = ?", module)
|
|
|
|
|
userInfo := gmodel.FsUserInfo{}
|
|
|
|
|
if in.UserId > 0 {
|
|
|
|
|
userInfoGorm.Where("user_id = ?", in.UserId)
|
|
|
|
|
userInfo.UserId = &in.UserId
|
|
|
|
|
} else {
|
|
|
|
|
userInfoGorm.Where("guest_id = ?", in.GuestId)
|
|
|
|
|
userInfo.GuestId = &in.GuestId
|
|
|
|
|
}
|
|
|
|
|
resFirst := userInfoGorm.First(&userInfo)
|
|
|
|
|
err := resFirst.Error
|
|
|
|
|
if err != nil {
|
|
|
|
|
if err != gorm.ErrRecordNotFound {
|
|
|
|
|
logc.Errorf(ctx, "FsUserInfo First err:%+v", err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-09-18 09:53:05 +00:00
|
|
|
|
if userInfo.Id != 0 {
|
|
|
|
|
tmp := string(*userInfo.Metadata)
|
|
|
|
|
userInfoMetadata = &tmp
|
|
|
|
|
}
|
2023-09-07 08:25:29 +00:00
|
|
|
|
var userMaterialInfo gmodel.FsUserMaterial
|
2023-08-29 10:00:37 +00:00
|
|
|
|
userMaterialModel := gmodel.NewFsUserMaterialModel(l.MysqlConn)
|
2023-09-19 04:04:51 +00:00
|
|
|
|
var metadataUserInfo struct {
|
|
|
|
|
LogoSelected struct {
|
|
|
|
|
TemplateTagSelected *struct {
|
|
|
|
|
Color [][]string `json:"color"`
|
|
|
|
|
TemplateTag string `json:"template_tag"`
|
|
|
|
|
SelectedIndex int64 `json:"selected_index"`
|
|
|
|
|
} `json:"template_tag_selected"`
|
|
|
|
|
LogoSelectedId int64 `json:"logo_selected_id"`
|
|
|
|
|
} `json:"logo_selected"`
|
|
|
|
|
}
|
2023-09-19 06:50:54 +00:00
|
|
|
|
userInfo.Id = 0
|
2023-09-07 08:25:29 +00:00
|
|
|
|
if userInfo.Id == 0 {
|
|
|
|
|
userMaterialInfo, err = userMaterialModel.FindLatestOne(ctx, in.UserId, in.GuestId)
|
2023-09-19 06:50:54 +00:00
|
|
|
|
if err != nil && err != gorm.ErrRecordNotFound {
|
2023-09-07 08:25:29 +00:00
|
|
|
|
logc.Errorf(ctx, "FsUserMaterial FindLatestOne err:%+v", err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
|
if userInfo.Metadata != nil {
|
2023-09-19 04:04:51 +00:00
|
|
|
|
err = json.Unmarshal([]byte(*userInfo.Metadata), &metadataUserInfo)
|
2023-08-29 10:00:37 +00:00
|
|
|
|
if err != nil {
|
2023-09-07 08:25:29 +00:00
|
|
|
|
logc.Errorf(ctx, "userInfo.Metadata Unmarshal err:%+v", err)
|
2023-08-29 10:00:37 +00:00
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-09-19 04:04:51 +00:00
|
|
|
|
|
|
|
|
|
if metadataUserInfo.LogoSelected.LogoSelectedId != 0 {
|
|
|
|
|
userMaterialInfos, err := userMaterialModel.FindOneById(ctx, metadataUserInfo.LogoSelected.LogoSelectedId)
|
2023-09-07 08:25:29 +00:00
|
|
|
|
if err != nil {
|
|
|
|
|
if err != gorm.ErrRecordNotFound {
|
2023-09-19 04:04:51 +00:00
|
|
|
|
logc.Errorf(ctx, "userMaterial findOneById err:%+v", err)
|
2023-09-07 08:25:29 +00:00
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if userMaterialInfos.Id != 0 {
|
|
|
|
|
userMaterialInfo = *userMaterialInfos
|
|
|
|
|
}
|
|
|
|
|
}
|
2023-08-29 10:00:37 +00:00
|
|
|
|
}
|
2023-09-07 08:25:29 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if userMaterialInfo.Id == 0 {
|
|
|
|
|
userMaterialInfoDefault, err := userMaterialModel.FindOneById(ctx, 0)
|
|
|
|
|
if err != nil {
|
|
|
|
|
if err != gorm.ErrRecordNotFound {
|
|
|
|
|
logc.Errorf(ctx, "userMaterialModel FindOneById defaul err:%+v", err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
metadataB := string(*userMaterialInfoDefault.Metadata)
|
|
|
|
|
metadata = &metadataB
|
|
|
|
|
logoUrl = userMaterialInfoDefault.ResourceUrl
|
|
|
|
|
|
2023-08-29 10:00:37 +00:00
|
|
|
|
} else {
|
2023-09-01 02:27:28 +00:00
|
|
|
|
metadataB := string(*userMaterialInfo.Metadata)
|
|
|
|
|
metadata = &metadataB
|
2023-08-29 10:00:37 +00:00
|
|
|
|
logoUrl = userMaterialInfo.ResourceUrl
|
|
|
|
|
}
|
|
|
|
|
return &LogoInfoRes{
|
2023-09-18 09:53:05 +00:00
|
|
|
|
Metadata: metadata,
|
|
|
|
|
LogoUrl: logoUrl,
|
|
|
|
|
UserInfoMetadata: userInfoMetadata,
|
2023-08-29 10:00:37 +00:00
|
|
|
|
}, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* 获取logo最新信息 */
|
|
|
|
|
|
2023-08-23 03:09:14 +00:00
|
|
|
|
/* logo信息 */
|
|
|
|
|
type (
|
|
|
|
|
LogoInfoSetReq struct {
|
|
|
|
|
LogoUrl string `json:"logo_url"`
|
2023-10-07 03:32:11 +00:00
|
|
|
|
Version string `json:"version"`
|
2023-08-23 03:09:14 +00:00
|
|
|
|
}
|
2023-08-23 06:22:36 +00:00
|
|
|
|
LogoInfoSetRes struct {
|
|
|
|
|
Res string `json:"res"`
|
|
|
|
|
}
|
2023-08-23 03:09:14 +00:00
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func (l *defaultImageHandle) LogoInfoSet(ctx context.Context, in *LogoInfoSetReq) (*LogoInfoSetRes, error) {
|
|
|
|
|
var resultBLM constants.BLMServiceUrlResult
|
2023-10-07 03:32:11 +00:00
|
|
|
|
postMap := make(map[string]string, 2)
|
2023-08-23 03:09:14 +00:00
|
|
|
|
postMap["logo_url"] = in.LogoUrl
|
2023-10-07 03:32:11 +00:00
|
|
|
|
postMap["version"] = in.Version
|
2023-08-23 03:09:14 +00:00
|
|
|
|
|
2023-08-24 07:35:31 +00:00
|
|
|
|
logc.Infof(ctx, "算法请求--LOGO基础信息--开始时间:%v", time.Now().UTC())
|
2023-08-23 06:22:36 +00:00
|
|
|
|
err := curl.NewClient(ctx, &curl.Config{
|
2023-08-23 03:09:14 +00:00
|
|
|
|
BaseUrl: *l.BLMServiceUrl,
|
2023-08-23 07:10:22 +00:00
|
|
|
|
Url: constants.BLMServiceUrlLogoFeatureExtraction,
|
2023-08-23 06:22:36 +00:00
|
|
|
|
}).PostJson(postMap, &resultBLM)
|
2023-08-24 07:35:31 +00:00
|
|
|
|
logc.Infof(ctx, "算法请求--LOGO基础信息--结束时间:%v", time.Now().UTC())
|
2023-08-23 03:09:14 +00:00
|
|
|
|
if err != nil {
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-08-23 06:22:36 +00:00
|
|
|
|
if resultBLM.Code != "200" {
|
|
|
|
|
err = errors.New(resultBLM.Msg)
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-08-23 07:10:22 +00:00
|
|
|
|
return &LogoInfoSetRes{
|
|
|
|
|
Res: resultBLM.Data.(string),
|
|
|
|
|
}, nil
|
2023-08-23 03:09:14 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* logo信息 */
|
|
|
|
|
|
2023-08-14 09:56:06 +00:00
|
|
|
|
/* logo合图 */
|
|
|
|
|
type (
|
|
|
|
|
LogoCombineReq struct {
|
2023-09-19 07:59:44 +00:00
|
|
|
|
UserId int64 `json:"user_id"`
|
|
|
|
|
GuestId int64 `json:"guest_id"`
|
|
|
|
|
ProductTemplateV2Info *gmodel.FsProductTemplateV2 `json:"product_template_v2_info"`
|
|
|
|
|
ProductTemplateTagGroups interface{} `json:"product_template_tag_groups"`
|
|
|
|
|
TemplateTag string `json:"template_tag"`
|
|
|
|
|
Website string `json:"website"` // 合图参数
|
|
|
|
|
Slogan string `json:"slogan"` // 合图参数
|
|
|
|
|
Address string `json:"address"` // 合图参数
|
|
|
|
|
Phone string `json:"phone"` // 合图参数
|
|
|
|
|
Qrcode string `json:"qrcode"` // 合图参数
|
|
|
|
|
LogoUrl string `json:"logo_url"` // 合图参数
|
|
|
|
|
TemplateTagColor TemplateTagColor `json:"template_tag_color"` //合图颜色
|
2023-08-14 09:56:06 +00:00
|
|
|
|
}
|
|
|
|
|
LogoCombineRes struct {
|
2023-08-25 02:37:07 +00:00
|
|
|
|
ResourceId string
|
|
|
|
|
ResourceUrl *string
|
|
|
|
|
Metadata *string
|
2023-08-25 02:48:54 +00:00
|
|
|
|
DiffTimeLogoCombine int64
|
|
|
|
|
DiffTimeUploadFile int64
|
2023-08-14 09:56:06 +00:00
|
|
|
|
}
|
|
|
|
|
)
|
2023-09-19 06:39:33 +00:00
|
|
|
|
type TemplateTagColor struct {
|
|
|
|
|
Color [][]string `json:"color"`
|
|
|
|
|
Index int `json:"index"`
|
|
|
|
|
}
|
2023-08-14 09:56:06 +00:00
|
|
|
|
|
|
|
|
|
func (l *defaultImageHandle) LogoCombine(ctx context.Context, in *LogoCombineReq) (*LogoCombineRes, error) {
|
2023-09-21 03:35:00 +00:00
|
|
|
|
s := strings.Split(in.LogoUrl, "/")
|
|
|
|
|
if len(s) <= 1 {
|
|
|
|
|
return nil, errors.New("无效的logo")
|
|
|
|
|
}
|
|
|
|
|
logoResourceId := s[len(s)-1]
|
|
|
|
|
userMaterialModel := gmodel.NewFsUserMaterialModel(l.MysqlConn)
|
|
|
|
|
resLogoInfo, err := userMaterialModel.FindOneByLogoResourceId(ctx, logoResourceId)
|
2023-08-18 09:16:49 +00:00
|
|
|
|
if err != nil {
|
2023-08-29 10:00:37 +00:00
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
2023-08-18 09:16:49 +00:00
|
|
|
|
}
|
2023-08-14 09:56:06 +00:00
|
|
|
|
// 根据hash 查询数据资源
|
2023-08-16 08:33:25 +00:00
|
|
|
|
var hashKeyData = *in
|
|
|
|
|
hashKeyData.GuestId = 0
|
|
|
|
|
hashKeyData.UserId = 0
|
2023-09-21 03:35:00 +00:00
|
|
|
|
hashKeyData.LogoUrl = in.LogoUrl
|
2023-08-18 11:37:11 +00:00
|
|
|
|
var hashKeyDataMap map[string]interface{}
|
|
|
|
|
hashKeyDataB, _ := json.Marshal(hashKeyData)
|
|
|
|
|
json.Unmarshal(hashKeyDataB, &hashKeyDataMap)
|
|
|
|
|
var resourceId string = hash.JsonHashKey(hashKeyDataMap)
|
2023-08-15 02:58:24 +00:00
|
|
|
|
|
2023-08-14 09:56:06 +00:00
|
|
|
|
resourceModel := gmodel.NewFsResourceModel(l.MysqlConn)
|
|
|
|
|
resourceInfo, err := resourceModel.FindOneById(ctx, resourceId)
|
|
|
|
|
if err == nil && resourceInfo.ResourceId != "" {
|
|
|
|
|
return &LogoCombineRes{
|
|
|
|
|
ResourceId: resourceId,
|
|
|
|
|
ResourceUrl: resourceInfo.ResourceUrl,
|
|
|
|
|
}, nil
|
|
|
|
|
} else {
|
|
|
|
|
if err != nil {
|
|
|
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var groupOptions map[string]interface{}
|
|
|
|
|
var materialList []interface{}
|
2023-09-19 07:09:48 +00:00
|
|
|
|
if in.ProductTemplateV2Info.TemplateInfo != nil {
|
2023-08-14 09:56:06 +00:00
|
|
|
|
var templateInfo map[string]interface{}
|
2023-09-19 07:09:48 +00:00
|
|
|
|
err = json.Unmarshal([]byte(*in.ProductTemplateV2Info.TemplateInfo), &templateInfo)
|
2023-08-14 09:56:06 +00:00
|
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-09-12 09:54:31 +00:00
|
|
|
|
mapMaterialList, existMaterialList := templateInfo["materialList"]
|
|
|
|
|
if !existMaterialList {
|
2023-09-12 09:49:40 +00:00
|
|
|
|
err = errors.New("materialList is null")
|
|
|
|
|
logc.Errorf(ctx, "materialList err:%v", err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-09-12 09:54:31 +00:00
|
|
|
|
materialList = mapMaterialList.([]interface{})
|
|
|
|
|
|
|
|
|
|
mapGroupOptions, existGroupOptions := templateInfo["groupOptions"]
|
|
|
|
|
if !existGroupOptions {
|
2023-09-12 09:49:40 +00:00
|
|
|
|
err = errors.New("groupOptions is null")
|
|
|
|
|
logc.Errorf(ctx, "groupOptions err:%v", err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-09-12 09:54:31 +00:00
|
|
|
|
groupOptions = mapGroupOptions.(map[string]interface{})
|
2023-08-14 09:56:06 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var moduleDataMap = make(map[string]interface{}, 4)
|
2023-09-19 07:09:48 +00:00
|
|
|
|
moduleDataMap["id"] = in.ProductTemplateV2Info.Id
|
|
|
|
|
moduleDataMap["material"] = in.ProductTemplateV2Info.MaterialImg
|
2023-08-14 09:56:06 +00:00
|
|
|
|
moduleDataMap["groupOptions"] = groupOptions
|
|
|
|
|
moduleDataMap["materialList"] = materialList
|
|
|
|
|
|
|
|
|
|
var combineParam map[string]interface{}
|
2023-08-29 10:00:37 +00:00
|
|
|
|
json.Unmarshal([]byte(*resLogoInfo.Metadata), &combineParam)
|
2023-08-15 02:58:24 +00:00
|
|
|
|
combineParam["template_tagid"] = in.TemplateTag
|
|
|
|
|
combineParam["website"] = in.Website
|
|
|
|
|
combineParam["slogan"] = in.Slogan
|
|
|
|
|
combineParam["phone"] = in.Phone
|
|
|
|
|
combineParam["address"] = in.Address
|
2023-08-15 03:20:09 +00:00
|
|
|
|
combineParam["qrcode"] = in.Qrcode
|
2023-09-19 06:39:33 +00:00
|
|
|
|
combineParam["template_tag_selected"] = map[string]interface{}{
|
|
|
|
|
"template_tag": in.TemplateTag,
|
|
|
|
|
"color": in.TemplateTagColor.Color,
|
|
|
|
|
"index": in.TemplateTagColor.Index,
|
|
|
|
|
}
|
2023-10-07 03:32:11 +00:00
|
|
|
|
var postMap = make(map[string]interface{}, 3)
|
2023-08-14 09:56:06 +00:00
|
|
|
|
postMap["module_data"] = moduleDataMap
|
2023-09-19 07:59:44 +00:00
|
|
|
|
postMap["tag_data"] = in.ProductTemplateTagGroups
|
2023-08-14 09:56:06 +00:00
|
|
|
|
postMap["param_data"] = combineParam
|
|
|
|
|
|
2023-08-25 02:37:07 +00:00
|
|
|
|
logc.Infof(ctx, "合图--算法请求--合图--开始时间:%v", time.Now().UTC())
|
2023-08-25 02:48:54 +00:00
|
|
|
|
var startTimeLogoCombine = time.Now().UnixMilli() //合图--处理--开始时间
|
2023-08-25 02:37:07 +00:00
|
|
|
|
|
2023-08-23 06:22:36 +00:00
|
|
|
|
var resultBLM constants.BLMServiceUrlResult
|
|
|
|
|
err = curl.NewClient(ctx, &curl.Config{
|
2023-09-22 04:05:05 +00:00
|
|
|
|
BaseUrl: *l.BLMServiceUrl,
|
|
|
|
|
Url: constants.BLMServiceUrlLogoCombine,
|
2023-09-22 07:12:44 +00:00
|
|
|
|
RequireTimeout: time.Second * 15,
|
2023-08-23 06:22:36 +00:00
|
|
|
|
}).PostJson(postMap, &resultBLM)
|
2023-08-25 02:37:07 +00:00
|
|
|
|
|
|
|
|
|
logc.Infof(ctx, "合图--算法请求--合图--结束时间:%v", time.Now().UTC())
|
2023-08-25 02:48:54 +00:00
|
|
|
|
endTimeLogoCombine := time.Now().UnixMilli() //合图--处理--开始时间
|
|
|
|
|
diffTimeLogoCombine := endTimeLogoCombine - startTimeLogoCombine //合图--处理--中间差
|
|
|
|
|
logc.Infof(ctx, "合图--算法请求--合图--业务耗时:%d", diffTimeLogoCombine)
|
2023-08-14 09:56:06 +00:00
|
|
|
|
if err != nil {
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-08-22 02:02:50 +00:00
|
|
|
|
|
2023-08-23 06:22:36 +00:00
|
|
|
|
if resultBLM.Code != "200" {
|
|
|
|
|
err = errors.New(resultBLM.Msg)
|
2023-08-15 03:59:35 +00:00
|
|
|
|
logx.Error(err)
|
2023-08-14 09:56:06 +00:00
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-23 06:22:36 +00:00
|
|
|
|
var resultStr string = resultBLM.Data.(string)
|
2023-08-14 09:56:06 +00:00
|
|
|
|
var resultData map[string]interface{}
|
2023-08-15 10:35:16 +00:00
|
|
|
|
err = json.Unmarshal([]byte(resultStr), &resultData)
|
|
|
|
|
if err != nil || resultData == nil {
|
2023-08-14 09:56:06 +00:00
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-08-18 10:41:51 +00:00
|
|
|
|
|
2023-08-17 09:20:43 +00:00
|
|
|
|
var fileBase = resultData["result"].(string)
|
2023-08-14 09:56:06 +00:00
|
|
|
|
|
|
|
|
|
// 上传文件
|
|
|
|
|
var upload = file.Upload{
|
|
|
|
|
Ctx: ctx,
|
|
|
|
|
MysqlConn: l.MysqlConn,
|
|
|
|
|
AwsSession: l.AwsSession,
|
|
|
|
|
}
|
2023-08-25 02:37:07 +00:00
|
|
|
|
|
|
|
|
|
logc.Infof(ctx, "合图--上传文件--开始时间:%v", time.Now().UTC())
|
2023-08-25 02:48:54 +00:00
|
|
|
|
var startTimeUploadFile = time.Now().UnixMilli() //合图--上传--开始时间
|
2023-08-25 02:37:07 +00:00
|
|
|
|
|
2023-08-14 09:56:06 +00:00
|
|
|
|
uploadRes, err := upload.UploadFileByBase64(&file.UploadBaseReq{
|
2023-08-16 09:33:34 +00:00
|
|
|
|
Source: "combine-image",
|
2023-08-14 09:56:06 +00:00
|
|
|
|
FileHash: resourceId,
|
2023-08-17 09:20:43 +00:00
|
|
|
|
FileData: fileBase,
|
2023-08-14 09:56:06 +00:00
|
|
|
|
UploadBucket: 1,
|
|
|
|
|
ApiType: 2,
|
|
|
|
|
UserId: in.UserId,
|
|
|
|
|
GuestId: in.GuestId,
|
|
|
|
|
})
|
2023-08-25 02:49:52 +00:00
|
|
|
|
logc.Infof(ctx, "合图--上传文件--结束时间:%v", time.Now().UTC())
|
2023-08-25 02:48:54 +00:00
|
|
|
|
endTimeUploadFile := time.Now().UnixMilli() //合图--处理--开始时间
|
|
|
|
|
diffTimeUploadFile := endTimeUploadFile - startTimeUploadFile //合图--处理--中间差
|
2023-08-25 02:37:07 +00:00
|
|
|
|
|
2023-08-25 02:48:54 +00:00
|
|
|
|
logc.Infof(ctx, "合图--上传文件--业务耗时:%d", diffTimeUploadFile)
|
2023-08-14 09:56:06 +00:00
|
|
|
|
if err != nil {
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-08-25 02:37:07 +00:00
|
|
|
|
|
2023-08-14 09:56:06 +00:00
|
|
|
|
return &LogoCombineRes{
|
2023-08-25 02:37:07 +00:00
|
|
|
|
ResourceId: uploadRes.ResourceId,
|
|
|
|
|
ResourceUrl: &uploadRes.ResourceUrl,
|
|
|
|
|
DiffTimeLogoCombine: diffTimeLogoCombine,
|
|
|
|
|
DiffTimeUploadFile: diffTimeUploadFile,
|
2023-08-14 09:56:06 +00:00
|
|
|
|
}, nil
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* logo合图 */
|
2023-08-18 08:47:22 +00:00
|
|
|
|
type (
|
|
|
|
|
LogoStandardReq struct {
|
|
|
|
|
IsRemoveBg string `json:"is_remove_bg"`
|
|
|
|
|
LogoFile string `json:"logo_file"`
|
|
|
|
|
Width string `json:"width"`
|
|
|
|
|
Height string `json:"height"`
|
2023-08-23 03:09:14 +00:00
|
|
|
|
Proportion string `json:"proportion"`
|
2023-08-18 08:47:22 +00:00
|
|
|
|
}
|
2023-08-18 09:16:49 +00:00
|
|
|
|
LogoStandardRes struct {
|
|
|
|
|
ResourceId string
|
|
|
|
|
ResourceUrl string
|
|
|
|
|
IsmaxProportion bool
|
|
|
|
|
ImgColor []string
|
|
|
|
|
}
|
2023-08-21 06:39:03 +00:00
|
|
|
|
LogoStandardMetaData struct {
|
|
|
|
|
Param LogoStandardReq `json:"param"`
|
|
|
|
|
Result LogoStandardRes `json:"result"`
|
|
|
|
|
}
|
2023-08-18 08:47:22 +00:00
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
/* 图片裁剪 */
|
|
|
|
|
func (l *defaultImageHandle) LogoStandard(ctx context.Context, in *LogoStandardReq) (*LogoStandardRes, error) {
|
2023-08-21 06:39:03 +00:00
|
|
|
|
var ismaxProportion bool
|
|
|
|
|
var imgColor []string
|
|
|
|
|
var logoStandardMetaData LogoStandardMetaData
|
|
|
|
|
|
2023-08-21 02:09:26 +00:00
|
|
|
|
var hashKeyDataMap map[string]interface{}
|
|
|
|
|
hashKeyDataB, _ := json.Marshal(in)
|
|
|
|
|
json.Unmarshal(hashKeyDataB, &hashKeyDataMap)
|
|
|
|
|
var resourceId string = hash.JsonHashKey(hashKeyDataMap)
|
|
|
|
|
|
2023-08-21 06:39:03 +00:00
|
|
|
|
resourceModel := gmodel.NewFsResourceModel(l.MysqlConn)
|
|
|
|
|
resourceInfo, err := resourceModel.FindOneById(ctx, resourceId)
|
|
|
|
|
if err == nil && resourceInfo.ResourceId != "" {
|
|
|
|
|
if resourceInfo.Metadata != nil {
|
|
|
|
|
json.Unmarshal([]byte(*resourceInfo.Metadata), &logoStandardMetaData)
|
|
|
|
|
}
|
|
|
|
|
// 取出参数
|
2023-08-21 06:59:21 +00:00
|
|
|
|
ismaxProportion = logoStandardMetaData.Result.IsmaxProportion
|
|
|
|
|
imgColor = logoStandardMetaData.Result.ImgColor
|
2023-08-21 06:39:03 +00:00
|
|
|
|
|
|
|
|
|
return &LogoStandardRes{
|
|
|
|
|
ResourceId: resourceInfo.ResourceId,
|
|
|
|
|
ResourceUrl: *resourceInfo.ResourceUrl,
|
|
|
|
|
IsmaxProportion: ismaxProportion,
|
|
|
|
|
ImgColor: imgColor,
|
|
|
|
|
}, nil
|
|
|
|
|
} else {
|
|
|
|
|
if err != nil {
|
|
|
|
|
if !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-23 03:09:14 +00:00
|
|
|
|
var resultBLM constants.BLMServiceUrlResult
|
2023-08-18 09:51:51 +00:00
|
|
|
|
var postMap = make(map[string]interface{}, 5)
|
2023-08-18 09:16:49 +00:00
|
|
|
|
postMap["is_remove_bg"] = in.IsRemoveBg
|
|
|
|
|
postMap["logo_file"] = in.LogoFile
|
|
|
|
|
postMap["width"] = in.Width
|
|
|
|
|
postMap["height"] = in.Height
|
|
|
|
|
postMap["proportion"] = in.Proportion
|
|
|
|
|
|
2023-08-24 07:35:31 +00:00
|
|
|
|
logc.Infof(ctx, "算法请求--去背景--开始时间:%v", time.Now().UTC())
|
2023-08-23 03:09:14 +00:00
|
|
|
|
err = curl.NewClient(ctx, &curl.Config{
|
|
|
|
|
BaseUrl: *l.BLMServiceUrl,
|
|
|
|
|
Url: constants.BLMServiceUrlLogoRemovebg,
|
|
|
|
|
}).PostJson(postMap, &resultBLM)
|
2023-08-24 07:35:31 +00:00
|
|
|
|
logc.Infof(ctx, "算法请求--去背景--结束时间:%v", time.Now().UTC())
|
2023-08-18 09:16:49 +00:00
|
|
|
|
if err != nil {
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-08-23 06:22:36 +00:00
|
|
|
|
|
|
|
|
|
if resultBLM.Code != "200" {
|
|
|
|
|
err = errors.New(resultBLM.Msg)
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-23 03:09:14 +00:00
|
|
|
|
var resultStr string = resultBLM.Data.(string)
|
2023-08-18 09:16:49 +00:00
|
|
|
|
var resultData map[string]interface{}
|
|
|
|
|
err = json.Unmarshal([]byte(resultStr), &resultData)
|
|
|
|
|
if err != nil || resultData == nil {
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
2023-08-21 06:39:03 +00:00
|
|
|
|
|
2023-08-18 09:38:12 +00:00
|
|
|
|
var fileBase = resultData["nobg_url"].(string)
|
2023-08-21 06:39:03 +00:00
|
|
|
|
ismaxProportion = resultData["ismax_proportion"].(bool)
|
2023-08-18 09:38:12 +00:00
|
|
|
|
|
|
|
|
|
for _, v := range resultData["img_color"].([]interface{}) {
|
|
|
|
|
imgColor = append(imgColor, v.(string))
|
|
|
|
|
}
|
|
|
|
|
|
2023-08-21 06:59:21 +00:00
|
|
|
|
var logoStandardRes LogoStandardRes
|
|
|
|
|
logoStandardRes.IsmaxProportion = ismaxProportion
|
|
|
|
|
logoStandardRes.ImgColor = imgColor
|
|
|
|
|
|
|
|
|
|
logoStandardMetaData.Param = *in
|
|
|
|
|
logoStandardMetaData.Result = logoStandardRes
|
|
|
|
|
metadataB, err := json.Marshal(logoStandardMetaData)
|
|
|
|
|
if err != nil {
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
var metadata = string(metadataB)
|
|
|
|
|
|
2023-08-18 09:38:12 +00:00
|
|
|
|
// 上传文件
|
|
|
|
|
var upload = file.Upload{
|
|
|
|
|
Ctx: ctx,
|
|
|
|
|
MysqlConn: l.MysqlConn,
|
|
|
|
|
AwsSession: l.AwsSession,
|
|
|
|
|
}
|
|
|
|
|
uploadRes, err := upload.UploadFileByBase64(&file.UploadBaseReq{
|
|
|
|
|
Source: "combine-removebg",
|
|
|
|
|
FileHash: resourceId,
|
|
|
|
|
FileData: fileBase,
|
|
|
|
|
UploadBucket: 1,
|
|
|
|
|
ApiType: 2,
|
2023-08-21 06:59:21 +00:00
|
|
|
|
Metadata: metadata,
|
2023-08-18 09:38:12 +00:00
|
|
|
|
})
|
|
|
|
|
if err != nil {
|
|
|
|
|
logx.Error(err)
|
|
|
|
|
return nil, err
|
|
|
|
|
}
|
|
|
|
|
return &LogoStandardRes{
|
|
|
|
|
ResourceId: uploadRes.ResourceId,
|
|
|
|
|
ResourceUrl: uploadRes.ResourceUrl,
|
|
|
|
|
IsmaxProportion: ismaxProportion,
|
|
|
|
|
ImgColor: imgColor,
|
|
|
|
|
}, nil
|
2023-08-18 08:47:22 +00:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/* 图片裁剪 */
|