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" "os" "os/signal" "syscall" "vamicro/config" "vamicro/scene-service/controllers" "vamicro/scene-service/models" "net/http" _ "net/http/pprof" ) var ( procName = "scene-service" proc = &bhomeclient.ProcInfo{ Name: procName, //进程名称 ID: procName, //进程id Info: "", //进程的描述信息,用于区分同一进程名称下多个进程 } env = flag.String("e", "pro", "") ) func init() { flag.Parse() 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() { // pprof 用于分析性能 go func() { logger.Info(http.ListenAndServe("0.0.0.0:6089", nil)) }() flag.Parse() vaversion.Usage() 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{}, } 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) <-q ms.DeRegister() cancel() ms.Free() } const urlPrefix = "/data/api-v" func initFuncMap() (map[string]bhomeclient.MicroFunc, []string) { funcMap := make(map[string]bhomeclient.MicroFunc) crc := new(controllers.CameraRuleController) ctc := new(controllers.CameraTimeRuleController) cpc := new(controllers.CameraPolygonController) cgc := new(controllers.CameraGroupController) tc := new(controllers.CameraTaskController) funcMap[urlPrefix+"/camera/rule/findAll"] = crc.FindAll funcMap[urlPrefix+"/camera/rule/save"] = crc.Save funcMap[urlPrefix+"/camera/rule/saveLinkRulesByGroup"] = crc.SaveLinkRulesByGroup funcMap[urlPrefix+"/camera/rule/getLinkRulesByCameraIds"] = crc.GetLinkRulesByCameraIds funcMap[urlPrefix+"/camera/rule/findByCameraId"] = crc.FindByCameraId funcMap[urlPrefix+"/camera/rule/existRunningTask"] = crc.ExistRunningTask funcMap[urlPrefix+"/camera/pasteRules"] = crc.PasteRules funcMap[urlPrefix+"/camera/rule/updateAlarmLevel"] = crc.UpdateAlarmLevel funcMap[urlPrefix+"/camera/rule/deleteGroup"] = crc.DeleteByGroup funcMap[urlPrefix+"/camera/findAllCameraLink"] = crc.FindAllCameraLink funcMap[urlPrefix+"/camera/rule/findAllTaskById"] = crc.FindAllTaskById funcMap[urlPrefix+"/camera/findLinksByCameraIds"] = crc.FindLinksByCameraIds funcMap[urlPrefix+"/camera/rule/dispatch"] = crc.Dispatch funcMap[urlPrefix+"/camera/rule/syncDelBySaas"] = crc.SyncDelBySaas funcMap[urlPrefix+"/camera/alltask"] = tc.FindTasksByCameraIds /****************时间设置******************/ funcMap[urlPrefix+"/cameraTimerule/save"] = ctc.Save funcMap[urlPrefix+"/cameraTimerule/delete"] = ctc.Delete funcMap[urlPrefix+"/cameraTimerule/findAll"] = ctc.FindAll /*********************区域配置***********************/ funcMap[urlPrefix+"/polygon/save"] = cpc.Save //funcMap[urlPrefix+"/polygon/delete"] = cpc.Delete funcMap[urlPrefix+"/polygon/findByCameraId"] = cpc.FindByCameraId funcMap[urlPrefix+"/polygon/findAll"] = cpc.FindAll funcMap[urlPrefix+"/polygon/updateDefenceStateByPolygonId"] = cpc.UpdateDefenceStateByPolygonId funcMap[urlPrefix+"/polygon/updateDefenceStateByCameraId"] = cpc.UpdateDefenceStateByCameraId funcMap[urlPrefix+"/polygon/findByCamGroup"] = cpc.FindByCamGroup /*************************saas同步***************************/ funcMap[urlPrefix+"/polygon/getSyncSceneData"] = cpc.GetSyncSceneData /**********************关联区域*************************/ funcMap[urlPrefix+"/polygon/relation/findAll"] = cpc.FindAllRelation funcMap[urlPrefix+"/polygon/relations"] = cpc.Relations funcMap[urlPrefix+"/polygon/relation/save"] = cpc.SaveRelation funcMap[urlPrefix+"/polygon/relation/del"] = cpc.DelRelation funcMap[urlPrefix+"/polygon/relation/findByGroup"] = cpc.FindRelationByGroup /**********************太原营业厅需求*************************/ acpC := new(controllers.AreaCamPgnController) funcMap[urlPrefix+"/acp/findAll"] = acpC.FindAll funcMap[urlPrefix+"/acp/save"] = acpC.Save funcMap[urlPrefix+"/acp/delete"] = acpC.Delete /*************************全景图***************************/ funcMap[urlPrefix+"/panorama/show"] = cpc.ShowPanorama //获取全景图地址 funcMap[urlPrefix+"/panorama/upload"] = cpc.UploadPanorama //上传新的全景图地址 funcMap[urlPrefix+"/camera/group/findAll"] = cgc.FindAll funcMap[urlPrefix+"/camera/group/save"] = cgc.Save funcMap[urlPrefix+"/camera/group/del"] = cgc.Delete var pubTopics []string for key, _ := range funcMap { pubTopics = append(pubTopics, key) } return funcMap, pubTopics }