使用golang編寫一個并發(fā)工作隊列
更新時間：2021年05月08日 10:36:05 作者：u010278923
這篇文章主要介紹了使用golang編寫一個并發(fā)工作隊列的操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
其實golang用一個函數(shù)可以構建一個并發(fā)隊列，現(xiàn)在編寫一個靈活可控的隊列程序
先定義一個工作
type Worker struct {
ID int
RepJobs chan int64
SM *SM
quit chan bool
}
包含了workid和執(zhí)行任務的id，上面的SM只是任務具體內容，這個和具體業(yè)務相關，大家自己編寫自己的SM業(yè)務邏輯
然后定義工作池
type workerPool struct {
workerChan chan *Worker
workerList []*Worker
}
這個里面定義了一個工作隊列的切片，可以自定義工作隊列的個數(shù)，甚至后期還可以添加work，還定義了一個隊列類型的管道。
定義完成過后就可以初始化工作池了
func InitWorkerPool() error {
n := 3
WorkerPool = &workerPool{
workerChan: make(chan *Worker, n),
workerList: make([]*Worker, 0, n),
}
for i := 0; i < n; i++ {
worker := NewWorker(i)
WorkerPool.workerList = append(WorkerPool.workerList, worker)
worker.Start()
log.Debugf("worker %d started", worker.ID)
}
return nil
}
這個里面我寫死了worker的個數(shù)是3，當然這個可以通過讀取配置文件或者參數(shù)傳遞的方式；這個里面逐一啟動work
worker.Start()，這個是關鍵
func (w *Worker) Start() {
go func() {
for {
WorkerPool.workerChan <- w
select {
case jobID := <-w.RepJobs:
log.Debugf("worker: %d, will handle job: %d", w.ID, jobID)
w.handleRepJob(jobID)
case q := <-w.quit:
if q {
log.Debugf("worker: %d, will stop.", w.ID)
return
}
}
}
}()
}
這個就是go 啟動一個協(xié)程，先把自己放到workerChan中，然后不斷從w.RepJobs管道中獲取任務并執(zhí)行，如果執(zhí)行完成后又把自己放回到隊列中。
所以如果你要有任務需要執(zhí)行，放到這個管道中即可
func Dispatch() {
for {
select {
case job := <-jobQueue:
go func(jobID int64) {
println("Trying to dispatch job: %d", jobID)
worker := <-WorkerPool.workerChan
worker.RepJobs <- jobID
}(job)
}
}
}
從管道中拿出一個worker并把任務id放到worker中去執(zhí)行。
當然你可以停止worker，甚至可以停止job
func (w *Worker) Stop() {
go func() {
w.quit <- true
}()
}
func (wp *workerPool) StopJobs(jobs []int64) {
log.Debugf("Works working on jobs: %v will be stopped", jobs)
for _, id := range jobs {
for _, w := range wp.workerList {
if w.SM.JobID == id {
log.Debugf("found a worker whose job ID is %d, will try to stop it", id)
w.SM.Stop(id)
}
}
}
}
補充一下，int64和字符串轉換。
string到int
int,err:=strconv.Atoi(string)
string到int64
int64, err := strconv.ParseInt(string, 10, 64)
int到string
string:=strconv.Itoa(int)
int64到string
string:=strconv.FormatInt(int64,10)
以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關文章
Golang多線程排序實現(xiàn)快速高效地處理大規(guī)模數(shù)據(jù)
Golang多線程排序是一種快速高效地處理大規(guī)模數(shù)據(jù)的方法，通過使用Golang的協(xié)程和通道，可以將排序任務分配到多個線程中并行處理，提高了排序的效率和速度，需要詳細了解可以參考下文2023-05-05
這篇文章主要給大家介紹了基于Go編寫一個可視化Navicat本地密碼解析器的方法,文中有詳細的代碼示例和圖文介紹,有需要的朋友可以參考閱讀本文2023-08-08
這篇文章主要為大家介紹了GO語言中常見排序算法的使用示例，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步早日升職加薪2022-04-04
golang中實現(xiàn)給gif、png、jpeg圖片添加文字水印
這篇文章主要介紹了golang中實現(xiàn)給gif、png、jpeg圖片添加文字水印，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-04-04
本文主要介紹了Golang String字符串類型轉Json格式的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友們下面隨著小編來一起學習學習吧2024-05-05
這篇文章主要為大家詳細介紹了如何利用Golang實現(xiàn)將視頻按照時間維度進行剪切，文中的示例代碼講解詳細，感興趣的小伙伴可以跟隨小編一起了解一下2022-12-12
這篇文章主要為大家介紹了LRU?LFU?TinyLFU緩存算法實例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-09-09
這篇文章主要介紹了使用golang實現(xiàn)在屏幕上打印進度條的操作,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-03-03
在ubuntu下安裝go開發(fā)環(huán)境的全過程
Go語言是谷歌公司開發(fā)的編程語言,雖然安裝和配置go很簡單,但是很多初學者在第一次安裝go環(huán)境時會遇到各種坑,下面這篇文章主要給大家介紹了關于在ubuntu下安裝go開發(fā)環(huán)境的相關資料,文中通過圖文介紹的非常詳細,需要的朋友可以參考下2022-08-08
在Go語言中,map和sync.Map都是用于存儲鍵值對的數(shù)據(jù)結構,但它們在并發(fā)安全性、性能和使用場景上存在顯著差異,接下來將深入探討這兩種數(shù)據(jù)結構的異同,感興趣的朋友可以參考下2024-01-01