Business code

[!TIP] This document is machine-translated by Google. If you find grammatical and semantic errors, and the document description is not clear, please PR

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.

Add Mysql configuration

  1. $ vim service/user/cmd/api/internal/config/config.go
  1. package config
  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. }

Improve yaml configuration

  1. $ vim service/user/cmd/api/etc/user-api.yaml
  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
  9.     Type: node

[!TIP] $user: mysql database user

$password: mysql database password

$url: mysql database connection address

$db: mysql database db name, that is, the database where the user table is located

$host: Redis connection address Format: ip:port, such as: 127.0.0.1:6379

$pass: redis password

For more configuration information, please refer to api configuration introduction

Improve service dependence

  1. $ vim service/user/cmd/api/internal/svc/servicecontext.go
  1. type ServiceContext struct {
  2.     Config    config.Config
  3.     UserModel model.UserModel
  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. }

Fill in the login logic

  1. $ vim service/user/cmd/api/internal/logic/loginlogic.go
  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.     }
  15.     if userInfo.Password != req.Password {
  16.         return nil, errors.New("User password is incorrect")
  17.     }
  19.     // ---start---
  20.     now := time.Now().Unix()
  21.     accessExpire := l.svcCtx.Config.Auth.AccessExpire
  22.     jwtToken, err := l.getJwtToken(l.svcCtx.Config.Auth.AccessSecret, now, l.svcCtx.Config.Auth.AccessExpire, userInfo.Id)
  23.     if err != nil {
  24.         return nil, err
  25.     }
  26.     // ---end---
  28.     return &types.LoginReply{
  29.         Id:           userInfo.Id,
  30.         Name:         userInfo.Name,
  31.         Gender:       userInfo.Gender,
  32.         AccessToken:  jwtToken,
  33.         AccessExpire: now + accessExpire,
  34.         RefreshAfter: now + accessExpire/2,
  35.     }, nil
  36. }

[!TIP] For the code implementation of [start]-[end] in the above code, please refer to the Jwt Authentication chapter

Guess you wants