package controllers
|
|
import (
|
"sort"
|
"strconv"
|
"strings"
|
"vamicro/camera-common/models"
|
"vamicro/camera-service/cache"
|
"vamicro/extend/util"
|
|
"basic.com/pubsub/protomsg.git"
|
"basic.com/valib/bhomeclient.git"
|
"basic.com/valib/bhomedbapi.git"
|
"basic.com/valib/logger.git"
|
uuid "github.com/satori/go.uuid"
|
)
|
|
type AreaController struct {
|
bk bhomeclient.Broker
|
}
|
|
// @Summary 显示树形结构
|
// @Description 显示左侧所有区域和摄像机
|
// @Produce json
|
// @Tags menu
|
// @Param parentId query string true "区域的id"
|
// @Param searchType query int true "查询类型（0:全部，1:分析摄像机，2:监控摄像机, 3:联动摄像机）"
|
// @Param cameraName query string false "摄像机名称"
|
// @Success 200 {string} json "{"code":200, msg:"目录结构数据"}"
|
// @Failure 500 {string} json "{"code":500, msg:"返回错误信息"}"
|
// @Router /data/api-v/area/localmenu [get]
|
func (ac AreaController) CameraTree(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var aram models.Area
|
var searchType = 0
|
var err error
|
|
parentId := c.Query("parentId")
|
if parentId == "" {
|
parentId = "0"
|
}
|
searchTypeStr := c.Query("searchType")
|
cameraName := c.Query("cameraName")
|
isPlatformStr := c.Query("isPlatform")
|
isPlatform, _ := strconv.Atoi(isPlatformStr) //单机获取本机摄像机，平台获取所有摄像机
|
|
treeType := models.TYPE_LOCAL_TREE //默认是本地树
|
|
if searchTypeStr != "" {
|
searchType, err = strconv.Atoi(searchTypeStr)
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误", Data: ""}
|
}
|
}
|
|
areas, err := aram.Find(parentId, cameraName, util.FiledbyStuct(models.TreeNode{}), treeType)
|
if err != nil {
|
logger.Debug("area.Find err:", err)
|
return &bhomeclient.Reply{Success: false, Msg: err.Error()}
|
}
|
logger.Debug("len(areas):", len(areas))
|
|
lkMap := make(map[string]string)
|
linkWhere := ""
|
if searchType == 3 { //查联动
|
var crApi bhomedbapi.CameraRuleApi
|
lb, lkList := crApi.FindAllCameraLink()
|
if !lb {
|
return &bhomeclient.Reply{Msg: "查询联动配置失败"}
|
}
|
if len(lkList) == 0 {
|
return &bhomeclient.Reply{Success: true, Data: []struct{}{}}
|
}
|
for _, lc := range lkList {
|
cArr := strings.Split(lc.CameraIds, ",")
|
for _, lcId := range cArr {
|
if _, ok := lkMap[lcId]; !ok {
|
lkMap[lcId] = lcId
|
}
|
}
|
}
|
for k := range lkMap {
|
linkWhere += "'" + k + "',"
|
}
|
linkWhere = strings.TrimRight(linkWhere, ",")
|
}
|
var cam models.Camera
|
cams, err := cam.Find(searchType, cameraName, models.TYPE_LOCAL_CAMERA, isPlatform, linkWhere)
|
if err != nil {
|
logger.Debug("cam.Find err:", err)
|
return &bhomeclient.Reply{Success: false, Msg: err.Error()}
|
}
|
|
var totalCams models.CamTreeNodeList
|
totalCams = cams
|
sort.Sort(totalCams)
|
rAreas := models.RecursiveArea(areas, cams, searchType)
|
|
allRules, e2 := cache.GetCameraRules()
|
ruleM := make(map[string]protomsg.CameraAndRules)
|
if e2 == nil && allRules != nil {
|
for _, r := range allRules {
|
if r.CameraInfo != nil {
|
ruleM[r.CameraInfo.Id] = r
|
}
|
}
|
}
|
|
//获取督查任务
|
tasks := models.GetTasks()
|
|
var totalAreas models.AreaList
|
totalAreas = rAreas
|
sort.Sort(totalAreas)
|
logger.Debug("len(rAreas):", len(rAreas))
|
arr := menuFormat(totalAreas, parentId, totalCams, treeType, tasks)
|
|
for _, camE := range totalCams {
|
if camE.Areaid == "0" {
|
isRunnig := false
|
isAI := false
|
camera := models.Camera{}
|
rows, e := camera.SelectById(camE.Id)
|
|
if rows > 0 && e == nil {
|
isRunnig = camera.IsRunning
|
if taskInfo, ok := tasks[camera.Id]; ok && taskInfo != nil && len(taskInfo) > 0 && (camera.RunType == models.TYPE_RUNTYPE_POLL || camera.RunType == models.TYPE_RUNTYPE_REALTIME) {
|
isAI = true
|
}
|
}
|
arr = append(arr, models.TreeMenu{
|
Id: camE.Id,
|
Type: "4",
|
Name: camE.Name,
|
IsAI: isAI, //是否AI摄像机
|
IsRunning: isRunnig,
|
Rtsp: camE.Rtsp,
|
CameraType: treeType,
|
})
|
}
|
}
|
return &bhomeclient.Reply{Success: true, Data: arr}
|
}
|
|
// 将区域和摄像机数组转换目录树形结构
|
func menuFormat(dbArr []models.Area, parentId string, dbCams []models.CameraTreeNode, treeType int, tasks map[string][]interface{}) []models.TreeMenu {
|
var arr []models.TreeMenu
|
for _, area := range dbArr {
|
if parentId == area.Parentid {
|
tmp := models.TreeMenu{}
|
tmp.Id = area.Id
|
tmp.Type = "MENU"
|
tmp.Name = area.Name
|
|
sontmp := menuFormat(dbArr, area.Id, dbCams, treeType, tasks)
|
tmp.Areanodes = sontmp
|
|
for _, val := range dbCams {
|
if val.Areaid == area.Id {
|
camtmp := models.TreeMenu{}
|
camtmp.Id = val.Id
|
camtmp.Name = val.Name
|
camtmp.Type = "4"
|
isRunnig := false
|
isAI := false
|
camera := models.Camera{}
|
rows, e := camera.SelectById(val.Id)
|
if rows > 0 && e == nil {
|
isRunnig = camera.IsRunning
|
if taskInfo, ok := tasks[camera.Id]; ok && taskInfo != nil && len(taskInfo) > 0 && (camera.RunType == models.TYPE_RUNTYPE_POLL || camera.RunType == models.TYPE_RUNTYPE_REALTIME) {
|
isAI = true
|
}
|
//if camera.RunType == models.TYPE_RUNTYPE_POLL || camera.RunType == models.TYPE_RUNTYPE_REALTIME {
|
// isAI = true
|
//}
|
}
|
camtmp.IsAI = isAI
|
camtmp.IsRunning = isRunnig
|
camtmp.Rtsp = val.Rtsp
|
camtmp.CameraType = treeType
|
tmp.Areanodes = append(tmp.Areanodes, camtmp)
|
}
|
}
|
arr = append(arr, tmp)
|
}
|
}
|
return arr
|
}
|
|
// @Summary 添加menu的区域
|
// @Description 添加目录上区域
|
// @Produce json
|
// @Tags menu
|
// @Param name formData string true "区域名字"
|
// @Param parentid formData string true "上一级父id"
|
// @Success 200 {string} json "{"code":200, data:"添加的区域信息",msg:"请求成功", success:true}"
|
// @Failure 200 {string} json "{"code":"错误码", data:"出错信息",msg:"请求失败", success:false}"
|
// @Router /data/api-v/area/add [post]
|
func (ac AreaController) AreaAdd(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var pArea models.Area
|
var cArea models.Area
|
name := c.PostForm("name")
|
parentId := c.PostForm("parentId")
|
if parentId == "" || parentId == "0" {
|
parentId = "0"
|
cArea.Parentids = ",0,"
|
} else {
|
if _, err := pArea.SelectbyId(parentId); err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
cArea.Parentids = pArea.Parentids + parentId + ","
|
}
|
cArea.Id = uuid.NewV4().String()
|
cArea.Name = name
|
cArea.Parentid = parentId
|
cArea.Type = models.TYPE_LOCAL_TREE
|
|
if cArea.IsUnique("", cArea.Name) {
|
return &bhomeclient.Reply{Msg: "名称不允许重复"}
|
}
|
|
if err := cArea.Insert(); err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: err.Error()}
|
}
|
return &bhomeclient.Reply{Success: true, Data: cArea}
|
}
|
|
// @Summary 修改名字
|
// @Description 修改区域名字
|
// @Accept json
|
// @Produce json
|
// @Tags menu
|
// @Param id formData string true "区域id"
|
// @Param name formData string true "区域名字"
|
// @Param parentId formData string true "上一级父id"
|
// @Param alias formData string false "备注名称"
|
// @Success 200 {string} json "{"code":200, data:"删除的区域信息",msg:"请求成功", success:true}"
|
// @Failure 200 {string} json "{"code":"错误码", data:"出错信息",msg:"请求失败", success:false}"
|
// @Router /data/api-v/area/update [post]
|
func (ac AreaController) AreaUpdate(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var aram models.Area
|
|
id := c.PostForm("id")
|
if id == "" {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
name := c.PostForm("name")
|
parentId := c.PostForm("parentId")
|
if parentId == "" {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
alias := c.PostForm("alias")
|
|
if aram.IsUnique(id, name) {
|
return &bhomeclient.Reply{Msg: "名称不允许重复"}
|
}
|
|
rows, err := aram.SelectbyId(id)
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: err.Error()}
|
}
|
if rows == 0 {
|
return &bhomeclient.Reply{Success: false, Msg: "没有记录"}
|
}
|
|
err = aram.Update(id, name, parentId, alias)
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: err.Error()}
|
}
|
return &bhomeclient.Reply{Success: true, Data: aram}
|
}
|
|
// @Summary 删除一个区域
|
// @Description 点击删除按钮时删除一个区域
|
// @Produce json
|
// @Tags menu
|
// @Param id formData string true "当前id"
|
// @Success 200 {string} json "{"code":200, data:"删除的区域信息",msg:"请求成功", success:true}"
|
// @Failure 200 {string} json "{"code":"500", data:"出错信息",msg:"请求失败", success:false}"
|
// @Router /data/api-v/area/del [post]
|
func (ac AreaController) AreaDelete(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var aram models.Area
|
id := c.PostForm("id")
|
logger.Debug("AreaDelete id:", id)
|
if id == "" {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
|
rows, err := aram.SelectbyId(id)
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
if rows == 0 {
|
return &bhomeclient.Reply{Success: false, Msg: "没有记录"}
|
}
|
|
//如果存在子节点或者摄像机
|
msg, ishas := aram.IsHasCnode(id)
|
if ishas {
|
return &bhomeclient.Reply{Success: false, Msg: msg}
|
}
|
|
if err := aram.Delete(id); err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: err.Error()}
|
} else {
|
return &bhomeclient.Reply{Success: true, Msg: "删除成功"}
|
}
|
}
|
