55 lines
971 B
Go
55 lines
971 B
Go
package hostPool
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"gemini-grc/logging"
|
|
)
|
|
|
|
var hostPool = HostPool{hostnames: make(map[string]struct{})} //nolint:gochecknoglobals
|
|
|
|
type HostPool struct {
|
|
hostnames map[string]struct{}
|
|
Lock sync.RWMutex
|
|
}
|
|
|
|
func (p *HostPool) Add(key string) {
|
|
p.Lock.Lock()
|
|
defer p.Lock.Unlock()
|
|
p.hostnames[key] = struct{}{}
|
|
}
|
|
|
|
func (p *HostPool) Get(key string) bool {
|
|
p.Lock.RLock()
|
|
defer p.Lock.RUnlock()
|
|
_, ok := p.hostnames[key]
|
|
return ok
|
|
}
|
|
|
|
func (p *HostPool) Delete(key string) {
|
|
p.Lock.Lock()
|
|
defer p.Lock.Unlock()
|
|
delete(p.hostnames, key)
|
|
}
|
|
|
|
func AddHostToHostPool(key string) {
|
|
for {
|
|
// Sleep until the host doesn't exist in pool,
|
|
// then add it.
|
|
if hostPool.Get(key) {
|
|
time.Sleep(1 * time.Second) // Avoid flood-retrying
|
|
logging.LogInfo("Waiting to add %s to pool...", key)
|
|
} else {
|
|
hostPool.Add(key)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func RemoveHostFromHostPool(key string) {
|
|
if hostPool.Get(key) {
|
|
hostPool.Delete(key)
|
|
}
|
|
}
|