Golang限制请求频率 + 接受超时信号后停止发送请求

直接跑一下看看呢

package main

import (
    "context"
    "fmt"
    "strconv"
    "sync"
    "time"
)

func main() {
    timeCtx, cancle := context.WithTimeout(context.Background(), time.Second*10)
    defer cancle()
    Test(timeCtx)
}

func Test(ctx context.Context) {
    finish := make(chan bool)
    list := []string{}
    for i := 0; i < 100000; i++ {
        list = append(list, strconv.Itoa(i))
    }
    taskList := make(chan string, len(list))
    // 为啥要close管道-因为不close下面的range是不会出来的
    defer close(taskList)
    for _, v := range list {
        taskList <- v
    }
    timeOut := make(chan bool)
    go func() {
        var wg sync.WaitGroup
        wg.Add(len(list))
        // 限频3个消费者
        for i := 0; i < 100; i++ {
            // 起3个协程去跑
            go func() {
                for v := range taskList {
                    select {
                    case <-timeOut:
                        func() {
                            defer wg.Done()
                        }()

                    default:
                        func() {
                            defer wg.Done()
                            ajax(v)
                        }()
                    }
                }
            }()
        }
        wg.Wait()
        fmt.Println(51)
        // 全部执行完成
        finish<-true
    }()
    fmt.Println(60)
    select {
    case <-ctx.Done():
        close(timeOut)
        // 超时
    case <-finish:
        // 执行完成
        fmt.Println("222")
        close(timeOut)        
    }
    // done
}

func ajax(v string) {
    time.Sleep(time.Second * 1)
    fmt.Println(v)
}

 

以上用于组装api接口请求

在最顶层的 timeout 抛出以后，下层会立即返回结果，并且终止待请求任务的发送；

通过消费者数量限定并行请求的数量 ；

实在没能想到其他更简单一些的方式、以及代码更简单清晰的方式 ～

context对象应用于多个层级的子协程，我这边的是一个按照配置的走的无限级递归调用的方式~

 

TRANSLATE with x

English

Arabic	Hebrew	Polish
Bulgarian	Hindi	Portuguese
Catalan	Hmong Daw	Romanian
Chinese Simplified	Hungarian	Russian
Chinese Traditional	Indonesian	Slovak
Czech	Italian	Slovenian
Danish	Japanese	Spanish
Dutch	Klingon	Swedish
English	Korean	Thai
Estonian	Latvian	Turkish
Finnish	Lithuanian	Ukrainian
French	Malay	Urdu
German	Maltese	Vietnamese
Greek	Norwegian	Welsh
Haitian Creole	Persian

 

TRANSLATE with

COPY THE URL BELOW

Back

EMBED THE SNIPPET BELOW IN YOUR SITE

Enable collaborative features and customize widget: Bing Webmaster Portal

Back