You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

100 lines
2.4 KiB

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
}