64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package order_id
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/tx7do/go-utils/trans"
|
|
)
|
|
|
|
func TestGenerateOrderIdWithRandom(t *testing.T) {
|
|
fmt.Println(GenerateOrderIdWithRandom("PT", "-", trans.Time(time.Now())))
|
|
|
|
tm := time.Now()
|
|
var ids map[string]bool
|
|
ids = make(map[string]bool)
|
|
count := 100
|
|
for i := 0; i < count; i++ {
|
|
ids[GenerateOrderIdWithRandom("PT", "", trans.Time(tm))] = true
|
|
}
|
|
assert.Equal(t, count, len(ids))
|
|
}
|
|
|
|
func TestGenerateOrderIdWithIndex(t *testing.T) {
|
|
tm := time.Now()
|
|
|
|
fmt.Println(GenerateOrderIdWithIncreaseIndex("PT", "", trans.Time(tm)))
|
|
|
|
ids := make(map[string]bool)
|
|
count := 100
|
|
for i := 0; i < count; i++ {
|
|
ids[GenerateOrderIdWithIncreaseIndex("PT", "", trans.Time(tm))] = true
|
|
}
|
|
assert.Equal(t, count, len(ids))
|
|
}
|
|
|
|
func TestGenerateOrderIdWithIndexThread(t *testing.T) {
|
|
tm := time.Now()
|
|
|
|
var wg sync.WaitGroup
|
|
var ids sync.Map
|
|
for i := 0; i < 10; i++ {
|
|
wg.Add(1)
|
|
go func() {
|
|
for i := 0; i < 100; i++ {
|
|
id := GenerateOrderIdWithIncreaseIndex("PT", "", trans.Time(tm))
|
|
ids.Store(id, true)
|
|
}
|
|
wg.Done()
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
|
|
aLen := 0
|
|
ids.Range(func(k, v interface{}) bool {
|
|
aLen++
|
|
return true
|
|
})
|
|
assert.Equal(t, 1000, aLen)
|
|
}
|