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 }