业务逻辑封装

    数据校验

    实现代码

    1. package user
    3. import (
    4.     "errors"
    5.     "fmt"
    6.     "github.com/gogf/gf-demos/app/model/user"
    7.     "github.com/gogf/gf/net/ghttp"
    8.     "github.com/gogf/gf/os/gtime"
    9.     "github.com/gogf/gf/util/gconv"
    10.     "github.com/gogf/gf/util/gvalid"
    11. )
    13. const (
    14.     USER_SESSION_MARK = "user_info"
    15. )
    17. // 注册输入参数
    18. type SignUpInput struct {
    19.     Passport  string `v:"required|length:6,16#账号不能为空|账号长度应当在:min到:max之间"`
    20.     Password  string `v:"required|length:6,16#请输入确认密码|密码长度应当在:min到:max之间"`
    21.     Password2 string `v:"required|length:6,16|same:Password#密码不能为空|密码长度应当在:min到:max之间|两次密码输入不相等"`
    22.     Nickname  string
    23. }
    25. // 用户注册
    26. func SignUp(data *SignUpInput) error {
    27.     // 输入参数检查
    28.     if e := gvalid.CheckStruct(data, nil); e != nil {
    29.         return errors.New(e.String())
    31.     if data.Nickname == "" {
    32.         data.Nickname = data.Passport
    33.     }
    34.     // 账号唯一性数据检查
    35.     if !CheckPassport(data.Passport) {
    36.         return errors.New(fmt.Sprintf("账号 %s 已经存在", data.Passport))
    37.     }
    38.     // 昵称唯一性数据检查
    39.     if !CheckNickName(data.Nickname) {
    40.         return errors.New(fmt.Sprintf("昵称 %s 已经存在", data.Nickname))
    41.     }
    42.     // 将输入参数赋值到数据库实体对象上
    43.     var entity *user.Entity
    44.     if err := gconv.Struct(data, &entity); err != nil {
    45.         return err
    46.     }
    47.     // 记录账号创建/注册时间
    48.     entity.CreateTime = gtime.Now()
    49.     if _, err := user.Save(entity); err != nil {
    50.         return err
    51.     }
    52.     return nil
    53. }
    55. // 判断用户是否已经登录
    56. func IsSignedIn(session *ghttp.Session) bool {
    57.     return session.Contains(USER_SESSION_MARK)
    58. }
    60. // 用户登录，成功返回用户信息，否则返回nil; passport应当会md5值字符串
    61. func SignIn(passport, password string, session *ghttp.Session) error {
    62.     one, err := user.FindOne("passport=? and password=?", passport, password)
    63.     if err != nil {
    65.     if one == nil {
    66.         return errors.New("账号或密码错误")
    67.     }
    68.     return session.Set(USER_SESSION_MARK, one)
    69. }
    71. // 用户注销
    72. func SignOut(session *ghttp.Session) error {
    73.     return session.Remove(USER_SESSION_MARK)
    74. }
    76. // 检查账号是否符合规范(目前仅检查唯一性),存在返回false,否则true
    77. func CheckPassport(passport string) bool {
    78.     if i, err := user.FindCount("passport", passport); err != nil {
    79.         return false
    80.     } else {
    81.         return i == 0
    82.     }
    83. }
    85. // 检查昵称是否符合规范(目前仅检查唯一性),存在返回false,否则true
    86. func CheckNickName(nickname string) bool {
    87.     if i, err := user.FindCount("nickname", nickname); err != nil {
    88.         return false
    89.     } else {
    90.         return i == 0
    91.     }
    92. }
    94. // 获得用户信息详情
    95. func GetProfile(session *ghttp.Session) (u *user.Entity) {
    96.     _ = session.GetStruct(USER_SESSION_MARK, &u)
    97.     return