「让我们一起Golang」怎样出让协程资源和设置可用CPU核心数

2021-09-01 01:59:00

「让我们一起Golang」怎样出让协程资源和设置可用CPU核心数

前面了解了协程的有关基础知识，了解了CPS并发模型，见识了Golang的百万级并发，下面我们来实现一下出让协程资源和设置可用CPU核心数。

出让协程资源

先看看执行结果：

子协程0 0
子协程0 1
子协程0 2
子协程0 3
子协程0 4
子协程0 5
子协程2 0
子协程2 1
子协程2 2
子协程2 3
子协程2 4
子协程2 5
子协程2 6
子协程2 7
子协程1 0
子协程0 6
子协程0 7
子协程1 1
子协程1 2
子协程1 3
子协程1 4
子协程1 5
子协程1 6
子协程1 7

我们可以看到先是子协程0打印内容，然后字协程2打印内容，然后出现了子协程1打印结果，之后又突然出现了子协程0，然后是子协程1打印结果。

为什么是这样呢？先别急，先看一看代码想一想为什么！

func main() {
    for i:=0;i<3;i++{
        go func(index int) {
            task("子协程"+strconv.Itoa(index))
        }(i)
    }
    time.Sleep(5 * time.Second)
}

func task(name string)  {
    for i:=0;i<8;i++{
        if name == "子协程1"{
            //Gosched yields the processor, allowing other goroutines to run. It does not suspend the current goroutine, so execution resumes automatically.
            runtime.Gosched()
        }
        fmt.Println(name,i)
    }
}

这段代码是先激活3条子协程，分别是子协程0、子协程1、子协程2。然后每条协程执行task函数，但是对子协程1执行runtime.Gosched()操作。

下面来介绍一下runtime.Gosched()，Gosched 生成一个处理器，允许其他goroutine先运行。它不会中止当前的 goroutine，因此当前的 goroutine 会自动恢复运行。它的作用就是会把当前协程的优先级降低。

我们知道三条协程是并发的。但是这里对协程1实施了runtime.Gosched()，这让子协程1出让了协程资源。所以最后打印输出的一定是子协程1。但是，并不是所有的子协程1都是最后执行打印操作。这里有一个子协程1在子协程0之前打印输出了。我们把协程的数量加大。将主函数里面的for循环范围从0~3改为0~108，我们会发现在最后打印的还是子协程1.（由于输出结果过多，在此不展示输出结果）但是要注意的是，如果是百万级并发，同时开辟一百万条协程，最后的可能就不是子协程1了，因为runtime.Gosched()虽然会降低协程的优先级，出让协程资源，但是并不一定让协程绝对的最后执行完毕。

设置可用CPU核心数

我们打开任务管理器，进入“性能”栏目，在CPU处右键选择“将图形更改为”，将“总体利用率”改为“逻辑处理器”。

这里可以看到图片里面的CPU核数为八。

然后我们可用利用GO语言查看电脑CPU的核数。

package main

import (
    "fmt"
    "runtime"
)

func main() {

    fmt.Println("可用CPU核心数为",runtime.NumCPU())
    
    // GOMAXPROCS sets the maximum number of CPUs that can be executing
    // simultaneously and returns the previous setting. If n < 1, it does not
    // change the current setting.
    // The number of logical CPUs on the local machine can be queried with NumCPU.
    // This call will go away when the scheduler improves.
    fmt.Println("设置CPU的可用核数为1，先前的设置为",runtime.GOMAXPROCS(1))
}

使用runtime.NumCPU()打印即可。

而使用runtime.GOMAXPROCS(n)可以设置运行的最大核数。这里是设置CPU的可用最大核心数为1.而且该函数会有返回值返回先前的可用最大核心数。如果 n < 1，则不更改当前设置。

码农公寓