58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
package rotater
|
|
|
|
import (
|
|
"sync"
|
|
"sync/atomic"
|
|
)
|
|
|
|
// RotateSyncerSet is registry of RotateSyncer
|
|
type RotateSyncerSet struct {
|
|
storage sync.Map
|
|
}
|
|
|
|
// Delete deletes the value for a key.
|
|
func (s *RotateSyncerSet) Delete(key RotateSyncer) {
|
|
s.storage.Delete(key)
|
|
}
|
|
|
|
// Exist returns whether value was found in the map.
|
|
func (s *RotateSyncerSet) Exist(key RotateSyncer) (ok bool) {
|
|
_, ok = s.storage.Load(key)
|
|
return
|
|
}
|
|
|
|
// SetNx returns false value was found in the map.
|
|
// Otherwise, it stores and returns true.
|
|
func (s *RotateSyncerSet) SetNx(key RotateSyncer) bool {
|
|
_, exist := s.storage.LoadOrStore(key, 0)
|
|
return !exist
|
|
}
|
|
|
|
// Range calls f sequentially for each key and value present in the map.
|
|
// If f returns false, range stops the iteration.
|
|
func (s *RotateSyncerSet) Range(f func(key RotateSyncer)) {
|
|
s.storage.Range(s.rangeWrap(f))
|
|
}
|
|
|
|
// Store sets the value for a key.
|
|
func (s *RotateSyncerSet) Store(key RotateSyncer) {
|
|
s.storage.Store(key, 0)
|
|
}
|
|
|
|
// Len returns sizeof the map.
|
|
func (s *RotateSyncerSet) Len() int {
|
|
var count uint64
|
|
s.Range(func(conn RotateSyncer) {
|
|
atomic.AddUint64(&count, 1)
|
|
})
|
|
return int(count)
|
|
}
|
|
|
|
func (s *RotateSyncerSet) rangeWrap(f func(key RotateSyncer)) func(key, value any) bool {
|
|
ok := true
|
|
return func(key, value any) bool {
|
|
f(key.(RotateSyncer))
|
|
return ok
|
|
}
|
|
}
|