cicv_data_closed_loop/pji/master/pkg/cfg/master_trigger_cfg.go

package cfg

import (
	"cicv-data-closedloop/pji/common/cfg"
	"cicv-data-closedloop/pji/common/cutil"
	"cicv-data-closedloop/pji/common/log"
	"cicv-data-closedloop/pji_msgs"
	"github.com/bluenviron/goroslib/v2/pkg/msgs/diagnostic_msgs"
	"github.com/bluenviron/goroslib/v2/pkg/msgs/nav_msgs"
	"github.com/bluenviron/goroslib/v2/pkg/msgs/sensor_msgs"
	"github.com/bluenviron/goroslib/v2/pkg/msgs/std_msgs"
	"plugin"
	"strconv"
	"strings"
)

var (
	TopicOfOdom              = "/odom"
	RuleOfOdom               []func(data *nav_msgs.Odometry) string
	TopicOfObstacleDetection = "/obstacle_detection"
	RuleOfObstacleDetection  []func(data *std_msgs.UInt8) string
	TopicOfSysInfo           = "/sys_info"
	RuleOfSysInfo            []func(data *pji_msgs.SysInfo) string
	TopicOfLocateInfo        = "/locate_info"
	RuleOfLocateInfo         []func(data *pji_msgs.LocateInfo) string
	TopicOfImu               = "/imu"
	RuleOfImu                []func(data *sensor_msgs.Imu) string
	TopicOfDiagnostics       = "/diagnostics"
	RuleOfDiagnostics        []func(data *diagnostic_msgs.DiagnosticArray) string
	LabelMapTriggerId        = make(map[string]string)
)

func InitTriggerConfig() {
	// 下载所有触发器的文件
	for _, trigger := range cfg.PlatformConfig.TaskTriggers {
		// 获取文件名
		slice := strings.Split(trigger.TriggerScriptPath, "/")
		fileName := slice[len(slice)-1]
		// 下载
		triggerLocalPath := cfg.CloudConfig.TriggersDir + fileName
		cutil.CreateParentDir(triggerLocalPath)
		log.GlobalLogger.Info("下载触发器插件从", trigger.TriggerScriptPath, "到", triggerLocalPath)
		err := cfg.OssBucket.GetObjectToFile(trigger.TriggerScriptPath, triggerLocalPath)
		if err != nil {
			log.GlobalLogger.Error("下载oss上的触发器插件失败：", err)
			continue
		}
		// 载入插件到数组
		{
			open, err := plugin.Open(triggerLocalPath)
			if err != nil {
				log.GlobalLogger.Error("加载本地插件", triggerLocalPath, "失败。", err)
			}
			topic0, err := open.Lookup("Topic")
			if err != nil {
				log.GlobalLogger.Error("加载本地插件", triggerLocalPath, "中的Topic方法失败。", err)
				continue
			}
			topic1, ok := topic0.(func() string)
			if ok != true {
				log.GlobalLogger.Error("插件", triggerLocalPath, "中的Topic方法必须是(func() string):", err)
				continue
			}
			topic2 := topic1()
			rule, err := open.Lookup("Rule")
			if err != nil {
				log.GlobalLogger.Error("加载本地插件", triggerLocalPath, "中的Rule方法失败。", err)
				continue
			}
			//TODO 类型断言需要处理异常，防止程序崩溃
			if TopicOfOdom == topic2 {
				f := rule.(func(data *nav_msgs.Odometry) string)
				RuleOfOdom = append(RuleOfOdom, f)
			} else if TopicOfObstacleDetection == topic2 {
				f := rule.(func(data *std_msgs.UInt8) string)
				RuleOfObstacleDetection = append(RuleOfObstacleDetection, f)
			} else if TopicOfSysInfo == topic2 {
				f := rule.(func(data *pji_msgs.SysInfo) string)
				RuleOfSysInfo = append(RuleOfSysInfo, f)
			} else if TopicOfLocateInfo == topic2 {
				f := rule.(func(data *pji_msgs.LocateInfo) string)
				RuleOfLocateInfo = append(RuleOfLocateInfo, f)
			} else if TopicOfImu == topic2 {
				f := rule.(func(data *sensor_msgs.Imu) string)
				RuleOfImu = append(RuleOfImu, f)
			} else if TopicOfDiagnostics == topic2 {
				f := rule.(func(data *diagnostic_msgs.DiagnosticArray) string)
				RuleOfDiagnostics = append(RuleOfDiagnostics, f)
			} else {
				log.GlobalLogger.Error("未知的topic：", topic2)
				continue
			}

			label, err := open.Lookup("Label")
			if err != nil {
				log.GlobalLogger.Error("加载本地插件", triggerLocalPath, "中的TriggerName方法失败。", err)
				continue
			}
			labelFunc := label.(func() string)
			labelString := labelFunc()
			LabelMapTriggerId[labelString] = strconv.Itoa(trigger.TriggerId)
			log.GlobalLogger.Info("主节点加载触发器插件：【ros topic】=", topic2, "，【触发器label】=", labelString)
		}

	}
}