Go 外部 API 接口的固定并发请求

package main

import (
    "fmt"
    "net/http"
    "sync"
    "time"
)

// 等待组
var wg = sync.WaitGroup{}

// 互斥锁
var mux sync.Mutex

// 同时并发数
var channelNum = 100

// 发送请求总数
var sendNum = 10000

// 成功数
var successNum = 0

// 失败数
var failNum = 0

// 请求的网址
var url = "https://tools.hongfs.cn/v2/ip"

func main() {
    // 创建带缓存的通道，缓存数是并发数
    ch := make(chan bool, channelNum)

    for i := 0; i < sendNum; i++ {
        wg.Add(1)

        go get(ch, int64(i))
    }

    // 阻塞
    wg.Wait()
}

func get(ch chan bool, i int64) {
    // 返回时，等待组计数自减
    defer wg.Done()

    // 发送到通道，如果满了那这里会阻塞
    ch <- true

    client := http.Client{
        Timeout: 1 * time.Second, // 请求超时
    }

    resp, err := client.Get(url)

    // 请求没有 err 时需要关闭连接
    if err == nil {
        defer resp.Body.Close()
    }

    // 锁一下，避免成功失败数的统计有问题
    mux.Lock()

    defer mux.Unlock()

    if err != nil {
        // 请求失败
        failNum++
    } else if resp.StatusCode != 200 {
        // 请求状态码异常
        failNum++
    } else {
        // 成功请求
        successNum++
    }

    fmt.Printf("I: %d, 成功数: %d，失败数：%d\n", i+1, successNum, failNum)

    // 释放当前的通道
    <-ch
}