mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-31 09:02:34 +02:00
4a8bd48ad5
Some checks are pending
Docker Image CI (socket-proxy) / build (push) Waiting to run
* refactor: simplify io code and make utils module independent * fix(docker): agent and socket-proxy docker event flushing with modified reverse proxy handler * refactor: remove unused code * refactor: remove the use of logging module in most code * refactor: streamline domain mismatch check in certState function * tweak: use ecdsa p-256 for autocert * fix(tests): update health check tests for invalid host and add case for port in host * feat(acme): custom acme directory * refactor: code refactor and improved context and error handling * tweak: optimize memory usage under load * fix(oidc): restore old user matching behavior * docs: add ChatGPT assistant to README --------- Co-authored-by: yusing <yusing@6uo.me>
65 lines
1.1 KiB
Go
65 lines
1.1 KiB
Go
package pool
|
|
|
|
import (
|
|
"sort"
|
|
|
|
"github.com/puzpuzpuz/xsync/v4"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
type (
|
|
Pool[T Object] struct {
|
|
m *xsync.Map[string, T]
|
|
name string
|
|
}
|
|
Object interface {
|
|
Key() string
|
|
Name() string
|
|
}
|
|
)
|
|
|
|
func New[T Object](name string) Pool[T] {
|
|
return Pool[T]{xsync.NewMap[string, T](), name}
|
|
}
|
|
|
|
func (p Pool[T]) Name() string {
|
|
return p.name
|
|
}
|
|
|
|
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())
|
|
}
|
|
|
|
func (p Pool[T]) Del(obj T) {
|
|
p.m.Delete(obj.Key())
|
|
log.Info().Msgf("%s: removed %s", p.name, obj.Name())
|
|
}
|
|
|
|
func (p Pool[T]) Get(key string) (T, bool) {
|
|
return p.m.Load(key)
|
|
}
|
|
|
|
func (p Pool[T]) Size() int {
|
|
return p.m.Size()
|
|
}
|
|
|
|
func (p Pool[T]) Clear() {
|
|
p.m.Clear()
|
|
}
|
|
|
|
func (p Pool[T]) Iter(fn func(k string, v T) bool) {
|
|
p.m.Range(fn)
|
|
}
|
|
|
|
func (p Pool[T]) Slice() []T {
|
|
slice := make([]T, 0, p.m.Size())
|
|
for _, v := range p.m.Range {
|
|
slice = append(slice, v)
|
|
}
|
|
sort.Slice(slice, func(i, j int) bool {
|
|
return slice[i].Name() < slice[j].Name()
|
|
})
|
|
return slice
|
|
}
|