sparrow/pkg/ruleEngine/nodes/log_node.go

package nodes

import (
	"encoding/json"
	"github.com/Sirupsen/logrus"
	"sparrow/pkg/protocol"
	"sparrow/pkg/ruleEngine"
)

type LogNodeConfig struct {
	Stdout bool `json:"stdout"` // 标准控制台输出
}

// LogNode 日志打印节点
type LogNode struct {
	config *LogNodeConfig
	mLog   *logrus.Logger
}

func (l *LogNode) Init(ctx ruleEngine.Context, config string) error {
	if config == "" {
		l.config = &LogNodeConfig{
			Stdout: false,
		}
	} else {
		c := new(LogNodeConfig)
		err := json.Unmarshal([]byte(config), c)
		if err != nil {
			return err
		}
		l.config = c
	}
	l.mLog = logrus.New()
	return nil
}

func (l *LogNode) OnMessage(ctx ruleEngine.Context, message *protocol.Message) error {
	if l.config.Stdout {
		l.mLog.Infof("日志节点：[设备ID：%s][消息类型：%s][消息内容：%s]", message.MetaData["device_id"], message.Type, message.Data)
	}
	ctx.TellNext(message, protocol.Success)
	return nil
}