feat(pool): add checkExists method to debug build to detect unexpected behavior

internal/utils/pool/pool.go

@@ -3,25 +3,23 @@ package pool
 import (
 	"sort"
 
+	"github.com/puzpuzpuz/xsync/v3"
 	"github.com/yusing/go-proxy/internal/logging"
-	"github.com/yusing/go-proxy/internal/utils"
-	"github.com/yusing/go-proxy/internal/utils/functional"
 )
 
 type (
 	Pool[T Object] struct {
-		m    functional.Map[string, T]
+		m    *xsync.MapOf[string, T]
 		name string
 	}
 	Object interface {
 		Key() string
 		Name() string
-		utils.MapMarshaler
 	}
 )
 
 func New[T Object](name string) Pool[T] {
-	return Pool[T]{functional.NewMapOf[string, T](), name}
+	return Pool[T]{xsync.NewMapOf[string, T](), name}
 }
 
 func (p Pool[T]) Name() string {
@@ -29,6 +27,7 @@ func (p Pool[T]) Name() string {
 }
 
 func (p Pool[T]) Add(obj T) {
+	p.checkExists(obj.Key())
 	p.m.Store(obj.Key(), obj)
 	logging.Info().Msgf("%s: added %s", p.name, obj.Name())
 }
@@ -50,8 +49,8 @@ func (p Pool[T]) Clear() {
 	p.m.Clear()
 }
 
-func (p Pool[T]) Base() functional.Map[string, T] {
+func (p Pool[T]) Iter(fn func(k string, v T) bool) {
-	return p.m
+	p.m.Range(fn)
 }
 
 func (p Pool[T]) Slice() []T {
@@ -64,11 +63,3 @@ func (p Pool[T]) Slice() []T {
 	})
 	return slice
 }
-
-func (p Pool[T]) Iter(fn func(k string, v T) bool) {
-	p.m.Range(fn)
-}
-
-func (p Pool[T]) IterAll(fn func(k string, v T)) {
-	p.m.RangeAll(fn)
-}

internal/utils/pool/pool_debug.go

@@ -0,0 +1,15 @@
+//go:build debug
+
+package pool
+
+import (
+	"runtime/debug"
+
+	"github.com/yusing/go-proxy/internal/logging"
+)
+
+func (p Pool[T]) checkExists(key string) {
+	if _, ok := p.m.Load(key); ok {
+		logging.Warn().Msgf("%s: key %s already exists\nstacktrace: %s", p.name, key, string(debug.Stack()))
+	}
+}

internal/utils/pool/pool_prod.go

@@ -0,0 +1,7 @@
+//go:build !debug
+
+package pool
+
+func (p Pool[T]) checkExists(key string) {
+	// no-op in production
+}