使用golang編寫一個并發(fā)工作隊列
更新時間：2021年05月08日 10:36:05 作者：u010278923
這篇文章主要介紹了使用golang編寫一個并發(fā)工作隊列的操作，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
其實golang用一個函數可以構建一個并發(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
}
這個里面定義了一個工作隊列的切片，可以自定義工作隊列的個數，甚至后期還可以添加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的個數是3，當然這個可以通過讀取配置文件或者參數傳遞的方式；這個里面逐一啟動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)
以上為個人經驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。
相關文章
Gin是一個用Go語言編寫的web框架,它是一個類似于martini但擁有更好性能的API框架, 由于使用了httprouter,速度提高了近40倍,本文給大家介紹了Go語言web框架Gin響應客戶端有哪些方式,文中有詳細的代碼示例供大家參考,需要的朋友可以參考下2024-10-10
這篇文章主要給大家講解了Go語言的基礎語法和常用特性解析，比較適合入門小白，文中通過代碼示例介紹的非常詳細，對我們學習Go語言有一定的幫助,需要的朋友可以參考下2023-07-07
這篇文章主要介紹了go 語言字符類型 byte 與 rune案例詳解,本文通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-03-03
這篇文章主要介紹了golang 字符串比較是否相等的方法示例，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-02-02
深入理解Golang?make和new的區(qū)別及實現(xiàn)原理
在Go語言中，有兩個比較雷同的內置函數，分別是new和make方法，二者都可以用來分配內存，那他們有什么區(qū)別呢？下面我們就從底層來分析一下二者的不同。感興趣的小伙伴們可以參考借鑒，希望對大家能有所幫助2022-10-10
這篇文章主要介紹了基于gin的golang web開發(fā)：路由示例詳解,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-10-10
Golang對struct字段重新排序優(yōu)化數據結構性能實踐
這篇文章主要為大家介紹了Golang對struct字段重新排序優(yōu)化數據結構性能實踐,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-12-12
這篇文章主要介紹了解決golang時間字符串轉time.Time的坑，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-04-04
golang簡易實現(xiàn)?k8s?的yaml上傳并應用示例方案
這篇文章主要為大家介紹了golang簡易實現(xiàn)?k8s?的yaml上傳并應用示例方案,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪2023-07-07
這篇文章主要介紹了使用go mod導入本地自定義包問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-07-07