uuid generator supports generating more than 1 uuid in one time

2026-05-16 16:07:33 +08:00 · 2023-03-26 23:58:30 +08:00
parent 1d6dbf63c0
commit 7443e8a532
5 changed files with 170 additions and 16 deletions
@@ -121,3 +121,126 @@ func TestGenerateUuid_1000000TimesConcurrent(t *testing.T) {

 	waitGroup.Wait()
 }
+
+func TestGenerateUuids_Count0(t *testing.T) {
+	expectedUuidServerId := uint8(90)
+	expectedUuidType := UUID_TYPE_TRANSACTION
+
+	generator, _ := NewInternalUuidGenerator(&settings.Config{UuidServerId: expectedUuidServerId})
+	uuids := generator.GenerateUuids(expectedUuidType, 0)
+
+	assert.NotEqual(t, nil, uuids)
+	assert.Equal(t, 0, len(uuids))
+}
+
+func TestGenerateUuids_Count255(t *testing.T) {
+	expectedUnixTime := time.Now().Unix()
+	expectedUuidServerId := uint8(90)
+	expectedUuidType := UUID_TYPE_TRANSACTION
+	expectedUuidCount := uint8(255)
+
+	generator, _ := NewInternalUuidGenerator(&settings.Config{UuidServerId: expectedUuidServerId})
+	uuids := generator.GenerateUuids(expectedUuidType, expectedUuidCount)
+
+	for i := 0; i < int(expectedUuidCount); i++ {
+		uuidInfo := generator.ParseUuidInfo(uuids[i])
+
+		actualUnixTime := uuidInfo.UnixTime
+		assert.Equal(t, uint32(expectedUnixTime), actualUnixTime)
+
+		actualUuidServerId := uuidInfo.UuidServerId
+		assert.Equal(t, expectedUuidServerId, actualUuidServerId)
+
+		actualUuidType := uuidInfo.UuidType
+		assert.Equal(t, uint8(expectedUuidType), actualUuidType)
+
+		expectedSeqId := i
+		actualSeqId := uuidInfo.SequentialId
+		assert.Equal(t, uint32(expectedSeqId), actualSeqId)
+	}
+
+	assert.Equal(t, int(expectedUuidCount), len(uuids))
+}
+
+func TestGenerateUuids_30TimesIn3Seconds(t *testing.T) {
+	expectedUuidServerId := uint8(90)
+	expectedUuidType := UUID_TYPE_TRANSACTION
+	expectedUuidCount := uint8(255)
+
+	generator, _ := NewInternalUuidGenerator(&settings.Config{UuidServerId: expectedUuidServerId})
+
+	for cycle := 0; cycle < 30; cycle++ {
+		expectedUnixTime := time.Now().Unix()
+		uuids := generator.GenerateUuids(expectedUuidType, expectedUuidCount)
+		var firstSeqId uint32
+
+		for i := 0; i < int(expectedUuidCount); i++ {
+			uuidInfo := generator.ParseUuidInfo(uuids[i])
+
+			actualUnixTime := uuidInfo.UnixTime
+			assert.Equal(t, uint32(expectedUnixTime), actualUnixTime)
+
+			actualUuidServerId := uuidInfo.UuidServerId
+			assert.Equal(t, expectedUuidServerId, actualUuidServerId)
+
+			actualUuidType := uuidInfo.UuidType
+			assert.Equal(t, uint8(expectedUuidType), actualUuidType)
+
+			if i == 0 {
+				firstSeqId = uuidInfo.SequentialId
+			} else {
+				expectedSeqId := firstSeqId + uint32(i)
+				actualSeqId := uuidInfo.SequentialId
+				assert.Equal(t, expectedSeqId, actualSeqId)
+			}
+		}
+
+		time.Sleep(100 * time.Millisecond)
+	}
+}
+
+func TestGenerateUuids_1000000TimesConcurrent(t *testing.T) {
+	concurrentCount := 50
+	expectedUuidCount := uint8(100)
+	generator, _ := NewInternalUuidGenerator(&settings.Config{UuidServerId: 3})
+	var mutex sync.Mutex
+	var generatedIds sync.Map
+	var waitGroup sync.WaitGroup
+
+	for routineIndex := 0; routineIndex < concurrentCount; routineIndex++ {
+		go func() {
+			waitGroup.Add(1)
+
+			for cycle := 0; cycle < 400; cycle++ {
+				if cycle%100 == 0 { // each server can only generate 500,000 (50 * 10000) uuids in one second
+					time.Sleep(1000 * time.Millisecond)
+				}
+
+				expectedUnixTime := time.Now().Unix()
+				uuids := generator.GenerateUuids(UUID_TYPE_USER, expectedUuidCount)
+
+				for i := 0; i < int(expectedUuidCount); i++ {
+					uuidInfo := generator.ParseUuidInfo(uuids[i])
+
+					if uint32(expectedUnixTime) != uuidInfo.UnixTime {
+						mutex.Lock()
+						assert.Equal(t, uint32(expectedUnixTime), uuidInfo.UnixTime)
+						mutex.Unlock()
+					}
+
+					if existedUnixTime, exists := generatedIds.Load(uuids[i]); exists {
+						mutex.Lock()
+						assert.Fail(t, fmt.Sprintf("uuid \"%d\" conflicts, seq id is %d, existed unixtime is %d, current unix time is %d", uuids[i], uuidInfo.SequentialId, existedUnixTime, uuidInfo.UnixTime))
+						mutex.Unlock()
+					}
+
+					generatedIds.Store(uuids[i], uuidInfo.UnixTime)
+				}
+			}
+
+			waitGroup.Done()
+		}()
+	}
+
+	waitGroup.Wait()
+}