mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-20 12:42:34 +02:00
54 lines
1.5 KiB
Go
54 lines
1.5 KiB
Go
package config
|
|
|
|
import (
|
|
"github.com/yusing/go-proxy/agent/pkg/agent"
|
|
"github.com/yusing/go-proxy/internal/gperr"
|
|
"github.com/yusing/go-proxy/internal/route/provider"
|
|
"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) AddAgent(host string, ca agent.PEMPair, client agent.PEMPair) (int, gperr.Error) {
|
|
var agentCfg agent.AgentConfig
|
|
agentCfg.Addr = host
|
|
err := agentCfg.StartWithCerts(cfg.Task(), ca.Cert, client.Cert, client.Key)
|
|
if err != nil {
|
|
return 0, gperr.Wrap(err, "failed to start agent")
|
|
}
|
|
|
|
provider := provider.NewAgentProvider(&agentCfg)
|
|
if err := cfg.errIfExists(provider); err != nil {
|
|
return 0, err
|
|
}
|
|
cfg.storeProvider(provider)
|
|
return provider.NumRoutes(), nil
|
|
}
|
|
|
|
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
|
|
}
|