package logic import ( "errors" "fusenapi/model/gmodel" "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 module = "logo" 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, Ctime: &nowTime, } 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{}{ "user_material_id": materialInfo.Id, }) } // 处理逻辑后 w,r 如:重定向, resp 必须重新处理 // func (l *UserLogoDataSetLogic) AfterLogic(w http.ResponseWriter, r *http.Request, resp *basic.Response) { // // httpx.OkJsonCtx(r.Context(), w, resp) // }