Geliebte
f9cdeb5d64
- 移除管理员初始化/登录/JWT 流程，改为固定进入密钥（19735500）， 首次启动自动初始化到数据库 settings 表（可改库更新） - analyze 接口支持可选密钥鉴权：X-Access-Key 请求头或 access_key 字段， 带密钥强制校验（错误 401），不带密钥保持免鉴权 - 解锁后顶栏"复制 API 文档"按钮：以当前访问 URL 为基础地址动态生成 Markdown 文档（免鉴权/带密钥两种 curl 示例，供智能体直接调用） - 白昼/黑夜主题切换：CSS 变量驱动，localStorage 持久化，默认跟随系统 - 修复 hidden 属性被 display:flex 覆盖导致解锁层无法隐藏的问题 - 页面去除"免费/Agnes"文案，页脚署名青梧映星软件开发工作室 - build.bat/build.sh 简化为 Linux 交叉编译（CGO_ENABLED=0） - 依赖精简：移除 golang-jwt/bcrypt，直接依赖仅 gin/sqlite/gorm/yaml Co-Authored-By: Claude <noreply@anthropic.com>
75 lines
1.7 KiB
Go
75 lines
1.7 KiB
Go
package middleware
|
|
|
|
import (
|
|
"math"
|
|
"net/http"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"vision-tool/internal/model"
|
|
)
|
|
|
|
// CORS 跨域中间件（开发环境放开，生产环境应限制为具体域名）。
|
|
func CORS() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
c.Header("Access-Control-Allow-Origin", "*")
|
|
c.Header("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS")
|
|
c.Header("Access-Control-Allow-Headers", "Content-Type, Authorization")
|
|
if c.Request.Method == http.MethodOptions {
|
|
c.AbortWithStatus(http.StatusNoContent)
|
|
return
|
|
}
|
|
c.Next()
|
|
}
|
|
}
|
|
|
|
// ============================================================================
|
|
// 内存令牌桶限流（全局限流，防匿名刷接口）
|
|
// ============================================================================
|
|
|
|
// RateLimiter 基于时间的令牌桶。
|
|
type RateLimiter struct {
|
|
mu sync.Mutex
|
|
rate float64 // 每秒补充令牌数
|
|
burst float64 // 桶容量
|
|
tokens float64
|
|
lastTime time.Time
|
|
}
|
|
|
|
func NewRateLimiter(rps, burst int) *RateLimiter {
|
|
return &RateLimiter{
|
|
rate: float64(rps),
|
|
burst: float64(burst),
|
|
tokens: float64(burst),
|
|
lastTime: time.Now(),
|
|
}
|
|
}
|
|
|
|
func (rl *RateLimiter) Allow() bool {
|
|
rl.mu.Lock()
|
|
defer rl.mu.Unlock()
|
|
|
|
now := time.Now()
|
|
rl.tokens = math.Min(rl.burst, rl.tokens+now.Sub(rl.lastTime).Seconds()*rl.rate)
|
|
rl.lastTime = now
|
|
|
|
if rl.tokens >= 1 {
|
|
rl.tokens--
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
// Middleware 返回限流 gin 中间件。
|
|
func (rl *RateLimiter) Middleware() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
if !rl.Allow() {
|
|
c.AbortWithStatusJSON(http.StatusTooManyRequests, model.Err(429, "请求过于频繁，请稍后再试"))
|
|
return
|
|
}
|
|
c.Next()
|
|
}
|
|
}
|