package main
|
|
import (
|
"context"
|
"flag"
|
"fmt"
|
"os"
|
"os/signal"
|
"sync"
|
"syscall"
|
|
"vamicro/config"
|
"vamicro/version-control/api"
|
"vamicro/version-control/controller"
|
"vamicro/version-control/service"
|
|
"basic.com/valib/bhomeclient.git"
|
"basic.com/valib/bhomedbapi.git"
|
"basic.com/valib/logger.git"
|
"basic.com/valib/version.git"
|
)
|
|
var (
|
env = flag.String("e", "pro", "")
|
procName = service.ProcName
|
proc = &bhomeclient.ProcInfo{
|
Name: procName, //进程名称
|
ID: procName, //进程id
|
Info: "", //进程的描述信息，用于区分同一进程名称下多个进程
|
}
|
)
|
|
func init() {
|
flag.Parse()
|
vaversion.Usage()
|
|
config.Init(*env)
|
|
// 日志初始化
|
var logFile = config.LogConf.Path + "vamicro-" + procName + ".log"
|
logger.InitLogger(logFile, config.LogConf.Level, config.LogConf.MaxSize, config.LogConf.MaxBackups, config.LogConf.MaxAge)
|
logger.Info("log init success !")
|
}
|
|
func main() {
|
//防止多进程运行
|
lockFile := "/tmp/run/version-control.lock"
|
var lock *os.File
|
lock, err := os.Create(lockFile)
|
if err != nil {
|
fmt.Println("create lock file error", err)
|
os.Exit(1)
|
}
|
|
defer os.Remove(lockFile)
|
defer lock.Close()
|
|
err = syscall.Flock(int(lock.Fd()), syscall.LOCK_EX|syscall.LOCK_NB)
|
if err != nil {
|
fmt.Println("report service is running ")
|
os.Exit(1)
|
}
|
|
defer syscall.Flock(int(lock.Fd()), syscall.LOCK_UN)
|
|
wg := &sync.WaitGroup{}
|
q := make(chan os.Signal, 1)
|
signal.Notify(q, os.Interrupt, os.Kill, syscall.SIGTERM)
|
|
fm, pubTopics := initFuncMap()
|
ctx, cancel := context.WithCancel(context.Background())
|
|
var reg = &bhomeclient.RegisterInfo{
|
Proc: *proc,
|
Channel: nil,
|
PubTopic: pubTopics,
|
SubTopic: []string{},
|
}
|
ms, err := bhomeclient.NewMicroNode(ctx, q, config.Server.AnalyServerId, reg, nil)
|
if err != nil {
|
return
|
}
|
bhomedbapi.InitGetNetNode(ms.GetLocalNetNodeByTopic)
|
bhomedbapi.InitDoReq(ms.RequestOnly)
|
bhomedbapi.InitLog(logger.Debug)
|
|
server := api.InitRouter(wg, ms)
|
|
go ms.StartServer(fm)
|
|
go service.SelfUpdateStart(ctx, ms) //自动更新
|
|
go service.NoticeTick(ctx) //更新提醒
|
|
go service.AuthorizationUpdate(ctx, ms) //更新授权
|
|
go service.BakSchedule(ctx)
|
|
<-q
|
logger.Info("user signalled...")
|
|
// shutdown http service gracefully
|
if err := server.Shutdown(ctx); nil != err {
|
logger.Fatal("server shutdown failed, err:", err)
|
}
|
|
cancel()
|
wg.Wait()
|
logger.Info("all sub threads exited...")
|
|
//upgrade.CheckVersion()
|
|
//fmt.Println("appver.BuildTime:", appver.BuildTime, "IvyTest:", IvyTest)
|
//ctx, cancel := context.WithCancel(context.Background())
|
}
|
|
const urlPrefix = "/data/api-v"
|
|
func initFuncMap() (map[string]bhomeclient.MicroFunc, []string) {
|
funcMap := make(map[string]bhomeclient.MicroFunc)
|
vController := new(controller.VersionController)
|
|
funcMap[urlPrefix+"/version/check"] = vController.VersionCheck
|
funcMap[urlPrefix+"/version/upgrade"] = vController.UpgradeVersion
|
funcMap[urlPrefix+"/version/snBus"] = vController.SnWithBus
|
|
var pubTopics []string
|
for key, _ := range funcMap {
|
pubTopics = append(pubTopics, key)
|
}
|
return funcMap, pubTopics
|
}
|
