在使用 Go-Zero 框架实现 JWT 校验的 API 接口时,你可以按照以下步骤进行:
1. 安装依赖
首先,确保你已经安装了 go-zero
和 jwt-go
库。
go get github.com/zeromicro/go-zero
go get github.com/dgrijalva/jwt-go
2. 创建 JWT 中间件
你需要创建一个中间件来处理 JWT 的校验逻辑。以下是一个简单的 JWT 校验中间件示例:
package middleware
import (
"net/http"
"strings"
"github.com/dgrijalva/jwt-go"
)
var jwtSecret = []byte("your_secret_key") // 替换为你的密钥
func JwtMiddleware(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
tokenString := r.Header.Get("Authorization")
if tokenString == "" || !strings.HasPrefix(tokenString, "Bearer ") {
http.Error(w, "Unauthorized", http.StatusUnauthorized)
return
}
tokenString = strings.TrimPrefix(tokenString, "Bearer ")
claims := &jwt.StandardClaims{}
token, err := jwt.ParseWithClaims(tokenString, claims, func(token *jwt.Token) (interface{}, error) {
return jwtSecret, nil
})
if err != nil || !token.Valid {
http.Error(w, "Unauthorized", http.StatusUnauthorized)
return
}
next.ServeHTTP(w, r)
})
}
3. 创建 API 接口
接下来,创建你的 API 接口,并将 JWT 中间件应用到这些接口上。
package main
import (
"net/http"
"github.com/gorilla/mux"
"your_project_path/middleware" // 修改为实际路径
)
func main() {
r := mux.NewRouter()
// 应用JWT中间件到某些路由上
r.Handle("/api/protected", middleware.JwtMiddleware(http.HandlerFunc(protectedHandler))).Methods("GET")
// 启动HTTP服务器
http.ListenAndServe(":8080", r)
}
func protectedHandler(w http.ResponseWriter, r *http.Request) {
w.Write([]byte("This is a protected route"))
}
4. 使用 Postman 或 Curl 测试
在发送请求时,请确保在 Authorization 头中包含 Bearer Token。例如:
curl -H "Authorization: Bearer your_jwt_token" http://localhost:8080/api/protected
如果 token 有效,你将会看到 "This is a protected route"
的响应。
注意事项
- 安全性:请勿硬编码敏感信息,如密钥。应使用环境变量或配置文件管理它们。
- Token生成:确保有一套机制去生成和签名 JWT Token。
- 错误处理:根据需要增强错误处理,以便提供更多上下文信息给客户端。
这样你就能使用 Go-Zero 实现一个带有 JWT 验证的 API 接口了。如果有其他具体需求或者问题,可以继续问我!
内容由零声教学AI助手提供,问题来源于学员提问