Business code

    In the previous section, we have written user.api based on the preliminary requirements to describe which services the user service provides to the outside world. In this section, we will continue with the previous steps. Use business coding to tell how go-zero is used in actual business.

    3. import "github.com/tal-tech/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("Invalid parameter")
    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("Username does not exist")
    11.     default:
    12.         return nil, err
    13.     }
    14.     if userInfo.Password != req.Password {
    15.         return nil, errors.New("User password is incorrect")
    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. }

    Guess you wants