使用Golang輕松實現JWT身份驗證的示例代碼

更新時間：2024年02月23日 09:41:20 作者：愛發(fā)白日夢的后端

JSON Web Tokens (JWT)是一種流行的安全方法,用于在兩個方之間表示聲明,本文主要為大家詳細介紹了實現Go應用程序中的JWT身份驗證過程,需要的可以參考下

什么是JWT

JSON Web Token (JWT)是一種緊湊且URL安全的方式，用于在兩個方之間傳輸聲明。JWT中的聲明被編碼為一個JSON對象，并使用JSON Web Signature (JWS)進行數字簽名。

JWT通常的格式為：xxxxx.yyyyy.zzzzz

頭部：頭部（xxxxx）通常由兩部分組成：令牌類型JWT和簽名算法。
負載：負載（yyyyy）包含了聲明。聲明是關于主題（用戶）的陳述。
簽名：要創(chuàng)建簽名（zzzzz）部分，您需要使用編碼后的頭部、編碼后的負載、一個密鑰以及頭部中指定的算法進行簽名。

Go環(huán)境設置

首先，您需要一個用于在Go中處理JWT的軟件包。我們將使用github.com/golang-jwt/jwt軟件包。[1]

在Go中生成JWT

讓我們創(chuàng)建一個生成JWT的函數：

package main

import (
 "fmt"
 "github.com/golang-jwt/jwt/v4"
 "time"
)
var mySigningKey = []byte("secretpassword")
func GenerateJWT() (string, error) {
 token := jwt.New(jwt.SigningMethodHS256)
 claims := token.Claims.(jwt.MapClaims)
 claims["authorized"] = true
 claims["user"] = "John Doe"
 claims["exp"] = time.Now().Add(time.Minute * 30).Unix()
 tokenString, err := token.SignedString(mySigningKey)
 if err != nil {
  fmt.Errorf("Something went wrong: %s", err.Error())
  return "", err
 }
 return tokenString, nil
}

在Go中驗證JWT

現在，讓我們驗證JWT：

func ValidateToken(tokenString string) (*jwt.Token, error) {
 token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
  if _, ok := token.Method.(*jwt.SigningMethodHMAC); !ok {
   return nil, fmt.Errorf("There was an error")
  }
  return mySigningKey, nil
 })

 if err != nil {
   return nil, err
 }
 return token, nil
}

在Go Web應用程序中使用JWT進行身份驗證

以下是一個簡單示例，在Go的HTTP服務器中集成了JWT生成和驗證：

package main

import (
 "fmt"
 "log"
 "net/http"
)

func HomePage(w http.ResponseWriter, r *http.Request) {
 validToken, err := GenerateJWT()
 if err != nil {
  fmt.Fprintf(w, err.Error())
 }
 clientToken := r.Header.Get("Token")
 if clientToken != validToken {
  w.WriteHeader(http.StatusUnauthorized)
  fmt.Fprintf(w, "Token is not valid")
  return
 }
 fmt.Fprintf(w, "Hello, World!")
}

func handleRequests() {
 http.HandleFunc("/", HomePage)
 log.Fatal(http.ListenAndServe(":9000", nil))
}

func main() {
 handleRequests()
}

使用此設置：