初始化仓库
This commit is contained in:
@@ -0,0 +1,62 @@
|
||||
package core
|
||||
|
||||
import "fmt"
|
||||
|
||||
// PlatformFactory creates a Platform from config options.
|
||||
type PlatformFactory func(opts map[string]any) (Platform, error)
|
||||
|
||||
// AgentFactory creates an Agent from config options.
|
||||
type AgentFactory func(opts map[string]any) (Agent, error)
|
||||
|
||||
var (
|
||||
platformFactories = make(map[string]PlatformFactory)
|
||||
agentFactories = make(map[string]AgentFactory)
|
||||
)
|
||||
|
||||
func RegisterPlatform(name string, factory PlatformFactory) {
|
||||
platformFactories[name] = factory
|
||||
}
|
||||
|
||||
func RegisterAgent(name string, factory AgentFactory) {
|
||||
agentFactories[name] = factory
|
||||
}
|
||||
|
||||
func CreatePlatform(name string, opts map[string]any) (Platform, error) {
|
||||
f, ok := platformFactories[name]
|
||||
if !ok {
|
||||
available := make([]string, 0, len(platformFactories))
|
||||
for k := range platformFactories {
|
||||
available = append(available, k)
|
||||
}
|
||||
return nil, fmt.Errorf("unknown platform %q, available: %v", name, available)
|
||||
}
|
||||
return f(opts)
|
||||
}
|
||||
|
||||
func ListRegisteredAgents() []string {
|
||||
names := make([]string, 0, len(agentFactories))
|
||||
for k := range agentFactories {
|
||||
names = append(names, k)
|
||||
}
|
||||
return names
|
||||
}
|
||||
|
||||
func ListRegisteredPlatforms() []string {
|
||||
names := make([]string, 0, len(platformFactories))
|
||||
for k := range platformFactories {
|
||||
names = append(names, k)
|
||||
}
|
||||
return names
|
||||
}
|
||||
|
||||
func CreateAgent(name string, opts map[string]any) (Agent, error) {
|
||||
f, ok := agentFactories[name]
|
||||
if !ok {
|
||||
available := make([]string, 0, len(agentFactories))
|
||||
for k := range agentFactories {
|
||||
available = append(available, k)
|
||||
}
|
||||
return nil, fmt.Errorf("unknown agent %q, available: %v", name, available)
|
||||
}
|
||||
return f(opts)
|
||||
}
|
||||
Reference in New Issue
Block a user