利用go語(yǔ)言編寫(xiě)一個(gè)并發(fā)包

更新時(shí)間：2023年10月22日 08:08:26 作者：nil

這篇文章主要為大家詳細(xì)介紹了如何利用go語(yǔ)言編寫(xiě)一個(gè)并發(fā)包,適合大部分并發(fā)任務(wù),開(kāi)箱即用,文中的示例代碼講解詳細(xì),有需要的小伙伴可以參考下

寫(xiě)在前面

這是一個(gè)只需要用50行代碼（核心代碼只有15行）實(shí)現(xiàn)的極其簡(jiǎn)單（原理簡(jiǎn)單、使用方法簡(jiǎn)單、功能簡(jiǎn)單）的go包mini_parallel_job，適合大部分并發(fā)任務(wù)，開(kāi)箱即用。

代碼

package mini_parallel_job

import (
    "fmt"
    "sync"
)

type JobType func()

type JobPool interface {
    AddJob(jobType JobType)
    Wait()
}

type jobPool struct {
    jobs []JobType
}

// 添加任務(wù)
func (j *jobPool) AddJob(job JobType) {
    j.jobs = append(j.jobs, job)
}

// 開(kāi)始并且等待任務(wù)
func (j *jobPool) Wait() {
    var wg sync.WaitGroup
    wg.Add(len(j.jobs))
    for i := range j.jobs {
       jJob := j.jobs[i]
       go func() {
          defer func() {
             wg.Done()
             if err := recover(); err != nil {
                fmt.Printf("err:%+v", err)
             }
          }()

          jJob()
       }()
    }
    wg.Wait()
}

func NewJobPool() JobPool {
    return &jobPool{
       jobs: make([]JobType, 0),
    }
}

壓測(cè)

package mini_parallel_job

import (
    "testing"
)

const (
    Count = 10
)

// 并行任務(wù)
func parallelJob() {
    jobPool := NewJobPool()
    for i := 0; i < Count; i++ {
       jobPool.AddJob(func() {
          _ = fib(10)
       })
    }
    jobPool.Wait()
}

// 串行任務(wù)
func serialJob() {
    for i := 0; i < Count; i++ {
       _ = fib(10)
    }
}

// 任務(wù)
func fib(n int) int {
    if n == 0 || n == 1 {
       return n
    }
    return fib(n-2) + fib(n-1)
}

// 性能測(cè)試
func BenchmarkSerialJob(b *testing.B) {
    for i := 0; i < b.N; i++ {
       serialJob()
    }
}

func BenchmarkParallelJob(b *testing.B) {
    for i := 0; i < b.N; i++ {
       parallelJob()
    }
}

/*
BenchmarkSerialJob-12             298855              3756 ns/op
BenchmarkParallelJob-12           117189              8710 ns/op
*/

example

package main

import (
    "fmt"
    mini_parallel_job "mini-parallel-job"
    "time"
)

const (
    JobCount = 10
)

func main() {
    // 串行執(zhí)行
    begin1 := time.Now()
    for i := 0; i < JobCount; i++ {
       fib(40)
    }
    fmt.Println(time.Since(begin1))

    // 并行執(zhí)行
    begin2 := time.Now()
    parallelJob := mini_parallel_job.NewJobPool()
    for i := 0; i < JobCount; i++ {
       parallelJob.AddJob(func() {
          fib(40)
       })
    }
    parallelJob.Wait()
    fmt.Println(time.Since(begin2))

    /*
       結(jié)果：
       7.335989407s
       1.112108503s
    */
}

// 任務(wù)
func fib(n int) int {
    if n == 0 || n == 1 {
       return n
    }
    return fib(n-2) + fib(n-1)
}

總結(jié)

這段代碼僅僅實(shí)現(xiàn)使用go rountine實(shí)現(xiàn)并發(fā)，sync.WaitGroup實(shí)現(xiàn)等待。

在大多數(shù)場(chǎng)景中，只需要并發(fā)，并不關(guān)心并發(fā)量是多少，大多數(shù)程序員也是使用Wait函數(shù)那段代碼實(shí)現(xiàn)的（至少作者在項(xiàng)目中看到都是這樣的，并且有多處相同的代碼，基于此場(chǎng)景封裝了一下）。

如果要實(shí)現(xiàn)復(fù)雜一點(diǎn)的場(chǎng)景，比如控制最大并發(fā)量，可以稍微對(duì)上述代碼做一些修改，Wait函數(shù)中加一個(gè)指定大小的chan來(lái)控制?；蛘邊⒖甲髡吡硗庖粋€(gè)對(duì)go并發(fā)的封裝gopool，使用master-worker模式實(shí)現(xiàn)的并發(fā)控制。

到此這篇關(guān)于利用go語(yǔ)言編寫(xiě)一個(gè)并發(fā)包的文章就介紹到這了,更多相關(guān)go并發(fā)包內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: