基于golang channel實現(xiàn)的輕量級異步任務(wù)分發(fā)器示例代碼

前言

有時候我們?yōu)榱烁玫睦糜嬎銠C(jī)資源，可以把一些耗時長的任務(wù)隊列化異步執(zhí)行。舉個對應(yīng)簡單的生活中例子就是大多數(shù)餐廳里面點菜都是先找地方做，看了菜單選好菜之后找服務(wù)員點菜，此時再等待菜做好送上來。這里餐廳廚房就是計算機(jī)的底層資源，菜就是待執(zhí)行的任務(wù)，而服務(wù)員就是我們的go channel。

關(guān)于消息隊列有很多好用的框架，如nsq，nats，kafka等等。但有時我們只需要輕量級的異步任務(wù)工具，而不需要太過于"復(fù)雜"的框架相對于我們的需求來說。于是借鑒一些項目框架，做了一個小小的封裝。

項目地址： https://github.com/chenhg5/go...  （本地下載）

go-task的使用很簡單，只需要初始化一個任務(wù)處理器，然后往處理器里面添加任務(wù)，然后處理器就會異步地去執(zhí)行了。

舉餐廳的例子，代碼如下：

package main

import (
 "runtime"
 "fmt"
 "time"
 "github.com/chenhg5/go-task"
)

func main() {

 // init
 task.InitTaskReceiver(runtime.NumCPU())

 // 有十個人同時點菜
 for i := 0; i  10; i++ {
  task.AddTask(task.NewTask(
   map[string]interface{}{
    "paramA" : "value",
   }, // 參數(shù)
   []task.FacFunc{ordering, cooking, deliverying}, // 任務(wù)列表
   -1), // -1代表任務(wù)不超時
  )
 }

 time.Sleep(time.Second * 50)
}

// 下單任務(wù)
func ordering(uuid string, param map[string]interface{}) (string, error) {
 fmt.Println("i am ordering")
 time.Sleep(time.Second * 1)
 return uuid, nil
}

// 做菜任務(wù)
func cooking(uuid string, param map[string]interface{}) (string, error) {
 fmt.Println("i am cooking")
 time.Sleep(time.Second * 1)
 return uuid, nil
}

// 配送任務(wù)
func deliverying(uuid string, param map[string]interface{}) (string, error) {
 fmt.Println("i am deliverying")
 time.Sleep(time.Second * 1)
 return uuid, nil
}

跑一遍，就會看到：

i am ordering
i am ordering
i am ordering
i am ordering
i am ordering
i am ordering
i am ordering
i am ordering
i am cooking
i am cooking
i am cooking
i am cooking
i am cooking
i am cooking
i am cooking
i am cooking
i am deliverying
i am deliverying
i am deliverying
i am deliverying
i am deliverying
i am deliverying
i am deliverying
i am deliverying
i am ordering
i am ordering
i am cooking
i am cooking
i am deliverying
i am deliverying

總結(jié)

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。