main.go 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  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. "os"
  8. "sync"
  9. "time"
  10. )
  11. type HeartBeatInfo struct {
  12. msg.Package `ros:"diag_msgs"`
  13. ModuleName string `rosname:"module_name"`
  14. Rate int8 `rosname:"rate"`
  15. timestamp float64 `rosname:"timestamp"`
  16. }
  17. var lastTime time.Time
  18. var mutex sync.Mutex
  19. func main() {
  20. lastTime = time.Now()
  21. // 1 启动 shell 脚本 dpi_start.sh
  22. _, s, err := util.ExecuteWithDirSync("/data", "sh", "dpi_start.sh")
  23. if err != nil {
  24. fmt.Println("执行启动命令报错,执行结果为:", s, ",错误信息为:", err)
  25. os.Exit(1)
  26. }
  27. fmt.Println("执行启动命令 sh dpi_start.sh 成功,执行结果为:", s)
  28. // 2 监听 ros 话题 /heartbeat_info
  29. rosNode, _ := goroslib.NewNode(goroslib.NodeConf{
  30. Name: "dpi_with_heartbeatInfo",
  31. MasterAddress: "127.0.0.1:11311",
  32. })
  33. _, _ = goroslib.NewSubscriber(goroslib.SubscriberConf{
  34. Node: rosNode,
  35. Topic: "/heartbeat_info",
  36. Callback: func(data *HeartBeatInfo) {
  37. fmt.Println("监听到数据:", data)
  38. lastTime = time.Now()
  39. }})
  40. // 3 超过5秒没有接收到数据则执行 dpi_stop.sh 和 dpi_start.sh
  41. for {
  42. time.Sleep(1 * time.Second)
  43. if time.Since(lastTime).Seconds() > 5.00 {
  44. fmt.Println("超过5秒没有监听到/heartbeat_info数据,即将重启dpi。")
  45. _, s, err = util.ExecuteWithDirSync("/data", "sh", "dpi_stop.sh")
  46. if err != nil {
  47. fmt.Println("执行停止命令报错,执行结果为:", s, ",错误信息为:", err)
  48. continue
  49. }
  50. fmt.Println("执行启动命令 sh dpi_stop.sh 成功,执行结果为:", s)
  51. _, s, err = util.ExecuteWithDirSync("/data", "sh", "dpi_start.sh")
  52. if err != nil {
  53. fmt.Println("执行启动命令报错,执行结果为:", s, ",错误信息为:", err)
  54. continue
  55. }
  56. fmt.Println("执行启动命令 sh dpi_start.sh 成功,执行结果为:", s)
  57. time.Sleep(5 * time.Second)
  58. lastTime = time.Now()
  59. }
  60. }
  61. }