main.go 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. package main
  2. import (
  3. "cicv-data-closedloop/common/util"
  4. "fmt"
  5. "github.com/bluenviron/goroslib/v2"
  6. "github.com/bluenviron/goroslib/v2/pkg/msg"
  7. "time"
  8. )
  9. type HeartBeatInfo struct {
  10. msg.Package `ros:"diag_msgs"`
  11. ModuleName string `rosname:"module_name"`
  12. Rate int8 `rosname:"rate"`
  13. timestamp float64 `rosname:"timestamp"`
  14. }
  15. var lastTime time.Time
  16. func main() {
  17. // 1 启动 shell 脚本 dpi_start.sh
  18. _, s, err := util.ExecuteWithDirSync("/data", "sh", "dpi_start.sh")
  19. if err != nil {
  20. fmt.Println("执行启动命令报错,执行结果为:", s, ",错误信息为:", err)
  21. }
  22. // 2 监听 ros 话题 /heartbeat_info
  23. rosNode, _ := goroslib.NewNode(goroslib.NodeConf{
  24. Name: "dpi_with_heartbeatInfo",
  25. MasterAddress: "127.0.0.1:11311",
  26. })
  27. _, _ = goroslib.NewSubscriber(goroslib.SubscriberConf{
  28. Node: rosNode,
  29. Topic: "/heartbeat_info",
  30. Callback: func(data *HeartBeatInfo) {
  31. lastTime = time.Now()
  32. }})
  33. // 3 超过5秒没有接收到数据则执行 dpi_stop.sh 和 dpi_start.sh
  34. for {
  35. time.Sleep(1 * time.Second)
  36. if time.Since(lastTime) > 5.00 {
  37. fmt.Println("超过5秒没有监听到/heart_info数据,即将重启dpi。")
  38. _, s, err = util.ExecuteWithDirSync("/data", "sh", "dpi_stop.sh")
  39. if err != nil {
  40. fmt.Println("执行停止命令报错,执行结果为:", s, ",错误信息为:", err)
  41. }
  42. _, s, err = util.ExecuteWithDirSync("/data", "sh", "dpi_start.sh")
  43. if err != nil {
  44. fmt.Println("执行启动命令报错,执行结果为:", s, ",错误信息为:", err)
  45. }
  46. }
  47. }
  48. }