consumer constructor

consumer.go

@@ -7,6 +7,23 @@ import (
 	"log"
 )
 
+type Consumer interface {
+	Start() chan []byte
+}
+
+type consumeHandler struct {
+	ctx     context.Context
+	client  *Client
+	chanLen int
+}
+
+func (c *consumeHandler) Start() chan []byte {
+	msgCh := make(chan []byte, c.chanLen)
+	go runConsumer(c.ctx, c.client, msgCh)
+
+	return msgCh
+}
+
 func runConsumer(ctx context.Context, client *Client, msgCh chan []byte) {
 	runCtx, cancel := context.WithCancel(ctx)
 	defer cancel()

handler.go

@@ -64,13 +64,10 @@ func NewClient(addr, queueName string, opts ...Option) (*Client, error) {
 	return &client, nil
 }
 
-func StartConsumer(ctx context.Context, client *Client, chanLen int) chan []byte {
-	msgCh := make(chan []byte, chanLen)
-	go runConsumer(ctx, client, msgCh)
-
-	return msgCh
-}
-
 func NewPublisher(client *Client) Publisher {
-	return &PubHandler{client: client}
+	return &pubHandler{client: client}
+}
+
+func NewConsumer(ctx context.Context, client *Client, chanLen int) Consumer {
+	return &consumeHandler{ctx: ctx, client: client, chanLen: chanLen}
 }