mirror of
https://github.com/xiaoqidun/limit.git
synced 2025-10-11 06:20:24 +08:00
69 lines
2.1 KiB
Go
69 lines
2.1 KiB
Go
// Copyright 2025 肖其顿
|
||
//
|
||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
// you may not use this file except in compliance with the License.
|
||
// You may obtain a copy of the License at
|
||
//
|
||
// http://www.apache.org/licenses/LICENSE-2.0
|
||
//
|
||
// Unless required by applicable law or agreed to in writing, software
|
||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
// See the License for the specific language governing permissions and
|
||
// limitations under the License.
|
||
|
||
package limit_test
|
||
|
||
import (
|
||
"fmt"
|
||
"time"
|
||
|
||
"github.com/xiaoqidun/limit"
|
||
"golang.org/x/time/rate"
|
||
)
|
||
|
||
// ExampleLimiter 演示了 limit 包的基本用法。
|
||
func ExampleLimiter() {
|
||
// 创建一个使用默认配置的 Limiter 实例
|
||
limiter := limit.New()
|
||
// 程序退出前，优雅地停止后台任务，这非常重要
|
||
defer limiter.Stop()
|
||
// 为一个特定的测试键获取一个速率限制器
|
||
// 限制为每秒2个请求，最多允许3个并发（桶容量）
|
||
testKey := "testKey"
|
||
rateLimiter := limiter.Get(testKey, rate.Limit(2), 3)
|
||
// 模拟连续的请求
|
||
for i := 0; i < 5; i++ {
|
||
if rateLimiter.Allow() {
|
||
fmt.Printf("请求 %d: 已允许\n", i+1)
|
||
} else {
|
||
fmt.Printf("请求 %d: 已拒绝\n", i+1)
|
||
}
|
||
time.Sleep(100 * time.Millisecond)
|
||
}
|
||
// 手动移除一个不再需要的限制器
|
||
limiter.Del(testKey)
|
||
// Output:
|
||
// 请求 1: 已允许
|
||
// 请求 2: 已允许
|
||
// 请求 3: 已允许
|
||
// 请求 4: 已拒绝
|
||
// 请求 5: 已拒绝
|
||
}
|
||
|
||
// ExampleNewWithConfig 展示了如何使用自定义配置。
|
||
func ExampleNewWithConfig() {
|
||
// 自定义配置
|
||
config := limit.Config{
|
||
ShardCount: 64, // 分片数量，必须是2的幂
|
||
GCInterval: 5 * time.Minute, // GC 检查周期
|
||
Expiration: 15 * time.Minute, // 限制器过期时间
|
||
}
|
||
// 使用自定义配置创建一个 Limiter 实例
|
||
customLimiter := limit.NewWithConfig(config)
|
||
defer customLimiter.Stop()
|
||
fmt.Println("使用自定义配置的限制器已成功创建")
|
||
// Output:
|
||
// 使用自定义配置的限制器已成功创建
|
||
}
|