Limit work

util/atomic_counter.go

@@ -0,0 +1,32 @@
+package util
+
+import "sync"
+
+type AtomicCounter[T int | int32 | int64] struct {
+	value T
+	mu    sync.Mutex
+}
+
+func NewAtomicCounter[T int | int32 | int64](value T) *AtomicCounter[T] {
+	return &AtomicCounter[T]{
+		value: value,
+	}
+}
+func (c *AtomicCounter[T]) Inc() T {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	c.value++
+	return c.value
+}
+
+func (c *AtomicCounter[T]) Value() T {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	return c.value
+}
+
+func (c *AtomicCounter[T]) Reset() {
+	c.mu.Lock()
+	defer c.mu.Unlock()
+	c.value = 0
+}