extend models

Author: markus621

data.go

@@ -19,6 +19,13 @@ func NewMap[K comparable, V any](size uint) *Map[K, V] {
 	}
 }
 
+func (v *Map[K, V]) Size() int {
+	v.mux.RLock()
+	defer v.mux.RUnlock()
+
+	return len(v.data)
+}
+
 func (v *Map[K, V]) Set(key K, val V) {
 	v.mux.Lock()
 	defer v.mux.Unlock()
@@ -77,16 +84,23 @@ func (v *Map[K, V]) Reset() {
 
 type Slice[V any] struct {
 	data []V
-	mux  sync.Mutex
+	mux  sync.RWMutex
 }
 
 func NewSlice[V any](size uint) *Slice[V] {
 	return &Slice[V]{
 		data: make([]V, 0, size),
-		mux:  sync.Mutex{},
+		mux:  sync.RWMutex{},
 	}
 }
 
+func (v *Slice[V]) Size() int {
+	v.mux.RLock()
+	defer v.mux.RUnlock()
+
+	return len(v.data)
+}
+
 func (v *Slice[V]) Append(val ...V) {
 	v.mux.Lock()
 	defer v.mux.Unlock()

group.go

@@ -25,8 +25,7 @@ type (
 		mux       sync.RWMutex
 		globalCtx context.Context
 		cancelCtx context.CancelFunc
-
-		onPanic func(err error)
+		onPanic   func(err error)
 	}
 )
 
@@ -51,29 +50,20 @@ func (v *_group) Wait() {
 
 func (v *_group) Cancel() {
 	v.cancelCtx()
+	v.wg.Wait()
 }
 
 func (v *_group) Background(name string, call func(ctx context.Context)) {
 	v.wg.Add(1)
-
-	go func() {
-		defer func() {
-			if err := recover(); err != nil {
-				v.mux.RLock()
-				if v.onPanic != nil {
-					v.onPanic(fmt.Errorf("%s: %v", name, err))
-				}
-				v.mux.RUnlock()
-			}
-			v.wg.Done()
-		}()
-
-		call(v.globalCtx)
-	}()
+	go v.launch(name, call)
 }
 
 func (v *_group) Run(name string, call func(ctx context.Context)) {
 	v.wg.Add(1)
+	v.launch(name, call)
+}
+
+func (v *_group) launch(name string, call func(ctx context.Context)) {
 	defer func() {
 		if err := recover(); err != nil {
 			v.mux.RLock()

limiter.go

@@ -28,14 +28,14 @@ func NewLimiter(ctx context.Context, count int, interval time.Duration) Limiter
 		count = 1
 	}
 
-	rl := &limiter{
+	lim := &limiter{
 		count: count,
 		pause: interval / time.Duration(count),
 		ch:    make(chan struct{}, count),
 		ctx:   ctx,
 	}
 
-	go rl.refill()
+	go lim.refill()
 	go func() {
 		tik := time.NewTicker(interval)
 		defer tik.Stop()
@@ -46,12 +46,12 @@ func NewLimiter(ctx context.Context, count int, interval time.Duration) Limiter
 				return
 
 			case <-tik.C:
-				rl.refill()
+				lim.refill()
 			}
 		}
 	}()
 
-	return rl
+	return lim
 }
 
 func (l *limiter) Tap(ctx context.Context) bool {