mirror of
https://github.com/binwiederhier/ntfy.git
synced 2024-06-02 02:24:45 +12:00
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
|
|
}
|