怎么利用go-zero在Go中實(shí)現(xiàn)JWT認(rèn)證

這篇文章主要介紹了怎么利用go-zero在Go中實(shí)現(xiàn)JWT認(rèn)證，具有一定借鑒價(jià)值，感興趣的朋友可以參考下，希望大家閱讀完這篇文章之后大有收獲，下面讓小編帶著大家一起了解一下。

1. 客戶端獲取JWT Token

我們定義一個(gè)協(xié)議供客戶端調(diào)用獲取JWT token，我們新建一個(gè)目錄jwt然后在目錄中執(zhí)行 goctl api -o jwt.api，將生成的jwt.api改成如下：

type JwtTokenRequest struct {
}

type JwtTokenResponse struct {
  AccessToken  string `json:"access_token"`
  AccessExpire int64  `json:"access_expire"`
  RefreshAfter int64  `json:"refresh_after"` // 建議客戶端刷新token的絕對(duì)時(shí)間
}

type GetUserRequest struct { 
  UserId string `json:"userId"`
}

type GetUserResponse struct {
  Name string `json:"name"`
}

service jwt-api {
  @handler JwtHandler
  post /user/token(JwtTokenRequest) returns (JwtTokenResponse)
}

@server(
  jwt: JwtAuth
)
service jwt-api {
  @handler JwtHandler
  post /user/info(GetUserRequest) returns (GetUserResponse)
}

在服務(wù)jwt目錄中執(zhí)行：goctl api go -api jwt.api -dir . 打開(kāi)jwtlogic.go文件，修改 func (l *JwtLogic) Jwt(req types.JwtTokenRequest) (*types.JwtTokenResponse, error) { 方法如下：

func (l *JwtLogic) Jwt(req types.JwtTokenRequest) (*types.JwtTokenResponse, error) {
	var accessExpire = l.svcCtx.Config.JwtAuth.AccessExpire

	now := time.Now().Unix()
	accessToken, err := l.GenToken(now, l.svcCtx.Config.JwtAuth.AccessSecret, nil, accessExpire)
	if err != nil {
		return nil, err
	}

	return &types.JwtTokenResponse{
    AccessToken:  accessToken,
    AccessExpire: now + accessExpire,
    RefreshAfter: now + accessExpire/2,
  }, nil
}

func (l *JwtLogic) GenToken(iat int64, secretKey string, payloads map[string]interface{}, seconds int64) (string, error) {
	claims := make(jwt.MapClaims)
	claims["exp"] = iat + seconds
	claims["iat"] = iat
	for k, v := range payloads {
		claims[k] = v
	}

	token := jwt.New(jwt.SigningMethodHS256)
	token.Claims = claims

	return token.SignedString([]byte(secretKey))
}

在啟動(dòng)服務(wù)之前，我們需要修改etc/jwt-api.yaml文件如下：

Name: jwt-api
Host: 0.0.0.0
Port: 8888
JwtAuth:
  AccessSecret: xxxxxxxxxxxxxxxxxxxxxxxxxxxxx
  AccessExpire: 604800

啟動(dòng)服務(wù)器，然后測(cè)試下獲取到的token。

? curl --location --request POST '127.0.0.1:8888/user/token'
{"access_token":"eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE2MDEyNjE0MjksImlhdCI6MTYwMDY1NjYyOX0.6u_hpE_4m5gcI90taJLZtvfekwUmjrbNJ-5saaDGeQc","access_expire":1601261429,"refresh_after":1600959029}

2. 服務(wù)器驗(yàn)證JWT token

1. 在api文件中通過(guò)jwt: JwtAuth標(biāo)記的service表示激活了jwt認(rèn)證。

2. 可以閱讀rest/handler/authhandler.go文件了解服務(wù)器jwt實(shí)現(xiàn)。

3. 修改getuserlogic.go如下：

func (l *GetUserLogic) GetUser(req types.GetUserRequest) (*types.GetUserResponse, error) {
	return &types.GetUserResponse{Name: "kim"}, nil
}

• 我們先不帶JWT Authorization header請(qǐng)求頭測(cè)試下，返回http status code是401，符合預(yù)期。

? curl -w  "\nhttp: %{http_code} \n" --location --request POST '127.0.0.1:8888/user/info' \
--header 'Content-Type: application/json' \
--data-raw '{
    "userId": "a"
}'

http: 401

• 加上Authorization header請(qǐng)求頭測(cè)試。

? curl -w  "\nhttp: %{http_code} \n" --location --request POST '127.0.0.1:8888/user/info' \
--header 'Authorization: eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE2MDEyNjE0MjksImlhdCI6MTYwMDY1NjYyOX0.6u_hpE_4m5gcI90taJLZtvfekwUmjrbNJ-5saaDGeQc' \
--header 'Content-Type: application/json' \
--data-raw '{
    "userId": "a"
}'
{"name":"kim"}
http: 200

綜上所述：基于go-zero的JWT認(rèn)證完成，在真實(shí)生產(chǎn)環(huán)境部署時(shí)候，AccessSecret, AccessExpire, RefreshAfter根據(jù)業(yè)務(wù)場(chǎng)景通過(guò)配置文件配置，RefreshAfter 是告訴客戶端什么時(shí)候該刷新JWT token了，一般都需要設(shè)置過(guò)期時(shí)間前幾天。

感謝你能夠認(rèn)真閱讀完這篇文章，希望小編分享的“怎么利用go-zero在Go中實(shí)現(xiàn)JWT認(rèn)證”這篇文章對(duì)大家有幫助，同時(shí)也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，更多相關(guān)知識(shí)等著你來(lái)學(xué)習(xí)!