qbit/go
1
0
Fork 0
mirror of https://github.com/golang/go synced 2024-11-26 11:48:03 -07:00
Code Releases Activity

sync: add HashTrieMap to Map tests and benchmarks

Browse Source 
Also, rename Map benchmarks to make them easier to single out via
regexp.

Change-Id: I4dcb066745aba1c340f56050d08539ae2976274d
Reviewed-on: https://go-review.googlesource.com/c/go/+/606461
Reviewed-by: David Chase <drchase@google.com>
LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com>
Auto-Submit: Michael Knyszek <mknyszek@google.com>
This commit is contained in:
Michael Anthony Knyszek 2024-08-13 20:51:43 +00:00 committed by Gopher Robot
parent 5c7b7c7d60
commit 50f1888814
3 changed files with 42 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
src/sync/map_bench_test.go
View File

@ -6,6 +6,7 @@ package sync_test
import ( import (
"fmt" "fmt"
isync "internal/sync"
"reflect" "reflect"
"sync" "sync"
"sync/atomic" "sync/atomic"
@ -18,13 +19,14 @@ type bench struct {
} }
func benchMap(b *testing.B, bench bench) { func benchMap(b *testing.B, bench bench) {
for _, m := range [...]mapInterface{&DeepCopyMap{}, &RWMutexMap{}, &sync.Map{}} { for _, m := range [...]mapInterface{&DeepCopyMap{}, &RWMutexMap{}, &isync.HashTrieMap[any, any]{}, &sync.Map{}} {
b.Run(fmt.Sprintf("%T", m), func(b *testing.B) { b.Run(fmt.Sprintf("%T", m), func(b *testing.B) {
m = reflect.New(reflect.TypeOf(m).Elem()).Interface().(mapInterface) m = reflect.New(reflect.TypeOf(m).Elem()).Interface().(mapInterface)
if bench.setup != nil { if bench.setup != nil {
bench.setup(b, m) bench.setup(b, m)
} }
b.ReportAllocs()
b.ResetTimer() b.ResetTimer()
var i int64 var i int64
@ -36,7 +38,7 @@ func benchMap(b *testing.B, bench bench) {
} }
} }
func BenchmarkLoadMostlyHits(b *testing.B) { func BenchmarkMapLoadMostlyHits(b *testing.B) {
const hits, misses = 1023, 1 const hits, misses = 1023, 1
benchMap(b, bench{ benchMap(b, bench{
@ -58,7 +60,7 @@ func BenchmarkLoadMostlyHits(b *testing.B) {
}) })
} }
func BenchmarkLoadMostlyMisses(b *testing.B) { func BenchmarkMapLoadMostlyMisses(b *testing.B) {
const hits, misses = 1, 1023 const hits, misses = 1, 1023
benchMap(b, bench{ benchMap(b, bench{
@ -80,7 +82,7 @@ func BenchmarkLoadMostlyMisses(b *testing.B) {
}) })
} }
func BenchmarkLoadOrStoreBalanced(b *testing.B) { func BenchmarkMapLoadOrStoreBalanced(b *testing.B) {
const hits, misses = 128, 128 const hits, misses = 128, 128
benchMap(b, bench{ benchMap(b, bench{
@ -114,7 +116,7 @@ func BenchmarkLoadOrStoreBalanced(b *testing.B) {
}) })
} }
func BenchmarkLoadOrStoreUnique(b *testing.B) { func BenchmarkMapLoadOrStoreUnique(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
setup: func(b *testing.B, m mapInterface) { setup: func(b *testing.B, m mapInterface) {
if _, ok := m.(*DeepCopyMap); ok { if _, ok := m.(*DeepCopyMap); ok {
@ -130,7 +132,7 @@ func BenchmarkLoadOrStoreUnique(b *testing.B) {
}) })
} }
func BenchmarkLoadOrStoreCollision(b *testing.B) { func BenchmarkMapLoadOrStoreCollision(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
setup: func(_ *testing.B, m mapInterface) { setup: func(_ *testing.B, m mapInterface) {
m.LoadOrStore(0, 0) m.LoadOrStore(0, 0)
@ -144,7 +146,7 @@ func BenchmarkLoadOrStoreCollision(b *testing.B) {
}) })
} }
func BenchmarkLoadAndDeleteBalanced(b *testing.B) { func BenchmarkMapLoadAndDeleteBalanced(b *testing.B) {
const hits, misses = 128, 128 const hits, misses = 128, 128
benchMap(b, bench{ benchMap(b, bench{
@ -174,7 +176,7 @@ func BenchmarkLoadAndDeleteBalanced(b *testing.B) {
}) })
} }
func BenchmarkLoadAndDeleteUnique(b *testing.B) { func BenchmarkMapLoadAndDeleteUnique(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
setup: func(b *testing.B, m mapInterface) { setup: func(b *testing.B, m mapInterface) {
if _, ok := m.(*DeepCopyMap); ok { if _, ok := m.(*DeepCopyMap); ok {
@ -190,7 +192,7 @@ func BenchmarkLoadAndDeleteUnique(b *testing.B) {
}) })
} }
func BenchmarkLoadAndDeleteCollision(b *testing.B) { func BenchmarkMapLoadAndDeleteCollision(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
setup: func(_ *testing.B, m mapInterface) { setup: func(_ *testing.B, m mapInterface) {
m.LoadOrStore(0, 0) m.LoadOrStore(0, 0)
@ -206,7 +208,7 @@ func BenchmarkLoadAndDeleteCollision(b *testing.B) {
}) })
} }
func BenchmarkRange(b *testing.B) { func BenchmarkMapRange(b *testing.B) {
const mapSize = 1 << 10 const mapSize = 1 << 10
benchMap(b, bench{ benchMap(b, bench{
@ -224,12 +226,12 @@ func BenchmarkRange(b *testing.B) {
}) })
} }
// BenchmarkAdversarialAlloc tests performance when we store a new value // BenchmarkMapAdversarialAlloc tests performance when we store a new value
// immediately whenever the map is promoted to clean and otherwise load a // immediately whenever the map is promoted to clean and otherwise load a
// unique, missing key. // unique, missing key.
// //
// This forces the Load calls to always acquire the map's mutex. // This forces the Load calls to always acquire the map's mutex.
func BenchmarkAdversarialAlloc(b *testing.B) { func BenchmarkMapAdversarialAlloc(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
perG: func(b *testing.B, pb *testing.PB, i int, m mapInterface) { perG: func(b *testing.B, pb *testing.PB, i int, m mapInterface) {
var stores, loadsSinceStore int64 var stores, loadsSinceStore int64
@ -245,12 +247,12 @@ func BenchmarkAdversarialAlloc(b *testing.B) {
}) })
} }
// BenchmarkAdversarialDelete tests performance when we periodically delete // BenchmarkMapAdversarialDelete tests performance when we periodically delete
// one key and add a different one in a large map. // one key and add a different one in a large map.
// //
// This forces the Load calls to always acquire the map's mutex and periodically // This forces the Load calls to always acquire the map's mutex and periodically
// makes a full copy of the map despite changing only one entry. // makes a full copy of the map despite changing only one entry.
func BenchmarkAdversarialDelete(b *testing.B) { func BenchmarkMapAdversarialDelete(b *testing.B) {
const mapSize = 1 << 10 const mapSize = 1 << 10
benchMap(b, bench{ benchMap(b, bench{
@ -276,7 +278,7 @@ func BenchmarkAdversarialDelete(b *testing.B) {
}) })
} }
func BenchmarkDeleteCollision(b *testing.B) { func BenchmarkMapDeleteCollision(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
setup: func(_ *testing.B, m mapInterface) { setup: func(_ *testing.B, m mapInterface) {
m.LoadOrStore(0, 0) m.LoadOrStore(0, 0)
@ -290,7 +292,7 @@ func BenchmarkDeleteCollision(b *testing.B) {
}) })
} }
func BenchmarkSwapCollision(b *testing.B) { func BenchmarkMapSwapCollision(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
setup: func(_ *testing.B, m mapInterface) { setup: func(_ *testing.B, m mapInterface) {
m.LoadOrStore(0, 0) m.LoadOrStore(0, 0)
@ -304,7 +306,7 @@ func BenchmarkSwapCollision(b *testing.B) {
}) })
} }
func BenchmarkSwapMostlyHits(b *testing.B) { func BenchmarkMapSwapMostlyHits(b *testing.B) {
const hits, misses = 1023, 1 const hits, misses = 1023, 1
benchMap(b, bench{ benchMap(b, bench{
@ -332,7 +334,7 @@ func BenchmarkSwapMostlyHits(b *testing.B) {
}) })
} }
func BenchmarkSwapMostlyMisses(b *testing.B) { func BenchmarkMapSwapMostlyMisses(b *testing.B) {
const hits, misses = 1, 1023 const hits, misses = 1, 1023
benchMap(b, bench{ benchMap(b, bench{
@ -360,7 +362,7 @@ func BenchmarkSwapMostlyMisses(b *testing.B) {
}) })
} }
func BenchmarkCompareAndSwapCollision(b *testing.B) { func BenchmarkMapCompareAndSwapCollision(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
setup: func(_ *testing.B, m mapInterface) { setup: func(_ *testing.B, m mapInterface) {
m.LoadOrStore(0, 0) m.LoadOrStore(0, 0)
@ -376,7 +378,7 @@ func BenchmarkCompareAndSwapCollision(b *testing.B) {
}) })
} }
func BenchmarkCompareAndSwapNoExistingKey(b *testing.B) { func BenchmarkMapCompareAndSwapNoExistingKey(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
perG: func(b *testing.B, pb *testing.PB, i int, m mapInterface) { perG: func(b *testing.B, pb *testing.PB, i int, m mapInterface) {
for ; pb.Next(); i++ { for ; pb.Next(); i++ {
@ -388,7 +390,7 @@ func BenchmarkCompareAndSwapNoExistingKey(b *testing.B) {
}) })
} }
func BenchmarkCompareAndSwapValueNotEqual(b *testing.B) { func BenchmarkMapCompareAndSwapValueNotEqual(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
setup: func(_ *testing.B, m mapInterface) { setup: func(_ *testing.B, m mapInterface) {
m.Store(0, 0) m.Store(0, 0)
@ -402,7 +404,7 @@ func BenchmarkCompareAndSwapValueNotEqual(b *testing.B) {
}) })
} }
func BenchmarkCompareAndSwapMostlyHits(b *testing.B) { func BenchmarkMapCompareAndSwapMostlyHits(b *testing.B) {
const hits, misses = 1023, 1 const hits, misses = 1023, 1
benchMap(b, bench{ benchMap(b, bench{
@ -432,7 +434,7 @@ func BenchmarkCompareAndSwapMostlyHits(b *testing.B) {
}) })
} }
func BenchmarkCompareAndSwapMostlyMisses(b *testing.B) { func BenchmarkMapCompareAndSwapMostlyMisses(b *testing.B) {
const hits, misses = 1, 1023 const hits, misses = 1, 1023
benchMap(b, bench{ benchMap(b, bench{
@ -458,7 +460,7 @@ func BenchmarkCompareAndSwapMostlyMisses(b *testing.B) {
}) })
} }
func BenchmarkCompareAndDeleteCollision(b *testing.B) { func BenchmarkMapCompareAndDeleteCollision(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
setup: func(_ *testing.B, m mapInterface) { setup: func(_ *testing.B, m mapInterface) {
m.LoadOrStore(0, 0) m.LoadOrStore(0, 0)
@ -474,7 +476,7 @@ func BenchmarkCompareAndDeleteCollision(b *testing.B) {
}) })
} }
func BenchmarkCompareAndDeleteMostlyHits(b *testing.B) { func BenchmarkMapCompareAndDeleteMostlyHits(b *testing.B) {
const hits, misses = 1023, 1 const hits, misses = 1023, 1
benchMap(b, bench{ benchMap(b, bench{
@ -506,7 +508,7 @@ func BenchmarkCompareAndDeleteMostlyHits(b *testing.B) {
}) })
} }
func BenchmarkCompareAndDeleteMostlyMisses(b *testing.B) { func BenchmarkMapCompareAndDeleteMostlyMisses(b *testing.B) {
const hits, misses = 1, 1023 const hits, misses = 1, 1023
benchMap(b, bench{ benchMap(b, bench{
@ -534,7 +536,7 @@ func BenchmarkCompareAndDeleteMostlyMisses(b *testing.B) {
}) })
} }
func BenchmarkClear(b *testing.B) { func BenchmarkMapClear(b *testing.B) {
benchMap(b, bench{ benchMap(b, bench{
perG: func(b *testing.B, pb *testing.PB, i int, m mapInterface) { perG: func(b *testing.B, pb *testing.PB, i int, m mapInterface) {
for ; pb.Next(); i++ { for ; pb.Next(); i++ {

2
src/sync/map_reference_test.go
View File

@ -5,6 +5,7 @@
package sync_test package sync_test
import ( import (
isync "internal/sync"
"sync" "sync"
"sync/atomic" "sync/atomic"
) )
@ -28,6 +29,7 @@ type mapInterface interface {
var ( var (
_ mapInterface = &RWMutexMap{} _ mapInterface = &RWMutexMap{}
_ mapInterface = &DeepCopyMap{} _ mapInterface = &DeepCopyMap{}
_ mapInterface = &isync.HashTrieMap[any, any]{}
) )
// RWMutexMap is an implementation of mapInterface using a sync.RWMutex. // RWMutexMap is an implementation of mapInterface using a sync.RWMutex.

11
src/sync/map_test.go
View File

@ -5,6 +5,7 @@
package sync_test package sync_test
import ( import (
isync "internal/sync"
"internal/testenv" "internal/testenv"
"math/rand" "math/rand"
"reflect" "reflect"
@ -133,6 +134,10 @@ func applyDeepCopyMap(calls []mapCall) ([]mapResult, map[any]any) {
return applyCalls(new(DeepCopyMap), calls) return applyCalls(new(DeepCopyMap), calls)
} }
func applyHashTrieMap(calls []mapCall) ([]mapResult, map[any]any) {
return applyCalls(new(isync.HashTrieMap[any, any]), calls)
}
func TestMapMatchesRWMutex(t *testing.T) { func TestMapMatchesRWMutex(t *testing.T) {
if err := quick.CheckEqual(applyMap, applyRWMutexMap, nil); err != nil { if err := quick.CheckEqual(applyMap, applyRWMutexMap, nil); err != nil {
t.Error(err) t.Error(err)
@ -145,6 +150,12 @@ func TestMapMatchesDeepCopy(t *testing.T) {
} }
} }
func TestMapMatchesHashTrieMap(t *testing.T) {
if err := quick.CheckEqual(applyMap, applyHashTrieMap, nil); err != nil {
t.Error(err)
}
}
func TestConcurrentRange(t *testing.T) { func TestConcurrentRange(t *testing.T) {
const mapSize = 1 << 10 const mapSize = 1 << 10