停止在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