test_batch_consumer.go

package rmq

type TestBatchConsumer struct {
	LastBatch     Deliveries
	ConsumedCount int
	AutoFinish    bool

	finish chan int
}

func NewTestBatchConsumer() *TestBatchConsumer {
	return &TestBatchConsumer{
		finish: make(chan int),
	}
}

func (consumer *TestBatchConsumer) Consume(batch Deliveries) {
	// log.Printf("TestBatchConsumer.Consume(%d)", len(batch))
	consumer.LastBatch = batch
	consumer.ConsumedCount += len(batch)
	if !consumer.AutoFinish {
		<-consumer.finish
		// log.Printf("TestBatchConsumer.Consume() finished")
	}
}

func (consumer *TestBatchConsumer) Finish() {
	// log.Printf("TestBatchConsumer.Finish()")
	consumer.LastBatch = nil
	consumer.finish <- 1
}