并发执行固定数量的 goroutine

大家好，今天本人给大家带来文章《并发执行固定数量的 goroutine》，文中内容主要涉及到，如果你对Golang方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

我不完全确定这里发生了什么，所以很难概括我的问题，但我会尽力而为。

在几年前的一段视频中，马特·帕克 (matt parker) 向观众发起挑战，要求观众找出 2 的幂，其中不包含任何 2 的幂。 （例如，2^16 = 65536。这些数字都不是 2 的幂）。最近我开始学习 go，我认为这将是一个很好的入门练习，可以帮助您习惯这门语言。

我很快就创建了这个，然后我决定尝试使其并发以充分利用我的四核处理器。这就是事情走下坡路的地方。

这里的目标是运行恒定数量的 goroutine，每个 goroutine 处理不同批次的数字。我像这样实现了该程序：

package main

import (
    "log"
    "math/big"
    "runtime"
)

//The maximum amount of goroutines
const routineAmt int = 3

//The amount of numbers for each routine to check
const rangeSize int64 = 5000

//The current start of the range to start checking
var rangeIndex int64 = 0

func main() {
    //loop forever
    for {
        //if we have less routines running than the maximum
        if runtime.NumGoroutine() < routineAmt {
            c := make(chan bool)
            // start a new one to check the next range:
            go checkRange(rangeIndex, rangeIndex+rangeSize, c)
            // wait for the signal that the values have been copied to the function, so that we can increment them safely:
            <-c
            close(c)
            // increment the rangeIndex for the next routine which will start:
            rangeIndex += rangeSize
        }
    }
}

// Function to check a range of powers of two, whether they contain any power-of-two-digits
func checkRange(from, to int64, c chan bool) {
    c <- true // signal to the main routine that the parameter values have been copied

    // Loop through the range for powers of two, which do not contain any power-of-two-digits
    for i := from; i < to; i++ {
        num := big.NewInt(2)
        num.Exp(num, big.NewInt(i), nil)
        if !hasStringPowerOfTwo(num.String()) {
            log.Println("Found 2 ^", i)
        }
    }
    log.Printf("Checked range %d-%d\n", from, to)
}

// Function to check if a string contains any number which is a power of two
func hasStringPowerOfTwo(input string) bool {
    powersOfTwo := [4]rune{'1', '2', '4', '8'}
    for _, char := range input {
        if runeInArray(char, powersOfTwo) {
            return true
        }
    }
    return false
}

// Function to check if a list of runes contains a certain rune
func runeInArray(a rune, list [4]rune) bool {
    for _, b := range list {
        if b == a {
            return true
        }
    }
    return false
}

等待了大约15分钟左右，程序仍然没有完成一个go例程（即没有看到log.printf("checked range %d-%d\n", from, to)在控制台中）

我尝试将范围大小降低到 5，这导致一些 goroutine 完成，但它突然停止在范围 2840-2845 处。我认为这可能是由于数字变大并且计算花费更多时间，但这没有意义，因为停止非常突然。如果是这种情况，我预计放缓至少会是渐进的。

解决方案

您不应该使用 for 循环来检查 runtime.numgoroutine 以确保您没有运行太多例程，因为循环将阻止 goruntime 正确安排您的例程，它会减慢整个过程。

相反，您应该使用缓冲通道，该通道在例程完成时发出信号，以便您可以开始新的例程。

我已经调整了您的 main 函数和 checkrange 函数：

func main() {
        var done = make(chan struct{}, routineAmt)
        //loop forever
        for i := 0; i < routineAmt; i++ {
                // start a new one to check the next range:
                go checkRange(done, rangeIndex, rangeIndex+rangeSize)
                // increment the rangeIndex for the next routine which will start:
                rangeIndex += rangeSize
        }

        for range done {
                // start a new one to check the next range:
                go checkRange(done, rangeIndex, rangeIndex+rangeSize)
                // increment the rangeIndex for the next routine which will start:
                rangeIndex += rangeSize
        }
}

// Function to check a range of powers of two, whether they contain any power-of-two-digits
func checkRange(done chan<- struct{}, from, to int64) {
        // Loop through the range for powers of two, which do not contain any power-of-two-digits
        for i := from; i < to; i++ {
                num := big.NewInt(2)
                num.Exp(num, big.NewInt(i), nil)
                if !hasStringPowerOfTwo(num.String()) {
                        log.Println("Found 2 ^", i)
                }
        }
        log.Printf("Checked range %d-%d\n", from, to)

        // let our main go routine know we're done with this one
        done <- struct{}{}
}

终于介绍完啦！小伙伴们，这篇关于《并发执行固定数量的 goroutine》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！