ryan
de7a52e81f
Motivation: 通知规则的事件名此前仅支持精确匹配，无法用一条规则覆盖多类事件（如 trade.* 下的全部交易事件），规则配置维护成本高。本次引入通配符匹配，让规则可按模式批量命中事件，同时保证精确规则优先。 Changes: * 事件名支持 * 和 ? 通配符匹配 * 匹配优先级：精确匹配 > 更具体的通配 > 泛匹配，同分时取更小 ID * 精确命中未命中时回退为列出源下启用规则并按优先级挑选 * 抽取缓存预热逻辑为独立方法 * 更新文档说明事件名通配规则
71 lines
1.7 KiB
Go
71 lines
1.7 KiB
Go
package engine
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"aiaa-notification-service/internal/cache"
|
|
"aiaa-notification-service/internal/model"
|
|
"aiaa-notification-service/internal/store"
|
|
)
|
|
|
|
type Matcher struct {
|
|
store *store.Store
|
|
cache *cache.Cache
|
|
}
|
|
|
|
func NewMatcher(s *store.Store, c *cache.Cache) *Matcher {
|
|
return &Matcher{store: s, cache: c}
|
|
}
|
|
|
|
func (m *Matcher) Match(ctx context.Context, sourceID int, event string) (*model.Rule, error) {
|
|
// Try cache first
|
|
if m.cache != nil {
|
|
cr, err := m.cache.GetRule(ctx, sourceID, event)
|
|
if err == nil {
|
|
rule := &model.Rule{ID: cr.RuleID, TemplateID: cr.TemplateID, SourceID: sourceID, Event: event}
|
|
if cr.Conditions != "" && cr.Conditions != "null" {
|
|
raw := json.RawMessage(cr.Conditions)
|
|
rule.Conditions = &raw
|
|
}
|
|
return rule, nil
|
|
}
|
|
}
|
|
|
|
rule, err := m.store.GetRuleBySourceEvent(ctx, sourceID, event)
|
|
if err != nil {
|
|
rules, listErr := m.store.ListEnabledRulesBySource(ctx, sourceID)
|
|
if listErr != nil {
|
|
return nil, fmt.Errorf("match rule: %w", listErr)
|
|
}
|
|
picked := PickEventRule(event, rules)
|
|
if picked == nil {
|
|
return nil, fmt.Errorf("match rule: %w", err)
|
|
}
|
|
return picked, nil
|
|
}
|
|
|
|
m.warmRuleCache(ctx, sourceID, event, rule)
|
|
return rule, nil
|
|
}
|
|
|
|
func (m *Matcher) warmRuleCache(ctx context.Context, sourceID int, event string, rule *model.Rule) {
|
|
if m.cache == nil {
|
|
return
|
|
}
|
|
tmpl, err := m.store.GetTemplate(ctx, rule.TemplateID)
|
|
if err != nil {
|
|
return
|
|
}
|
|
cr := &cache.CachedRule{
|
|
RuleID: rule.ID,
|
|
TemplateID: rule.TemplateID,
|
|
Content: tmpl.Content,
|
|
}
|
|
if rule.Conditions != nil {
|
|
cr.Conditions = string(*rule.Conditions)
|
|
}
|
|
_ = m.cache.SetRule(ctx, sourceID, event, cr)
|
|
}
|