在 Go 语言中用限流算法保护你的后端服务

后端

2024-01-19 19:44:27

** 限流算法简介及 Go 语言示例 **

** 理解限流算法 **

限流算法是一种策略，它限制客户端在给定时间内向服务器发出的请求数量。它的目的是防止服务因过载而崩溃，并确保服务的稳定性。

** 限流算法类型 **

令牌桶算法： 将请求视为令牌，并以恒定的速率向桶中添加令牌。当客户端发送请求时，算法会检查桶中是否有可用令牌。如果没有令牌，则请求会被丢弃。
漏桶算法： 与令牌桶类似，但桶有一个泄漏速率。当客户端发送请求时，算法会检查桶中是否有空间。如果没有空间，则请求会被丢弃。
信号量： 一种简单高效的限流算法。信号量会跟踪可用的资源数量，当客户端发送请求时，算法会减少信号量。当信号量为零时，则请求会被丢弃。

** Go 语言中实现限流算法 **

Go 语言提供了一个名为 golang.org/x/time/rate 的库来实现限流算法。该库提供了以下方法：

NewLimiter：创建一个限流器，它会以恒定的速率添加令牌到桶中。
Wait：阻塞直到桶中有可用的令牌。
Allow：检查桶中是否有可用的令牌，如果有，则返回 true，否则返回 false。

** Go 语言示例 **

package main

import (
    "context"
    "fmt"
    "log"
    "time"

    "golang.org/x/time/rate"
)

func main() {
    // 创建一个限流器，以每秒 10 个令牌的速率添加令牌。
    limiter := rate.NewLimiter(rate.Every(time.Second), 10)

    // 模拟处理请求的 goroutine。
    for i := 0; i < 100; i++ {
        go func(i int) {
            // 尝试从桶中获取令牌。
            if err := limiter.Wait(context.Background()); err != nil {
                log.Printf("请求 %d 被限流", i)
                return
            }

            // 处理请求。
            fmt.Printf("处理请求 %d\n", i)
        }(i)
    }

    // 等待所有 goroutine 完成。
    time.Sleep(10 * time.Second)
}