package logic import ( "errors" "fmt" "fusenapi/model/gmodel" "fusenapi/service/repositories" "fusenapi/utils/auth" "fusenapi/utils/basic" "fusenapi/utils/s3url_to_s3id" "time" "context" "fusenapi/server/home-user-auth/internal/svc" "fusenapi/server/home-user-auth/internal/types" "github.com/zeromicro/go-zero/core/logc" "github.com/zeromicro/go-zero/core/logx" "gorm.io/gorm" ) type UserLogoDataSetLogic struct { logx.Logger ctx context.Context svcCtx *svc.ServiceContext } func NewUserLogoDataSetLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UserLogoDataSetLogic { return &UserLogoDataSetLogic{ Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx, } } // 处理进入前逻辑w,r // func (l *UserLogoDataSetLogic) BeforeLogic(w http.ResponseWriter, r *http.Request) { // } func (l *UserLogoDataSetLogic) UserLogoDataSet(req *types.UserLogoDataSetReq, userinfo *auth.UserInfo) (resp *basic.Response) { // 返回值必须调用Set重新返回, resp可以空指针调用 resp.SetStatus(basic.CodeOK, data) // userinfo 传入值时, 一定不为null if userinfo.IsOnlooker() { // 如果是,返回未授权的错误码 return resp.SetStatus(basic.CodeUnAuth) } var userId int64 var guestId int64 // 检查用户是否是游客 if userinfo.IsGuest() { // 如果是,使用游客ID和游客键名格式 guestId = userinfo.GuestId } else { // 否则,使用用户ID和用户键名格式 userId = userinfo.UserId } var logoData = gmodel.FsPreprocessLogo{} result := l.svcCtx.MysqlConn.Model(&gmodel.FsPreprocessLogo{}).Where("id = ?", req.LogoDataId).Take(&logoData) err := result.Error if err != nil { if errors.Is(err, gorm.ErrRecordNotFound) { logc.Errorf(l.ctx, "UserLogoDataSet logoData find err:%+v", err) basic.CodeApiErr.Message = "logo data not find" return resp.SetStatus(basic.CodeApiErr) } return resp.SetStatus(basic.CodeApiErr) } var materialInfoOld gmodel.FsUserMaterial result1 := l.svcCtx.MysqlConn.Model(&gmodel.FsUserMaterial{}).Where("logo_id = ?", req.LogoDataId).Take(&materialInfoOld) err = result1.Error if err != nil { if !errors.Is(err, gorm.ErrRecordNotFound) { logc.Errorf(l.ctx, "FsUserMaterial logoData find err:%+v", err) return resp.SetStatus(basic.CodeApiErr) } } else { // 返回成功的响应和上传URL return resp.SetStatus(basic.CodeOK, map[string]interface{}{ "id": materialInfoOld.Id, "module": materialInfoOld.Module, "user_id": materialInfoOld.UserId, "guest_id": materialInfoOld.GuestId, "resource_id": materialInfoOld.ResourceId, "resource_url": materialInfoOld.ResourceUrl, "metadata": materialInfoOld.Metadata, "ctime": materialInfoOld.Ctime, "resource_info": nil, }) } var userMaterialMetadata []byte if logoData.Metadata == nil { var resultStr string resLogoStandard, err := l.svcCtx.Repositories.ImageHandle.LogoInfoSet(l.ctx, &repositories.LogoInfoSetReq{ LogoUrl: *logoData.ResourceUrl, Version: l.svcCtx.Config.BLMService.Version, Debug: userinfo.Debug, }) if err != nil { logx.Error(err) basic.CodeServiceErr.Message = fmt.Sprintf("算法请求--LOGO信息--错误:%+v", err) return resp.SetStatus(basic.CodeServiceErr, fmt.Sprintf("算法请求--LOGO信息--错误:%+v", err)) } resultStr = resLogoStandard.Res userMaterialMetadata = []byte(resultStr) } else { userMaterialMetadata = *logoData.Metadata } // 新增素材--预备logo var module = "logo-prepare" var nowTime = time.Now().UTC() var resourceId = s3url_to_s3id.GetS3ResourceIdFormUrl(*logoData.ResourceUrl) // 新增素材记录 materialInfo := gmodel.FsUserMaterial{ Module: &module, UserId: &userId, GuestId: &guestId, ResourceId: &resourceId, ResourceUrl: logoData.ResourceUrl, Metadata: &userMaterialMetadata, Ctime: &nowTime, LogoId: &req.LogoDataId, } resCreate := l.svcCtx.MysqlConn.Create(&materialInfo) err = resCreate.Error if err != nil { logc.Errorf(l.ctx, "UserLogoDataSet logoData find err:%+v", err) return resp.SetStatus(basic.CodeApiErr) } // 返回成功的响应和上传URL return resp.SetStatus(basic.CodeOK, map[string]interface{}{ "id": materialInfo.Id, "module": materialInfo.Module, "user_id": materialInfo.UserId, "guest_id": materialInfo.GuestId, "resource_id": materialInfo.ResourceId, "resource_url": materialInfo.ResourceUrl, "metadata": materialInfo.Metadata, "ctime": materialInfo.Ctime, "resource_info": nil, }) } // 处理逻辑后 w,r 如:重定向, resp 必须重新处理 // func (l *UserLogoDataSetLogic) AfterLogic(w http.ResponseWriter, r *http.Request, resp *basic.Response) { // // httpx.OkJsonCtx(r.Context(), w, resp) // }