optimized_inverted_bitmap_test.go

package matching

import (
	"testing"

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

func TestOptimizedInvertedBitmapMatcher(t *testing.T) {
	assert := assert.New(t)
	var (
		ib = NewOptimizedInvertedBitmapMatcher(5)
		s0 = 0
		s1 = 1
		s2 = 2
	)

	sub0, err := ib.Subscribe("forex.*", s0)
	assert.NoError(err)
	sub1, err := ib.Subscribe("*.usd", s0)
	assert.NoError(err)
	sub2, err := ib.Subscribe("forex.eur", s0)
	assert.NoError(err)
	sub3, err := ib.Subscribe("*.eur", s1)
	assert.NoError(err)
	sub4, err := ib.Subscribe("forex.*", s1)
	assert.NoError(err)
	sub5, err := ib.Subscribe("trade", s1)
	assert.NoError(err)
	sub6, err := ib.Subscribe("*", s2)
	assert.NoError(err)

	assertEqual(assert, []Subscriber{s0, s1}, ib.Lookup("forex.eur"))
	assertEqual(assert, []Subscriber{s2}, ib.Lookup("forex"))
	assertEqual(assert, []Subscriber{}, ib.Lookup("trade.jpy"))
	assertEqual(assert, []Subscriber{s0, s1}, ib.Lookup("forex.jpy"))
	assertEqual(assert, []Subscriber{s1, s2}, ib.Lookup("trade"))

	ib.Unsubscribe(sub0)
	ib.Unsubscribe(sub1)
	ib.Unsubscribe(sub2)
	ib.Unsubscribe(sub3)
	ib.Unsubscribe(sub4)
	ib.Unsubscribe(sub5)
	ib.Unsubscribe(sub6)

	assertEqual(assert, []Subscriber{}, ib.Lookup("forex.eur"))
	assertEqual(assert, []Subscriber{}, ib.Lookup("forex"))
	assertEqual(assert, []Subscriber{}, ib.Lookup("trade.jpy"))
	assertEqual(assert, []Subscriber{}, ib.Lookup("forex.jpy"))
	assertEqual(assert, []Subscriber{}, ib.Lookup("trade"))
}

func BenchmarkOptimizedInvertedBitmapMatcherSubscribe(b *testing.B) {
	var (
		ib = NewOptimizedInvertedBitmapMatcher(5)
		s0 = 0
	)
	populateMatcher(ib, 1000, 5)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		ib.Subscribe("foo.*.baz.qux.quux", s0)
	}
}

func BenchmarkOptimizedInvertedBitmapMatcherUnsubscribe(b *testing.B) {
	var (
		ib = NewOptimizedInvertedBitmapMatcher(5)
		s0 = 0
	)
	id, _ := ib.Subscribe("foo.*.baz.qux.quux", s0)
	populateMatcher(ib, 1000, 5)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		ib.Unsubscribe(id)
	}
}

func BenchmarkOptimizedInvertedBitmapMatcherLookup(b *testing.B) {
	var (
		ib = NewOptimizedInvertedBitmapMatcher(5)
		s0 = 0
	)
	ib.Subscribe("foo.*.baz.qux.quux", s0)
	populateMatcher(ib, 1000, 5)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		ib.Lookup("foo.bar.baz.qux.quux")
	}
}

func BenchmarkOptimizedInvertedBitmapMatcherSubscribeCold(b *testing.B) {
	var (
		ib = NewOptimizedInvertedBitmapMatcher(5)
		s0 = 0
	)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		ib.Subscribe("foo.*.baz.qux.quux", s0)
	}
}

func BenchmarkOptimizedInvertedBitmapMatcherUnsubscribeCold(b *testing.B) {
	var (
		ib = NewOptimizedInvertedBitmapMatcher(5)
		s0 = 0
	)
	id, _ := ib.Subscribe("foo.*.baz.qux.quux", s0)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		ib.Unsubscribe(id)
	}
}

func BenchmarkOptimizedInvertedBitmapMatcherLookupCold(b *testing.B) {
	var (
		ib = NewOptimizedInvertedBitmapMatcher(5)
		s0 = 0
	)
	ib.Subscribe("foo.*.baz.qux.quux", s0)

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		ib.Lookup("foo.bar.baz.qux.quux")
	}
}

func BenchmarkMultithreaded1Thread5050OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 1
	benchmark5050(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded2Thread5050OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 2
	benchmark5050(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded4Thread5050OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 4
	benchmark5050(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded8Thread5050OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 8
	benchmark5050(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded12Thread5050OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 12
	benchmark5050(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded16Thread5050OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 16
	benchmark5050(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded1Thread9010OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 1
	benchmark9010(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded2Thread9010OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 2
	benchmark9010(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded4Thread9010OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 4
	benchmark9010(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded8Thread9010OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 8
	benchmark9010(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded12Thread9010OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 12
	benchmark9010(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}

func BenchmarkMultithreaded16Thread9010OptimizedInvertedBitmap(b *testing.B) {
	numItems := 1000
	numThreads := 16
	benchmark9010(b, numItems, numThreads, func(items [][]string) Matcher {
		return NewOptimizedInvertedBitmapMatcher(uint(numItems))
	})
}