我的消費者(從 運行)支持上下文取消和通過語句main從通道讀取。case我可以使用上下文關閉消費者,效果很好。但是,當我在一個案例語句中生成多個工作人員時,每個工作人員都會從 獲得相同的工作(消息)jobsChan,這不是我想要的:func (app *App) consumer() { for { select { case <-app.ctx.Done(): app.infoLog.Print("Caught SIGINT, stopping.") app.wg.Wait() app.doneChan <- struct{}{} # main uses this channel to block itself until all goroutines are stopped app.infoLog.Print("Shutting down the consumer...") return case job := <-app.jobsChan: // PROBLEM here: wrong, each worker is given the same job for workerNumber := 0; workerNumber < app.config.workers; workerNumber++ { app.wg.Add(1) go app.workerFunc(workerNumber, job) } } }}func (app *App) workerFunc(id int, job Job) { defer app.wg.Done() ... actual worker code here ...}如何重寫此代碼以便我可以保留select頻道app.ctx.Done并同時生成工人,以便每個工人從頻道中選擇下一條消息作為作業?我需要繼續for/select監聽ctx取消,但同時我需要生成 X 工作人員來讀取來自jobsChan消費者的消息。這可能嗎?想到的唯一選擇是將 channel 直接傳遞給 spawnedworkerFunc并for job := range app.jobsChan在workerFunc. 但隨后case job := <-app.jobsChan:消費者的整體變得毫無意義,我不知道如何重寫它。澄清一下:當我運行應用程序時,我希望每個工作人員都有一個新的工作 id 從jobsChan- 但他們都處理相同的,例如 1,然后他們都處理下一個,例如 2#wrongWorker 0: start processing item 1Worker 2: start processing item 1Worker 1: start processing item 1
1 回答

米脂
TA貢獻1836條經驗 獲得超3個贊
您現有的代碼明確地將相同的工作分配給所有工作人員。如果您有固定數量的工作人員,請為他們創建 goroutine(在初始化期間),并讓他們收聽頻道:
for workerNumber:0;workerNumber<app.config.workers;workerNumber++ {
go app.workerFunc(ctx,workerNumber,app.jobsChan)
}
在每個工作人員中,只需檢查 jobQueue 和上下文取消。
換句話說,您不需要consumer, 將工作直接傳遞給工人。
- 1 回答
- 0 關注
- 90 瀏覽
添加回答
舉報
0/150
提交
取消