feat: order id generator.

order_id/id_test.go

@@ -0,0 +1,63 @@
+package order_id
+
+import (
+	"fmt"
+	"sync"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+
+	"github.com/tx7do/kratos-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)
+}