PandaX/pkg/rule_engine/instance.go

package rule_engine

import (
	"pandax/pkg/rule_engine/manifest"
	"pandax/pkg/rule_engine/nodes"
    "fmt"
	"github.com/sirupsen/logrus"
)

type RuleChainInstance struct {
	ruleId          string
	firstRuleNodeID string
	nodes           map[string]nodes.Node
}

func NewRuleChainInstance(ruleId string, data []byte) (*RuleChainInstance, error) {
	manifest, err := manifest.New(data)
	if err != nil {
		logrus.WithError(err).Errorf("invalid manifest file")
		return nil, err
	}
	withManifest, err := newInstanceWithManifest(manifest)
	if err != nil {
		return nil, err
	}
	withManifest.ruleId = ruleId
	return withManifest, nil
}

func newInstanceWithManifest(m *manifest.Manifest) (*RuleChainInstance, error) {
	nodes, err := nodes.GetNodes(m)
	if err != nil {
		return nil, err
	}
	r := &RuleChainInstance{
		firstRuleNodeID: m.FirstRuleNodeId,
		nodes:           nodes,
	}
	return r, nil
}
// 获取规则实体指定节点
func (rule *RuleChainInstance) GetNode(nodeId string) (nodes.Node, error) {
	if node,ok := rule.nodes[nodeId];ok {
		return node,nil
	}else {
		return nil,fmt.Errorf("节点不存在")
	}
}