go 創建攜程池，開啟並發

地址；https://github.com/panjf2000/ants

采用螞蟻池開源的SDK，ants 是一個高性能且低損耗的 goroutine 池

 

package main

import (
   "fmt"
   "sync"
   "sync/atomic"
   "time"

   "github.com/panjf2000/ants"
)

var sum int32

func myFunc(i interface{}) {
   n := i.(int32)
   atomic.AddInt32(&sum, n)
   fmt.Printf("run with %d\n", n)
}

func demoFunc() {
   time.Sleep(10 * time.Millisecond)
   fmt.Println("Hello World!")
}

func main() {
   defer ants.Release()

   runTimes := 1000

   // Use the common pool.
   var wg sync.WaitGroup
   syncCalculateSum := func() {
      demoFunc()
      wg.Done()
   }
   for i := 0; i < runTimes; i++ {
      wg.Add(1)
      _ = ants.Submit(syncCalculateSum)
   }
   wg.Wait()
   fmt.Printf("running goroutines: %d\n", ants.Running())
   fmt.Printf("finish all tasks.\n")

   // Use the pool with a method,
   // set 10 to the capacity of goroutine pool and 1 second for expired duration.
   fn := func(i interface{}) {
      myFunc(i)
      wg.Done()
   }
   var mmm []func(i ants.Options)

   fno := func(i ants.Options) {
      i.PreAlloc = false
      i.MaxBlockingTasks = 100
   }
   mmm = append(mmm, fno)

   var hhh ants.Option  // 設置攜程
   lll :=  &ants.Options{}
   lll.PreAlloc = false
   lll.MaxBlockingTasks = 100
   hhh(lll)
   p, _ := ants.NewPoolWithFunc(10, fn, hhh)
   defer p.Release()
   // Submit tasks one by one.
   for i := 0; i < runTimes; i++ {
      wg.Add(1)
      _ = p.Invoke(int32(i))
   }
   wg.Wait()
   fmt.Printf("running goroutines: %d\n", p.Running())
   fmt.Printf("finish all tasks, result is %d\n", sum)
   if sum != 499500 {
      panic("the final result is wrong!!!")
   }
}