提交一些公共函数

internal/utils/sync/rw_mutex_test.go

@@ -0,0 +1,47 @@
+// Copyright 2023 Liuxiangchao iwind.liu@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package syncutils_test
+
+import (
+	syncutils "github.com/TeaOSLab/EdgeNode/internal/utils/sync"
+	"runtime"
+	"sync"
+	"testing"
+	"time"
+)
+
+func TestNewRWMutex(t *testing.T) {
+	var locker = syncutils.NewRWMutex(runtime.NumCPU())
+	locker.Lock(1)
+	t.Log(locker.TryLock(1))
+	locker.Unlock(1)
+	t.Log(locker.TryLock(1))
+}
+
+func BenchmarkRWMutex_Lock(b *testing.B) {
+	var locker = syncutils.NewRWMutex(runtime.NumCPU())
+
+	b.RunParallel(func(pb *testing.PB) {
+		var i = 0
+		for pb.Next() {
+			i++
+			locker.Lock(i)
+			time.Sleep(1 * time.Millisecond)
+			locker.Unlock(i)
+		}
+	})
+}
+
+func BenchmarkRWMutex_Lock2(b *testing.B) {
+	var locker = &sync.Mutex{}
+
+	b.RunParallel(func(pb *testing.PB) {
+		var i = 0
+		for pb.Next() {
+			i++
+			locker.Lock()
+			time.Sleep(1 * time.Millisecond)
+			locker.Unlock()
+		}
+	})
+}