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)
|
}
|
}
|
}
|
