bench.go

// .... modified *_test to use an interface
func BenchmarkLoadOrStoreAddOnValue(b *testing.B) {
	benchMap(b, bench{
		setup: func(b *testing.B, m mapInterface) {
			if _, ok := m.(*DeepCopyMap); ok {
				b.Skip("DeepCopyMap has quadratic running time.")
			}
		},

		perG: func(b *testing.B, pb *testing.PB, i int, m mapInterface) {
			for ; pb.Next(); i++ {
				v, _ := m.LoadOrStore(i%10, new(uint64))
				atomic.AddUint64(v.(*uint64), 1)
			}
		},
	})
}

func BenchmarkLoadOrStoreFnAddOnValue(b *testing.B) {
	fn := func() interface{} { return new(uint64) }
	benchMap(b, bench{
		setup: func(b *testing.B, m mapInterface) {
			if _, ok := m.(*DeepCopyMap); ok {
				b.Skip("DeepCopyMap has quadratic running time.")
			}
		},

		perG: func(b *testing.B, pb *testing.PB, i int, m mapInterface) {
			for ; pb.Next(); i++ {
				v, _ := m.LoadOrStoreFn(i%10, fn)
				atomic.AddUint64(v.(*uint64), 1)
			}
		},
	})
}