95 lines
2.6 KiB
Go
95 lines
2.6 KiB
Go
package runner
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sort"
|
|
"sync"
|
|
|
|
"github.com/ssdomei232/goodBaby/drivers/bilibili"
|
|
"github.com/ssdomei232/goodBaby/drivers/dingtalk"
|
|
"github.com/ssdomei232/goodBaby/drivers/email"
|
|
"github.com/ssdomei232/goodBaby/drivers/fwalert"
|
|
"github.com/ssdomei232/goodBaby/drivers/github"
|
|
"github.com/ssdomei232/goodBaby/drivers/onebot"
|
|
"github.com/ssdomei232/goodBaby/model"
|
|
)
|
|
|
|
// RuleExecutor 规则执行器接口
|
|
type RuleExecutor interface {
|
|
// Execute 执行规则，ctx 决定了重试的最长时间
|
|
Execute(ctx context.Context, rule *model.Rule) error
|
|
// GetType 获取执行器支持的规则类型
|
|
GetType() string
|
|
}
|
|
|
|
// ExecutorRegistry 执行器注册表
|
|
type ExecutorRegistry struct {
|
|
executors map[string]RuleExecutor
|
|
}
|
|
|
|
// NewExecutorRegistry 创建新的执行器注册表
|
|
func NewExecutorRegistry() *ExecutorRegistry {
|
|
return &ExecutorRegistry{
|
|
executors: make(map[string]RuleExecutor),
|
|
}
|
|
}
|
|
|
|
// Register 注册规则执行器
|
|
func (er *ExecutorRegistry) Register(executor RuleExecutor) {
|
|
er.executors[executor.GetType()] = executor
|
|
}
|
|
|
|
// Execute 根据规则类型执行规则
|
|
func (er *ExecutorRegistry) Execute(ctx context.Context, rule *model.Rule) error {
|
|
executor, exists := er.executors[rule.Type]
|
|
if !exists {
|
|
return fmt.Errorf("不支持的规则类型: %s", rule.Type)
|
|
}
|
|
|
|
return executor.Execute(ctx, rule)
|
|
}
|
|
|
|
// GetSupportedTypes 获取所有支持的规则类型
|
|
func (er *ExecutorRegistry) GetSupportedTypes() []string {
|
|
types := make([]string, 0, len(er.executors))
|
|
for t := range er.executors {
|
|
types = append(types, t)
|
|
}
|
|
sort.Strings(types)
|
|
return types
|
|
}
|
|
|
|
var (
|
|
registryOnce sync.Once
|
|
globalExecutorRegistry *ExecutorRegistry
|
|
)
|
|
|
|
// InitExecutorRegistry 初始化执行器注册表并注册所有执行器
|
|
func InitExecutorRegistry() *ExecutorRegistry {
|
|
registryOnce.Do(func() {
|
|
registry := NewExecutorRegistry()
|
|
|
|
// 注册所有规则执行器
|
|
registry.Register(&bilibili.BilibiliDynamicExecutor{})
|
|
registry.Register(&bilibili.BilibiliPrivateMessageExecutor{})
|
|
registry.Register(&email.EmailExecutor{})
|
|
registry.Register(&github.GithubMakeRepoPublicExecutor{})
|
|
registry.Register(&onebot.OneBotExecutor{})
|
|
registry.Register(&dingtalk.DingTalkExecutor{})
|
|
registry.Register(&fwalert.FwalertExecutor{})
|
|
// 未来添加新规则类型时，在这里注册即可
|
|
|
|
globalExecutorRegistry = registry
|
|
})
|
|
return globalExecutorRegistry
|
|
}
|
|
|
|
// GetGlobalExecutorRegistry 获取全局执行器注册表
|
|
func GetGlobalExecutorRegistry() *ExecutorRegistry {
|
|
if globalExecutorRegistry == nil {
|
|
panic("执行器注册表未初始化，请先调用 InitExecutorRegistry()")
|
|
}
|
|
return globalExecutorRegistry
|
|
}
|