47 lines
935 B
Go
47 lines
935 B
Go
package logger
|
|
|
|
import (
|
|
"sync"
|
|
"sync/atomic"
|
|
)
|
|
|
|
/**
|
|
logger set
|
|
*/
|
|
type RotateSyncerSet struct {
|
|
storage sync.Map
|
|
}
|
|
|
|
func (s *RotateSyncerSet) Delete(key RotateSyncer) {
|
|
s.storage.Delete(key)
|
|
}
|
|
func (s *RotateSyncerSet) Exist(key RotateSyncer) (ok bool) {
|
|
_, ok = s.storage.Load(key)
|
|
return
|
|
}
|
|
func (s *RotateSyncerSet) SetNx(key RotateSyncer) (bool) {
|
|
_, exist := s.storage.LoadOrStore(key, 0)
|
|
return !exist
|
|
}
|
|
func (s *RotateSyncerSet) Range(f func(key RotateSyncer)) {
|
|
s.storage.Range(s.rangeWrap(f))
|
|
}
|
|
func (s *RotateSyncerSet) Store(key RotateSyncer) {
|
|
s.storage.Store(key, 0)
|
|
}
|
|
func (s *RotateSyncerSet) rangeWrap(f func(key RotateSyncer)) func(key, value interface{}) bool {
|
|
ok := true
|
|
return func(key, value interface{}) bool {
|
|
f(key.(RotateSyncer))
|
|
return ok
|
|
}
|
|
}
|
|
|
|
func (s *RotateSyncerSet) Len() int {
|
|
var count uint64
|
|
s.Range(func(conn RotateSyncer) {
|
|
atomic.AddUint64(&count, 1)
|
|
})
|
|
return int(count)
|
|
}
|