zchengo
118 lines
2.9 KiB
Go
118 lines
2.9 KiB
Go
package api
|
|
|
|
import (
|
|
"crm/models"
|
|
"crm/response"
|
|
"crm/service"
|
|
"log"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type UserApi struct {
|
|
userService *service.UserService
|
|
}
|
|
|
|
func NewUserApi() *UserApi {
|
|
userApi := UserApi{
|
|
userService: &service.UserService{},
|
|
}
|
|
return &userApi
|
|
}
|
|
|
|
// 用户注册
|
|
func (u *UserApi) Register(context *gin.Context) {
|
|
var param models.UserCreateParam
|
|
if err := context.ShouldBind(¶m); err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
log.Printf("[error]UserApi:Register:%s", err)
|
|
return
|
|
}
|
|
errCode := u.userService.Register(¶m)
|
|
response.Result(errCode, nil, context)
|
|
}
|
|
|
|
// 用户登录
|
|
func (u *UserApi) Login(context *gin.Context) {
|
|
var param models.UserLoginParam
|
|
if err := context.ShouldBind(¶m); err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
userInfo, errCode := u.userService.Login(¶m)
|
|
if userInfo == nil {
|
|
response.Result(errCode, nil, context)
|
|
return
|
|
}
|
|
response.Result(errCode, userInfo, context)
|
|
}
|
|
|
|
// 获取验证码
|
|
func (u *UserApi) GetVerifyCode(context *gin.Context) {
|
|
var param models.UserVerifyCodeParam
|
|
if err := context.ShouldBind(¶m); err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
errCode := u.userService.GetVerifyCode(param.Email)
|
|
response.Result(errCode, nil, context)
|
|
}
|
|
|
|
// 忘记密码
|
|
func (u *UserApi) ForgotPass(context *gin.Context) {
|
|
var param models.UserPassParam
|
|
if err := context.ShouldBind(¶m); err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
errCode := u.userService.ForgotPass(¶m)
|
|
response.Result(errCode, nil, context)
|
|
}
|
|
|
|
// 修改邮箱
|
|
func (u *UserApi) UpdateMail(context *gin.Context) {
|
|
var param models.UserMailParam
|
|
if err := context.ShouldBind(¶m); err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
errCode := u.userService.UpdateMail(¶m)
|
|
response.Result(errCode, nil, context)
|
|
}
|
|
|
|
// 退出登录
|
|
func (u *UserApi) Logout(context *gin.Context) {
|
|
token := context.Request.Header.Get("token")
|
|
if token == "" {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
errCode := u.userService.Logout(token)
|
|
response.Result(errCode, nil, context)
|
|
}
|
|
|
|
// 注销账号
|
|
func (u *UserApi) Delete(context *gin.Context) {
|
|
var param models.UserDeleteParam
|
|
uid, _ := strconv.Atoi(context.Request.Header.Get("uid"))
|
|
err := context.ShouldBind(¶m)
|
|
if uid <= 0 || err != nil {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
param.Id = int64(uid)
|
|
errCode := u.userService.Delete(param)
|
|
response.Result(errCode, nil, context)
|
|
}
|
|
|
|
// 获取用户信息
|
|
func (u *UserApi) GetInfo(context *gin.Context) {
|
|
uid, _ := strconv.Atoi(context.Request.Header.Get("uid"))
|
|
if uid <= 0 {
|
|
response.Result(response.ErrCodeParamInvalid, nil, context)
|
|
return
|
|
}
|
|
userInfo, errCode := u.userService.GetInfo(int64(uid))
|
|
response.Result(errCode, userInfo, context)
|
|
} |