package user import ( "yunyan/comm" "yunyan/lego/sys/mysql" "yunyan/lego/sys/wordfilter" "yunyan/pb" "yunyan/sys/tencentyun/cos" "yunyan/utils" "os" "path/filepath" ) // @Summary 用户设置接口 // @Description 用户设置接口 // @Tags User // @Accept json // @Produce json // @Security BearerAuth // @Param user body pb.UserSettingReq true "发送请求" // @Success 200 {object} comm.HttpResult{data=pb.UserSettingResp} "成功返回" // @Router /api/home/user_setting [post] func (this *apiComp) Setting(session comm.IUserSession, req *pb.UserSettingReq) (resp *pb.UserSettingResp, errdata *pb.ErrorData) { var ( ok bool user *pb.DBUser filename, filePath string publicURL string file *os.File err error ) resp = &pb.UserSettingResp{} if req.Name != "" { if ok, _ = wordfilter.Validate(req.Name); !ok { errdata = &pb.ErrorData{ Code: pb.ErrorCode_NameInscriptionWords, Message: pb.ErrorCode_NameInscriptionWords.String(), } return } } if user, err = this.module.model.findforid(session.GetUserId()); err != nil && err != mysql.ErrNoDocuments { errdata = &pb.ErrorData{ Code: pb.ErrorCode_DBError, Message: err.Error(), } return } if req.Name != "" { user.Name = req.Name } if req.Language != "" { user.Language = req.Language } if req.Gender != "" { user.Gender = int32(utils.StringToInt64(req.Gender)) } filename = session.GetMateToString("file_name") filePath = session.GetMateToString("file_path") if filename != "" && filePath != "" { defer os.Remove(filePath) if file, err = os.Open(filePath); err != nil { errdata = &pb.ErrorData{ Code: pb.ErrorCode_FileNotFound, Message: err.Error(), } return } defer file.Close() // 获取文件扩展名 ext := filepath.Ext(filename) // 上传 - 使用用户ID+扩展名作为文件名避免冲突 publicURL, err = cos.Put("/userAvatar/"+session.GetUserId()+ext, file) if err != nil { errdata = &pb.ErrorData{ Code: pb.ErrorCode_StorageError, Message: err.Error(), } return } user.Avatar = publicURL } if err = this.module.model.save(user); err != nil { errdata = &pb.ErrorData{ Code: pb.ErrorCode_DBError, Message: err.Error(), } return } resp = &pb.UserSettingResp{ Name: req.Name, Language: req.Language, Avatar: user.Avatar, Gender: user.Gender, } return }