Go で単純な並行処理の使用例があるのですが、問題に対するエレガントな解決策が見つかりません。
リモート サーバーから不特定多数のリソースを並行してクエリするメソッドを作成したいと考えていますfetchAll。いずれかのフェッチが失敗した場合は、最初のエラーをすぐに返します。
私の最初の実装では、goroutine がリークします。
package main
import (
"fmt"
"math/rand"
"sync"
"time"
)
func fetchAll() error {
wg := sync.WaitGroup{}
errs := make(chan error)
leaks := make(map[int]struct{})
defer fmt.Println("these goroutines leaked:", leaks)
// run all the http requests in parallel
for i := 0; i < 4; i++ {
leaks[i] = struct{}{}
wg.Add(1)
go func(i int) {
defer wg.Done()
defer delete(leaks, i)
// pretend this does an http request and returns an error
time.Sleep(time.Duration(rand.Intn(100)) * time.Millisecond)
errs <- fmt.Errorf("goroutine %d's error returned", i)
}(i)
}
// wait until all the fetches are done and close the error
// channel so the loop below terminates
go func() {
wg.Wait()
close(errs)
}()
// return the first error
for err := range errs {
if err != nil {
return err
}
}
return nil
}
func main() {
fmt.Println(fetchAll())
}
遊び場：https://play.golang.org/p/Be93J514R5
読んで分かったhttps://blog.golang.org/pipelines他のスレッドをクリーンアップするためのシグナル チャネルを作成できます。 あるいは、 を使用することもできますcontext。 しかし、このような単純なユース ケースには、私が見逃しているより単純なソリューションがあるはずです。
ベストアンサー1
使用エラーグループこれをさらに簡単にします。これにより、提供されたすべての Go ルーチンが正常に完了するまで自動的に待機するか、いずれかのルーチンがエラーを返した場合には残りのルーチンをすべてキャンセルします (この場合、そのエラーは呼び出し元に返される唯一のエラーです)。
package main
import (
"context"
"fmt"
"math/rand"
"time"
"golang.org/x/sync/errgroup"
)
func fetchAll(ctx context.Context) error {
errs, ctx := errgroup.WithContext(ctx)
// run all the http requests in parallel
for i := 0; i < 4; i++ {
errs.Go(func() error {
// pretend this does an http request and returns an error
time.Sleep(time.Duration(rand.Intn(100)) * time.Millisecond)
return fmt.Errorf("error in go routine, bailing")
})
}
// Wait for completion and return the first error (if any)
return errs.Wait()
}
func main() {
fmt.Println(fetchAll(context.Background()))
}