go - 第一个完成时如何安全地绕过其他 goroutines 的结果

Question

我想向多个服务器请求数据(例如多个只读副本)。在这个任务中最重要的是速度，所以应该提供第一个结果其他的都可以忽略。

我对绕过这些数据的惯用方法有疑问。一切这个问题在它退出时没问题(所有较慢的 goroutines 都不是完成他们的工作，因为主要过程存在)。但是当我们取消注释时最后一行(带 Sleep)我们可以看到其他 goroutines 也在做他们的工作。

现在我正在通过 channel 推送数据，有什么办法可以不推送它们吗？

处理此类问题的安全方法是什么？

package main

import (
    "fmt"
    "log"
    "math/rand"
    "time"
)

type Result int

type Conn struct {
    Id int
}

func (c *Conn) DoQuery(params string) Result {
    log.Println("Querying start", params, c.Id)
    time.Sleep(time.Duration(rand.Int31n(1000)) * time.Millisecond)
    log.Println("Querying end", params, c.Id)

    return Result(1000 + c.Id*c.Id)
}

func Query(conns []Conn, query string) Result {
    ch := make(chan Result)
    for _, conn := range conns {
        go func(c Conn) {
            ch <- c.DoQuery(query)
        }(conn)
    }

    return <-ch
}

func main() {
    conns := []Conn{Conn{1}, Conn{2}, Conn{3}, Conn{4}, Conn{5}}
    result := Query(conns, "query!")
    fmt.Println(result)
    // time.Sleep(time.Minute)
}

Answer 1

我的建议是让 ch 成为每个查询一个空格的缓冲 channel :ch := make(chan Result, len(conns))。这样每个查询都可以运行完成，并且不会阻塞 channel 写入。

Query 可以读取一次并返回第一个结果。当所有其他 goroutines 完成时， channel 最终将被垃圾收集，一切都会消失。使用无缓冲 channel ，您可以创建许多永远不会终止的 goroutine。

编辑:如果你想取消飞行中的请求，它会变得更加困难。一些操作和 api 提供取消，而另一些则不提供。对于 http 请求，您可以在请求结构中使用 Cancel 字段。只需提供一个您可以关闭以取消的 channel :

func (c *Conn) DoQuery(params string, cancel chan struct{}) Result {
    //error handling omitted. It is important to handle errors properly. 
    req, _ := http.NewRequest(...)
    req.Cancel = cancel
    resp, _ := http.DefaultClient.Do(req)
    //On Cancellation, the request will return an error of some kind.
    return readData(resp)
}
func Query(conns []Conn, query string) Result {
    ch := make(chan Result)
    cancel := make(chan struct{})
    for _, conn := range conns {
        go func(c Conn) {
            ch <- c.DoQuery(query,cancel)
        }(conn)
    }

    first := <-ch
    close(cancel)
    return first
}

如果有大量您不关心的读取请求，这可能会有所帮助，但它可能会或可能不会真正取消远程服务器上的请求。如果您的查询不是 http，而是数据库调用或其他查询，您将需要研究是否可以使用类似的取消机制。