chore(queue): add queue message interface (#602)

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>

queue/nsq/nsq.go

@@ -102,7 +102,7 @@ func (s *Worker) Run(quit chan struct{}) error {
 	s.q.AddHandler(nsq.HandlerFunc(func(msg *nsq.Message) error {
 		wg.Add(1)
 		defer wg.Done()
-		var notification gorush.PushNotification
+		var notification *gorush.PushNotification
 		if err := json.Unmarshal(msg.Body, &notification); err != nil {
 			return err
 		}
@@ -136,8 +136,8 @@ func (s *Worker) Usage() int {
 }
 
 // Queue send notification to queue
-func (s *Worker) Queue(job interface{}) error {
-	v, ok := job.(gorush.PushNotification)
+func (s *Worker) Queue(job queue.QueuedMessage) error {
+	v, ok := job.(*gorush.PushNotification)
 	if !ok {
 		return errors.New("wrong type of job")
 	}

queue/queue.go

@@ -59,7 +59,7 @@ func (q *Queue) Wait() {
 }
 
 // Queue to queue all job
-func (q *Queue) Queue(job interface{}) error {
+func (q *Queue) Queue(job QueuedMessage) error {
 	return q.worker.Queue(job)
 }
 

queue/simple/simple.go

@@ -17,7 +17,7 @@ var errMaxCapacity = errors.New("max capacity reached")
 
 // Worker for simple queue using channel
 type Worker struct {
-	queueNotification chan gorush.PushNotification
+	queueNotification chan queue.QueuedMessage
 }
 
 // BeforeRun run script before start worker
@@ -56,9 +56,9 @@ func (s *Worker) Usage() int {
 }
 
 // Queue send notification to queue
-func (s *Worker) Queue(job interface{}) error {
+func (s *Worker) Queue(job queue.QueuedMessage) error {
 	select {
-	case s.queueNotification <- job.(gorush.PushNotification):
+	case s.queueNotification <- job:
 		return nil
 	default:
 		return errMaxCapacity
@@ -68,14 +68,14 @@ func (s *Worker) Queue(job interface{}) error {
 // WithQueueNum setup the capcity of queue
 func WithQueueNum(num int) Option {
 	return func(w *Worker) {
-		w.queueNotification = make(chan gorush.PushNotification, num)
+		w.queueNotification = make(chan queue.QueuedMessage, num)
 	}
 }
 
 // NewWorker for struc
 func NewWorker(opts ...Option) *Worker {
 	w := &Worker{
-		queueNotification: make(chan gorush.PushNotification, runtime.NumCPU()<<1),
+		queueNotification: make(chan queue.QueuedMessage, runtime.NumCPU()<<1),
 	}
 
 	// Loop through each option

queue/simple/simple_test.go

@@ -14,7 +14,7 @@ func TestQueueUsage(t *testing.T) {
 	assert.Equal(t, runtime.NumCPU()<<1, w.Capacity())
 	assert.Equal(t, 0, w.Usage())
 
-	w.Queue(gorush.PushNotification{})
+	w.Queue(&gorush.PushNotification{})
 	assert.Equal(t, 1, w.Usage())
 }
 
@@ -23,13 +23,13 @@ func TestMaxCapacity(t *testing.T) {
 	assert.Equal(t, 2, w.Capacity())
 	assert.Equal(t, 0, w.Usage())
 
-	assert.NoError(t, w.Queue(gorush.PushNotification{}))
+	assert.NoError(t, w.Queue(&gorush.PushNotification{}))
 	assert.Equal(t, 1, w.Usage())
-	assert.NoError(t, w.Queue(gorush.PushNotification{}))
+	assert.NoError(t, w.Queue(&gorush.PushNotification{}))
 	assert.Equal(t, 2, w.Usage())
-	assert.Error(t, w.Queue(gorush.PushNotification{}))
+	assert.Error(t, w.Queue(&gorush.PushNotification{}))
 	assert.Equal(t, 2, w.Usage())
 
-	err := w.Queue(gorush.PushNotification{})
+	err := w.Queue(&gorush.PushNotification{})
 	assert.Equal(t, errMaxCapacity, err)
 }

queue/worker.go

@@ -7,7 +7,12 @@ type Worker interface {
 	AfterRun() error
 
 	Shutdown() error
-	Queue(job interface{}) error
+	Queue(job QueuedMessage) error
 	Capacity() int
 	Usage() int
 }
+
+// QueuedMessage ...
+type QueuedMessage interface {
+	Bytes() []byte
+}