package main
|
|
import (
|
"context"
|
"flag"
|
"os"
|
"os/signal"
|
"syscall"
|
"vamicro/camera-service/cache"
|
"vamicro/camera-service/controllers"
|
"vamicro/camera-service/models"
|
"vamicro/camera-service/service"
|
"vamicro/config"
|
|
"basic.com/valib/bhomeclient.git"
|
"basic.com/valib/bhomedbapi.git"
|
"basic.com/valib/logger.git"
|
vaversion "basic.com/valib/version.git"
|
)
|
|
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()
|
models.InitVectorDb()
|
defer models.CloseDBClient()
|
defer models.CloseDB()
|
|
ctx, cancel := context.WithCancel(context.Background())
|
fm, pubTopics := initFuncMap()
|
var reg = &bhomeclient.RegisterInfo{
|
Proc: *proc,
|
Channel: nil,
|
PubTopic: pubTopics,
|
SubTopic: []string{bhomeclient.Proc_Scene_Service, bhomeclient.Proc_System_Service},
|
}
|
|
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 dealSubMsg(ctx, ms)
|
|
go ms.StartServer(fm)
|
cache.InitDataCache()
|
|
<-q
|
|
ms.DeRegister()
|
cancel()
|
ms.Free()
|
}
|
|
// 处理订阅消息
|
func dealSubMsg(ctx context.Context, ms *bhomeclient.MicroNode) {
|
|
for {
|
select {
|
case <-ctx.Done():
|
return
|
case msg := <-ms.SubCh:
|
logger.Debug("recv sub msg:", msg)
|
|
cache.UpdateBySub(msg)
|
}
|
}
|
}
|
|
const urlPrefix = "/data/api-v"
|
|
func initFuncMap() (map[string]bhomeclient.MicroFunc, []string) {
|
funcMap := make(map[string]bhomeclient.MicroFunc)
|
cc := new(controllers.CameraController)
|
ac := new(controllers.AreaController)
|
|
funcMap[urlPrefix+"/area/localmenu"] = ac.CameraTree
|
funcMap[urlPrefix+"/area/add"] = ac.AreaAdd
|
funcMap[urlPrefix+"/area/update"] = ac.AreaUpdate
|
funcMap[urlPrefix+"/area/del"] = ac.AreaDelete
|
|
funcMap[urlPrefix+"/camera/add"] = cc.CameraAdd
|
funcMap[urlPrefix+"/camera/update"] = cc.CameraUpdate
|
funcMap[urlPrefix+"/camera/del"] = cc.CameraDel
|
funcMap[urlPrefix+"/camera/show"] = cc.CameraSel
|
funcMap[urlPrefix+"/camera/showAll"] = cc.GetAllCameras
|
funcMap[urlPrefix+"/camera/getCamerasByRunType"] = cc.GetCamerasByRunType
|
funcMap[urlPrefix+"/camera/updateRunEnable"] = cc.UpdateRunEnable
|
funcMap[urlPrefix+"/camera/changeRunType"] = cc.ChangeRunType
|
funcMap[urlPrefix+"/camera/updateIsRunningState"] = cc.UpdateIsRunningState
|
funcMap[urlPrefix+"/camera/updateIsRunningAll"] = cc.UpdateIsRunningAll
|
funcMap[urlPrefix+"/camera/updateSnapshotUrl"] = cc.UpdateSnapshot
|
funcMap[urlPrefix+"/camera/updateSnapshotSaaS"] = cc.UpdateSnapshotSaaS
|
funcMap[urlPrefix+"/camera/updateCameraArea"] = cc.UpdateCameraArea
|
|
funcMap[urlPrefix+"/camera/findOnMap"] = cc.FindOnMap
|
funcMap[urlPrefix+"/camera/updatePos"] = cc.UpdatePos
|
funcMap[urlPrefix+"/camera/nodeCamera"] = cc.NodeCamera
|
funcMap[urlPrefix+"/camera/coord"] = cc.Coord
|
funcMap[urlPrefix+"/camera/updateCoord"] = cc.UpdateCoord
|
funcMap[urlPrefix+"/camera/getSyncCameraData"] = cc.GetSyncCameraData
|
funcMap[urlPrefix+"/camera/getSyncSensorData"] = cc.GetSyncSensorData
|
funcMap[urlPrefix+"/camera/saveNotExistOrUpdate"] = cc.SaveNotExistOrUpdate
|
|
//获取抓拍预警信息
|
funcMap[urlPrefix+"/camera/getRecord"] = cc.GetRecord
|
//获取任务根据摄像机ids
|
funcMap[urlPrefix+"/camera/getTasks"] = cc.GetTasks
|
|
funcMap[urlPrefix+"/task/aggregateTaskList"] = cc.GetAllTasks
|
|
var pubTopics []string
|
for key, _ := range funcMap {
|
pubTopics = append(pubTopics, key)
|
}
|
return funcMap, pubTopics
|
}
|
