停止在goroutine中的所有递归函数

Question

凝视运行递归函数的goroutine，我想发送一个信号来停止这些递归函数。这是函数（功能并不重要）：

func RecursiveFunc(x int, depth int, quit chan bool) int { 

    if depth == 0 { 
     return 1 
    } 

    if quit != nil { 
     select { 
     case <-quit: 
      return 0 
     default: 
     } 
    } 

    total := 0 

    for i := 0; i < x; i++ { 

     y := RecursiveFunc(x, depth - 1, quit) 

     if y > 0 { 
      total += y 
     } 

    } 

    return total 
} 

该函数可以采用做很长一段时间，我想发送的退出信号，之后停止和使用的结果（不管它是什么）。要运行它：

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

func main() { 

    quit := make(chan bool) 
    wg := &sync.WaitGroup{} 
    result := -1 

    go func() { 
     defer wg.Done() 
     wg.Add(1) 
     result = RecursiveFunc(5, 20, quit) 
    }() 

    time.Sleep(10 * time.Millisecond) 

    close(quit) // Using `quit <- true` doesn't work 

    wg.Wait() 

    fmt.Println(result) 
} 

要停止够程，我使用的信道说quit和关闭它后，程序效果很好，但我不希望真的关闭通道，我想只是发信号quit <- true。但是，quit <- true不起作用，我可能只退出递归的一个实例。

如何通过发送退出信号来停止递归函数的所有实例？

Answer 1

你可以使用context来做你想要做的事情。

您可以传递一个context.Context对象作为第一个参数，你需要从外部停止功能，并调用相应的cancel函数来发送一个“消除信号”的功能，这将导致的对Done()通道context.Context将被关闭，并且因此被叫功能将在select声明中被通知取消信号。

下面是函数如何处理使用context.Context消除信号：

func RecursiveFunc(ctx context.Context, x int, depth int) int { 

    if depth == 0 { 
     return 1 
    } 

    select { 
    case <-ctx.Done(): 
     return 0 
    default: 
    } 

    total := 0 

    for i := 0; i < x; i++ { 

     y := RecursiveFunc(ctx, x, depth-1) 

     if y > 0 { 
      total += y 
     } 

    } 

    return total 
} 

这里是你如何调用该函数与新签名：

func main() { 

    wg := &sync.WaitGroup{} 
    result := -1 

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

    go func() { 
     defer wg.Done() 
     wg.Add(1) 
     result = RecursiveFunc(ctx, 5, 20) 
    }() 

    time.Sleep(10 * time.Millisecond) 

    cancel() 

    wg.Wait() 

    fmt.Println(result) 
} 

Answer 2

尝试添加标志以继续执行，但它可能不是线程安全的。

var finishIt bool 

func RecursiveFunc(x int, depth int, quit chan bool) int { 
    if finishIt { 
    return 0 
    } 
//other code here 
} 


//some code here, but than we decide to stop it 
finishIt = true