diff --git a/README.md b/README.md index 9caf153..dc69946 100644 --- a/README.md +++ b/README.md @@ -18,7 +18,7 @@ Example usage: package main import ( - "github.com/jrallison/go-workers" + "github.com/jrallison/go-workers" ) func myJob(message *workers.Msg) { diff --git a/manager.go b/manager.go index 0e4ee6f..5e1f9b3 100644 --- a/manager.go +++ b/manager.go @@ -54,13 +54,14 @@ func (m *manager) manage() { go m.fetch.Fetch() +Lmanage: for { select { case message := <-m.confirm: m.fetch.Acknowledge(message) case <-m.stop: m.exit <- true - break + break Lmanage } } } diff --git a/worker.go b/worker.go index f0d79a2..de661ce 100644 --- a/worker.go +++ b/worker.go @@ -23,6 +23,7 @@ func (w *worker) quit() { } func (w *worker) work(messages chan *Msg) { +Lwork: for { select { case message := <-messages: @@ -40,9 +41,10 @@ func (w *worker) work(messages chan *Msg) { // ready to accept a message case <-w.stop: w.exit <- true - return + break Lwork } } + } func (w *worker) process(message *Msg) (acknowledge bool) {