pulsar/producer_test.go - pulsar-client-go - Git at Google

 // Licensed to the Apache Software Foundation (ASF) under one
 // or more contributor license agreements.  See the NOTICE file
 // distributed with this work for additional information
 // regarding copyright ownership.  The ASF licenses this file
 // to you under the Apache License, Version 2.0 (the
 // "License"); you may not use this file except in compliance
 // with the License.  You may obtain a copy of the License at
 //
 //   http://www.apache.org/licenses/LICENSE-2.0
 //
 // Unless required by applicable law or agreed to in writing,
 // software distributed under the License is distributed on an
 // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 // KIND, either express or implied.  See the License for the
 // specific language governing permissions and limitations
 // under the License.

 package pulsar

 import (
 	"context"
 	"fmt"
 	"github.com/apache/pulsar-client-go/pulsar/internal"
 	"net/http"
 	"sync"
 	"testing"
 	"time"

 	"github.com/stretchr/testify/assert"

 	log "github.com/sirupsen/logrus"
 )

 func TestInvalidURL(t *testing.T) {
 	client, err := NewClient(ClientOptions{})

 	if client != nil || err == nil {
 		t.Fatal("Should have failed to create client")
 	}
 }

 func TestProducerConnectError(t *testing.T) {
 	client, err := NewClient(ClientOptions{
 		URL: "pulsar://invalid-hostname:6650",
 	})

 	assert.Nil(t, err)

 	defer client.Close()

 	producer, err := client.CreateProducer(ProducerOptions{
 		Topic: newTopicName(),
 	})

 	// Expect error in creating producer
 	assert.Nil(t, producer)
 	assert.NotNil(t, err)

 	assert.Equal(t, err.Error(), "connection error")
 }

 func TestProducerNoTopic(t *testing.T) {
 	client, err := NewClient(ClientOptions{
 		URL: "pulsar://localhost:6650",
 	})

 	if err != nil {
 		t.Fatal(err)
 		return
 	}

 	defer client.Close()

 	producer, err := client.CreateProducer(ProducerOptions{})

 	// Expect error in creating producer
 	assert.Nil(t, producer)
 	assert.NotNil(t, err)

 	assert.Equal(t, err.(*Error).Result(), ResultInvalidTopicName)
 }

 func TestSimpleProducer(t *testing.T) {
 	client, err := NewClient(ClientOptions{
 		URL: serviceURL,
 	})
 	assert.NoError(t, err)
 	defer client.Close()

 	producer, err := client.CreateProducer(ProducerOptions{
 		Topic: newTopicName(),
 	})

 	assert.NoError(t, err)
 	assert.NotNil(t, producer)
 	defer producer.Close()

 	for i := 0; i < 10; i++ {
 		err = producer.Send(context.Background(), &ProducerMessage{
 			Payload: []byte("hello"),
 		})

 		assert.NoError(t, err)
 	}
 }

 func TestProducerAsyncSend(t *testing.T) {
 	client, err := NewClient(ClientOptions{
 		URL: serviceURL,
 	})
 	assert.NoError(t, err)
 	defer client.Close()

 	producer, err := client.CreateProducer(ProducerOptions{
 		Topic:                   newTopicName(),
 		BatchingMaxPublishDelay: 1 * time.Second,
 	})

 	assert.NoError(t, err)
 	assert.NotNil(t, producer)
 	defer producer.Close()

 	wg := sync.WaitGroup{}
 	wg.Add(10)
 	errors := internal.NewBlockingQueue(10)

 	for i := 0; i < 10; i++ {
 		producer.SendAsync(context.Background(), &ProducerMessage{
 			Payload: []byte("hello"),
 		}, func(id MessageID, message *ProducerMessage, e error) {
 			if e != nil {
 				log.WithError(e).Error("Failed to publish")
 				errors.Put(e)
 			} else {
 				log.Info("Published message ", id)
 			}
 			wg.Done()
 		})

 		assert.NoError(t, err)
 	}

 	err = producer.Flush()
 	assert.Nil(t, err)

 	wg.Wait()

 	assert.Equal(t, 0, errors.Size())
 }

 func TestProducerCompression(t *testing.T) {

 	type testProvider struct {
 		name            string
 		compressionType CompressionType
 	}

 	var providers = []testProvider{
 		{"zlib", ZLib},
 		{"lz4", LZ4},
 		{"zstd", ZSTD},
 	}

 	for _, provider := range providers {
 		p := provider
 		t.Run(p.name, func(t *testing.T) {
 			client, err := NewClient(ClientOptions{
 				URL: serviceURL,
 			})
 			assert.NoError(t, err)
 			defer client.Close()

 			producer, err := client.CreateProducer(ProducerOptions{
 				Topic:           newTopicName(),
 				CompressionType: p.compressionType,
 			})

 			assert.NoError(t, err)
 			assert.NotNil(t, producer)
 			defer producer.Close()

 			for i := 0; i < 10; i++ {
 				err = producer.Send(context.Background(), &ProducerMessage{
 					Payload: []byte("hello"),
 				})

 				assert.NoError(t, err)
 			}
 		})
 	}
 }

 func TestProducerLastSequenceID(t *testing.T) {
 	client, err := NewClient(ClientOptions{
 		URL: serviceURL,
 	})
 	assert.NoError(t, err)
 	defer client.Close()

 	producer, err := client.CreateProducer(ProducerOptions{
 		Topic: newTopicName(),
 	})

 	assert.NoError(t, err)
 	assert.NotNil(t, producer)
 	defer producer.Close()

 	assert.Equal(t, int64(-1), producer.LastSequenceID())

 	for i := 0; i < 10; i++ {
 		err = producer.Send(context.Background(), &ProducerMessage{
 			Payload: []byte("hello"),
 		})

 		assert.NoError(t, err)
 		assert.Equal(t, int64(i), producer.LastSequenceID())
 	}
 }

 func TestEventTime(t *testing.T) {
 	client, err := NewClient(ClientOptions{
 		URL: serviceURL,
 	})
 	assert.NoError(t, err)
 	defer client.Close()

 	topicName := "test-event-time"
 	producer, err := client.CreateProducer(ProducerOptions{
 		Topic: topicName,
 	})
 	assert.Nil(t, err)
 	defer producer.Close()

 	consumer, err := client.Subscribe(ConsumerOptions{
 		Topic:            topicName,
 		SubscriptionName: "subName",
 	})
 	assert.Nil(t, err)
 	defer consumer.Close()

 	eventTime := timeFromUnixTimestampMillis(uint64(1565161612))
 	err = producer.Send(context.Background(), &ProducerMessage{
 		Payload:   []byte(fmt.Sprintf("test-event-time")),
 		EventTime: &eventTime,
 	})
 	assert.Nil(t, err)

 	msg, err := consumer.Receive(context.Background())
 	assert.Nil(t, err)
 	actualEventTime := msg.EventTime()
 	assert.Equal(t, eventTime.Unix(), actualEventTime.Unix())
 }

 func TestFlushInProducer(t *testing.T) {
 	client, err := NewClient(ClientOptions{
 		URL: serviceURL,
 	})
 	assert.NoError(t, err)
 	defer client.Close()

 	topicName := "test-flush-in-producer"
 	subName := "subscription-name"
 	numOfMessages := 10
 	ctx := context.Background()

 	// set batch message number numOfMessages, and max delay 10s
 	producer, err := client.CreateProducer(ProducerOptions{
 		Topic:                   topicName,
 		DisableBatching:         false,
 		BatchingMaxMessages:     uint(numOfMessages),
 		BatchingMaxPublishDelay: time.Second * 10,
 		BlockIfQueueFull:        true,
 		Properties: map[string]string{
 			"producer-name": "test-producer-name",
 			"producer-id":   "test-producer-id",
 		},
 	})
 	assert.Nil(t, err)
 	defer producer.Close()

 	consumer, err := client.Subscribe(ConsumerOptions{
 		Topic:            topicName,
 		SubscriptionName: subName,
 	})
 	assert.Nil(t, err)
 	defer consumer.Close()

 	prefix := "msg-batch-async"
 	msgCount := 0

 	wg := sync.WaitGroup{}
 	wg.Add(5)
 	errors := internal.NewBlockingQueue(10)
 	for i := 0; i < numOfMessages/2; i++ {
 		messageContent := prefix + fmt.Sprintf("%d", i)
 		producer.SendAsync(ctx, &ProducerMessage{
 			Payload: []byte(messageContent),
 		}, func(id MessageID, producerMessage *ProducerMessage, e error) {
 			if e != nil {
 				log.WithError(e).Error("Failed to publish")
 				errors.Put(e)
 			} else {
 				log.Info("Published message ", id)
 			}
 			wg.Done()
 		})
 		assert.Nil(t, err)
 	}
 	err = producer.Flush()
 	assert.Nil(t, err)
 	wg.Wait()

 	for i := 0; i < numOfMessages/2; i++ {
 		_, err = consumer.Receive(ctx)
 		assert.Nil(t, err)
 		msgCount++
 	}

 	assert.Equal(t, msgCount, numOfMessages/2)

 	wg.Add(5)
 	for i := numOfMessages / 2; i < numOfMessages; i++ {
 		messageContent := prefix + fmt.Sprintf("%d", i)
 		producer.SendAsync(ctx, &ProducerMessage{
 			Payload: []byte(messageContent),
 		}, func(id MessageID, producerMessage *ProducerMessage, e error) {
 			if e != nil {
 				log.WithError(e).Error("Failed to publish")
 				errors.Put(e)
 			} else {
 				log.Info("Published message ", id)
 			}
 			wg.Done()
 		})
 		assert.Nil(t, err)
 	}

 	err = producer.Flush()
 	assert.Nil(t, err)
 	wg.Wait()

 	for i := numOfMessages / 2; i < numOfMessages; i++ {
 		_, err := consumer.Receive(ctx)
 		assert.Nil(t, err)
 		msgCount++
 	}
 	assert.Equal(t, msgCount, numOfMessages)
 }

 func TestFlushInPartitionedProducer(t *testing.T) {
 	topicName := "persistent://public/default/partition-testFlushInPartitionedProducer"

 	// call admin api to make it partitioned
 	url := adminURL + "/" + "admin/v2/" + topicName + "/partitions"
 	makeHTTPCall(t, http.MethodPut, url, "5")

 	numberOfPartitions := 5
 	numOfMessages := 10
 	ctx := context.Background()

 	// creat client connection
 	client, err := NewClient(ClientOptions{
 		URL: serviceURL,
 	})
 	assert.NoError(t, err)
 	defer client.Close()

 	// create consumer
 	consumer, err := client.Subscribe(ConsumerOptions{
 		Topic:            topicName,
 		SubscriptionName: "my-sub",
 		Type:             Exclusive,
 	})
 	assert.Nil(t, err)
 	defer consumer.Close()

 	// create producer and set batch message number numOfMessages, and max delay 10s
 	producer, err := client.CreateProducer(ProducerOptions{
 		Topic:                   topicName,
 		DisableBatching:         false,
 		BatchingMaxMessages:     uint(numOfMessages / numberOfPartitions),
 		BatchingMaxPublishDelay: time.Second * 10,
 		BlockIfQueueFull:        true,
 	})
 	defer producer.Close()

 	// send 5 messages
 	prefix := "msg-batch-async-"
 	wg := sync.WaitGroup{}
 	wg.Add(5)
 	errors := internal.NewBlockingQueue(5)
 	for i := 0; i < numOfMessages/2; i++ {
 		messageContent := prefix + fmt.Sprintf("%d", i)
 		producer.SendAsync(ctx, &ProducerMessage{
 			Payload: []byte(messageContent),
 		}, func(id MessageID, producerMessage *ProducerMessage, e error) {
 			if e != nil {
 				log.WithError(e).Error("Failed to publish")
 				errors.Put(e)
 			} else {
 				log.Info("Published message: ", id)
 			}
 			wg.Done()
 		})
 		assert.Nil(t, err)
 	}

 	// After flush, should be able to consume.
 	err = producer.Flush()
 	assert.Nil(t, err)

 	wg.Wait()

 	// Receive all messages
 	msgCount := 0
 	for i := 0; i < numOfMessages/2; i++ {
 		msg, err := consumer.Receive(ctx)
 		fmt.Printf("Received message msgId: %#v -- content: '%s'\n",
 			msg.ID(), string(msg.Payload()))
 		assert.Nil(t, err)
 		consumer.Ack(msg)
 		msgCount++
 	}
 	assert.Equal(t, msgCount, numOfMessages/2)
 }

 func TestMessageRouter(t *testing.T) {
 	// Create topic with 5 partitions
 	httpPut("http://localhost:8080/admin/v2/persistent/public/default/my-partitioned-topic/partitions", 5)
 	client, err := NewClient(ClientOptions{
 		URL: serviceURL,
 	})

 	assert.Nil(t, err)
 	defer client.Close()

 	// Only subscribe on the specific partition
 	consumer, err := client.Subscribe(ConsumerOptions{
 		Topic:            "my-partitioned-topic-partition-2",
 		SubscriptionName: "my-sub",
 	})

 	assert.Nil(t, err)
 	defer consumer.Close()

 	producer, err := client.CreateProducer(ProducerOptions{
 		Topic: "my-partitioned-topic",
 		MessageRouter: func(msg *ProducerMessage, tm TopicMetadata) int {
 			fmt.Println("Routing message ", msg, " -- Partitions: ", tm.NumPartitions())
 			return 2
 		},
 	})

 	assert.Nil(t, err)
 	defer producer.Close()

 	ctx := context.Background()

 	err = producer.Send(ctx, &ProducerMessage{
 		Payload: []byte("hello"),
 	})
 	assert.Nil(t, err)

 	fmt.Println("PUBLISHED")

 	// Verify message was published on partition 2
 	msg, err := consumer.Receive(ctx)
 	assert.Nil(t, err)
 	assert.NotNil(t, msg)
 	assert.Equal(t, string(msg.Payload()), "hello")
 }

 func TestNonPersistentTopic(t *testing.T) {
 	topicName := "non-persistent://public/default/testNonPersistentTopic"
 	client, err := NewClient(ClientOptions{
 		URL: serviceURL,
 	})
 	assert.Nil(t, err)
 	defer client.Close()

 	producer, err := client.CreateProducer(ProducerOptions{
 		Topic: topicName,
 	})

 	assert.Nil(t, err)
 	defer producer.Close()

 	consumer, err := client.Subscribe(ConsumerOptions{
 		Topic:            topicName,
 		SubscriptionName: "my-sub",
 	})
 	assert.Nil(t, err)
 	defer consumer.Close()
 }
	// Licensed to the Apache Software Foundation (ASF) under one
	// or more contributor license agreements. See the NOTICE file
	// distributed with this work for additional information
	// regarding copyright ownership. The ASF licenses this file
	// to you under the Apache License, Version 2.0 (the
	// "License"); you may not use this file except in compliance
	// with the License. You may obtain a copy of the License at
	//
	// http://www.apache.org/licenses/LICENSE-2.0
	//
	// Unless required by applicable law or agreed to in writing,
	// software distributed under the License is distributed on an
	// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
	// KIND, either express or implied. See the License for the
	// specific language governing permissions and limitations
	// under the License.

	package pulsar

	import (
	"context"
	"fmt"
	"github.com/apache/pulsar-client-go/pulsar/internal"
	"net/http"
	"sync"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"

	log "github.com/sirupsen/logrus"
	)

	func TestInvalidURL(t *testing.T) {
	client, err := NewClient(ClientOptions{})

	if client != nil \|\| err == nil {
	t.Fatal("Should have failed to create client")
	}
	}

	func TestProducerConnectError(t *testing.T) {
	client, err := NewClient(ClientOptions{
	URL: "pulsar://invalid-hostname:6650",
	})

	assert.Nil(t, err)

	defer client.Close()

	producer, err := client.CreateProducer(ProducerOptions{
	Topic: newTopicName(),
	})

	// Expect error in creating producer
	assert.Nil(t, producer)
	assert.NotNil(t, err)

	assert.Equal(t, err.Error(), "connection error")
	}

	func TestProducerNoTopic(t *testing.T) {
	client, err := NewClient(ClientOptions{
	URL: "pulsar://localhost:6650",
	})

	if err != nil {
	t.Fatal(err)
	return
	}

	defer client.Close()

	producer, err := client.CreateProducer(ProducerOptions{})

	// Expect error in creating producer
	assert.Nil(t, producer)
	assert.NotNil(t, err)

	assert.Equal(t, err.(*Error).Result(), ResultInvalidTopicName)
	}

	func TestSimpleProducer(t *testing.T) {
	client, err := NewClient(ClientOptions{
	URL: serviceURL,
	})
	assert.NoError(t, err)
	defer client.Close()

	producer, err := client.CreateProducer(ProducerOptions{
	Topic: newTopicName(),
	})

	assert.NoError(t, err)
	assert.NotNil(t, producer)
	defer producer.Close()

	for i := 0; i < 10; i++ {
	err = producer.Send(context.Background(), &ProducerMessage{
	Payload: []byte("hello"),
	})

	assert.NoError(t, err)
	}
	}

	func TestProducerAsyncSend(t *testing.T) {
	client, err := NewClient(ClientOptions{
	URL: serviceURL,
	})
	assert.NoError(t, err)
	defer client.Close()

	producer, err := client.CreateProducer(ProducerOptions{
	Topic: newTopicName(),
	BatchingMaxPublishDelay: 1 * time.Second,
	})

	assert.NoError(t, err)
	assert.NotNil(t, producer)
	defer producer.Close()

	wg := sync.WaitGroup{}
	wg.Add(10)
	errors := internal.NewBlockingQueue(10)

	for i := 0; i < 10; i++ {
	producer.SendAsync(context.Background(), &ProducerMessage{
	Payload: []byte("hello"),
	}, func(id MessageID, message *ProducerMessage, e error) {
	if e != nil {
	log.WithError(e).Error("Failed to publish")
	errors.Put(e)
	} else {
	log.Info("Published message ", id)
	}
	wg.Done()
	})

	assert.NoError(t, err)
	}

	err = producer.Flush()
	assert.Nil(t, err)

	wg.Wait()

	assert.Equal(t, 0, errors.Size())
	}

	func TestProducerCompression(t *testing.T) {

	type testProvider struct {
	name string
	compressionType CompressionType
	}

	var providers = []testProvider{
	{"zlib", ZLib},
	{"lz4", LZ4},
	{"zstd", ZSTD},
	}

	for _, provider := range providers {
	p := provider
	t.Run(p.name, func(t *testing.T) {
	client, err := NewClient(ClientOptions{
	URL: serviceURL,
	})
	assert.NoError(t, err)
	defer client.Close()

	producer, err := client.CreateProducer(ProducerOptions{
	Topic: newTopicName(),
	CompressionType: p.compressionType,
	})

	assert.NoError(t, err)
	assert.NotNil(t, producer)
	defer producer.Close()

	for i := 0; i < 10; i++ {
	err = producer.Send(context.Background(), &ProducerMessage{
	Payload: []byte("hello"),
	})

	assert.NoError(t, err)
	}
	})
	}
	}

	func TestProducerLastSequenceID(t *testing.T) {
	client, err := NewClient(ClientOptions{
	URL: serviceURL,
	})
	assert.NoError(t, err)
	defer client.Close()

	producer, err := client.CreateProducer(ProducerOptions{
	Topic: newTopicName(),
	})

	assert.NoError(t, err)
	assert.NotNil(t, producer)
	defer producer.Close()

	assert.Equal(t, int64(-1), producer.LastSequenceID())

	for i := 0; i < 10; i++ {
	err = producer.Send(context.Background(), &ProducerMessage{
	Payload: []byte("hello"),
	})

	assert.NoError(t, err)
	assert.Equal(t, int64(i), producer.LastSequenceID())
	}
	}

	func TestEventTime(t *testing.T) {
	client, err := NewClient(ClientOptions{
	URL: serviceURL,
	})
	assert.NoError(t, err)
	defer client.Close()

	topicName := "test-event-time"
	producer, err := client.CreateProducer(ProducerOptions{
	Topic: topicName,
	})
	assert.Nil(t, err)
	defer producer.Close()

	consumer, err := client.Subscribe(ConsumerOptions{
	Topic: topicName,
	SubscriptionName: "subName",
	})
	assert.Nil(t, err)
	defer consumer.Close()

	eventTime := timeFromUnixTimestampMillis(uint64(1565161612))
	err = producer.Send(context.Background(), &ProducerMessage{
	Payload: []byte(fmt.Sprintf("test-event-time")),
	EventTime: &eventTime,
	})
	assert.Nil(t, err)

	msg, err := consumer.Receive(context.Background())
	assert.Nil(t, err)
	actualEventTime := msg.EventTime()
	assert.Equal(t, eventTime.Unix(), actualEventTime.Unix())
	}

	func TestFlushInProducer(t *testing.T) {
	client, err := NewClient(ClientOptions{
	URL: serviceURL,
	})
	assert.NoError(t, err)
	defer client.Close()

	topicName := "test-flush-in-producer"
	subName := "subscription-name"
	numOfMessages := 10
	ctx := context.Background()

	// set batch message number numOfMessages, and max delay 10s
	producer, err := client.CreateProducer(ProducerOptions{
	Topic: topicName,
	DisableBatching: false,
	BatchingMaxMessages: uint(numOfMessages),
	BatchingMaxPublishDelay: time.Second * 10,
	BlockIfQueueFull: true,
	Properties: map[string]string{
	"producer-name": "test-producer-name",
	"producer-id": "test-producer-id",
	},
	})
	assert.Nil(t, err)
	defer producer.Close()

	consumer, err := client.Subscribe(ConsumerOptions{
	Topic: topicName,
	SubscriptionName: subName,
	})
	assert.Nil(t, err)
	defer consumer.Close()

	prefix := "msg-batch-async"
	msgCount := 0

	wg := sync.WaitGroup{}
	wg.Add(5)
	errors := internal.NewBlockingQueue(10)
	for i := 0; i < numOfMessages/2; i++ {
	messageContent := prefix + fmt.Sprintf("%d", i)
	producer.SendAsync(ctx, &ProducerMessage{
	Payload: []byte(messageContent),
	}, func(id MessageID, producerMessage *ProducerMessage, e error) {
	if e != nil {
	log.WithError(e).Error("Failed to publish")
	errors.Put(e)
	} else {
	log.Info("Published message ", id)
	}
	wg.Done()
	})
	assert.Nil(t, err)
	}
	err = producer.Flush()
	assert.Nil(t, err)
	wg.Wait()

	for i := 0; i < numOfMessages/2; i++ {
	_, err = consumer.Receive(ctx)
	assert.Nil(t, err)
	msgCount++
	}

	assert.Equal(t, msgCount, numOfMessages/2)

	wg.Add(5)
	for i := numOfMessages / 2; i < numOfMessages; i++ {
	messageContent := prefix + fmt.Sprintf("%d", i)
	producer.SendAsync(ctx, &ProducerMessage{
	Payload: []byte(messageContent),
	}, func(id MessageID, producerMessage *ProducerMessage, e error) {
	if e != nil {
	log.WithError(e).Error("Failed to publish")
	errors.Put(e)
	} else {
	log.Info("Published message ", id)
	}
	wg.Done()
	})
	assert.Nil(t, err)
	}

	err = producer.Flush()
	assert.Nil(t, err)
	wg.Wait()

	for i := numOfMessages / 2; i < numOfMessages; i++ {
	_, err := consumer.Receive(ctx)
	assert.Nil(t, err)
	msgCount++
	}
	assert.Equal(t, msgCount, numOfMessages)
	}

	func TestFlushInPartitionedProducer(t *testing.T) {
	topicName := "persistent://public/default/partition-testFlushInPartitionedProducer"

	// call admin api to make it partitioned
	url := adminURL + "/" + "admin/v2/" + topicName + "/partitions"
	makeHTTPCall(t, http.MethodPut, url, "5")

	numberOfPartitions := 5
	numOfMessages := 10
	ctx := context.Background()

	// creat client connection
	client, err := NewClient(ClientOptions{
	URL: serviceURL,
	})
	assert.NoError(t, err)
	defer client.Close()

	// create consumer
	consumer, err := client.Subscribe(ConsumerOptions{
	Topic: topicName,
	SubscriptionName: "my-sub",
	Type: Exclusive,
	})
	assert.Nil(t, err)
	defer consumer.Close()

	// create producer and set batch message number numOfMessages, and max delay 10s
	producer, err := client.CreateProducer(ProducerOptions{
	Topic: topicName,
	DisableBatching: false,
	BatchingMaxMessages: uint(numOfMessages / numberOfPartitions),
	BatchingMaxPublishDelay: time.Second * 10,
	BlockIfQueueFull: true,
	})
	defer producer.Close()

	// send 5 messages
	prefix := "msg-batch-async-"
	wg := sync.WaitGroup{}
	wg.Add(5)
	errors := internal.NewBlockingQueue(5)
	for i := 0; i < numOfMessages/2; i++ {
	messageContent := prefix + fmt.Sprintf("%d", i)
	producer.SendAsync(ctx, &ProducerMessage{
	Payload: []byte(messageContent),
	}, func(id MessageID, producerMessage *ProducerMessage, e error) {
	if e != nil {
	log.WithError(e).Error("Failed to publish")
	errors.Put(e)
	} else {
	log.Info("Published message: ", id)
	}
	wg.Done()
	})
	assert.Nil(t, err)
	}

	// After flush, should be able to consume.
	err = producer.Flush()
	assert.Nil(t, err)

	wg.Wait()

	// Receive all messages
	msgCount := 0
	for i := 0; i < numOfMessages/2; i++ {
	msg, err := consumer.Receive(ctx)
	fmt.Printf("Received message msgId: %#v -- content: '%s'\n",
	msg.ID(), string(msg.Payload()))
	assert.Nil(t, err)
	consumer.Ack(msg)
	msgCount++
	}
	assert.Equal(t, msgCount, numOfMessages/2)
	}

	func TestMessageRouter(t *testing.T) {
	// Create topic with 5 partitions
	httpPut("http://localhost:8080/admin/v2/persistent/public/default/my-partitioned-topic/partitions", 5)
	client, err := NewClient(ClientOptions{
	URL: serviceURL,
	})

	assert.Nil(t, err)
	defer client.Close()

	// Only subscribe on the specific partition
	consumer, err := client.Subscribe(ConsumerOptions{
	Topic: "my-partitioned-topic-partition-2",
	SubscriptionName: "my-sub",
	})

	assert.Nil(t, err)
	defer consumer.Close()

	producer, err := client.CreateProducer(ProducerOptions{
	Topic: "my-partitioned-topic",
	MessageRouter: func(msg *ProducerMessage, tm TopicMetadata) int {
	fmt.Println("Routing message ", msg, " -- Partitions: ", tm.NumPartitions())
	return 2
	},
	})

	assert.Nil(t, err)
	defer producer.Close()

	ctx := context.Background()

	err = producer.Send(ctx, &ProducerMessage{
	Payload: []byte("hello"),
	})
	assert.Nil(t, err)

	fmt.Println("PUBLISHED")

	// Verify message was published on partition 2
	msg, err := consumer.Receive(ctx)
	assert.Nil(t, err)
	assert.NotNil(t, msg)
	assert.Equal(t, string(msg.Payload()), "hello")
	}

	func TestNonPersistentTopic(t *testing.T) {
	topicName := "non-persistent://public/default/testNonPersistentTopic"
	client, err := NewClient(ClientOptions{
	URL: serviceURL,
	})
	assert.Nil(t, err)
	defer client.Close()

	producer, err := client.CreateProducer(ProducerOptions{
	Topic: topicName,
	})

	assert.Nil(t, err)
	defer producer.Close()

	consumer, err := client.Subscribe(ConsumerOptions{
	Topic: topicName,
	SubscriptionName: "my-sub",
	})
	assert.Nil(t, err)
	defer consumer.Close()
	}