mirror of
https://github.com/tnb-labs/panel.git
synced 2026-09-01 14:55:12 +08:00
耗子
443516a2cf
依赖注入：
- 移除 google/wire（含 wire.go/wire_gen.go 与全部 ProviderSet），
改用 samber/do v2 单注入器 + 双入口惰性构建
- 贡献模型替代命令式注册：路由 routes:、命令 commands:、
任务 jobs: 前缀经 internal/registry 收集与校验
- 构造函数统一 func NewXxx(i do.Injector) (T, error)
三层架构：
- 补全用例层：每个 biz.XxxRepo 配 XxxUsecase，
service/command/job 表现层只依赖用例，不再直接引用仓储
路由与文档：
- route/http.go 按域拆为声明式 Endpoint 贡献，
Endpoint 承载登录白名单与端点限流语义
- 调试模式下提供 OpenAPI 3.1 文档：/openapi.json 与 /docs（Scalar），
从 validate 标签生成
目录与依赖：
- internal/http/{middleware,request,rule} 拍平至 internal/*
- CLI 命令拆至 internal/command
- 日志轮转 timberjack 换为 libtnb/logrotate
- 升级 validator/cron/sessions/gormstore/sqlite/securecookie
- 数据库迁移保持 gormigrate 不变
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
29 lines
656 B
Go
29 lines
656 B
Go
package middleware
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/sethvargo/go-limiter/httplimit"
|
|
"github.com/sethvargo/go-limiter/memorystore"
|
|
)
|
|
|
|
// Throttle 限流器
|
|
func Throttle(ipHeader string, tokens uint64, interval time.Duration) func(next http.Handler) http.Handler {
|
|
store, err := memorystore.New(&memorystore.Config{
|
|
Tokens: tokens,
|
|
Interval: interval,
|
|
})
|
|
if err != nil {
|
|
log.Fatalf("failed to create throttle memorystore: %v", err)
|
|
}
|
|
|
|
limiter, err := httplimit.NewMiddleware(store, httplimit.IPKeyFunc(ipHeader))
|
|
if err != nil {
|
|
log.Fatalf("failed to initialize throttle middleware: %v", err)
|
|
}
|
|
|
|
return limiter.Handle
|
|
}
|