/usr/share/go-1.6/test/stress/maps.go is in golang-1.6-src 1.6.1-0ubuntu1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 | // Copyright 2013 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
package main
import (
"math/rand"
"runtime"
"sync"
)
func mapTypes() []MapType {
// TODO(bradfitz): bunch more map types of all different key and value types.
// Use reflect.MapOf and a program to generate lots of types & struct types.
// For now, just one:
return []MapType{intMapType{}}
}
type MapType interface {
NewMap() Map
}
type Map interface {
AddItem()
DelItem()
Len() int
GetItem()
RangeAll()
}
func stressMapType(mt MapType, done func()) {
defer done()
m := mt.NewMap()
for m.Len() < 10000 {
Println("map at ", m.Len())
if m.Len()%100 == 0 {
runtime.Gosched()
}
m.AddItem()
m.AddItem()
m.DelItem()
var wg sync.WaitGroup
const numGets = 10
wg.Add(numGets)
for i := 0; i < numGets; i++ {
go func(i int) {
if i&1 == 0 {
m.GetItem()
} else {
m.RangeAll()
}
wg.Done()
}(i)
}
wg.Wait()
}
for m.Len() > 0 {
m.DelItem()
}
}
type intMapType struct{}
func (intMapType) NewMap() Map {
return make(intMap)
}
var deadcafe = []byte("\xDE\xAD\xCA\xFE")
type intMap map[int][]byte
func (m intMap) AddItem() {
s0 := len(m)
for len(m) == s0 {
key := rand.Intn(s0 + 1)
m[key] = make([]byte, rand.Intn(64<<10))
}
}
func (m intMap) DelItem() {
for k := range m {
delete(m, k)
return
}
}
func (m intMap) GetItem() {
key := rand.Intn(len(m))
if s, ok := m[key]; ok {
copy(s, deadcafe)
}
}
func (m intMap) Len() int { return len(m) }
func (m intMap) RangeAll() {
for _ = range m {
}
for range m {
}
}
func stressMaps() {
for {
var wg sync.WaitGroup
for _, mt := range mapTypes() {
wg.Add(1)
go stressMapType(mt, wg.Done)
}
wg.Wait()
}
}
|