package main import ( "basic.com/valib/bhomeclient.git" "basic.com/valib/bhomedbapi.git" "basic.com/valib/logc.git" "basic.com/valib/logger.git" "basic.com/valib/version.git" "context" "flag" "os" "os/signal" "syscall" "vamicro/config" "vamicro/log-service/controllers" "vamicro/log-service/models" "vamicro/log-service/service" ) var ( procName = service.ProcName proc = &bhomeclient.ProcInfo{ Name: procName, //进程名称 ID: procName, //进程id Info: "", //进程的描述信息,用于区分同一进程名称下多个进程 } env = flag.String("e", "pro", "") dbPath string = "../config/log-service.db" logSetsDbPath string = "../config/op-log-sets.db" ) 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() { ctx, cancel := context.WithCancel(context.Background()) models.Init(dbPath, logSetsDbPath, ctx) defer models.CloseDB() fm, pubTopics := initFuncMap() var reg = &bhomeclient.RegisterInfo{ Proc: *proc, Channel: nil, PubTopic: pubTopics, SubTopic: []string{ logc.OperationLogTopic, logc.ScheduleLogTopic, logc.RuleServerLogTopic, logc.VaSystemLogTopic, }, } 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 handleSubMsg(ctx, ms) go ms.StartServer(fm) <-q _ = ms.DeRegister() cancel() ms.Free() } const urlPrefix = "/data/api-v" func initFuncMap() (map[string]bhomeclient.MicroFunc, []string) { funcMap := make(map[string]bhomeclient.MicroFunc) service := new(controllers.LogServiceController) funcMap[urlPrefix+"/log/queryOperationLog"] = service.OperationLog funcMap[urlPrefix+"/log/queryScheduleLog"] = service.ScheduleLog funcMap[urlPrefix+"/log/operations"] = service.Operations funcMap[urlPrefix+"/log/modules"] = service.Modules funcMap[urlPrefix+"/log/set_modules"] = service.SetModules funcMap[urlPrefix+"/log/set_operations"] = service.SetOperations funcMap[urlPrefix+"/log/find_all_modules_map"] = service.FindAllMap funcMap[urlPrefix+"/log/ruleServerLog"] = service.RuleServerLog funcMap[urlPrefix+"/log/vaSystemLog"] = service.VaSystemLog funcMap[urlPrefix+"/log/procNames"] = service.GetProcNames 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) topic := string(msg.Topic) models.PersistentWrapper(topic, msg.Data) } } }