Golang使用Token来验证

go get github.com/golang-jwt/jwt

package main

import (
    "fmt"
    "time"
    "github.com/golang-jwt/jwt"
)

// 定义一个密钥，用于加密Token
var jwtKey = []byte("my_secret_key")

// 定义Token中包含的信息
type Claims struct {
    Username string `json:"username"`
    jwt.StandardClaims
}

// 生成Token的函数
func GenerateToken(username string) (string, error) {
    // 设置Token的过期时间，比如24小时后过期
    expirationTime := time.Now().Add(24 * time.Hour)
    // 创建一个包含用户名和过期时间的声明
    claims := &Claims{
        Username: username,
        StandardClaims: jwt.StandardClaims{
            ExpiresAt: expirationTime.Unix(),
        },
    }
    // 使用声明创建一个Token
    token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
    // 使用密钥对Token进行签名
    tokenString, err := token.SignedString(jwtKey)
    if err != nil {
        return "", err
    }
    return tokenString, nil
}

func main() {
    // 假设用户名是"user1"
    token, err := GenerateToken("user1")
    if err != nil {
        fmt.Println("生成Token时出错:", err)
    } else {
        fmt.Println("生成的Token:", token)
    }
}

package main

import (
    "fmt"
    "github.com/golang-jwt/jwt"
    "net/http"
)

// 验证Token的函数
func ValidateToken(tokenString string) (*Claims, error) {
    claims := &Claims{}
    token, err := jwt.ParseWithClaims(tokenString, claims, func(token *jwt.Token) (interface{}, error) {
        return jwtKey, nil
    })
    if err != nil {
        if err == jwt.ErrSignatureInvalid {
            return nil, fmt.Errorf("无效的Token签名")
        }
        return nil, fmt.Errorf("无效的Token")
    }
    if !token.Valid {
        return nil, fmt.Errorf("无效的Token")
    }
    return claims, nil
}

func main() {
    http.HandleFunc("/protected", func(w http.ResponseWriter, r *http.Request) {
        // 从请求头中获取Token
        tokenString := r.Header.Get("Authorization")
        // 验证Token
        claims, err := ValidateToken(tokenString)
        if err != nil {
            http.Error(w, err.Error(), http.StatusUnauthorized)
            return
        }
        // 如果Token合法，返回欢迎信息
        fmt.Fprintf(w, "Hello, %s", claims.Username)
    })

    http.ListenAndServe(":8080", nil)
}