Util: Fix panic when generating UIDs concurrently (#69476)

This commit is contained in:
Jean-Philippe Quéméner
2023-06-05 11:21:11 +02:00
committed by GitHub
parent 1ed4c0382b
commit fe2103bfc8
2 changed files with 28 additions and 0 deletions
+20
View File
@@ -1,6 +1,7 @@
package util
import (
"sync"
"testing"
"github.com/google/uuid"
@@ -16,6 +17,25 @@ func TestAllowedCharMatchesUidPattern(t *testing.T) {
}
}
// Run with "go test -race -run ^TestThreadSafe$ github.com/grafana/grafana/pkg/util"
func TestThreadSafe(t *testing.T) {
// This test was used to showcase the bug, unfortunately there is
// no way to enable the -race flag programmatically.
t.Skip()
// Use 1000 go routines to create 100 UIDs each at roughly the same time.
var wg sync.WaitGroup
for i := 0; i < 1000; i++ {
go func() {
for ii := 0; ii < 100; ii++ {
_ = GenerateShortUID()
}
wg.Done()
}()
wg.Add(1)
}
wg.Wait()
}
func TestRandomUIDs(t *testing.T) {
for i := 0; i < 100; i++ {
v := GenerateShortUID()