utils_test.go

// Copyright (C) 2018 Tyler Treat <https://github.com/tylertreat>
// Licensed 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.

// Modifications copyright (C) 2018 Leandro Lugaresi

package hub

import (
	"math/rand"
	"strconv"
	"sync"
	"testing"

	"github.com/stretchr/testify/assert"
)

type discardSubscriber int

func (d discardSubscriber) Set(msg Message)    {}
func (d discardSubscriber) Ch() <-chan Message { return make(chan Message) }
func (d discardSubscriber) Close()             {}

var result []subscriber

func benchmarkMatcher(b *testing.B, numThreads int, m matcher, doSubs func(n int) bool) {
	numItems := 1000
	itemsToInsert := generateTopics(numThreads, numItems)
	sub := discardSubscriber(0)

	var wg sync.WaitGroup

	populateMatcher(m, 3)
	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		wg.Add(numThreads)

		for j := 0; j < numThreads; j++ {
			go func(j int) {
				var r []subscriber

				for n, key := range itemsToInsert[j] {
					if doSubs(n) {
						m.Subscribe([]string{key}, sub)
						continue
					}

					r = m.Lookup(key)
				}

				result = r

				wg.Done()
			}(j)
		}
		wg.Wait()
	}
}

func percentual5050(n int) bool {
	return n%2 == 0
}

func percentual9010(n int) bool {
	return n%10 == 0
}

func assertEqual(assert *assert.Assertions, expected, actual []subscriber) {
	assert.Len(actual, len(expected))

	for _, sub := range expected {
		assert.Contains(actual, sub)
	}
}

func generateTopics(numThreads, numItems int) [][]string {
	itemsToInsert := make([][]string, 0, numThreads)

	for i := 0; i < numThreads; i++ {
		items := make([]string, 0, numItems)

		for j := 0; j < numItems; j++ {
			topic := strconv.Itoa(j%10) + "." + strconv.Itoa(j%50) + "." + strconv.Itoa(j)
			items = append(items, topic)
		}

		itemsToInsert = append(itemsToInsert, items)
	}

	return itemsToInsert
}

func populateMatcher(m matcher, topicSize int) {
	num := 1000

	for i := 0; i < num; i++ {
		prefix := ""
		topic := ""

		for j := 0; j < topicSize; j++ {
			topic += prefix + strconv.Itoa(rand.Int())
			prefix = "."
		}
		m.Subscribe([]string{topic}, discardSubscriber(0))
	}
}