单元测试 goroutine 数与性能 CPU+1

1、

并发下载图片的goroutine数为CPU数加1

package goroutin
 
import (
    "fmt"
    "io"
    "net/http"
    "os"
    "sync"
)
 
func dlImg(threadPool, batchNum int) {
    var wg sync.WaitGroup
    URL := "http://intranet/images/a.png"
    s := func(i interface{}) string {
        return fmt.Sprint(i)
    }
    dir := "img" + s(threadPool) + "-" + s(batchNum)
    err := os.Mkdir(dir, 0777)
    if err != nil {
        logHelper.Error(err)
    }
    for j := 0; j < threadPool; j++ {
        wg.Add(1)
        go func() {
            for i := 0; i < batchNum; i++ {
                resp, err := http.Get(URL)
                if err != nil {
                    logHelper.Error(err)
                    continue
                }
                defer resp.Body.Close()
                bs, err := io.ReadAll(resp.Body)
                if err != nil {
                    logHelper.Error(err)
                    continue
                }
                name := dir + "/" + s(i) + ".png"
                os.WriteFile(name, bs, 0777)
            }
            wg.Done()
        }()
    }
    wg.Wait()
}

package goroutin
 
import "testing"
 
func Test_dlImg(t *testing.T) {
 
    type args struct {
        threadPool int
        batchNum   int
    }
    tests := []struct {
        name string
        args args
    }{
        // TODO: Add test cases.
        {name: "10-128", args: args{threadPool: 10, batchNum: 128}},
        {name: "3-128", args: args{threadPool: 3, batchNum: 128}},
        {name: "2-128", args: args{threadPool: 2, batchNum: 128}},
        {name: "1-128", args: args{threadPool: 1, batchNum: 128}},
    }
    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            dlImg(tt.args.threadPool, tt.args.batchNum)
        })
    }
}

测试结果

Running tool: C:\Program Files\Go\bin\go.exe test -timeout 30s -run ^Test_dlImg$ goroutin

=== RUN Test_dlImg

=== RUN Test_dlImg/10-128

=== RUN Test_dlImg/3-128

=== RUN Test_dlImg/2-128

=== RUN Test_dlImg/1-128

--- PASS: Test_dlImg (1.79s)

--- PASS: Test_dlImg/10-128 (0.65s)

--- PASS: Test_dlImg/3-128 (0.37s)

--- PASS: Test_dlImg/2-128 (0.51s)

--- PASS: Test_dlImg/1-128 (0.26s)

PASS

ok goroutin 2.010s

internet Img
--- PASS: Test_dlImg (16.33s)
--- PASS: Test_dlImg/10-128 (9.47s)
--- PASS: Test_dlImg/3-128 (2.94s)
--- PASS: Test_dlImg/2-128 (2.58s)
--- PASS: Test_dlImg/1-128 (1.34s)
PASS

--- PASS: Test_dlImg (17.33s)
--- PASS: Test_dlImg/10-128 (9.27s)
--- PASS: Test_dlImg/3-128 (3.22s)
--- PASS: Test_dlImg/2-128 (2.35s)
--- PASS: Test_dlImg/1-128 (2.48s)
PASS
--- PASS: Test_dlImg (19.78s)
--- PASS: Test_dlImg/10-128 (11.25s)
--- PASS: Test_dlImg/3-128 (3.38s)
--- PASS: Test_dlImg/2-128 (2.58s)
--- PASS: Test_dlImg/1-128 (2.56s)
PASS

--- PASS: Test_dlImg (19.22s)
--- PASS: Test_dlImg/10-256 (6.60s)
--- PASS: Test_dlImg/3-256 (4.26s)
--- PASS: Test_dlImg/2-256 (4.27s)
--- PASS: Test_dlImg/1-256 (4.09s)
PASS

intranet Img
--- PASS: Test_dlImg (5.10s)
--- PASS: Test_dlImg/10-512 (2.38s)
--- PASS: Test_dlImg/3-512 (0.97s)
--- PASS: Test_dlImg/2-512 (0.85s)
--- PASS: Test_dlImg/1-512 (0.91s)
PASS
ok goroutin 5.329s

--- PASS: Test_dlImg (9.60s)
--- PASS: Test_dlImg/10-1024 (3.25s)
--- PASS: Test_dlImg/3-1024 (2.92s)
--- PASS: Test_dlImg/2-1024 (1.85s)
--- PASS: Test_dlImg/1-1024 (1.57s)
PASS

--- PASS: Test_dlImg (24.36s)
--- PASS: Test_dlImg/10-2048 (10.53s)
--- PASS: Test_dlImg/3-2048 (5.27s)
--- PASS: Test_dlImg/2-2048 (4.60s)
--- PASS: Test_dlImg/1-2048 (3.96s)
PASS

--- PASS: Test_dlImg (20.01s)
--- PASS: Test_dlImg/3-4096 (6.90s)
--- PASS: Test_dlImg/2-4096 (6.43s)
--- PASS: Test_dlImg/1-4096 (6.69s)
PASS