使用上下文停止执行简单函数

Question

据我了解，上下文库的主要目的是停止执行所有子级的 RPC 和 HTTP 请求。或者用简单的语言，我们可以说在上下文库的帮助下，我们可以将触发器传递给函数调用。

这是我想使用上下文解决的抽象形式的问题。

package main

import (
    "context"
    "fmt"
    "time"
)

func PrintAndStop(ctx context.Context, numberOfTimePrint int, msg string) {
    communicationChannel := make(chan bool)
    go func() {
        for i := 0; i < numberOfTimePrint; i++ {
            fmt.Println(msg)
            time.Sleep(time.Second)
        }
        communicationChannel <- true
    }()
    select {
    case <-communicationChannel:
        fmt.Println("process has been completed")
    case <-ctx.Done():
        fmt.Println("closing call from the main function")
    }
}

func main() {
    ctx := context.Background()
    ctx, cancel := context.WithTimeout(ctx, 3*time.Second)
    defer cancel()
    PrintAndStop(ctx, 5*time.Second, "testing")

    time.Sleep(5 * time.Second)
}

以下代码的输出是：

testing
testing
testing
closing call from the main function
testing
testing

此处提供以下代码 sn-p 的链接：https://play.golang.org/p/4Ntwn3wYOiT PrintAndStop 是一种以 1 秒的间隔打印一些消息给定次数的方法。 在上下文库的帮助下，我想控制主函数以随时停止执行PrintAndStop。 main末尾的time.Sleep(5 * time.Second)只是表示main函数在函数调用后没有结束。

我知道以下重构，但实际问题不能这样重构。

func PrintAndStop(ctx context.Context, numberOfTimePrint int, msg string) {
    completed := true
    for i := 0; i < numberOfTimePrint; i++ {
        select {
        case <-time.After(1 * time.Second):
            fmt.Println(msg)
        case <-ctx.Done():
            fmt.Println("closing call from the main function")
            completed = false
            break
        }
    }
    if completed {
        fmt.Println("function execution is completed")
    }
}

注意：我很乐意使用 context 库的一些扩展或 Go 中一些全新的库来解决上述问题。

Answer 1

我想控制主函数以随时停止执行PrintAndStop

然后你需要检查上下文的取消在那个循环中：

    go func() {
        for i := 0; i < int(timeout/time.Second); i++ {
            select {
            case <-ctx.Done():
                return
            }
            fmt.Println(msg)
            time.Sleep(time.Second)
        }
        communicationChannel <- true
    }()