diff --git a/internal/middleware/limitrate/limitrate.go b/internal/middleware/limitrate/limitrate.go deleted file mode 100644 index 8cdb2d1..0000000 --- a/internal/middleware/limitrate/limitrate.go +++ /dev/null @@ -1,24 +0,0 @@ -// Package limitrate 提供基于令牌桶算法的请求速率限制功能。 -// -// 包含速率限制器相关的逻辑,用于控制请求处理速率。 -// -// 作者:xfy -package limitrate - -import ( - "rua.plus/lolly/internal/config" -) - -const ( - // LargeFileStrategySkip 大文件策略:跳过(不限制)。 - LargeFileStrategySkip = "skip" - // LargeFileStrategyCoarse 大文件策略:粗略限制。 - LargeFileStrategyCoarse = "coarse" -) - -// Middleware 速率限制中间件 -type Middleware struct { - config *config.LimitRateConfig -} - - diff --git a/internal/middleware/limitrate/limitrate_test.go b/internal/middleware/limitrate/limitrate_test.go deleted file mode 100644 index af8da5b..0000000 --- a/internal/middleware/limitrate/limitrate_test.go +++ /dev/null @@ -1,19 +0,0 @@ -// Package limitrate 提供响应速率限制中间件的测试。 -// -// 该文件测试速率限制模块的各项功能,包括: -// - 常量值 -// -// 作者:xfy -package limitrate - -import "testing" - -// TestConstants 测试常量值。 -func TestConstants(t *testing.T) { - if LargeFileStrategySkip != "skip" { - t.Errorf("LargeFileStrategySkip = %q, want %q", LargeFileStrategySkip, "skip") - } - if LargeFileStrategyCoarse != "coarse" { - t.Errorf("LargeFileStrategyCoarse = %q, want %q", LargeFileStrategyCoarse, "coarse") - } -} diff --git a/internal/middleware/limitrate/writer.go b/internal/middleware/limitrate/writer.go deleted file mode 100644 index 5573658..0000000 --- a/internal/middleware/limitrate/writer.go +++ /dev/null @@ -1,24 +0,0 @@ -// Package limitrate 提供基于令牌桶算法的请求速率限制功能。 -// -// 包含速率限制响应写入器相关的逻辑,用于处理被限制的请求响应。 -// -// 作者:xfy -package limitrate - -import ( - "io" - "sync" - "time" -) - -// RateLimitedWriter 限速写入器,使用令牌桶算法 -type RateLimitedWriter struct { - writer io.Writer - rate int64 // 字节/秒 - bucket int64 // 当前令牌数 - maxBucket int64 // 令牌桶最大容量 - lastTime time.Time // 上次更新时间 - mu sync.Mutex -} - -