dis

broker/worker.go

@@ -0,0 +1,30 @@
+package broker
+
+type Worker struct {
+	WorkerPool chan chan *Message
+	MsgChannel chan *Message
+	quit       chan bool
+}
+
+func (w Worker) Start() {
+	go func() {
+		for {
+			// register the current worker into the worker queue.
+			w.WorkerPool <- w.MsgChannel
+			select {
+			case msg := <-w.MsgChannel:
+				// we have received a work request.
+			case <-w.quit:
+				// we have received a signal to stop
+				return
+			}
+		}
+	}()
+}
+
+// Stop signals the worker to stop listening for work requests.
+func (w Worker) Stop() {
+	go func() {
+		w.quit <- true
+	}()
+}