在Go语言中如何解决并发网络请求的请求流量控制和限流问题？

package main

import (
    "fmt"
    "sync"
    "time"
)

type TokenBucket struct {
    capacity  int            // 令牌桶的容量
    rate      int            // 每秒钟产生的令牌数量
    timeUnit  time.Duration  // 令牌产生的时间间隔
    available int            // 当前可用令牌数量
    mu        sync.Mutex     // 互斥锁
}

func NewTokenBucket(capacity, rate int, timeUnit time.Duration) *TokenBucket {
    return &TokenBucket{
        capacity:  capacity,
        rate:      rate,
        timeUnit:  timeUnit,
        available: capacity,
    }
}

func (tb *TokenBucket) getToken() bool {
    tb.mu.Lock()
    defer tb.mu.Unlock()
    now := time.Now()
    // 计算令牌产生的数量
    delta := int(now.Sub(tb.lastTime) / tb.timeUnit) * tb.rate
    // 更新上次令牌产生的时间
    tb.lastTime = now
    // 重新计算当前可用令牌数量
    tb.available = tb.available + delta
    if tb.available > tb.capacity {
        tb.available = tb.capacity
    }
    if tb.available < 1 {
        return false
    }
    // 使用一个令牌
    tb.available--
    return true
}

func main() {
    // 创建一个容量为100，每秒钟产生10个令牌的令牌桶
    tb := NewTokenBucket(100, 10, time.Second)

    // 模拟1000个并发请求
    var wg sync.WaitGroup
    for i := 0; i < 1000; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            // 判断是否可以获取令牌
            if tb.getToken() {
                // 执行网络请求
                fmt.Println("执行网络请求")
            } else {
                // 请求被拒绝
                fmt.Println("请求被限制")
            }
        }()
    }

    wg.Wait()
}