泛型

collections/generics/concurrent_map.go

@@ -0,0 +1,102 @@
+package generics
+
+import (
+	"fmt"
+	"runtime"
+	"sync"
+
+	"github.com/charlienet/go-mixed/hash"
+)
+
+var _ Map[string, string] = &ConcurrnetMap[string, string]{}
+
+var defaultNumOfBuckets = runtime.GOMAXPROCS(runtime.NumCPU())
+
+type ConcurrnetMap[K comparable, V any] struct {
+	buckets      []Map[K, V]
+	numOfBuckets uint64
+}
+
+func NewConcurrnetMap[K comparable, V any]() Map[K, V] {
+	num := defaultNumOfBuckets
+
+	buckets := make([]Map[K, V], num)
+	for i := 0; i < num; i++ {
+		buckets[i] = NewRWLockMap[K, V]()
+	}
+
+	return &ConcurrnetMap[K, V]{
+		numOfBuckets: uint64(num),
+		buckets:      buckets,
+	}
+}
+
+func (m *ConcurrnetMap[K, V]) Set(key K, value V) {
+	m.getBucket(key).Set(key, value)
+}
+
+func (m *ConcurrnetMap[K, V]) Get(key K) (V, bool) {
+	return m.getBucket(key).Get(key)
+}
+
+func (m *ConcurrnetMap[K, V]) Delete(key K) {
+	im := m.getBucket(key)
+	im.Delete(key)
+}
+
+func (m *ConcurrnetMap[K, V]) ForEach(f func(K, V)) {
+	var wg sync.WaitGroup
+
+	num := int(m.numOfBuckets)
+
+	wg.Add(int(m.numOfBuckets))
+	for i := 0; i < num; i++ {
+		go func(i int) {
+			m.buckets[i].ForEach(f)
+			wg.Done()
+		}(i)
+	}
+
+	wg.Wait()
+}
+
+func (m *ConcurrnetMap[K, V]) Clone() Map[K, V] {
+
+	num := int(m.numOfBuckets)
+
+	buckets := make([]Map[K, V], m.numOfBuckets)
+	for i := 0; i < num; i++ {
+		buckets[i] = m.buckets[i].Clone()
+	}
+
+	return &ConcurrnetMap[K, V]{
+		buckets:      buckets,
+		numOfBuckets: m.numOfBuckets,
+	}
+}
+
+func (m *ConcurrnetMap[K, V]) Clear() {
+	for i := 0; i < int(m.numOfBuckets); i++ {
+		m.buckets[i].Clear()
+	}
+}
+
+func (m *ConcurrnetMap[K, V]) Count() int {
+	var count int
+	for i := 0; i < int(m.numOfBuckets); i++ {
+		count += m.buckets[i].Count()
+	}
+
+	return count
+}
+
+func (m *ConcurrnetMap[K, V]) getBucket(k K) Map[K, V] {
+	bytes := getBytes(k)
+
+	id := hash.XXHashUint64(bytes) % m.numOfBuckets
+	return m.buckets[id]
+}
+
+func getBytes(k any) []byte {
+	return []byte(fmt.Sprintf("%v", k))
+}