package main import ( "basic.com/valib/bhomeclient.git" "basic.com/valib/bhomedbapi.git" "basic.com/valib/logger.git" "basic.com/valib/version.git" "context" "flag" "net/http" _ "net/http/pprof" "os" "os/signal" "syscall" "vamicro/appcenter-service/controllers" "vamicro/appcenter-service/models" "vamicro/appcenter-service/service" "vamicro/config" versionControlS "vamicro/version-control/service" ) var ( procName = service.ProcName proc = &bhomeclient.ProcInfo{ Name: procName, //进程名称 ID: procName, //进程id Info: "", //进程的描述信息,用于区分同一进程名称下多个进程 } env = flag.String("e", "pro", "") ) 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(){ models.Init() defer models.CloseDB() ctx, cancel := context.WithCancel(context.Background()) fm,pubTopics := initFuncMap() var reg = &bhomeclient.RegisterInfo { Proc: *proc, Channel: nil, PubTopic: pubTopics, SubTopic: []string{versionControlS.SysUpdateConfigTopic}, } q := make(chan os.Signal, 1) signal.Notify(q, os.Interrupt, os.Kill, syscall.SIGTERM) ms, err := bhomeclient.NewMicroNode(ctx, q, config.Server.AnalyServerId, reg, logger.Debug) if err !=nil { return } bhomedbapi.InitLog(logger.Debug) bhomedbapi.InitGetNetNode(ms.GetLocalNetNodeByTopic) bhomedbapi.InitDoReq(ms.RequestOnly) go ms.StartServer(fm) go handleSubMsg(ctx, ms) go service.SelfUpdateStart(ctx) //自动更新 go service.NoticeTick(ctx)//更新提醒 // pprof 用于分析性能 go func() { logger.Info(http.ListenAndServe("0.0.0.0:7073", nil)) }() <-q ms.DeRegister() cancel() ms.Free() } const urlPrefix = "/data/api-v" func initFuncMap() (map[string]bhomeclient.MicroFunc,[]string) { funcMap := make(map[string]bhomeclient.MicroFunc) sc := new(controllers.SdkController) sa := new(controllers.SdkArgController) ac := new(controllers.AppController) funcMap[urlPrefix+"/sdk/register"] = sc.Register funcMap[urlPrefix+"/sdk/getById"] = sc.GetById funcMap[urlPrefix+"/sdk/findAllSdk"] = sc.FindAllSdk funcMap[urlPrefix+"/sdk/getAllSdkChanSet"] = sc.GetAllSdkChanSet funcMap[urlPrefix+"/sdk/findByTaskId"] = sc.FindByTaskId funcMap[urlPrefix+"/sdk/updateSdkArgConfig"] = sc.UpdateSdkArgConfig funcMap[urlPrefix+"/sdk/updateSdkChanSet"] = sc.UpdateSdkChanSet funcMap[urlPrefix+"/sdkArg/getSdkArgs"] = sa.GetSdkArgs funcMap[urlPrefix+"/sdk/upload"] = sc.Upload funcMap[urlPrefix+"/sdk/uploadPack"] = sc.UploadPack funcMap[urlPrefix+"/sdk/install"] = sc.Install funcMap[urlPrefix+"/sdk/unInstall"] = sc.UnInstall funcMap[urlPrefix+"/sdk/buyList"] = sc.BuyList funcMap[urlPrefix+"/sdk/active"] = sc.Active funcMap[urlPrefix+"/sdk/sdkDownload"] = sc.SdkDownLoad funcMap[urlPrefix+"/sdk/showInstallInfo"] = sc.ShowInstallInfo funcMap[urlPrefix+"/app/findAllApp"] = ac.FindAll funcMap[urlPrefix+"/app/getAppInfo"] = ac.GetAppInfo funcMap[urlPrefix+"/app/unInstall"] = ac.UnInstall funcMap[urlPrefix+"/app/buyList"] = ac.BuyList funcMap[urlPrefix+"/app/active"] = ac.Active funcMap[urlPrefix+"/app/detail"] = ac.Detail funcMap[urlPrefix+"/app/upgrade/notice"] = ac.UpgradeNotice funcMap[urlPrefix+"/app/upgrade/notice/delay"] = ac.DelayNotice var pubTopics []string for key,_ := range funcMap { pubTopics = append(pubTopics, key) } return funcMap, pubTopics } func handleSubMsg(ctx context.Context, ms *bhomeclient.MicroNode) { for { select { case <-ctx.Done(): return case msg := <-ms.SubCh: logger.Debug("handleSubMsg recv sub msg:", msg) service.PersistentWrapper(string(msg.Topic), msg.Data) } } }