benchmarks_test.go

// Copyright 2016 Canonical Ltd.
// Licensed under the LGPLv3, see LICENCE file for details.

package pubsub_test

import (
	"time"

	jc "github.com/juju/testing/checkers"
	gc "gopkg.in/check.v1"

	"github.com/juju/pubsub/v2"
)

type BenchmarkSuite struct{}

var _ = gc.Suite(&BenchmarkSuite{})

func (*BenchmarkSuite) BenchmarkStructuredNoConversions(c *gc.C) {
	hub := pubsub.NewStructuredHub(nil)
	topic := "benchmarking"
	counter := 0
	unsub, err := hub.SubscribeMatch(pubsub.MatchAll, func(topic string, data map[string]interface{}) {
		counter++
	})
	c.Assert(err, jc.ErrorIsNil)
	defer unsub()
	failedCount := 0
	for i := 0; i < c.N; i++ {
		done, err := hub.Publish(topic, nil)
		c.Assert(err, jc.ErrorIsNil)

		select {
		case <-pubsub.Wait(done):
		case <-time.After(time.Second):
			failedCount++
		}
	}
	c.Check(failedCount, gc.Equals, 0)
	c.Check(counter, jc.GreaterThan, 0)
}

func (*BenchmarkSuite) BenchmarkStructuredSerialize(c *gc.C) {
	hub := pubsub.NewStructuredHub(nil)
	topic := "benchmarking"
	counter := 0
	unsub, err := hub.SubscribeMatch(pubsub.MatchAll, func(topic string, data Emission, err error) {
		c.Assert(err, jc.ErrorIsNil)
		counter++
	})
	c.Assert(err, jc.ErrorIsNil)
	defer unsub()
	failedCount := 0
	data := Emission{
		Origin:  "master",
		Message: "hello world",
		ID:      42,
	}
	for i := 0; i < c.N; i++ {
		done, err := hub.Publish(topic, data)
		c.Assert(err, jc.ErrorIsNil)

		select {
		case <-pubsub.Wait(done):
		case <-time.After(time.Second):
			failedCount++
		}
	}
	c.Check(failedCount, gc.Equals, 0)
	c.Check(counter, jc.GreaterThan, 0)
}