go - 上下文因超时而取消但计算未中止？

Question

试图了解 go 上下文取消将如何中止后续代码的执行

实验细节：

1. main func 有一个超时的上下文2sec
2. main funcsum在单独的 go-routine 中调用另一个 func - 它1sec为 test-run-1 和4sec为 test-run-2休眠
3. 让 main sleep3sec让 spun go-routine 完成执行

package main

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

func main() {

    c := context.Background()
    childCtx, cancel := context.WithTimeout(c, 2*time.Second)
    defer cancel()

    ch := make(chan int, 1)
    go sum(5, 6, ch)

    var msg string

    select {
    case <-childCtx.Done():
        msg = "return from ctx done channel"
    case res := <-ch:
        msg = fmt.Sprintf("return from go routine: %v", res)
    }

    log.Print(msg)

    time.Sleep(3 * time.Second) //sleeping here to test if go-routine is still running
}


func sum(x int, y int, c chan<- int) {
    time.Sleep(1 * time.Second) 
    //testcase-1: sleep - 1s
    //testcase-2: sleep - 4s

    result := x + y

    log.Printf("print from sum fn: %v", result)

    c <- result
}

testcase-1 的响应：睡眠总和功能 1 秒：

2021/04/12 01:06:58 print from sum fn: 11
2021/04/12 01:06:58 return from go routine: 11

testcase-2 的响应：睡眠总和功能 4 秒：

2021/04/12 01:08:25 return from ctx done channel
2021/04/12 01:08:27 print from sum fn: 11

在 testcase-2 中，当 sum func 休眠 4 秒时，上下文已经在 2 秒后被超时取消，为什么它仍然在 diff go-routine 中执行 sum func 并打印print from sum fn: 1？

来自文档：取消此上下文会释放与其关联的资源。

我的假设是所有计算将在 2 秒后立即中止，包括旋转的 go-routine

让我知道如何正确执行此操作，提前谢谢

Answer 1

正如@AndySchweig 所指出的，context发出取消事件的信号，但不强制取消。在检测到取消后，任何可能阻塞的 goroutine 都会尽最大努力尝试取消/清理。

要更新您的sum功能以支持取消，您可以尝试：

// add context parameter as the first argument
// add a return error - to indicate any errors (i.e. function was interrupted due to cancelation)
func sum(ctx context.Context, x int, y int, c chan<- int) (err error) {

    wait := 1 * time.Second // testcase-1
    //wait := 4 * time.Second // testcase-2

    // any blocking called - even sleeps - should be interruptible
    select {
    case <-time.After(wait):
    case <-ctx.Done():
        err = ctx.Err()
        return
    }

    result := x + y

    log.Printf("print from sum fn: %v", result)

    select {
    case c <- result:
    case <-ctx.Done(): // check for ctx cancelation here - as no one may be listening on result channel
        err = ctx.Err()
    }
    return
}

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

Answer 2

上下文不会中止 go 例程。在您的情况下，如果上下文的时间到了，您就不会打印 go 例程的结果。go 例程对上下文一无所知。