From f73f610cdf4b0666dc139a51e72353b88f8f25ab Mon Sep 17 00:00:00 2001 From: zhangzengfei <zhangzengfei@smartai.com> Date: 星期四, 19 十月 2023 17:27:17 +0800 Subject: [PATCH] 为保证正确获取集群状态, 添加程序启动时读取system-service运行状态 --- main.go | 33 +++++++++++++++++++++++++-------- 1 files changed, 25 insertions(+), 8 deletions(-) diff --git a/main.go b/main.go index 569c9e7..8f852e5 100644 --- a/main.go +++ b/main.go @@ -25,28 +25,29 @@ return } - if err := crontask.InitTask(); err != nil { - logx.Errorf("crontab task Init err:%v", err) - return - } - //鍔犺浇plc鍐欏叆鍦板潃 plc_address.LoadAddressFromFile() // 鍚姩鏁版嵁鍚屾 var serfStartChan = make(chan bool) + // 闇�瑕佸悓姝ョ殑琛� var syncTables = []string{ "procedures", "process_model", "production_progress", "work_order", + "task_status_sync", } agent := serf.InitAgent("apsClient", syncTables, sqlitex.GetDB()) agent.RegisterClusterEvent(serfClusterEvent) + go agent.Serve(serfStartChan) - <-serfStartChan + if !<-serfStartChan { + logx.Errorf("serf Init err, exit") + return + } // 鍒ゆ柇褰撳墠闆嗙兢鐘舵�� if agent.ClusterStatus != "slave" { @@ -54,6 +55,12 @@ logx.Errorf("nsq Init err:%v", err) return } + + } + crontask.Once(agent.ClusterStatus != "slave") + if err := crontask.StartTask(agent.ClusterStatus != "slave"); err != nil { + logx.Errorf("crontab task Init err:%v", err) + return } logx.Infof("apsClient start serve...") @@ -69,12 +76,22 @@ func serfClusterEvent(stat int) { switch stat { case serf.EventCreateCluster, serf.EventSlave2Master, serf.EventLeaveCluster: - if err := nsq.Init(); err != nil { + if err := nsq.Init(); err != nil { //寮�鍚痭sq logx.Errorf("nsq Init err:%v", err) return } + crontask.Once(true) + if err := crontask.RestartTask(true); err != nil { //浠aster鏂瑰紡閲嶅惎task + logx.Errorf("crontab task Init err:%v", err) + return + } case serf.EventJoinCluster, serf.EventMaster2Slave: - nsq.Stop() + nsq.Stop() //鍏抽棴nsq + crontask.Once(false) + if err := crontask.RestartTask(false); err != nil { //浠ラ潪master鏂瑰紡閲嶅惎task + logx.Errorf("crontab task Init err:%v", err) + return + } } logx.Infof("serf cluster event: %v", stat) -- Gitblit v1.8.0