Use goroutine-safe version of rand.Source

LK4D4 · LK4D4 · commit d8661250e71a · 2015-05-19T12:32:40.000-07:00
Signed-off-by: Alexander Morozov &lt;lk4d4@docker.com&gt;
diff --git a/pkg/namesgenerator/names-generator.go b/pkg/namesgenerator/names-generator.go
@@ -3,7 +3,8 @@ package namesgenerator
 import (
 	"fmt"
 	"math/rand"
-	"time"
+
+	"github.com/docker/docker/pkg/random"
 )
 
 var (
@@ -309,7 +310,7 @@ var (
 		"yonath",
 	}
 
-	rnd = rand.New(rand.NewSource(time.Now().UnixNano()))
+	rnd = rand.New(random.NewSource())
 )
 
 func GetRandomName(retry int) string {
diff --git a/pkg/random/random.go b/pkg/random/random.go
@@ -0,0 +1,34 @@
+package random
+
+import (
+	"math/rand"
+	"sync"
+	"time"
+)
+
+// copypaste from standard math/rand
+type lockedSource struct {
+	lk  sync.Mutex
+	src rand.Source
+}
+
+func (r *lockedSource) Int63() (n int64) {
+	r.lk.Lock()
+	n = r.src.Int63()
+	r.lk.Unlock()
+	return
+}
+
+func (r *lockedSource) Seed(seed int64) {
+	r.lk.Lock()
+	r.src.Seed(seed)
+	r.lk.Unlock()
+}
+
+// NewSource returns math/rand.Source safe for concurrent use and initialized
+// with current unix-nano timestamp
+func NewSource() rand.Source {
+	return &lockedSource{
+		src: rand.NewSource(time.Now().UnixNano()),
+	}
+}
diff --git a/pkg/random/random_test.go b/pkg/random/random_test.go
@@ -0,0 +1,22 @@
+package random
+
+import (
+	"math/rand"
+	"sync"
+	"testing"
+)
+
+// for go test -v -race
+func TestConcurrency(t *testing.T) {
+	rnd := rand.New(NewSource())
+	var wg sync.WaitGroup
+
+	for i := 0; i < 10; i++ {
+		wg.Add(1)
+		go func() {
+			rnd.Int63()
+			wg.Done()
+		}()
+	}
+	wg.Wait()
+}
diff --git a/pkg/stringutils/stringutils.go b/pkg/stringutils/stringutils.go
@@ -2,17 +2,18 @@ package stringutils
 
 import (
 	"bytes"
-	mathrand "math/rand"
+	"math/rand"
 	"strings"
-	"time"
+
+	"github.com/docker/docker/pkg/random"
 )
 
 // Generate alpha only random stirng with length n
 func GenerateRandomAlphaOnlyString(n int) string {
 	// make a really long string
 	letters := []byte("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")
 	b := make([]byte, n)
-	r := mathrand.New(mathrand.NewSource(time.Now().UTC().UnixNano()))
+	r := rand.New(random.NewSource())
 	for i := range b {
 		b[i] = letters[r.Intn(len(letters))]
 	}
@@ -26,7 +27,7 @@ func GenerateRandomAsciiString(n int) string {
 		"~!@#$%^&*()-_+={}[]\\|<,>.?/\"';:` "
 	res := make([]byte, n)
 	for i := 0; i < n; i++ {
-		res[i] = chars[mathrand.Intn(len(chars))]
+		res[i] = chars[rand.Intn(len(chars))]
 	}
 	return string(res)
 }

Original file line number	Diff line number	Diff line change
`@@ -3,7 +3,8 @@ package namesgenerator`
`3`	`3`	`import (`
`4`	`4`	`"fmt"`
`5`	`5`	`"math/rand"`
`6`		`- "time"`
	`6`	`+`
	`7`	`+ "github.com/docker/docker/pkg/random"`
`7`	`8`	`)`
`8`	`9`
`9`	`10`	`var (`
`@@ -309,7 +310,7 @@ var (`
`309`	`310`	`"yonath",`
`310`	`311`	`}`
`311`	`312`
`312`		`- rnd = rand.New(rand.NewSource(time.Now().UnixNano()))`
	`313`	`+ rnd = rand.New(random.NewSource())`
`313`	`314`	`)`
`314`	`315`
`315`	`316`	`func GetRandomName(retry int) string {`