【Go实战 | 电商平台】(4) 用户注册

文章目录

1. 编写路由

2. 注册路由

3. 注册接口

3.1 service层

3.2 api层

3.2 serializer

4. 注册服务

用到的包

github.com/gin-contrib/sessions

github.com/gin-contrib/sessions/cookie

github.com/gin-gonic/gin

新建一个WSGI应用程序实例。

r := gin.Default()      

设置全局变量store

store := cookie.NewStore([]byte("something-very-secret"))      

在路由中使用中间件调用store

r.Use(sessions.Sessions("mysession", store))      

分一个基础路由组

v1 := r.Group("api/v1")      

在这个基础路由编写用户注册路由

v1 := r.Group("api/v1")
{
    //用户操作
    v1.POST("user/register", api.UserRegister)
}      

在service先创建一个user.go

再在service层上编写用户注册服务的结构体

//UserRegisterService 管理用户注册服务
type UserRegisterService struct {
    Nickname  string `form:"nickname" json:"nickname" binding:"required,min=2,max=10"`
    UserName  string `form:"user_name" json:"user_name" binding:"required,min=5,max=15"`
    Password  string `form:"password" json:"password" binding:"required,min=8,max=16"`
}      

在service上编写用户注册register方法

func (service *UserRegisterService) Register() {

}      

在api层创建user.go

申请一个 UserRegisterService用户注册服务对象。

var userRegisterService service.UserRegisterService      

但是这个服务我们还没有写，先把这个接口写完，再把这个服务补上。

上下文绑定数据

c.ShouldBind(&userRegisterService)      

调用这个服务的register方法

res := userRegisterService.Register()      

返回这个服务的处理结果

c.JSON(200, res)      

api 层的用户注册服务全部代码

func UserRegister(c *gin.Context) {
    var userRegisterService service.UserRegisterService 
    //相当于创建了一个UserRegisterService对象
    if err := c.ShouldBind(&userRegisterService); err == nil {
  res := userRegisterService.Register()
  //调用这个对象中的Register方法。
  c.JSON(200, res)
    } else {
  c.JSON(200, ErrorResponse(err))
  logging.Info(err)
    }
}      

创建一个common.go文件， 进行错误的返回

//返回错误信息 ErrorResponse
func ErrorResponse(err error) serializer.Response {
    if ve, ok := err.(validator.ValidationErrors); ok {
  for _, e := range ve {
    field := conf.T(fmt.Sprintf("Field.%s", e.Field))
    tag := conf.T(fmt.Sprintf("Tag.Valid.%s", e.Tag))
    return serializer.Response{
    Status: 40001,
    Msg:    fmt.Sprintf("%s%s", field, tag),
    Error:  fmt.Sprint(err),
    }
  }
    }
    if _, ok := err.(*json.UnmarshalTypeError); ok {
  return serializer.Response{
    Status: 40001,
    Msg:    "JSON类型不匹配",
    Error:  fmt.Sprint(err),
  }
    }
    return serializer.Response{
  Status: 40001,
  Msg:    "参数错误",
  Error:  fmt.Sprint(err),
    }
}      

创建一个基础的序列化返回结构体

// Response 基础序列化器
type Response struct {
    Status int         `json:"status"`
    Data   interface{} `json:"data"`
    Msg    string      `json:"msg"`
    Error  string      `json:"error"`
}      

接下来我们就可以编写register()注册服务了

先对传过来的用户名进行验证，查看是否已经存在了

model.DB.Model(&model.User{}).Where("user_name=?",service.UserName).Count(&count)
    if count == 1 {
  code = e.ErrorExistUser
  return serializer.Response{
    Status: code,
    Msg:    e.GetMsg(code),
  }
    }      

再对密码进行加密

if err := user.SetPassword(service.Password); err != nil {
  logging.Info(err)
  code = e.ErrorFailEncryption
  return serializer.Response{
    Status: code,
    Msg:    e.GetMsg(code),
  }
    }      

注意： 我们可以直接再model/user.go下编写密码加密方法

const (
    PassWordCost        = 12         //密码加密难度
    Active       string = "active"   //激活用户
)
//SetPassword 设置密码
func (user *User) SetPassword(password string) error {
    bytes, err := bcrypt.GenerateFromPassword([]byte(password), PassWordCost)
    if err != nil {
  return err
    }
    user.PasswordDigest = string(bytes)
    return nil
}      

确认无误之后对用户进行创建

if err := model.DB.Create(&user).Error; err != nil {
  logging.Info(err)
  code = e.ErrorDatabase
  return serializer.Response{
    Status: code,
    Msg:    e.GetMsg(code),
  }
    }      

服务层中用户注册的完整代码

func (service *UserRegisterService) Register() serializer.Response {
    var user model.User
    var count int
    code := e.SUCCESS
    model.DB.Model(&model.User{}).Where("user_name=?",service.UserName).Count(&count)
    if count == 1 {
  code = e.ErrorExistUser
  return serializer.Response{
    Status: code,
    Msg:    e.GetMsg(code),
  }
    }
    user = model.User{
  Nickname: service.Nickname,
  UserName: service.UserName,
  Status:   model.Active,
    }
    //加密密码
    if err := user.SetPassword(service.Password); err != nil {
  logging.Info(err)
  code = e.ErrorFailEncryption
  return serializer.Response{
    Status: code,
    Msg:    e.GetMsg(code),
  }
    }
    user.Avatar = "http://q1.qlogo.cn/g?b=qq&nk=294350394&s=640"
    //创建用户
    if err := model.DB.Create(&user).Error; err != nil {
  logging.Info(err)
  code = e.ErrorDatabase
  return serializer.Response{
    Status: code,
    Msg:    e.GetMsg(code),
  }
    }
    return serializer.Response{
  Status: code,
  Msg:    e.GetMsg(code),
    }
}      

下一章中，我们编写用户登录的业务逻辑。