zeromq_pipeline.go

package mq

import (

	"github.com/pebbe/zmq4"
	"github.com/tylertreat/mq-benchmarking/benchmark"
)

func NewZeromqPipeline(numberOfMessages int, testLatency bool) *Zeromq {
	ctx, _ := zmq4.NewContext()
	pub, _ := ctx.NewSocket(zmq4.PUSH)
	pub.Bind("tcp://*:5555")
	sub, _ := ctx.NewSocket(zmq4.PULL)
	sub.Connect("tcp://localhost:5555")

	var handler benchmark.MessageHandler
	if testLatency {
		handler = &benchmark.LatencyMessageHandler{
			NumberOfMessages: numberOfMessages,
			Latencies:        []float32{},
		}
	} else {
		handler = &benchmark.ThroughputMessageHandler{NumberOfMessages: numberOfMessages}
	}

	return &Zeromq{
		handler:  handler,
		sender:   pub,
		receiver: sub,
	}
}