Last active
April 12, 2018 11:59
-
-
Save nesv/9220339 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
package main | |
import ( | |
"fmt" | |
"time" | |
) | |
// NewWorker creates, and returns a new Worker object. Its only argument | |
// is a channel that the worker can add itself to whenever it is done its | |
// work. | |
func NewWorker(id int, workerQueue chan chan WorkRequest) Worker { | |
// Create, and return the worker. | |
worker := Worker{ | |
ID: id, | |
Work: make(chan WorkRequest), | |
WorkerQueue: workerQueue, | |
QuitChan: make(chan bool)} | |
return worker | |
} | |
type Worker struct { | |
ID int | |
Work chan WorkRequest | |
WorkerQueue chan chan WorkRequest | |
QuitChan chan bool | |
} | |
// This function "starts" the worker by starting a goroutine, that is | |
// an infinite "for-select" loop. | |
func (w *Worker) Start() { | |
go func() { | |
for { | |
// Add ourselves into the worker queue. | |
w.WorkerQueue <- w.Work | |
select { | |
case work := <-w.Work: | |
// Receive a work request. | |
fmt.Printf("worker%d: Received work request, delaying for %f seconds\n", w.ID, work.Delay.Seconds()) | |
time.Sleep(work.Delay) | |
fmt.Printf("worker%d: Hello, %s!\n", w.ID, work.Name) | |
case <-w.QuitChan: | |
// We have been asked to stop. | |
fmt.Printf("worker%d stopping\n", w.ID) | |
return | |
} | |
} | |
}() | |
} | |
// Stop tells the worker to stop listening for work requests. | |
// | |
// Note that the worker will only stop *after* it has finished its work. | |
func (w *Worker) Stop() { | |
go func() { | |
w.QuitChan <- true | |
}() | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment