包名约定

根据官方《Effective Go》建议，包名尽量采用言简意赅的名称(short, concise, evocative)。并且推荐通过不同的import路径来区分相同包名的包引入。

控制器实现

控制器主要用于接口的路由注册以及对于用户输入参数的处理。

我们这里使用执行对象注册方式。

/app/api/user/user.go

package user
import (
    "github.com/gogf/gf-demos/app/service/user"
    "github.com/gogf/gf-demos/library/response"
    "github.com/gogf/gf/g/net/ghttp"
    "github.com/gogf/gf/g/util/gvalid"
)
// 用户API管理对象
type Controller struct { }
// 用户注册接口
func (c *Controller) SignUp(r *ghttp.Request) {
    if err := user.SignUp(r.GetPostMap()); err != nil {
        response.Json(r, 1, err.Error())
    } else {
        response.Json(r, 0, "ok")
    }
}
// 用户登录接口
func (c *Controller) SignIn(r *ghttp.Request) {
    data  := r.GetPostMap()
    rules := map[string]string {
        "passport"  : "required",
        "password"  : "required",
    }
    msgs  := map[string]interface{} {
        "passport" : "账号不能为空",
        "password" : "密码不能为空",
    }
    if e := gvalid.CheckMap(data, rules, msgs); e != nil {
        response.Json(r, 1, e.String())
    }
    if err := user.SignIn(data["passport"], data["password"], r.Session); err != nil {
        response.Json(r, 1, err.Error())
    } else {
        response.Json(r, 0, "ok")
    }
}
// 判断用户是否已经登录
func (c *Controller) IsSignedIn(r *ghttp.Request) {
    if user.IsSignedIn(r.Session) {
        response.Json(r, 0, "ok")
    } else {
        response.Json(r, 1, "")
    }
}
// 用户注销/退出接口
func (c *Controller) SignOut(r *ghttp.Request) {
    user.SignOut(r.Session)
    response.Json(r, 0, "ok")
}
// 检测用户账号接口(唯一性校验)
func (c *Controller) CheckPassport(r *ghttp.Request) {
    passport := r.Get("passport")
    if e := gvalid.Check(passport, "required", "请输入账号"); e != nil {
        response.Json(r, 1, e.String())
    }
    if user.CheckPassport(passport) {
        response.Json(r, 0, "ok")
    }
    response.Json(r, 1, "账号已经存在")
}
// 检测用户昵称接口(唯一性校验)
func (c *Controller) CheckNickName(r *ghttp.Request) {
    nickname := r.Get("nickname")
    if e := gvalid.Check(nickname, "required", "请输入昵称"); e != nil {
        response.Json(r, 1, e.String())
    }
    if user.CheckNickName(r.Get("nickname")) {
        response.Json(r, 0, "ok")
    }
    response.Json(r, 1, "昵称已经存在")
}

逻辑封装实现

我们这里没有使用到数据模型，仅使用了 逻辑封装层+gdb 来操作数据库。

用户逻辑

主要用于用户接口的业务逻辑封装。

/app/service/user/user.go

package user
import (
    "errors"
    "fmt"
    "github.com/gogf/gf/g"
    "github.com/gogf/gf/g/net/ghttp"
    "github.com/gogf/gf/g/os/gtime"
    "github.com/gogf/gf/g/util/gvalid"
)
const (
    USER_SESSION_MARK = "user_info"
)
var (
    // 表对象
    table = g.DB().Table("user").Safe()
)
// 用户注册
func SignUp(data g.MapStrStr) error {
    // 数据校验
    rules := []string {
        "passport @required|length:6,16#账号不能为空|账号长度应当在:min到:max之间",
        "password2@required|length:6,16#请输入确认密码|密码长度应当在:min到:max之间",
        "password @required|length:6,16|same:password2#密码不能为空|密码长度应当在:min到:max之间|两次密码输入不相等",
    }
    if e := gvalid.CheckMap(data, rules); e != nil {
        return errors.New(e.String())
    }
    if _, ok := data["nickname"]; !ok {
        data["nickname"] = data["passport"]
    }
    // 唯一性数据检查
    if !CheckPassport(data["passport"]) {
        return errors.New(fmt.Sprintf("账号 %s 已经存在", data["passport"]))
    }
    if !CheckNickName(data["nickname"]) {
        return errors.New(fmt.Sprintf("昵称 %s 已经存在", data["nickname"]))
    }
    // 记录账号创建/注册时间
    if _, ok := data["create_time"]; !ok {
        data["create_time"] = gtime.Now().String()
    }
    if _, err := table.Filter().Data(data).Save(); err != nil {
        return err
    }
    return nil
}
// 判断用户是否已经登录
func IsSignedIn(session *ghttp.Session) bool {
    return session.Contains(USER_SESSION_MARK)
}
// 用户登录，成功返回用户信息，否则返回nil; passport应当会md5值字符串
func SignIn(passport, password string, session *ghttp.Session) error {
    record, err := table.Where("passport=? and password=?", passport, password).One()
    if err != nil {
        return err
    }
    if record == nil {
        return errors.New("账号或密码错误")
    }
    session.Set(USER_SESSION_MARK, record)
    return nil
}
// 用户注销
func SignOut(session *ghttp.Session) {
    session.Remove(USER_SESSION_MARK)
}
// 检查账号是否符合规范(目前仅检查唯一性),存在返回false,否则true
func CheckPassport(passport string) bool {
    if i, err := table.Where("passport", passport).Count(); err != nil {
        return false
    } else {
        return i == 0
    }
}
// 检查昵称是否符合规范(目前仅检查唯一性),存在返回false,否则true
func CheckNickName(nickname string) bool {
    if i, err := table.Where("nickname", nickname).Count(); err != nil {
        return false
    } else {
        return i == 0
    }
}

工具类包

主要用于返回JSON数据格式的统一。

/library/response/response.go

package response
import (
    "github.com/gogf/gf/g"
    "github.com/gogf/gf/g/net/ghttp"
)
// 标准返回结果数据结构封装。
// 返回固定数据结构的JSON:
// err:  错误码(0:成功, 1:失败, >1:错误码);
// msg:  请求结果信息;
// data: 请求结果,根据不同接口返回结果的数据结构不同;
func Json(r *ghttp.Request, err int, msg string, data...interface{}) {
    responseData := interface{}(nil)
    if len(data) > 0 {
        responseData = data[0]
    }
    r.Response.WriteJson(g.Map{
        "err"  : err,
        "msg"  : msg,
        "data" : responseData,
    })
    r.Exit()
}