Golang 搭建 WebSocket 应用 - jwt 认证

package main

import (
    "net/http"

    "github.com/dgrijalva/jwt-go"
    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
    CheckOrigin: func(r *http.Request) bool {
        // 在这里设置跨域请求
        return true
    },
}

// jwtTokenAuth 是一个中间件，用于验证JWT token
func jwtTokenAuth(h http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        // 从请求中获取token
        tokenString := r.URL.Query().Get("token")

        token, err := jwt.Parse(tokenString, func(token *jwt.Token) (interface{}, error) {
            // 这里应该是你的秘钥
            return []byte("your_secret_key"), nil
        })

        if err != nil || !token.Valid {
            http.Error(w, "Invalid token", http.StatusUnauthorized)
            return
        }

        h.ServeHTTP(w, r)
    }
}

func wsHandler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        // 处理错误
        return
    }
    defer conn.Close()

    // 使用conn进行通信
}

func main() {
    http.HandleFunc("/ws", jwtTokenAuth(wsHandler))
    http.ListenAndServe(":8080", nil)
}