如何停止多个 go 例程

Question

我必须从循环中调用戈鲁图因/线程。由于循环，有许多 go 例程并行执行。如果任何例程/线程成功执行，那么我必须停止所有其他线程/例程。有没有办法实现这个？

Answer 1

除了Burak回答的上下文之外，您还可以使用退出频道。

package main

import (

    "fmt"

    "math/rand"

    "time"

)

func foo(channel, quit chan string, i int) {

    channel <- fmt.Sprintf("goroutine %d started!", i)

    for {

        rand.Seed(time.Now().UnixNano())

        time.Sleep(time.Duration(rand.Intn(500)+500) * time.Millisecond)

        quit <- fmt.Sprintf("goRoutine %d completed!", i)

    }

}

func main() {

    channel := make(chan string)

    quit := make(chan string)

    for i := 0; i < 3; i++ {

        go foo(channel, quit, i)

    }

    

    for {

        select {

        case update:= <-channel:

            fmt.Println(update)

        case quit:= <-quit:

            fmt.Println(quit)

            return

        }

    }

}

Answer 2

您可以使用上下文：

ctx, cancel:= context.WithCancel(context.Background())

for ... {

   go func() {

      defer cancel() // cancel context once this goroutine ends

      doStuff(ctx)

   }()

}

您必须在戈鲁丁中检查上下文取消：

func doStuff(ctx context.Context) {

   ...

   if ctx.Err()!=nil {

       // Canceled, return

       return

   }

  ...

}

这并不能保证一旦上下文被取消，其他戈鲁廷将立即结束，但它保证，如果您正确检查上下文取消，所有戈鲁廷最终都会结束。

Answer 3

您可以使用上下文和错误组。一些类似的东西...

package main

import (

    "context"

    "fmt"

    "math/rand"

    "os"

    "time"

    "golang.org/x/sync/errgroup"

)

func doStuff(ctx context.Context, i int) error {

    count := 0

    for ctx.Err() == nil {

        // do the stuff

        time.Sleep(time.Millisecond * time.Duration(rand.Intn(500)))

        count++

        if count > 6 { // error condition

            fmt.Fprintf(os.Stdout, "%v reached count %v\n", i, count)

            return fmt.Errorf("Error %v, count %v\n", i, count)

        }

    }

    fmt.Fprintf(os.Stdout, "Killed %v @ count %v\n", i, count)

    return ctx.Err()

}

func main() {

    rand.Seed(int64(time.Now().Nanosecond()))

    ctxWc, _ := context.WithCancel(context.Background())

    g, ctx := errgroup.WithContext(ctxWc)

    for i := 0; i < 5; i++ {

        i := i

        g.Go(func() error {

            return doStuff(ctx, i)

        })

    }

    err := g.Wait()

    fmt.Println("The End")

    if err != nil {

        fmt.Println(err)

    }

}