golang 单个api限流

限流是什么

在高并发的系统中，限流是一种常用的手段，用于保护系统资源，提高系统的可靠性和稳定性。通过限制系统处理请求的速率，可以防止系统超载，避免系统崩溃或响应变慢。

Golang中的限流机制

Golang是一门强大的编程语言，它提供了丰富的标准库和第三方库，其中就包括了限流相关的工具。Golang中最常用的限流库是"golang.org/x/time/rate"，它是Go语言团队开发的库，提供了简单而灵活的限流功能。

使用rate库进行限流

首先，我们需要导入rate库：

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

接下来，我们可以创建一个限流器对象：

limiter := rate.NewLimiter(rate.Limit(10), 100)

上述代码中，"rate.Limit(10)"表示每秒最多处理10个请求，"100"表示漏桶最大容量为100个请求。

然后，我们可以使用限流器对象来控制请求的处理速率：

func handleRequest(w http.ResponseWriter, r *http.Request) {
    if limiter.Allow() {
        // 处理请求
    } else {
        // 请求被限流
    }
}

在上述代码中，我们首先调用limiter.Allow()函数来判断当前请求是否被允许处理。如果返回true，说明请求通过限流器的限制，可以继续处理；如果返回false，说明请求被限流。

限制每秒请求数

通过设置 rate.Limit(10)，我们可以限制每秒最多处理10个请求。这样做的好处是，可以避免系统被大量的请求同时冲击，导致系统负载过高。

limiter := rate.NewLimiter(rate.Limit(10), 100)

在上述代码中，第一个参数表示每秒最多处理10个请求，第二个参数表示漏桶的容量为100个请求。

限制并发请求数

除了限制每秒请求数，我们还可以通过设置漏桶的容量来限制系统的并发请求数。通过设置漏桶的容量，我们可以避免系统被大量并发请求压垮。

limiter := rate.NewLimiter(rate.Limit(10), 100)

在上述代码中，第一个参数表示每秒最多处理10个请求，第二个参数表示漏桶的容量为100个请求。当请求超过100个时，多余的请求数将被拒绝。

超过限流处理

当请求被限流时，我们可以采取一些措施来处理这种情况。例如，可以返回一个友好的错误信息给用户，或者将请求放入队列中等待处理。

func handleRequest(w http.ResponseWriter, r *http.Request) {
    if limiter.Allow() {
        // 处理请求
    } else {
        // 请求被限流，返回错误信息给用户
        w.WriteHeader(http.StatusTooManyRequests)
        fmt.Fprint(w, "Too many requests. Please try again later.")
    }
}

在上述代码中，我们使用w.WriteHeader()函数设置Http状态码为429，表示请求过多。然后，使用fmt.Fprint()函数向客户端返回错误信息。

总结

通过使用golang.org/x/time/rate库，我们可以轻松实现对API的限流功能。限流可以保护系统资源，提高系统的可靠性和稳定性。通过限制请求的处理速率，我们可以避免系统被大量请求同时冲击，导致系统负载过高。在限流时，我们可以采取一些措施来处理被限流的请求，例如返回友好的错误信息给用户或将请求放入队列中等待处理。