starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-04-23 08:48:32 +02:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

feat(healthcheck): allow health checking for excluded routes

Browse Source
This commit is contained in:
yusing
2025-06-02 23:19:30 +08:00
parent 4705989f4b
commit 9087c4f195
9 changed files with 138 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
internal/utils/pool/pool.go
View File

@@ -9,8 +9,9 @@ import (
type (
Pool[T Object] struct {
m *xsync.Map[string, T]
name string
m *xsync.Map[string, T]
name string
disableLog bool
}
Object interface {
Key() string
@@ -19,41 +20,54 @@ type (
)
func New[T Object](name string) Pool[T] {
return Pool[T]{xsync.NewMap[string, T](), name}
return Pool[T]{xsync.NewMap[string, T](), name, false}
}
func (p Pool[T]) Name() string {
func (p *Pool[T]) DisableLog() {
p.disableLog = true
}
func (p *Pool[T]) Name() string {
return p.name
}
func (p Pool[T]) Add(obj T) {
func (p *Pool[T]) Add(obj T) {
p.checkExists(obj.Key())
p.m.Store(obj.Key(), obj)
log.Info().Msgf("%s: added %s", p.name, obj.Name())
if !p.disableLog {
log.Info().Msgf("%s: added %s", p.name, obj.Name())
}
}
func (p Pool[T]) Del(obj T) {
func (p *Pool[T]) AddIfNotExists(obj T) (actual T, added bool) {
actual, loaded := p.m.LoadOrStore(obj.Key(), obj)
return actual, !loaded
}
func (p *Pool[T]) Del(obj T) {
p.m.Delete(obj.Key())
log.Info().Msgf("%s: removed %s", p.name, obj.Name())
if !p.disableLog {
log.Info().Msgf("%s: removed %s", p.name, obj.Name())
}
}
func (p Pool[T]) Get(key string) (T, bool) {
func (p *Pool[T]) Get(key string) (T, bool) {
return p.m.Load(key)
}
func (p Pool[T]) Size() int {
func (p *Pool[T]) Size() int {
return p.m.Size()
}
func (p Pool[T]) Clear() {
func (p *Pool[T]) Clear() {
p.m.Clear()
}
func (p Pool[T]) Iter(fn func(k string, v T) bool) {
func (p *Pool[T]) Iter(fn func(k string, v T) bool) {
p.m.Range(fn)
}
func (p Pool[T]) Slice() []T {
func (p *Pool[T]) Slice() []T {
slice := make([]T, 0, p.m.Size())
for _, v := range p.m.Range {
slice = append(slice, v)