PandaX/pkg/rule_engine/engine_data.go

package rule_engine

import (
	"pandax/pkg/rule_engine/message"
	"pandax/pkg/rule_engine/nodes"
)

func GetCategory() []map[string]interface{} {
	return nodes.GetCategory()
}

func GetDebugData(ruleId, nodeId string) []message.DebugData {
	if data, ok := ruleChainDebugData.Data[ruleId]; ok {
		return data.Get(nodeId).Items
	}
	return nil
}

func ClearDebugData(ruleId, nodeId string) {
	if data, ok := ruleChainDebugData.Data[ruleId]; ok {
		data.Clear(nodeId)
	}
}

func GetDebugDataPage(page, pageSize int, ruleId, nodeId string) (int64, []message.DebugData) {
	if page < 1 {
		page = 1
	}
	offset := pageSize * (page - 1)
	if data, ok := ruleChainDebugData.Data[ruleId]; ok {
		nodeData := data.Get(nodeId)
		if nodeData != nil {
			total := len(nodeData.Items)
			end := offset + pageSize
			if end >= total {
				end = total
			}
			return int64(total), nodeData.Items[offset:end]
		}
	}
	return 0, nil
}