如何使用redis做限流(golang实现小样)

在实际开发中,限流(Rate Limiting)是一种保护服务、避免接口被恶意刷流的常见技术。常用的限流算法有令牌桶、漏桶、固定窗口、滑动窗口等。由于Redis具备高性能和原子性操作,常常被用来实现分布式限流。

下面给出使用Golang结合Redis实现简单限流的几种常见方式(以“固定窗口计数”和“滑动窗口”为例)。使用的Go Redis库为go-redis


1. 固定窗口计数(Fixed Window Counter)

在单位时间窗口(如1秒、1分钟)内计数,超过阈值则限流。

伪代码思路:

  1. 用Redis的INCR自增某个key记录当前窗口内的访问次数。
  2. 使用EXPIRE设置Key过期时间为窗口大小。
  3. 如果计数超过阈值,说明被限流。

Go代码示例:

package main

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

var ctx = context.Background()

func FixedWindowRateLimit(client *redis.Client, key string, limit int, window time.Duration) (bool, error) {
   
    // 每次请求自增计数
    cnt, err := client.Incr(ctx, key).Result()
    if err != nil {
   
       
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值