package main

import (
    "github.com/gin-gonic/gin"
    "thinkgin/app"
    "thinkgin/framework"
    "thinkgin/route"
)

func main() {
    // 1. 注册自定义中间件（必须在 framework.New 之前）
    route.RegisterMiddleware("audit", func() gin.HandlerFunc {
        return func(c *gin.Context) {
            app.GetLogger().Infof("audit: %s %s", c.Request.Method, c.Request.URL.Path)
            c.Next()
        }
    })

    // 2. 也可以覆盖内置中间件（例如替换默认 access_log）
    route.RegisterMiddleware("access_log", func() gin.HandlerFunc {
        return myaccesslog.New()
    })

    // 3. 然后在 config/middleware.yaml 写：
    //    middleware:
    //      global: [recovery, audit, access_log]

    a, _ := framework.New()
    _ = a.Run(nil)
}

// 默认：窗口 100 / 50% 错误率 / 10s 冷却 / 5 次试探
r.Use(middleware.CircuitBreaker())

// 自定义
r.Use(middleware.CircuitBreakerWithConfig(middleware.CircuitBreakerConfig{
    WindowSize:            200,
    ErrorThresholdPercent: 30,
    CooldownDuration:      5 * time.Second,
    HalfOpenMaxRequests:   3,
}))

r.Use(middleware.Timeout())                          // 默认 30s
r.Use(middleware.TimeoutWithDuration(5 * time.Second)) // 自定义

r.Use(middleware.BodyLimit())                       // 默认 10MB
r.Use(middleware.BodyLimitWithSize(5 << 20))         // 5MB
h, _ := middleware.BodyLimitFromString("512KB")      // 字符串解析
r.Use(h)

if c := middleware.BusinessCounter("api_calls", []string{"endpoint"}); c != nil {
    c.WithLabelValues("/api/users").Inc()
}
if h := middleware.BusinessHistogram("api_duration", []string{"endpoint"}, nil); h != nil {
    h.WithLabelValues("/api/users").Observe(elapsed.Seconds())
}

svc := app.SvcFromGin(c)
svc.Log.Infof("当前应用: %s", svc.Config.App.Name)

import (
    "thinkgin/app/cache"
    "thinkgin/app/database"
)

db := database.Default()
pg, _ := database.Get("pgsql")
rdb := cache.Default()

// GORM + OTel：传递 context 自动纳入 trace
database.Default().WithContext(c.Request.Context()).Find(&users)

import "thinkgin/extend/websocket"

// Echo
r.GET("/ws", websocket.Handler(func(conn *websocket.Conn) {
    defer conn.Close()
    for {
        mt, msg, err := conn.Conn.ReadMessage()
        if err != nil { break }
        conn.WriteSafeMessage(mt, msg)
    }
}))

// Hub 广播
hub := websocket.NewHub()
hub.Broadcast(ws.TextMessage, msg)
hub.BroadcastJSON(map[string]string{"event": "join"})

// 心跳保活（v3.14.0+）：自动 ping/pong + 读写超时，及时回收死连接
r.GET("/push", websocket.KeepAliveHandler(func(conn *websocket.Conn) {
    for {
        if _, _, err := conn.ReadMessage(); err != nil { break }
    }
}, websocket.KeepAliveConfig{})) // 零值使用默认 10s/60s/54s

import "thinkgin/app/session"

s := session.From(c)
s.Set("user_id", 42)
_ = s.Save(c.Request.Context())

// 签发
token, err := middleware.JWTIssue("42", map[string]any{"role": "admin"}, 0)

// 保护路由
api := r.Group("/api/v1", middleware.JWTAuth())
api.GET("/me", func(c *gin.Context) {
    claims := middleware.JWTFromContext(c)
    c.JSON(200, gin.H{"uid": claims.Subject})
})