feat(auth): 添加完整的用户认证API项目
- 实现用户注册、登录、JWT令牌认证功能 - 集成Gin、GORM、Viper、Zap等框架 - 添加密码加密、数据库操作、中间件等完整功能 - 配置多环境支持、日志轮转、CORS处理 - 创建完整的项目结构和配置文件体系
This commit is contained in:
73
Web开发/02go-gin-middleware/middleware_complete.go
Normal file
73
Web开发/02go-gin-middleware/middleware_complete.go
Normal file
@@ -0,0 +1,73 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
|
||||
// 日志中间件
|
||||
func Logger() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
start := time.Now()
|
||||
c.Next()
|
||||
duration := time.Since(start)
|
||||
fmt.Printf("[%s] %s %d (%v)\n",
|
||||
c.Request.Method, c.Request.URL.Path, c.Writer.Status(), duration)
|
||||
}
|
||||
}
|
||||
|
||||
// 认证中间件
|
||||
func Auth() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
token := c.GetHeader("Authorization")
|
||||
if token != "Bearer secret" {
|
||||
c.JSON(401, gin.H{"error": "Unauthorized"})
|
||||
c.Abort()
|
||||
return
|
||||
}
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
// Recovery 中间件
|
||||
func Recovery() gin.HandlerFunc {
|
||||
return func(c *gin.Context) {
|
||||
defer func() {
|
||||
if err := recover(); err != nil {
|
||||
c.JSON(500, gin.H{"error": "Internal Server Error"})
|
||||
c.Abort()
|
||||
}
|
||||
}()
|
||||
c.Next()
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
r := gin.New()
|
||||
|
||||
// 全局中间件
|
||||
r.Use(Logger())
|
||||
r.Use(Recovery())
|
||||
|
||||
// 公开路由
|
||||
r.GET("/", func(c *gin.Context) {
|
||||
c.JSON(200, gin.H{"message": "Public"})
|
||||
})
|
||||
|
||||
// 需要认证的路由组
|
||||
auth := r.Group("/api")
|
||||
auth.Use(Auth())
|
||||
{
|
||||
auth.GET("/users", func(c *gin.Context) {
|
||||
c.JSON(200, gin.H{"users": []string{"Alice", "Bob"}})
|
||||
})
|
||||
|
||||
auth.POST("/users", func(c *gin.Context) {
|
||||
c.JSON(201, gin.H{"message": "User created"})
|
||||
})
|
||||
}
|
||||
|
||||
r.Run(":9999")
|
||||
}
|
||||
Reference in New Issue
Block a user