EdgeAPI/internal/db/models/sys_locker_dao_test.go

package models

import (
	"sync"
	"testing"
	"time"

	_ "github.com/go-sql-driver/mysql"
	"github.com/iwind/TeaGo/dbs"
	"github.com/iwind/TeaGo/types"
)

func TestSysLockerDAO_Lock(t *testing.T) {
	var tx *dbs.Tx

	var dao = NewSysLockerDAO()

	isOk, err := dao.Lock(tx, "test", 600)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(isOk)

	if isOk {
		err = dao.Unlock(tx, "test")
		if err != nil {
			t.Fatal(err)
		}
	}
}

func TestSysLocker_Increase_SQL(t *testing.T) {
	var dao = NewSysLockerDAO()
	value, err := dao.Read(nil, "hello")
	if err != nil {
		t.Fatal(err)
	}
	t.Log("before:", value)
	v, err := dao.Increase(nil, "hello", 0)
	if err != nil {
		t.Log("err:", err)
		return
	}
	t.Log("after:", v)
}

func TestSysLocker_Increase_New_Key(t *testing.T) {
	var key = "KEY" + types.String(time.Now().Unix())

	var dao = NewSysLockerDAO()
	value, err := dao.Read(nil, key)
	if err != nil {
		t.Fatal(err)
	}
	t.Log("before:", value)
	for i := 0; i < 2; i++ {
		v, err := dao.Increase(nil, key, 0)
		if err != nil {
			t.Log("err:", err)
			return
		}
		t.Log("after:", v)
	}
}

func TestSysLocker_Increase_Cache(t *testing.T) {
	var dao = NewSysLockerDAO()
	for i := 0; i < 11; i++ {
		v, err := dao.Increase(nil, "hello", 0)
		if err != nil {
			t.Log("err:", err)
			return
		}
		t.Log("hello", i, "after:", v)
	}

	for i := 0; i < 11; i++ {
		v, err := dao.Increase(nil, "hello2", 0)
		if err != nil {
			t.Log("err:", err)
			return
		}
		t.Log("hello2", i, "after:", v)
	}
}

func TestSysLocker_Increase(t *testing.T) {
	dbs.NotifyReady()

	var dao = NewSysLockerDAO()
	dao.Instance.Raw().SetMaxOpenConns(64)

	var count = 1000

	value, err := dao.Read(nil, "hello")
	if err != nil {
		t.Fatal(err)
	}
	t.Log("before", value)

	var locker = sync.Mutex{}
	var allValueMap = map[int64]bool{}

	var before = time.Now()

	var wg = sync.WaitGroup{}
	wg.Add(count)

	for i := 0; i < count; i++ {
		go func(i int) {
			defer wg.Done()

			var key = "hello"
			v, err := dao.Increase(nil, key, 0)
			if err != nil {
				t.Log("err:", err)
				return
			}

			locker.Lock()
			if allValueMap[v] {
				t.Log("duplicated:", v)
			} else {
				allValueMap[v] = true
			}
			locker.Unlock()

			//t.Log("v:", v)
			_ = v
		}(i)
	}

	wg.Wait()

	t.Log("cost:", time.Since(before).Seconds()*1000, "ms")

	value, err = dao.Read(nil, "hello")
	if err != nil {
		t.Fatal(err)
	}

	t.Log("after", value, "values:", len(allValueMap))
}

func TestSysLocker_Increase_Performance(t *testing.T) {
	dbs.NotifyReady()

	var count = 1000

	var dao = NewSysLockerDAO()

	var before = time.Now()

	var wg = sync.WaitGroup{}
	wg.Add(count)

	for i := 0; i < count; i++ {
		go func(i int) {
			defer wg.Done()

			var key = "hello" + types.String(i%10)
			v, err := dao.Increase(nil, key, 0)
			if err != nil {
				t.Log("err:", err)
				return
			}
			//t.Log("v:", v)
			_ = v
		}(i)
	}

	wg.Wait()

	t.Log("cost:", time.Since(before).Seconds()*1000, "ms")
}

func BenchmarkSysLockerDAO_Increase(b *testing.B) {
	var dao = NewSysLockerDAO()
	dao.Instance.Raw().SetMaxOpenConns(64)

	_, _ = dao.Increase(nil, "hello", 0)

	b.ResetTimer()

	b.RunParallel(func(pb *testing.PB) {
		for pb.Next() {
			_, err := dao.Increase(nil, "hello", 0)
			if err != nil {
				b.Fatal(err)
			}
		}
	})
}