Redis_Rate 限速器使用指南

最新推荐文章于 2024-08-20 20:04:06 发布

卫标尚

最新推荐文章于 2024-08-20 20:04:06 发布

阅读量319

点赞数 5

本文链接：https://blog.csdn.net/gitblog_00956/article/details/141078624

版权

Redis_Rate 限速器使用指南

redis_rateRate limiting for go-redis项目地址:https://gitcode.com/gh_mirrors/re/redis_rate

1. 项目介绍

redis_rate 是一个基于 Go 语言和 Redis 的率限制库，它实现了 GCRA（也称为漏桶算法）以进行速率限制。此项目依赖于 Redis 3.2 或更高版本的 replicate_commands 功能。它主要用于帮助开发者限制应用程序中的请求频率，从而防止滥用或保护服务稳定性。

2. 项目快速启动

首先，确保你的开发环境已经安装了 Go 和 Redis。接下来，初始化一个新的 Go 模块并导入 redis_rate 库：

go mod init example.com/myapp
go get github.com/go-redis/redis_rate/v9

以下是一个简单的示例，展示了如何创建一个限速器以及如何检查是否超过限制：

package main

import (
	"context"
	"fmt"
	"github.com/go-redis/redis/v8"
	"github.com/go-redis/redis_rate/v9"
)

func main() {
	rdb := redis.NewClient(&redis.Options{
		Addr: "localhost:6379",
	})
	limiter := redis_rate.NewLimiter(rdb)
	res, err := limiter.Allow(context.Background(), "project:123", redis_rate.PerSecond(10))
	if err != nil {
		panic(err)
	}
	fmt.Printf("allowed: %t, remaining: %d\n", res.Allowed, res.Remaining)
}

运行上述代码，你会看到允许的请求结果以及剩余的请求数量。

3. 应用案例和最佳实践

在实际应用中，可以将 redis_rate 集成到 HTTP 请求处理流程中，例如使用 BunRouter:

func rateLimit(next bunrouter.HandlerFunc) bunrouter.HandlerFunc {
	return func(w http.ResponseWriter, req bunrouter.Request) error {
		res, err := limiter.Allow(req.Context(), "project:123", redis_rate.PerMinute(10))
		if err != nil {
			return err
		}

		h := w.Header()
		h.Set("RateLimit-Remaining", strconv.Itoa(res.Remaining))
		if res.Allowed == 0 {
			// 我们被限速了
			seconds := int(res.RetryAfter / time.Second)
			h.Set("RateLimit-RetryAfter", strconv.Itoa(seconds))
			// 停止进一步处理请求
			return http.ErrTooManyRequests
		}

		return next(w, req)
	}
}

这个中间件会在每次请求时检查速率限制，并在响应头中提供剩余请求数及重试时间。