go - 如何将 goroutine 包含到上下文中？

Question

我正在开发一个需要initFunction在单独的 goroutine 中调用启动函数 ( ) 的 Go 项目（以确保此函数不会干扰项目的其余部分）。initFunction 不得超过 30 秒，所以我想我会使用 context.WithTimeout。最后，initFunction必须能够将错误通知给调用者，所以我想到了创建一个错误通道并从匿名函数中调用 initFunction 来接收并报告错误。

func RunInitGoRoutine(initFunction func(config string)error) error {

    initErr := make(chan error)
    ctx, cancel := context.WithTimeout(context.Background(), 30*time.Seconds)

    go func() {
        <-ctx.Done()  // Line 7
        err := initFunction(config)
        initErr <-err
    }()

    select {
    case res := <-initErr:
        return res.err
    case <-ctx.Done():
        err := errors.New("Deadline")
    return err
    }
}

我对 Go 很陌生，所以我要求对上述代码提供反馈。

1. 我对第 7 行有一些疑问。我用它来确保匿名函数被“包含”在下面ctx，因此被杀死和释放，一旦超时到期，一切都将结束，但我不确定我是否做了正确的事情。
2. 第二件事是，我知道我应该打电话给cancel( )某个地方，但我无法确定在哪里。
3. 最后，真的欢迎任何反馈，无论是关于效率、风格、正确性或任何其他方面。

Answer 1

在 Go 中，实践是通过通道进行通信。所以最好的事情可能是在您的上下文中共享一个频道，以便其他人可以从该频道消费。

正如你所说的你是 Go 新手，我写了一大堆关于 Go（初学者级别）的文章https://marcofranssen.nl/categories/golang。

从旧到新阅读以熟悉该语言。

关于频道细节，你应该看看这篇文章。

https://marcofranssen.nl/concurrency-in-go

这篇博客文章中描述了一个 Web 服务器监听ctrl+c然后使用通道优雅地关闭服务器的实际示例。

https://marcofranssen.nl/improved-graceful-shutdown-webserver

本质上，我们在后台程序中运行服务器

go func() {
    if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
      srv.l.Fatal("Could not listen on", zap.String("addr", srv.Addr), zap.Error(err))
    }
  }()

然后我们有一些代码通过在通道上监听关闭信号来阻塞主程序。

quit := make(chan os.Signal, 1)

  signal.Notify(quit, os.Interrupt)
  sig := <-quit
  srv.l.Info("Server is shutting down", zap.String("reason", sig.String()))

  ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
  defer cancel()

  srv.SetKeepAlivesEnabled(false)
  if err := srv.Shutdown(ctx); err != nil {
    srv.l.Fatal("Could not gracefully shutdown the server", zap.Error(err))
  }
  srv.l.Info("Server stopped")

这与您的用例非常相似。因此，在后台例程中运行您的 init，然后使用等待此 init 例程结果的通道。

package main

import (
    "fmt"
    "time"
)

type InitResult struct {
    Message string
}

func main() {
    initResult := make(chan InitResult, 0)
    go func(c chan<- InitResult) {
        time.Sleep(5 * time.Second)
        // here we are publishing the result on the channel
        c <- InitResult{Message: "Initialization succeeded"}
    }(initResult)

    fmt.Println("Started initializing")

    // here we have a blocking operation consuming the channel
    res := <-initResult

    fmt.Printf("Init result: %s", res.Message)
}

https://play.golang.org/p/_YGIrdNVZx6

您还可以在结构上添加一个错误字段，以便您可以按照通常的方式进行错误检查。