定时同步任务状态

zhangqian

2023-10-17 56307a744dc94f9b1eed3f7513a4be16a130e984

 main.go

@@ -6,14 +6,12 @@
   "apsClient/model"
   "apsClient/nsq"
   "apsClient/pkg/logx"
   "apsClient/pkg/sqlitex"
   "apsClient/router"
   "apsClient/service"
   "apsClient/serf"
   "apsClient/service/plc_address"
   "fmt"
   "net/http"
   "os"
   "os/signal"
   "syscall"
   "time"
)

@@ -27,22 +25,38 @@
      return
   }

   if err := nsq.Init(); err != nil {
      logx.Errorf("nsq Init err:%v", err)
      return
   }

   if err := crontask.InitTask(); err != nil {
      logx.Errorf("crontab task Init err:%v", err)
      return
   }
   //加载plc写入地址
   plc_address.LoadAddressFromFile()

   //提前加载任务
   service.NewTaskService().GetTask()
   // 启动数据同步
   var serfStartChan = make(chan bool)
   // 需要同步的表
   var syncTables = []string{
      "procedures",
      "process_model",
      "production_progress",
      "work_order",
      "task_status_sync",
   }

   go shutdown()
   agent := serf.InitAgent("apsClient", syncTables, sqlitex.GetDB())
   agent.RegisterClusterEvent(serfClusterEvent)
   go agent.Serve(serfStartChan)
   <-serfStartChan

   // 判断当前集群状态
   if agent.ClusterStatus != "slave" {
      if err := nsq.Init(); err != nil {
         logx.Errorf("nsq Init err:%v", err)
         return
      }
      crontask.Once()
      if err := crontask.StartTask(); err != nil {
         logx.Errorf("crontab task Init err:%v", err)
         return
      }
   }

   logx.Infof("apsClient start serve...")
   server := &http.Server{
      Addr:         fmt.Sprintf(":%d", conf.Conf.System.Port),
@@ -53,11 +67,22 @@
   logx.Error(server.ListenAndServe().Error())
}

func shutdown() {
   quit := make(chan os.Signal, 1)
   signal.Notify(quit, syscall.SIGKILL, syscall.SIGQUIT, syscall.SIGINT, syscall.SIGTERM)
   <-quit
func serfClusterEvent(stat int) {
   switch stat {
   case serf.EventCreateCluster, serf.EventSlave2Master, serf.EventLeaveCluster:
      if err := nsq.Init(); err != nil {
         logx.Errorf("nsq Init err:%v", err)
         return
      }
      crontask.Once()
      if err := crontask.StartTask(); err != nil {
         logx.Errorf("crontab task Init err:%v", err)
         return
      }
   case serf.EventJoinCluster, serf.EventMaster2Slave:
      nsq.Stop()
      crontask.StopTask()
   }

   logx.Infof("apsClient exited...")
   os.Exit(0)
   logx.Infof("serf cluster event: %v", stat)
}