利用go語言編寫一個并發(fā)包

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

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

寫在前面

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

代碼

package mini_parallel_job

import (
    "fmt"
    "sync"
)

type JobType func()

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

type jobPool struct {
    jobs []JobType
}

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

// 開始并且等待任務
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),
    }
}

壓測

package mini_parallel_job

import (
    "testing"
)

const (
    Count = 10
)

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

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

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

// 性能測試
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
    */
}

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