map_benchmark_test.go

package ordered_sync_map_test

import (
	"fmt"
	"testing"

	mp "github.com/m-murad/ordered-sync-map"
)

func getPopulatedOrderedSyncMap(size int) *mp.Map[any, any] {
	m := mp.New[any, any]()
	populateOrderedSyncMap(m, size)
	return m
}

func populateOrderedSyncMap(m *mp.Map[any, any], size int) {
	for i := 0; i < size; i++ {
		m.Put(i, i)
	}
}

func BenchmarkOrderedSyncMapGet(b *testing.B) {
	mapSize := 1000
	m := initMap()
	populateOrderedSyncMap(m, mapSize)
	for n := 1; n <= 10; n++ {
		b.Run(fmt.Sprintf("Get form ordered_sync_map - %d", n), func(b *testing.B) {
			for i := 0; i < b.N; i++ {
				_, _ = m.Get(b.N % mapSize)
			}
		})
	}
}

func BenchmarkOrderedSyncMapPut(b *testing.B) {
	for n := 1; n <= 10; n++ {
		m := mp.New[any, any]()
		b.Run(fmt.Sprintf("Put in ordered_sync_map - %d", n), func(b *testing.B) {
			populateOrderedSyncMap(m, b.N)
		})
	}
}

func BenchmarkOrderedSyncMapDelete(b *testing.B) {
	for n := 1; n < 10; n++ {
		b.Run(fmt.Sprintf("Delete form ordered_sync_map - %d", n), func(b *testing.B) {
			size := b.N
			m := getPopulatedOrderedSyncMap(size)
			b.ResetTimer()
			for i := 0; i < size; i++ {
				m.Delete(i)
			}
		})
	}
}

func BenchmarkOrderedSyncMapUnorderedTraversal(b *testing.B) {
	for n := 1; n < 5; n++ {
		b.Run(fmt.Sprintf("Traverse ordered_sync_map randomly - %d", n), func(b *testing.B) {
			m := getPopulatedOrderedSyncMap(b.N)
			b.ResetTimer()
			m.UnorderedRange(func(key interface{}, value interface{}) {})
		})
	}
}

func BenchmarkOrderedSyncMapOrderedTraversal(b *testing.B) {
	for n := 1; n < 5; n++ {
		b.Run(fmt.Sprintf("Traverse ordered_sync_map in order - %d", n), func(b *testing.B) {
			m := getPopulatedOrderedSyncMap(b.N)
			b.ResetTimer()
			m.OrderedRange(func(key interface{}, value interface{}) {})
		})
	}
}