package main
|
|
import (
|
"apsClient/conf"
|
"apsClient/constvar"
|
"apsClient/crontask"
|
"apsClient/model"
|
"apsClient/nsq"
|
"apsClient/pkg/logx"
|
"apsClient/pkg/sqlitex"
|
"apsClient/router"
|
"apsClient/serf"
|
"apsClient/service"
|
"fmt"
|
"log"
|
"net/http"
|
_ "net/http/pprof"
|
"time"
|
)
|
|
func main() {
|
|
logx.Init(conf.Conf.Log)
|
defer logx.Sync()
|
|
if err := model.Init(); err != nil {
|
logx.Errorf("model Init err:%v", err)
|
return
|
}
|
|
//加载plc写入地址
|
//plc_address.LoadAddressFromFile()
|
|
// 启动数据同步
|
var serfStartChan = make(chan bool)
|
|
// 需要同步的表
|
var syncTables = []string{
|
"procedures",
|
"process_model",
|
"work_order",
|
"task_status_sync",
|
"device",
|
"device_plc",
|
"system_status",
|
"process_model_plc_address",
|
"reports_to_cloud",
|
"report_work",
|
}
|
|
agent := serf.InitAgent("apsClient", syncTables, sqlitex.GetDB())
|
agent.RegisterClusterEvent(serfClusterEvent)
|
|
go agent.Serve(serfStartChan)
|
if !<-serfStartChan {
|
logx.Errorf("serf Init err, exit")
|
return
|
}
|
|
//从文件里读取当前生产设备id获取设备列表第一个
|
err := service.InitCurrentDeviceID(serf.Vasystem.ServerID)
|
if err != nil {
|
logx.Errorf("InitCurrentDeviceID error: %v, exit", err)
|
return
|
}
|
|
go func() {
|
service.ReportsSystemDeviceToCloud(serf.Vasystem.ServerID)
|
}()
|
|
// 判断当前集群状态
|
logx.Infof("current agent.ClusterStatus:%v", agent.ClusterStatus)
|
log.Println("current agent.ClusterStatus:", agent.ClusterStatus)
|
|
conf.Conf.SerfClusterStatus = agent.ClusterStatus
|
|
if agent.ClusterStatus != "slave" {
|
if err := nsq.Init(); err != nil {
|
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...")
|
server := &http.Server{
|
Addr: fmt.Sprintf(":%d", conf.Conf.System.Port),
|
Handler: router.InitRouter(),
|
ReadTimeout: 5 * time.Second,
|
WriteTimeout: 5 * time.Second,
|
}
|
logx.Error(server.ListenAndServe().Error())
|
}
|
|
func serfClusterEvent(stat int) {
|
ChangeClusterStatus(stat)
|
switch stat {
|
case serf.EventCreateCluster, serf.EventSlave2Master, serf.EventLeaveCluster:
|
if err := nsq.Init(); err != nil { //开启nsq
|
logx.Errorf("nsq Init err:%v", err)
|
return
|
}
|
crontask.Once(true)
|
if err := crontask.RestartTask(true); err != nil { //以master方式重启task
|
logx.Errorf("crontab task Init err:%v", err)
|
return
|
}
|
case serf.EventJoinCluster, serf.EventMaster2Slave:
|
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)
|
}
|
|
func ChangeClusterStatus(stat int) {
|
switch stat {
|
case serf.EventSlave2Master, serf.EventCreateCluster:
|
conf.Conf.SerfClusterStatus = constvar.SerfClusterStatusMaster
|
case serf.EventLeaveCluster:
|
conf.Conf.SerfClusterStatus = constvar.SerfClusterStatusNull
|
case serf.EventJoinCluster, serf.EventMaster2Slave:
|
conf.Conf.SerfClusterStatus = constvar.SerfClusterStatusSlave
|
}
|
}
|
