33 lines
511 B
Go
33 lines
511 B
Go
|
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
|
||
|
|
}
|