go - 如何在Go中对多个网址进行长轮询？

Question

到目前为止，这是我所拥有的:

package main

import (
    "encoding/json"
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
)

func subscribe(urls Urls) []byte {
    req, err := http.NewRequest("GET", urls.Url, nil)
    if err != nil {
        log.Fatal(err)
    }

    req.Header.Set("authentication", "Bearer " + urls.Token)

    http_client := &http.Client{}
    res, err := http_client.Do(req)
    if err != nil {
        log.Fatal(err)
    }

    defer res.Body.Close()

    resourceResp, err := ioutil.ReadAll(res.Body)
    if err != nil {
        log.Fatal(err)
    }

    fmt.Println(string(resourceResp))

    var data map[string]interface{}

    error := json.Unmarshal([]byte(resourceResp), &data)
    if error != nil {
        log.Fatal(error)
    }

    return subscribe(urls)
}

type Urls struct {
    Url string
    Token string
}

func main() {
    var urls [2]Urls

    urls[0] = Urls{
        Url: "https://example.com/users/8",
        Token: "abcdefg",
    }

    urls[1] = Urls{
        Url:   "https://example.com/users/9",
        Token: "hijklmnop",
    }

    subscribe(urls[0])
    subscribe(urls[1])
}

最终目标是“订阅”多个URL并提取所有更新的数据(最终将其添加到队列中，但一次只添加一个步骤)。之后，重新建立连接。现在，只有第一个订阅运行。谢谢!

Answer 1

我认为您要的是让subscribe函数并行运行。一种方法是将它们包装在goroutine中，然后等待所有goroutine完成:

func main() {
    ...
    ...
    var wg sync.WaitGroup

    wg.Add(len(urls))

    for _, url := range(urls) {
        go func() {
            defer wg.Done()
            subscribe(url)
        }()
    }

    wg.Wait()
}