cicv_data_closed_loop/aarch64/pjisuv/master/service/for_competition.go

package svc

import (
	commonConfig "cicv-data-closedloop/aarch64/pjisuv/common/config"
	"cicv-data-closedloop/common/config/c_log"
	"cicv-data-closedloop/common/util"
	"github.com/bluenviron/goroslib/v2"
	"github.com/bluenviron/goroslib/v2/pkg/msgs/std_msgs"
	"os"
	"strings"
	"sync"
	"time"
)

var (
	dir                    = "/userdata/competition/"
	commandArgs            = []string{"record", "--split", "--duration=1", "/pji_gps", "/data_read"}
	topic                  = "/cicv_competition"
	urlExamBegin           = "36.110.106.142/exam/begin"
	urlExamEnd             = "36.110.106.142/exam/end"
	cacheMutex             sync.Mutex
	cacheTeamName          = make(map[string]time.Time)
	heartBeatTimeThreshold = 5 * time.Second // 心跳时间
)

// todo 实车比赛临时使用
// history record命令无法录制（）
func ForCompetition() {
	go dataCollection()
	go examBegin()
	go examEnd()
}

// 全量数据采集
func dataCollection() {
	c_log.GlobalLogger.Info("开始采集实车算法比赛全量数据。")
	util.CreateDir(dir)
	// 1 打包
	c_log.GlobalLogger.Info("采集实车算法比赛全量数据的环境变量为：", commonConfig.RosbagEnvs)
	command, err := util.ExecuteWithEnvAndDirAsync(commonConfig.RosbagEnvs, dir, commonConfig.RosbagPath, commandArgs...)
	if err != nil {
		c_log.GlobalLogger.Error("程序崩溃。执行record命令", command, "出错：", err)
		os.Exit(-1)
	}
	// 2 扫描目录文件
	for {
		time.Sleep(time.Duration(2) * time.Second)
		files, _ := util.ListAbsolutePathAndSort(dir)
		if len(files) >= 2 {
			c_log.GlobalLogger.Info("扫描试车比赛数据采集目录，", files)
			for i := range files {
				if i == len(files)-1 { // 最后一个包在录制中，不上传
					break
				}
				c_log.GlobalLogger.Debug("上传实车算法比赛全量数据包", files[i])
				bagSlice := strings.Split(files[i], "/")
				commonConfig.OssMutex.Lock()
				_ = commonConfig.OssBucket.PutObjectFromFile("competition/"+bagSlice[len(bagSlice)-1], files[i])
				commonConfig.OssMutex.Unlock()
				_ = util.DeleteFile(files[i])
			}
		}
	}
}

// data格式为队伍编号
// 保存单次考试时间区间
func examBegin() {
	_, _ = goroslib.NewSubscriber(goroslib.SubscriberConf{
		Node:  commonConfig.RosNode,
		Topic: topic,
		Callback: func(data *std_msgs.String) {
			teamName := data.Data
			cacheMutex.Lock()
			{
				if !util.ContainsKey(cacheTeamName, teamName) { // 1 如果缓存数组中没有此队名，代表考试开始，缓存此队名，和当前时间戳
					examBeginTime := time.Now()
					cacheTeamName[teamName] = examBeginTime
					_, _ = util.PostJsonResponseString(urlExamBegin, map[string]string{"teamName": teamName})
					c_log.GlobalLogger.Infof("队伍 %v 的考试开始。", teamName)
				} else { // 2 如果缓存数组中有此队名，代表考试进行中，刷新时间戳
					cacheTeamName[teamName] = time.Now()
				}
			}
			cacheMutex.Unlock()
		},
	})
}

func examEnd() {
	for {
		time.Sleep(time.Duration(1) * time.Second)
		cacheMutex.Lock()
		{
			var keysToDelete []string
			for teamName, heartBeatTime := range cacheTeamName {
				if time.Since(heartBeatTime) > heartBeatTimeThreshold { // 检查缓存中的队名，如果超过心跳时间，则代表考试结束，删除缓存中的队名
					keysToDelete = append(keysToDelete, teamName)
				}
			}
			for _, teamName := range keysToDelete { // 检查缓存中的队名，如果超过心跳时间，则代表考试结束，删除缓存中的队名
				delete(cacheTeamName, teamName)
				_, _ = util.PostJsonResponseString(urlExamEnd, map[string]string{"teamName": teamName})
				c_log.GlobalLogger.Infof("队伍 %v 的考试结束。", teamName)
			}
		}
		cacheMutex.Unlock()
	}
}