mt-rabbit/publisher.go

package rabbit

import (
	"context"
	"errors"
	"log"
	"time"
)

type Publisher interface {
	Start(ctx context.Context, chanLen uint) chan<- []byte
}

type pubHandler struct {
	client *Client
}

func (p *pubHandler) Start(ctx context.Context, chanLen uint) chan<- []byte {
	ch := make(chan []byte, chanLen)
	go func() {
		for {
			select {
			case <-ctx.Done():
				for msg := range ch {
					if err := p.push(msg); err != nil {
						log.Printf("Error publishing message (shutdown): %s", err)
					}
				}
				log.Println("Publisher stopped")
				return

			case msg := <-ch:
				if err := p.push(msg); err != nil {
					log.Printf("Error publishing message: %s", err)
				}
			}
		}
	}()
	return ch
}

func (p *pubHandler) push(data []byte) error {
	p.client.mutex.Lock()
	if !p.client.isReady {
		p.client.mutex.Unlock()
		return errors.New("failed to push: not connected")
	}
	p.client.mutex.Unlock()
	for {
		err := p.client.unsafePush(data)
		if err != nil {
			p.client.logger.Println("Push failed. Retrying...")
			select {
			case <-p.client.done:
				return errShutdown
			case <-time.After(p.client.opts.resendDelay):
			}
			continue
		}
		confirm := <-p.client.notifyConfirm
		if confirm.Ack {
			p.client.logger.Printf("Push confirmed [%d]!", confirm.DeliveryTag)
			return nil
		}
	}
}