Go 函数并发编程中的限流与负载均衡

在分布式系统中，并发请求可能会导致服务器过载。为了避免这种情况，可以使用限流和负载均衡技术来管理并发请求。

限流

限流是一种技术，用于限制传入请求的数量。它可以防止服务器被过多的请求压垮。在 Golang 中，可以使用 [rate](https://godoc.org/golang.org/x/time/rate) 包来实现限流。

立即学习“go语言免费学习笔记（深入）”；

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

以下示例展示了如何使用 rate 包实现令牌桶算法的限流器：

package main

import (
    "context"
    "fmt"
    "log"
    "net/http"
    "time"

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

func main() {
    // 每秒允许 100 个请求
    limiter := rate.NewLimiter(rate.Limit(100), 100)

    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        ctx := context.Background()

        // 从限流器获取令牌
        if err := limiter.Wait(ctx); err != nil {
            http.Error(w, "Too many requests", http.StatusTooManyRequests)
            return
        }

        fmt.Fprintf(w, "Hello, World!")
    })

    log.Fatal(http.ListenAndServe(":8080", nil))
}

负载均衡

负载均衡是一种技术，用于将请求分布到多个服务器。它可以提高系统的可靠性和可扩展性。在 Golang 中，可以使用 httputil 和 reverseproxy 包来实现负载均衡。

以下示例展示了如何使用 httputil 和 reverseproxy 包实现反向代理的负载均衡器：

package main

import (
    "log"
    "net/http"
    "net/http/httputil"
    "net/url"
)

func main() {
    // 目标服务器的 URL 列表
    targets := []string{
        "http://localhost:8081",
        "http://localhost:8082",
    }

    // 创建反向代理 Director
    director := func(req *http.Request) {
        // 轮询选择目标服务器
        target := targets[len(targets)%len(targets)]

        // 设置请求的远端地址
        req.URL.Host = target
        req.URL.Scheme = "http"
    }

    // 创建反向代理
    proxy := httputil.NewSingleHostReverseProxy(&url.URL{Scheme: "http", Host: "localhost:8080"})
    proxy.Director = director

    log.Fatal(http.ListenAndServe(":8080", proxy))
}