主页 / 问题 / 1485362
Accepted
firet
Asked:2023-01-11 04:56:23 +0000 UTC2023-01-11 04:56:23 +0000 UTC

如果花费的时间太长，则中断 goroutine 函数

  • 772

任务，在函数中，需要运行几个带计算的goroutines等待它们完成，但是如果函数运行超过指定的秒数，中断它的执行

最小的例子

package main

import (
"fmt"
"sync"
)

func FibonacciRecursion(n int) int {
    if n <= 1 {
        return n
    }
    return FibonacciRecursion(n-1) + FibonacciRecursion(n-2)
}

func f(i int, wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Println(FibonacciRecursion(45 + i))
}

func fun() {
    var wg sync.WaitGroup
    for i := 0; i < 3; i++ {
        go f(i, &wg)
       wg.Add(1)
    }
    wg.Wait()
    println("Function ended")
}

func main() {
    fun()
}

需要中断fun函数的执行，怎么办？

golang

1 个回答

  1. Best Answer

    好吧， context包就是用于此类目的。

    这里在俄语中有详细的了解Golang中的上下文包

    package main

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

func main() {
    timeLimit := time.Second * 10
    ctx, cancel := context.WithTimeout(context.Background(), timeLimit)
    defer cancel()
    fun(ctx)
}

func isDone(ctx context.Context) bool {
    select {
    case <-ctx.Done():
        return true
    default:
        return false
    }
}

func FibonacciRecursion(ctx context.Context, n int) int {
    if n <= 1 || isDone(ctx) {
        return n
    }
    return FibonacciRecursion(ctx, n-1) + FibonacciRecursion(ctx, n-2)
}

func f(ctx context.Context, i int, wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Println(FibonacciRecursion(ctx, 45+i))
}

func fun(ctx context.Context) {
    var wg sync.WaitGroup
    for i := 0; i < 3; i++ {
        wg.Add(1)
        go f(ctx, i, &wg)
    }
    wg.Wait()
    println("Function ended")
}
    • 2

相关问题