feat: throttle (#72)

Author: CorrectRoadH

utils/utils.go

@@ -1,5 +1,10 @@
 package utils
 
+import (
+	"sync/atomic"
+	"time"
+)
+
 func CompareSlices(a, b []any) bool {
 	if len(a) != len(b) {
 		return false
@@ -31,3 +36,32 @@ func CompareStringSlices(a, b []string) bool {
 func Ptr[T any](v T) *T {
 	return &v
 }
+
+type Throttle struct {
+	lastCall time.Time
+	limit    time.Duration
+	timer    atomic.Pointer[time.Timer]
+}
+
+func NewThrottle(limit time.Duration) *Throttle {
+	return &Throttle{limit: limit}
+}
+
+func (t *Throttle) Do(f func()) {
+	now := time.Now()
+	if now.Sub(t.lastCall) >= t.limit {
+		t.lastCall = now
+		f()
+		if old := t.timer.Swap(nil); old != nil {
+			old.Stop()
+		}
+	} else {
+		lastFunc := f
+		if old := t.timer.Swap(time.AfterFunc(t.limit-now.Sub(t.lastCall), func() {
+			t.lastCall = time.Now()
+			lastFunc()
+		})); old != nil {
+			old.Stop()
+		}
+	}
+}