结束持续运行时间较长的 goroutine 的方法

在长时间运行的循环任务中，为了在任意一个任务失败时取消所有任务，可以将每个任务封装在管理器任务中。管理器任务生成实际的长时任务，并返回等待上下文以取消由于循环中其他任务错误而引起的取消。但是，由于实际任务没有引用父级的可取消情况，取消单个管理任务是否也会中止生成的长时间任务？本文旨在探讨这个问题，并通过代码示例提供一个明确的答案。

我想在循环中执行一组长时间运行的任务，并在任何一个失败时取消所有这些任务。为此，我将每个任务包装在一个管理器任务中，该任务生成实际的长时间运行的任务，然后返回等待上下文由于循环中任何其他任务中的错误而被取消。

我的问题是：“考虑到实际任务没有引用父级中的可取消上下文，取消单个管理器任务是否也会终止生成的长时间运行任务？”

下面的“工作”代码应该可以使我的问题清楚。我了解 readwithcancel 的所有实例都将返回，但每个相应的 readcsv 子项都会因此终止。

一如既往的感谢

func main() {
    files := []string{"../data/file1.csv", "../data/file2.csv", "../data/file3.csv"}

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

    var wg sync.WaitGroup
    for _, file := range files {
        wg.Add(1)
        fileLocal := file
        go func() {
            defer wg.Done()
            if err := readWithCancel(ctx, fileLocal); err != nil {
                fmt.Println(fileLocal, err)
                cancel()
            }
        }()
    }
    wg.Wait()
}

func readWithCancel(ctx context.Context, file string) error {

    ch := make(chan error)

    go func() {
        ch <- readCSV(file)
    }()

    select {
    case err := <-ch:
        return err
    case <-ctx.Done():
        return ctx.Err()
    }
}

func readCSV(file string) error {

    f, err := os.Open(file)
    if err != nil {
        return fmt.Errorf("opening file %w", err)
    }

    cr := csv.NewReader(f)
    linenum := 0
    for {
        record, err := cr.Read()
        if err != nil {
            if errors.Is(err, io.EOF) {
                return nil
            }
            return err
        }

        fmt.Printf("%s: %s\n", file, record)
        linenum++
    }
}

正确答案

简短的回答是“不”。 Goroutine 之间没有内在的关系（父子关系等）。 Goroutine 会一直运行，直到其顶级函数（go 语句中指定的函数）终止或程序终止。

在您提供的示例代码中，错误将导致程序终止，因为取消上下文将导致您正在等待的 goroutine 终止，从而导致 wg.Wait 返回。当发生这种情况时，运行 readCSV 的 goroutine 将随程序终止，但我认为这不是您正在寻找的解决方案。

到这里，我们也就讲完了《结束持续运行时间较长的 goroutine 的方法》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！