go 使用redis 完成分布式限流

原创

mob64ca12e4d52e 2024-07-02 05:52:51 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e4d52e的原创作品，请联系作者获取转载授权，否则将追究法律责任

实现“go 使用redis 完成分布式限流”

简介

在分布式系统中，限流是一种非常重要的技术，可以保护系统免受突发流量的冲击。在本文中，我将向你展示如何使用Go语言和Redis来实现分布式限流。

流程图

erDiagram
    LIMIT_FLOW {
        用户ID,
        时间戳,
    }

状态图

stateDiagram
    [*] --> 未通过限流: 未超出限流阈值
    未通过限流 --> [*]: 通过限流

实现步骤

下面是实现分布式限流的步骤：

步骤	描述
1	检查Redis中是否存在该用户ID的限流记录
2	如果不存在限流记录，则将用户ID和当前时间戳存入Redis，并设置过期时间
3	如果存在限流记录，判断当前时间与上次限流的时间间隔
4	如果时间间隔小于设定的限流时间间隔，则限流失败，否则更新限流记录

代码实现

package main

import (
    "fmt"
    "time"

    "github.com/go-redis/redis/v8"
)

var (
    ctx = context.Background()
    client = redis.NewClient(&redis.Options{
        Addr: "localhost:6379",
        Password: "",
        DB: 0,
    })
    limitKey = "limit_flow"
    limitDuration = 1 * time.Second
)

func limitFlow(userID string) bool {
    now := time.Now().Unix()
    res, err := client.Get(ctx, limitKey+":"+userID).Result()
    if err == redis.Nil {
        err := client.Set(ctx, limitKey+":"+userID, now, limitDuration).Err()
        if err != nil {
            panic(err)
        }
        return true
    }
    lastTime, _ := strconv.Atoi(res)
    if now - lastTime > limitDuration.Seconds() {
        err := client.Set(ctx, limitKey+":"+userID, now, limitDuration).Err()
        if err != nil {
            panic(err)
        }
        return true
    }
    return false
}

func main() {
    userID := "123"
    if limitFlow(userID) {
        fmt.Println("通过限流")
    } else {
        fmt.Println("未通过限流")
    }
}