业务编码

    3. import "github.com/zeromicro/go-zero/rest"
    5. type Config struct {
    6.     rest.RestConf
    7.     Mysql struct{
    8.         DataSource string
    9.     }
    11.     CacheRedis cache.CacheConf
    12. }
    1. Name: user-api
    2. Host: 0.0.0.0
    3. Port: 8888
    4. Mysql:
    5.   DataSource: $user:$password@tcp($url)/$db?charset=utf8mb4&parseTime=true&loc=Asia%2FShanghai
    6. CacheRedis:
    7.   - Host: $host
    8.     Pass: $pass
    1. type ServiceContext struct {
    2.     Config    config.Config
    4. }
    6. func NewServiceContext(c config.Config) *ServiceContext {
    7.     conn:=sqlx.NewMysql(c.Mysql.DataSource)
    8.     return &ServiceContext{
    9.         Config: c,
    10.         UserModel: model.NewUserModel(conn,c.CacheRedis),
    11.     }
    12. }
    1. func (l *LoginLogic) Login(req types.LoginReq) (*types.LoginReply, error) {
    2.     if len(strings.TrimSpace(req.Username)) == 0 || len(strings.TrimSpace(req.Password)) == 0 {
    3.         return nil, errors.New("参数错误")
    4.     }
    6.     userInfo, err := l.svcCtx.UserModel.FindOneByNumber(req.Username)
    7.     switch err {
    8.     case nil:
    9.     case model.ErrNotFound:
    10.         return nil, errors.New("用户名不存在")
    11.     default:
    12.         return nil, err
    13.     }
    14.     if userInfo.Password != req.Password {
    15.         return nil, errors.New("用户密码不正确")
    16.     }
    18.     // ---start---
    19.     now := time.Now().Unix()
    20.     accessExpire := l.svcCtx.Config.Auth.AccessExpire
    21.     jwtToken, err := l.getJwtToken(l.svcCtx.Config.Auth.AccessSecret, now, l.svcCtx.Config.Auth.AccessExpire, userInfo.Id)
    22.     if err != nil {
    23.         return nil, err
    24.     }
    25.     // ---end---
    27.     return &types.LoginReply{
    28.         Id:           userInfo.Id,
    29.         Name:         userInfo.Name,
    30.         Gender:       userInfo.Gender,
    31.         AccessToken:  jwtToken,
    32.         AccessExpire: now + accessExpire,
    33.         RefreshAfter: now + accessExpire/2,
    34.     }, nil
    35. }

    [!TIP] 上述代码中 [start]-[end]的代码实现见章节

    猜你想看