|
@@ -6,14 +6,27 @@ import (
|
|
|
"cicv-data-closedloop/aarch64/pjibot_guide/common/variable"
|
|
|
"cicv-data-closedloop/aarch64/pjibot_guide/control/pkg"
|
|
|
"cicv-data-closedloop/common/config/c_log"
|
|
|
+ "cicv-data-closedloop/common/entity"
|
|
|
"cicv-data-closedloop/common/util"
|
|
|
+ "fmt"
|
|
|
"net/rpc"
|
|
|
"os"
|
|
|
"runtime"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
-var applicationName = "pji-control"
|
|
|
+var (
|
|
|
+ applicationName = "pji-control"
|
|
|
+ localStatus = "idle"
|
|
|
+ cloudStatus = "NONE"
|
|
|
+ lastLocalStatus = "idle"
|
|
|
+ lastCloudStatus = "NONE"
|
|
|
+ launchedFlag = false
|
|
|
+ renewedFlag = false
|
|
|
+ configuredFlag = false
|
|
|
+ renewTimer *time.Timer
|
|
|
+ RenewDur = 5
|
|
|
+)
|
|
|
|
|
|
func init() {
|
|
|
runtime.GOMAXPROCS(1)
|
|
@@ -26,97 +39,182 @@ func init() {
|
|
|
|
|
|
commonConfig.InitCloudConfig()
|
|
|
|
|
|
+
|
|
|
+
|
|
|
+ commonConfig.InitWebsocketConfig()
|
|
|
}
|
|
|
|
|
|
-func main() {
|
|
|
- go judgeState1()
|
|
|
- go pkg.JudgeLocal()
|
|
|
- go judgeState3()
|
|
|
- select {}
|
|
|
+func renew() {
|
|
|
+ if renewTimer != nil {
|
|
|
+ renewTimer.Stop()
|
|
|
+ }
|
|
|
+ renewedFlag = true
|
|
|
+ renewTimer = time.AfterFunc(time.Duration(RenewDur)*time.Minute, func() {
|
|
|
+ renewedFlag = false
|
|
|
+ })
|
|
|
+ c_log.GlobalLogger.Infof("续约时间【%v】分钟 - 成功。", RenewDur)
|
|
|
+}
|
|
|
+
|
|
|
+func startMasterNode() {
|
|
|
+ c_log.GlobalLogger.Info("获取数据闭环平台最新配置。")
|
|
|
+ commonConfig.InitPlatformConfig()
|
|
|
+
|
|
|
+ if _, err := util.ExecuteWithPath(commonConfig.LocalConfig.RestartCmd.Dir, commonConfig.LocalConfig.RestartCmd.Name, commonConfig.LocalConfig.RestartCmd.Args...); err != nil {
|
|
|
+ c_log.GlobalLogger.Info("启动新程序失败,【path】=", commonConfig.LocalConfig.RestartCmd.Dir, "【cmd】=", commonConfig.LocalConfig.RestartCmd.Name, commonConfig.LocalConfig.RestartCmd.Args, ":", err)
|
|
|
+ os.Exit(-1)
|
|
|
+ }
|
|
|
+ c_log.GlobalLogger.Info("启动任务,本地执行启动命令:【path】=", commonConfig.LocalConfig.RestartCmd.Dir, "【cmd】=", commonConfig.LocalConfig.RestartCmd.Name, commonConfig.LocalConfig.RestartCmd.Args)
|
|
|
+
|
|
|
+ launchedFlag = true
|
|
|
+ c_log.GlobalLogger.Info("数采程序启动 - 成功。")
|
|
|
}
|
|
|
|
|
|
-func judgeState1() {
|
|
|
- init := true
|
|
|
- turnLength := 60
|
|
|
- lastStatus := "NONE"
|
|
|
-
|
|
|
+func stopMasterNode() {
|
|
|
+
|
|
|
+ var killArgs commonService.KillSignal
|
|
|
+ killArgs = commonService.KillSignal{NodeName: "master", DropUploadData: commonConfig.PlatformConfig.DropUploadData, Restart: false}
|
|
|
+ c_log.GlobalLogger.Info("杀死任务,发送rpc结束信号:", killArgs)
|
|
|
+ KillRpcClient, err := rpc.Dial("tcp", commonConfig.LocalConfig.Node.Ip+":"+commonConfig.CloudConfig.RpcPort)
|
|
|
+ if err != nil {
|
|
|
+
|
|
|
+ lastCloudStatus = "NONE"
|
|
|
+ c_log.GlobalLogger.Error("采集程序已经停止:", err)
|
|
|
+ return
|
|
|
+ }
|
|
|
+
|
|
|
+ reply := 0
|
|
|
+ if err = KillRpcClient.Call("KillService.Kill", killArgs, &reply); err != nil {
|
|
|
+ c_log.GlobalLogger.Error("发送 rpc 请求到 master 报错:", err)
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+ c_log.GlobalLogger.Info("结束任务后,将数据闭环平台配置置空。")
|
|
|
+ commonConfig.PlatformConfig = commonConfig.PlatformConfigStruct{}
|
|
|
+ if err = KillRpcClient.Close(); err != nil {
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+ launchedFlag = false
|
|
|
+ c_log.GlobalLogger.Info("数采程序关闭 - 成功。")
|
|
|
+}
|
|
|
+
|
|
|
+func main() {
|
|
|
+ localTurnLength := 2
|
|
|
+ cloudTurnLength := 60
|
|
|
+ waitStopLength := 1
|
|
|
+
|
|
|
+
|
|
|
+ go pkg.GetLocalStatus(&localStatus, &lastLocalStatus, localTurnLength)
|
|
|
+
|
|
|
+ go pkg.GetCloudStatus(&cloudStatus, &lastCloudStatus, cloudTurnLength)
|
|
|
+
|
|
|
for {
|
|
|
- c_log.GlobalLogger.Errorf("一轮次扫描时间【%v】秒:", turnLength)
|
|
|
- if init {
|
|
|
- time.Sleep(time.Duration(1) * time.Second)
|
|
|
- init = false
|
|
|
+ if launchedFlag {
|
|
|
+ time.Sleep(time.Duration(cloudTurnLength) * time.Second)
|
|
|
} else {
|
|
|
- time.Sleep(time.Duration(turnLength) * time.Second)
|
|
|
+ time.Sleep(time.Duration(localTurnLength) * time.Second)
|
|
|
}
|
|
|
-
|
|
|
- status, err := commonConfig.GetStatus(commonConfig.PlatformConfig.TaskConfigId)
|
|
|
- if err != nil {
|
|
|
- c_log.GlobalLogger.Error("获取配置status失败:", err)
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
- if status == "UN_CHANGE" {
|
|
|
- lastStatus = "UN_CHANGE"
|
|
|
- continue
|
|
|
- } else if status == "CHANGE" || status == "NONE" {
|
|
|
- if lastStatus == "CHANGE" && status == "CHANGE" {
|
|
|
- commonConfig.InitPlatformConfig()
|
|
|
- continue
|
|
|
- }
|
|
|
- if lastStatus == "NONE" && status == "NONE" {
|
|
|
- continue
|
|
|
- }
|
|
|
-
|
|
|
- if lastStatus == "NONE" && status == "CHANGE" {
|
|
|
- if _, err := util.ExecuteWithPath(commonConfig.LocalConfig.RestartCmd.Dir, commonConfig.LocalConfig.RestartCmd.Name, commonConfig.LocalConfig.RestartCmd.Args...); err != nil {
|
|
|
- c_log.GlobalLogger.Info("启动新程序失败,【path】=", commonConfig.LocalConfig.RestartCmd.Dir, "【cmd】=", commonConfig.LocalConfig.RestartCmd.Name, commonConfig.LocalConfig.RestartCmd.Args, ":", err)
|
|
|
- os.Exit(-1)
|
|
|
- }
|
|
|
- c_log.GlobalLogger.Info("启动任务,本地执行启动命令:【path】=", commonConfig.LocalConfig.RestartCmd.Dir, "【cmd】=", commonConfig.LocalConfig.RestartCmd.Name, commonConfig.LocalConfig.RestartCmd.Args)
|
|
|
- lastStatus = status
|
|
|
- c_log.GlobalLogger.Info("获取数据闭环平台最新配置。")
|
|
|
- commonConfig.InitPlatformConfig()
|
|
|
- continue
|
|
|
- }
|
|
|
|
|
|
-
|
|
|
- var killArgs commonService.KillSignal
|
|
|
- if lastStatus == "UN_CHANGE" && status == "CHANGE" {
|
|
|
- killArgs = commonService.KillSignal{NodeName: "master", DropUploadData: commonConfig.PlatformConfig.DropUploadData, Restart: true}
|
|
|
- c_log.GlobalLogger.Info("更新任务,发送rpc重启信号:", killArgs)
|
|
|
- }
|
|
|
- if lastStatus == "UN_CHANGE" && status == "NONE" {
|
|
|
- killArgs = commonService.KillSignal{NodeName: "master", DropUploadData: commonConfig.PlatformConfig.DropUploadData, Restart: false}
|
|
|
- c_log.GlobalLogger.Info("杀死任务,发送rpc结束信号:", killArgs)
|
|
|
- }
|
|
|
+ fmt.Println("localStatus: ", localStatus, "lastLocalStatus: ", lastLocalStatus)
|
|
|
+ fmt.Println("cloudStatus: ", cloudStatus, "lastCloudStatus: ", lastCloudStatus)
|
|
|
|
|
|
- KillRpcClient, err := rpc.Dial("tcp", commonConfig.LocalConfig.Node.Ip+":"+commonConfig.CloudConfig.RpcPort)
|
|
|
- if err != nil {
|
|
|
-
|
|
|
- lastStatus = "NONE"
|
|
|
- c_log.GlobalLogger.Error("采集程序已经停止:", err)
|
|
|
- continue
|
|
|
- }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
|
|
|
- reply := 0
|
|
|
- if err = KillRpcClient.Call("KillService.Kill", killArgs, &reply); err != nil {
|
|
|
- c_log.GlobalLogger.Error("发送 rpc 请求到 master 报错:", err)
|
|
|
-
|
|
|
-
|
|
|
-
|
|
|
+
|
|
|
+ if localStatus == "running" {
|
|
|
+ if !launchedFlag {
|
|
|
+ c_log.GlobalLogger.Info("数采程序启动 - 开始。")
|
|
|
+ startMasterNode()
|
|
|
+ } else {
|
|
|
+ c_log.GlobalLogger.Info("设备仍处于运行状态,续约 - 开始。")
|
|
|
+
|
|
|
+ renew()
|
|
|
}
|
|
|
- lastStatus = status
|
|
|
- c_log.GlobalLogger.Info("结束任务后,将数据闭环平台配置置空。")
|
|
|
- commonConfig.PlatformConfig = commonConfig.PlatformConfigStruct{}
|
|
|
- if err = KillRpcClient.Close(); err != nil {
|
|
|
-
|
|
|
+ continue
|
|
|
+ } else if localStatus == "idle" {
|
|
|
+ if !renewedFlag && launchedFlag && len(entity.TimeWindowConsumerQueue) == 0 {
|
|
|
+ c_log.GlobalLogger.Info("设备不在运行状态且没有待处理的数据,数采程序关闭 - 开始。")
|
|
|
+ time.Sleep(time.Duration(waitStopLength) * time.Minute)
|
|
|
+ stopMasterNode()
|
|
|
}
|
|
|
+ continue
|
|
|
+ } else if localStatus == "error" {
|
|
|
+ c_log.GlobalLogger.Error("设备运行状态出错,停止数采程序。")
|
|
|
+ stopMasterNode()
|
|
|
} else {
|
|
|
- c_log.GlobalLogger.Error("未知的采集任务状态。【status】=", status)
|
|
|
+ c_log.GlobalLogger.Error("未知的设备运行状态。【status】=", localStatus)
|
|
|
}
|
|
|
+
|
|
|
+
|
|
|
+ go pkg.GetCloudConfig(cloudStatus, lastCloudStatus, cloudTurnLength)
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
}
|
|
|
+
|
|
|
}
|
|
|
-func judgeState3() {}
|