Initial commit (code only without large binaries)

EdgeNode/internal/utils/sync/values.go

@@ -0,0 +1,91 @@
+// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package syncutils
+
+import (
+	"github.com/TeaOSLab/EdgeNode/internal/utils/procs"
+	"runtime"
+	"sync"
+)
+
+type Values[T any] struct {
+	count  int
+	values []T
+	muList []*sync.RWMutex
+}
+
+func NewValues[T any](newFn func() T) *Values[T] {
+	var countProcs = runtime.GOMAXPROCS(0)
+	if countProcs < 0 {
+		countProcs = 32
+	}
+
+	var values []T
+	for i := 0; i < countProcs; i++ {
+		values = append(values, newFn())
+	}
+
+	var muList []*sync.RWMutex
+	for i := 0; i < countProcs; i++ {
+		muList = append(muList, &sync.RWMutex{})
+	}
+
+	return &Values[T]{
+		count:  countProcs,
+		values: values,
+		muList: muList,
+	}
+}
+
+func (this *Values[T]) Len() int {
+	return this.count
+}
+
+func (this *Values[T]) DoReadAll(f func(value T)) {
+	for i := 0; i < this.count; i++ {
+		var mu = this.muList[i]
+		var value = this.values[i]
+		mu.RLock()
+		f(value)
+		mu.RUnlock()
+	}
+}
+
+func (this *Values[T]) DoRead(f func(value T)) {
+	var procId = procs.Id()
+	var mu = this.muList[procId]
+	var value = this.values[procId]
+	mu.RLock()
+	f(value)
+	mu.RUnlock()
+}
+
+func (this *Values[T]) DoWriteAll(f func(value T)) {
+	for i := 0; i < this.count; i++ {
+		var mu = this.muList[i]
+		var value = this.values[i]
+		mu.Lock()
+		f(value)
+		mu.Unlock()
+	}
+}
+
+func (this *Values[T]) DoWrite(f func(value T)) {
+	var procId = procs.Id()
+	var mu = this.muList[procId]
+	var value = this.values[procId]
+	mu.Lock()
+	f(value)
+	mu.Unlock()
+}
+
+func (this *Values[T]) DoUnsafe(f func(value *T)) {
+	var procId = procs.Id()
+	f(&this.values[procId])
+}
+
+func (this *Values[T]) DoUnsafeAll(f func(value T)) {
+	for _, value := range this.values {
+		f(value)
+	}
+}