mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-20 00:25:02 +01:00
2717dc963a
- Introduced ContainerRuntime field in AgentConfig and AgentEnvConfig. - Added IterAgents and NumAgents functions for agent pool management. - Updated agent creation and verification endpoints to handle container runtime. - Enhanced Docker Compose template to support different container runtimes. - Added runtime endpoint to retrieve agent runtime information.
68 lines
1.3 KiB
Go
68 lines
1.3 KiB
Go
package agent
|
|
|
|
import (
|
|
"iter"
|
|
|
|
"github.com/puzpuzpuz/xsync/v4"
|
|
"github.com/yusing/go-proxy/internal/common"
|
|
)
|
|
|
|
var agentPool = xsync.NewMap[string, *AgentConfig](xsync.WithPresize(10))
|
|
|
|
func init() {
|
|
if common.IsTest {
|
|
agentPool.Store("test-agent", &AgentConfig{
|
|
Addr: "test-agent",
|
|
})
|
|
}
|
|
}
|
|
|
|
func GetAgent(agentAddrOrDockerHost string) (*AgentConfig, bool) {
|
|
if !IsDockerHostAgent(agentAddrOrDockerHost) {
|
|
return getAgentByAddr(agentAddrOrDockerHost)
|
|
}
|
|
return getAgentByAddr(GetAgentAddrFromDockerHost(agentAddrOrDockerHost))
|
|
}
|
|
|
|
func GetAgentByName(name string) (*AgentConfig, bool) {
|
|
for _, agent := range agentPool.Range {
|
|
if agent.Name == name {
|
|
return agent, true
|
|
}
|
|
}
|
|
return nil, false
|
|
}
|
|
|
|
func AddAgent(agent *AgentConfig) {
|
|
agentPool.Store(agent.Addr, agent)
|
|
}
|
|
|
|
func RemoveAgent(agent *AgentConfig) {
|
|
agentPool.Delete(agent.Addr)
|
|
}
|
|
|
|
func RemoveAllAgents() {
|
|
agentPool.Clear()
|
|
}
|
|
|
|
func ListAgents() []*AgentConfig {
|
|
agents := make([]*AgentConfig, 0, agentPool.Size())
|
|
for _, agent := range agentPool.Range {
|
|
agents = append(agents, agent)
|
|
}
|
|
return agents
|
|
}
|
|
|
|
func IterAgents() iter.Seq2[string, *AgentConfig] {
|
|
return agentPool.Range
|
|
}
|
|
|
|
func NumAgents() int {
|
|
return agentPool.Size()
|
|
}
|
|
|
|
func getAgentByAddr(addr string) (agent *AgentConfig, ok bool) {
|
|
agent, ok = agentPool.Load(addr)
|
|
return
|
|
}
|