chore: Add test worker.

Signed-off-by: Bo-Yi Wu <[email protected]>

Author: appleboy

worker.go

@@ -1,11 +1,5 @@
 package queue
 
-import (
-	"context"
-	"errors"
-	"time"
-)
-
 // Worker interface
 type Worker interface {
 	// BeforeRun is called before starting the worker
@@ -28,83 +22,3 @@ type Worker interface {
 type QueuedMessage interface {
 	Bytes() []byte
 }
-
-var (
-	_ Worker = (*emptyWorker)(nil)
-	_ Worker = (*messageWorker)(nil)
-)
-
-type emptyWorker struct{}
-
-func (w *emptyWorker) BeforeRun() error              { return nil }
-func (w *emptyWorker) AfterRun() error               { return nil }
-func (w *emptyWorker) Run() error                    { return nil }
-func (w *emptyWorker) Shutdown() error               { return nil }
-func (w *emptyWorker) Queue(job QueuedMessage) error { return nil }
-func (w *emptyWorker) Capacity() int                 { return 0 }
-func (w *emptyWorker) Usage() int                    { return 0 }
-
-type messageWorker struct {
-	messages chan QueuedMessage
-}
-
-func (w *messageWorker) BeforeRun() error { return nil }
-func (w *messageWorker) AfterRun() error  { return nil }
-func (w *messageWorker) Run() error {
-	for msg := range w.messages {
-		if string(msg.Bytes()) == "panic" {
-			panic("show panic")
-		}
-		time.Sleep(20 * time.Millisecond)
-	}
-	return nil
-}
-
-func (w *messageWorker) Shutdown() error {
-	close(w.messages)
-	return nil
-}
-
-func (w *messageWorker) Queue(job QueuedMessage) error {
-	select {
-	case w.messages <- job:
-		return nil
-	default:
-		return errors.New("max capacity reached")
-	}
-}
-func (w *messageWorker) Capacity() int { return cap(w.messages) }
-func (w *messageWorker) Usage() int    { return len(w.messages) }
-
-type taskWorker struct {
-	messages chan QueuedMessage
-}
-
-func (w *taskWorker) BeforeRun() error { return nil }
-func (w *taskWorker) AfterRun() error  { return nil }
-func (w *taskWorker) Run() error {
-	for msg := range w.messages {
-		if v, ok := msg.(Job); ok {
-			if v.Task != nil {
-				_ = v.Task(context.Background())
-			}
-		}
-	}
-	return nil
-}
-
-func (w *taskWorker) Shutdown() error {
-	close(w.messages)
-	return nil
-}
-
-func (w *taskWorker) Queue(job QueuedMessage) error {
-	select {
-	case w.messages <- job:
-		return nil
-	default:
-		return errors.New("max capacity reached")
-	}
-}
-func (w *taskWorker) Capacity() int { return cap(w.messages) }
-func (w *taskWorker) Usage() int    { return len(w.messages) }

worker_empty.go

@@ -0,0 +1,14 @@
+package queue
+
+var _ Worker = (*emptyWorker)(nil)
+
+// just for unit testing, don't use it.
+type emptyWorker struct{}
+
+func (w *emptyWorker) BeforeRun() error              { return nil }
+func (w *emptyWorker) AfterRun() error               { return nil }
+func (w *emptyWorker) Run() error                    { return nil }
+func (w *emptyWorker) Shutdown() error               { return nil }
+func (w *emptyWorker) Queue(job QueuedMessage) error { return nil }
+func (w *emptyWorker) Capacity() int                 { return 0 }
+func (w *emptyWorker) Usage() int                    { return 0 }

worker_message.go

@@ -0,0 +1,41 @@
+package queue
+
+import (
+	"errors"
+	"time"
+)
+
+var _ Worker = (*messageWorker)(nil)
+
+// just for unit testing, don't use it.
+type messageWorker struct {
+	messages chan QueuedMessage
+}
+
+func (w *messageWorker) BeforeRun() error { return nil }
+func (w *messageWorker) AfterRun() error  { return nil }
+func (w *messageWorker) Run() error {
+	for msg := range w.messages {
+		if string(msg.Bytes()) == "panic" {
+			panic("show panic")
+		}
+		time.Sleep(20 * time.Millisecond)
+	}
+	return nil
+}
+
+func (w *messageWorker) Shutdown() error {
+	close(w.messages)
+	return nil
+}
+
+func (w *messageWorker) Queue(job QueuedMessage) error {
+	select {
+	case w.messages <- job:
+		return nil
+	default:
+		return errors.New("max capacity reached")
+	}
+}
+func (w *messageWorker) Capacity() int { return cap(w.messages) }
+func (w *messageWorker) Usage() int    { return len(w.messages) }

worker_task.go

@@ -0,0 +1,42 @@
+package queue
+
+import (
+	"context"
+	"errors"
+)
+
+var _ Worker = (*taskWorker)(nil)
+
+// just for unit testing, don't use it.
+type taskWorker struct {
+	messages chan QueuedMessage
+}
+
+func (w *taskWorker) BeforeRun() error { return nil }
+func (w *taskWorker) AfterRun() error  { return nil }
+func (w *taskWorker) Run() error {
+	for msg := range w.messages {
+		if v, ok := msg.(Job); ok {
+			if v.Task != nil {
+				_ = v.Task(context.Background())
+			}
+		}
+	}
+	return nil
+}
+
+func (w *taskWorker) Shutdown() error {
+	close(w.messages)
+	return nil
+}
+
+func (w *taskWorker) Queue(job QueuedMessage) error {
+	select {
+	case w.messages <- job:
+		return nil
+	default:
+		return errors.New("max capacity reached")
+	}
+}
+func (w *taskWorker) Capacity() int { return cap(w.messages) }
+func (w *taskWorker) Usage() int    { return len(w.messages) }