github.com/coyove/sdss@v0.0.0-20231129015646-c2ec58cca6a2/future/clock_test.go

package future

import (
	"fmt"
	"math/rand"
	"sync"
	"sync/atomic"
	"testing"
	"time"
)

func TestNow(t *testing.T) {
	StartWatcher(func(e error) {
		fmt.Println(e)
	})

	dedup := map[Future]bool{}
	mu := sync.Mutex{}

	for i := 0; i < 20; i++ {
		wg := sync.WaitGroup{}
		var tot int64
		const N = 5000
		for i := 0; i < N; i++ {
			wg.Add(1)
			go func() {
				start := UnixNano()
				ts := Get(1) // int64(rand.Intn(Channels)))
				// fmt.Println(ts.Channel(), ts.IsFixed(), ts.Fixed(), Future(ts.Fixed()).IsFixed())
				v := uint16(rand.Int())
				v0, _ := ts.ToCookie(v).Cookie()
				if v0 != v {
					panic("invalid cookie")
				}

				mu.Lock()
				if dedup[ts] {
					panic("duplicated id")
				}
				dedup[ts] = true
				mu.Unlock()

				ts.Wait()
				diff := UnixNano() - start
				atomic.AddInt64(&tot, diff)
				wg.Done()
			}()
		}
		wg.Wait()
		fmt.Println(tot / N / 1e6)
	}
}

func TestFuzzyID(t *testing.T) {
	const N = 500
	rand.Seed(UnixNano())
	m := sync.Map{}
	test = true
	for i := 0; i < 5; i++ {
		wg := sync.WaitGroup{}
		for j := 0; j < N/5; j++ {
			wg.Add(1)
			go func(i int) {
				defer wg.Done()
				id := Get(1)
				_, loaded := m.LoadOrStore(id, 1)
				if loaded {
					panic("duplicated id")
				}
				time.Sleep(time.Duration(rand.Intn(1000)+500) * time.Millisecond)
			}(i*100 + j)
		}
		wg.Wait()
	}
}