mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-21 13:02:34 +02:00
36 lines
930 B
Go
36 lines
930 B
Go
package config
|
|
|
|
import (
|
|
"github.com/yusing/go-proxy/agent/pkg/agent"
|
|
"github.com/yusing/go-proxy/internal/utils/functional"
|
|
)
|
|
|
|
var agentPool = functional.NewMapOf[string, *agent.AgentConfig]()
|
|
|
|
func addAgent(agent *agent.AgentConfig) {
|
|
agentPool.Store(agent.Addr, agent)
|
|
}
|
|
|
|
func removeAllAgents() {
|
|
agentPool.Clear()
|
|
}
|
|
|
|
func GetAgent(addr string) (agent *agent.AgentConfig, ok bool) {
|
|
agent, ok = agentPool.Load(addr)
|
|
return
|
|
}
|
|
|
|
func (cfg *Config) GetAgent(agentAddrOrDockerHost string) (*agent.AgentConfig, bool) {
|
|
if !agent.IsDockerHostAgent(agentAddrOrDockerHost) {
|
|
return GetAgent(agentAddrOrDockerHost)
|
|
}
|
|
return GetAgent(agent.GetAgentAddrFromDockerHost(agentAddrOrDockerHost))
|
|
}
|
|
|
|
func (cfg *Config) ListAgents() []*agent.AgentConfig {
|
|
agents := make([]*agent.AgentConfig, 0, agentPool.Size())
|
|
agentPool.RangeAll(func(key string, value *agent.AgentConfig) {
|
|
agents = append(agents, value)
|
|
})
|
|
return agents
|
|
}
|