package models
|
|
import (
|
"encoding/json"
|
"fmt"
|
"strconv"
|
"strings"
|
"vamicro/camera-common/pub"
|
"vamicro/config"
|
|
"basic.com/pubsub/protomsg.git"
|
"basic.com/valib/logger.git"
|
)
|
|
const (
|
Default_Layer = -9999 //摄像机添加进来的默认楼层
|
TYPE_LOCAL_CAMERA = 0 //本地摄像机
|
TYPE_GB28181_CAMERA = 1 //国标摄像机
|
|
TYPE_RUNTYPE_VIDEO = -1 //单纯的监控，不做分析
|
TYPE_RUNTYPE_POLL = 0 //轮询做任务
|
TYPE_RUNTYPE_REALTIME = 1 //实时做任务
|
|
Camera_Status_NoRule = 0 //未配规则
|
Camera_Status_Wait = 1 //等待处理
|
Camera_Status_Doing = 2 //处理中
|
Camera_Status_Other = -1 //其他情况
|
Camera_Status_Err = -2 //故障
|
)
|
|
type Camera struct {
|
VideoId int64 `gorm:"-;primaryKey;autoIncrement"` //摄像机id
|
Id string `gorm:"column:id;type:varchar(100);unique;" json:"id"`
|
Name string `gorm:"column:name" json:"name" bind:"required"`
|
Alias string `gorm:"column:alias" json:"alias"` //摄像机的别名
|
SnapshotUrl string `gorm:"column:snapshot_url" json:"snapshot_url"` //快照地址
|
Type int `gorm:"column:type" json:"type" `
|
Addr string `gorm:"column:addr" json:"addr"`
|
Longitude float32 `gorm:"column:longitude" json:"longitude"`
|
Latitude float32 `gorm:"column:latitude" json:"latitude"`
|
Floor int `gorm:"column:floor" json:"floor"` //楼层
|
Rtsp string `gorm:"column:rtsp" json:"rtsp"`
|
Ip string `gorm:"column:ip" json:"ip"`
|
Port int `gorm:"column:port" json:"port"`
|
Username string `gorm:"column:username" json:"username"`
|
Password string `gorm:"column:password" json:"password"`
|
Brand string `gorm:"column:brand" json:"brand"`
|
Reserved string `gorm:"column:reserved" json:"reserved"`
|
IsRunning bool `gorm:"column:is_running" json:"is_running"` //是否正在解码
|
RunEnable bool `gorm:"column:run_enable" json:"run_enable"` //控制实时处理或轮询处理的开关
|
RunType int `gorm:"column:run_type" json:"run_type"` //处理类型：0：轮询，1：实时,-1:无任务，不做分析或者分析任务被关了
|
RunServerId string `gorm:"column:run_server_id" json:"run_server_id"` //当前正在处理的分析服务器id
|
|
ResolutionWidth int `gorm:"column:resolution_width;default:0" json:"resolution_width"` //分辨率宽
|
ResolutionHeight int `gorm:"column:resolution_height;default:0" json:"resolution_height"` //分辨率高
|
|
VoiceEnable bool `gorm:"column:voiceEnable;default:0" json:"voiceEnable"`
|
VoiceId string `gorm:"column:voiceId" json:"voiceId"`
|
CoordTransform string `gorm:"column:coordTransform" json:"coordTransform"` //坐标转换,长春追踪reid用
|
|
}
|
|
type CameraDto struct {
|
VideoId int64 `gorm:"primaryKey;autoIncrement"` //摄像机id
|
Id string `gorm:"column:id;type:varchar(100);unique;" json:"id"`
|
Name string `gorm:"column:name" json:"name" bind:"required"`
|
Alias string `gorm:"column:alias" json:"alias"` //摄像机的别名
|
SnapshotUrl string `gorm:"column:snapshot_url" json:"snapshot_url"` //快照地址
|
Type int `gorm:"column:type" json:"type" `
|
Addr string `gorm:"column:addr" json:"addr"`
|
Longitude float32 `gorm:"column:longitude" json:"longitude"`
|
Latitude float32 `gorm:"column:latitude" json:"latitude"`
|
Floor int `gorm:"column:floor" json:"floor"` //楼层
|
Rtsp string `gorm:"column:rtsp" json:"rtsp"`
|
Ip string `gorm:"column:ip" json:"ip"`
|
Port int `gorm:"column:port" json:"port"`
|
Username string `gorm:"column:username" json:"username"`
|
Password string `gorm:"column:password" json:"password"`
|
Brand string `gorm:"column:brand" json:"brand"`
|
Reserved string `gorm:"column:reserved" json:"reserved"`
|
IsRunning bool `gorm:"column:is_running" json:"is_running"` //是否正在解码
|
RunEnable bool `gorm:"column:run_enable" json:"run_enable"` //控制实时处理或轮询处理的开关
|
RunType int `gorm:"column:run_type" json:"run_type"` //处理类型：0：轮询，1：实时,-1:无任务，不做分析或者分析任务被关了
|
RunServerId string `gorm:"column:run_server_id" json:"run_server_id"` //当前正在处理的分析服务器id
|
|
ResolutionWidth int `gorm:"column:resolution_width;default:0" json:"resolution_width"` //分辨率宽
|
ResolutionHeight int `gorm:"column:resolution_height;default:0" json:"resolution_height"` //分辨率高
|
|
VoiceEnable bool `gorm:"column:voiceEnable;default:0" json:"voiceEnable"`
|
VoiceId string `gorm:"column:voiceId" json:"voiceId"`
|
CoordTransform string `gorm:"column:coordTransform" json:"coordTransform"` //坐标转换,长春追踪reid用
|
|
}
|
|
func (Camera) TableName() string {
|
return "cameras"
|
}
|
|
type CameraTreeNode struct {
|
Id string `json:"id"`
|
Areaid string `json:"areaid"`
|
Name string `json:"name"`
|
CameraType int `json:"cameratype"` //是本地还是国标
|
Rtsp string `json:"rtsp"`
|
}
|
|
type Resolution struct {
|
Width int `json:"width"`
|
Height int `json:"height"`
|
}
|
|
func (camera *Camera) GetIdNameMap() map[string]string {
|
cameraEMap := make(map[string]string, 0)
|
allCams, err := camera.FindAll()
|
if err == nil && allCams != nil && len(allCams) > 0 {
|
for _, model := range allCams {
|
cameraEMap[model.Id] = model.Name
|
}
|
}
|
|
return cameraEMap
|
}
|
|
// cameraType (0:查全部,1:分析摄像机,2:监控摄像机)
|
// cameraName
|
func (camera *Camera) Find(cameraType int, cameraName string, camType int, isPlatform int, linkStr string) (camMenus []CameraTreeNode, err error) {
|
logger.Debug("camType:", camType)
|
cType := strconv.Itoa(camType)
|
var sql = "select c.id,ca.areaid as areaid,case ifnull(c.alias,'') when '' then c.name else c.alias end as name,c.type as cameratype,c.rtsp from camera_area ca join cameras c on ca.cameraid=c.id where c.type=" + cType + ""
|
|
if cameraName != "" {
|
sql += " and c.name like '%" + cameraName + "%'"
|
}
|
if cameraType == 1 { //查分析摄像机
|
sql += " and (c.run_type=" + strconv.Itoa(TYPE_RUNTYPE_POLL) + " or c.run_type=" + strconv.Itoa(TYPE_RUNTYPE_REALTIME) + ")"
|
} else if cameraType == 2 { //查监控摄像机
|
sql += " and c.run_type=" + strconv.Itoa(TYPE_RUNTYPE_VIDEO) + ""
|
} else if cameraType == 3 { //查联动摄像机
|
//sql += " and c.id in (select crga.camera_id from camera_rule_group_arg crga join camera_task_link ctl on crga.group_id=ctl.link_task_id)"
|
sql += " and c.id in (" + linkStr + ")"
|
}
|
if camType == TYPE_LOCAL_CAMERA { //国标的始终显示整棵树，本地树本机查本机，平台查所有
|
if isPlatform == 0 { //查本机
|
sql += " and c.run_server_id='" + config.Server.AnalyServerId + "'"
|
}
|
}
|
|
sql += " order by c.id asc"
|
|
if err := db.Raw(sql).Scan(&camMenus).Error; err != nil {
|
return nil, err
|
}
|
return camMenus, nil
|
}
|
|
func (camera *Camera) FindAllByServer(serverId string, cameraName string, cameraType int, cameraId string, areaId string, linkWhere string) (list []Camera, err error) {
|
var sql = "select * from cameras where 1=1 "
|
if serverId != "" {
|
sql += " and (run_server_id='" + serverId + "' or (run_server_id ='' or run_server_id is NULL))"
|
}
|
if cameraName != "" {
|
sql += " and name like '%" + cameraName + "%'"
|
}
|
if cameraId != "" {
|
sql += " and id='" + cameraId + "'"
|
}
|
if areaId != "" {
|
sql += " and id in (select cameraId from camera_area where areaId='" + areaId + "')"
|
}
|
if cameraType == 1 { //查分析摄像机
|
sql += " and (run_type=" + strconv.Itoa(TYPE_RUNTYPE_POLL) + " or run_type=" + strconv.Itoa(TYPE_RUNTYPE_REALTIME) + ")"
|
} else if cameraType == 2 { //查监控摄像机
|
sql += " and run_type=" + strconv.Itoa(TYPE_RUNTYPE_VIDEO) + ""
|
} else if cameraType == 3 { //查联动摄像机
|
//规则配置进程独立后，此条件需要到规则进程中查询
|
sql += " and id in (" + linkWhere + ")"
|
}
|
if err := db.Raw(sql).Scan(&list).Error; err != nil {
|
return nil, err
|
}
|
return list, nil
|
}
|
|
func (camera *Camera) GetCamerasByRunType(runType int, cameraName string) (list []Camera, err error) {
|
analyServerId := config.Server.AnalyServerId //当前分析服务器的id(analyServerId,在配置文件中)
|
sql := "select * from cameras where run_type=? and run_server_id=? "
|
if cameraName != "" {
|
sql = sql + " and name like ?"
|
}
|
|
if err := db.Raw(sql, runType, analyServerId, "%"+cameraName+"%").Scan(&list).Error; err != nil {
|
return nil, err
|
}
|
return list, nil
|
}
|
|
func (camera *Camera) FindCamerasByType(typ int, runServerId string) (list []Camera, err error) {
|
sql := "select * from cameras where type=? "
|
if runServerId != "" {
|
sql += " and run_server_id='" + runServerId + "'"
|
}
|
if err := db.Raw(sql, typ).Scan(&list).Error; err != nil {
|
return nil, err
|
}
|
return list, nil
|
}
|
|
func (camera *Camera) Insert() (err error) {
|
tx := db.Table("cameras").Begin()
|
|
if tx.Error != nil {
|
return err
|
}
|
|
if err := tx.Create(&camera).Error; err != nil {
|
tx.Rollback()
|
return err
|
}
|
//新增摄像机成功,发布消息
|
|
err = tx.Commit().Error
|
return err
|
}
|
|
func (camera *Camera) SelectById(cameraId string) (rows int64, err error) {
|
dbselect := db.Table(camera.TableName()).Where("id = ?", cameraId).First(&camera)
|
if dbselect.Error != nil || dbselect.RowsAffected == 0 {
|
return 0, err
|
}
|
return dbselect.RowsAffected, nil
|
}
|
|
func (camera *Camera) FindByRtsp(rtsp string) (cams []Camera, err error) {
|
dbselect := db.Table(camera.TableName()).Where("rtsp = ?", rtsp).Find(&cams)
|
if dbselect.Error != nil {
|
return nil, dbselect.Error
|
}
|
return cams, nil
|
}
|
|
func (camera *Camera) FindAll() (cams []Camera, err error) {
|
dberr := db.Table("cameras").Find(&cams)
|
if dberr.Error != nil {
|
return nil, err
|
}
|
return cams, nil
|
}
|
|
func (camera *Camera) FindAllMap() map[string]Camera {
|
m := make(map[string]Camera, 0)
|
cams, err := camera.FindAll()
|
if err == nil && cams != nil {
|
for _, model := range cams {
|
m[model.Id] = model
|
}
|
}
|
return m
|
}
|
|
func (camera *Camera) Update() (err error) {
|
logger.Debug("camera:", camera)
|
longitude := fmt.Sprintf("%3.4f", camera.Longitude)
|
latitude := fmt.Sprintf("%3.4f", camera.Latitude)
|
voiceEnable := 0
|
if camera.VoiceEnable {
|
voiceEnable = 1
|
}
|
sql := "update cameras set name='" + camera.Name + "',alias='" + camera.Alias + "',type=" + strconv.Itoa(camera.Type) + ",addr='" + camera.Addr + "',longitude=" + longitude + ",latitude=" + latitude + ",rtsp='" + camera.Rtsp + "',ip='" + camera.Ip + "',port=" + strconv.Itoa(camera.Port) + ",username='" + camera.Username + "',password='" + camera.Password + "',brand='" + camera.Brand + "',reserved='" + camera.Reserved + "',run_server_id='" + camera.RunServerId + "',resolution_width=" + strconv.Itoa(camera.ResolutionWidth) + ",resolution_height=" + strconv.Itoa(camera.ResolutionHeight) + ",voiceEnable=" + strconv.Itoa(voiceEnable) + ",voiceId='" + camera.VoiceId + "' where id='" + camera.Id + "'"
|
|
if err := db.Exec(sql).Error; err != nil {
|
return err
|
}
|
camera.AddDbChangeMsg(camera.Id, protomsg.DbAction_Update)
|
return nil
|
}
|
|
func (camera *Camera) UpdateRunEnable(cameraId string, runEnable bool) bool {
|
result := db.Exec("update cameras set run_enable=? where id=?", runEnable, cameraId)
|
if result.Error != nil {
|
return false
|
}
|
return result.RowsAffected > 0
|
}
|
|
func (camera *Camera) ChangeRunType(cameraId string, runType int) bool {
|
result := db.Exec("update cameras set run_type=? where id=?", runType, cameraId)
|
if result.Error != nil {
|
return false
|
}
|
return result.RowsAffected > 0
|
}
|
|
func (camera *Camera) UpdateIsRunningState(cameraId string, isRunning bool) bool {
|
isRunningStr := "0"
|
if isRunning {
|
isRunningStr = "1"
|
}
|
result := db.Exec("update cameras set is_running=" + isRunningStr + " where id='" + cameraId + "'")
|
if result.Error != nil {
|
return false
|
}
|
return result.RowsAffected > 0
|
}
|
|
func (camera *Camera) UpdateIsRunningAll(camIds []string) bool {
|
analyServerId := config.Server.AnalyServerId
|
sql := "update cameras set is_running=0 where run_server_id='" + analyServerId + "'"
|
uIds := ""
|
if camIds != nil {
|
for _, id := range camIds {
|
uIds = uIds + "'" + id + "',"
|
}
|
}
|
uIds = strings.Trim(uIds, ",")
|
if uIds != "" {
|
sql += " and id not in (" + uIds + ");update cameras set is_running=1 where run_server_id='" + analyServerId + "' and id in (" + uIds + ");"
|
}
|
logger.Debug("UpdateIsRunningAll sql:", sql)
|
result := db.Exec(sql)
|
if result.Error != nil {
|
return false
|
}
|
return true
|
}
|
|
func (camera *Camera) UpdateSnapshot(cameraId string, snapshot string) bool {
|
result := db.Exec("update cameras set snapshot_url=? where id=?", snapshot, cameraId)
|
if result.Error != nil {
|
return false
|
}
|
return result.RowsAffected > 0
|
}
|
|
func (camera *Camera) Delete(cid string) (int64, error) {
|
var err error
|
tx := GetDB().Begin()
|
defer func() {
|
if err != nil && tx != nil {
|
tx.Rollback()
|
}
|
}()
|
dbdel := tx.Exec("delete from cameras where id=?", cid)
|
err = dbdel.Error
|
if err != nil || dbdel.RowsAffected == 0 {
|
return 0, err
|
}
|
//if err = tx.Exec("delete from camera_polygon where camera_id=?",cid).Error;err !=nil{
|
// return 0,err
|
//}
|
if err = tx.Exec("delete from camera_area where cameraId=?", cid).Error; err != nil {
|
return 0, err
|
}
|
if err = tx.Exec("delete from camera_sensor where camera_id=?", cid).Error; err != nil {
|
return 0, err
|
}
|
tx.Commit()
|
//发布数据更改消息
|
camera.AddDbChangeMsg(cid, protomsg.DbAction_Delete)
|
return dbdel.RowsAffected, nil
|
}
|
|
func (camera *Camera) FindBy(isOnMap bool, floor int, cameraName string) (camArr []Camera, err error) {
|
if isOnMap {
|
sql := "select * from cameras where floor!=" + strconv.Itoa(Default_Layer) + ""
|
if floor != Default_Layer {
|
sql = sql + " and floor =" + strconv.Itoa(Default_Layer) + ""
|
}
|
err = db.Raw(sql).Scan(&camArr).Error
|
} else {
|
err = db.Raw("select * from cameras where (floor=" + strconv.Itoa(Default_Layer) + " or longitude<=0 or latitude<=0)").Scan(&camArr).Error
|
}
|
if err != nil {
|
return nil, err
|
} else {
|
return camArr, nil
|
}
|
}
|
|
func (camera *Camera) UpdatePos(id string, floor int, longitude float32, latitude float32) bool {
|
|
result := db.Exec("update cameras set floor=?,longitude=?,latitude=? where id=?", floor, longitude, latitude, id)
|
if result.Error != nil {
|
return false
|
}
|
return result.RowsAffected > 0
|
}
|
|
func (camera *Camera) UpdateCoordTrans(id string, coords string) bool {
|
result := db.Exec("update cameras set coordTransform=? where id=?", coords, id)
|
if result.Error != nil {
|
return false
|
}
|
if result.RowsAffected > 0 {
|
camera.AddDbChangeMsg(id, protomsg.DbAction_Update)
|
return true
|
}
|
return false
|
}
|
|
func (camera *Camera) AddDbChangeMsg(chCamId string, action protomsg.DbAction) {
|
bytes, _ := json.Marshal(camera)
|
dbMsg := protomsg.DbChangeMessage{
|
Table: protomsg.TableChanged_T_Camera,
|
Id: chCamId,
|
Action: action,
|
Info: string(bytes),
|
}
|
pub.AddDbMessage(&dbMsg)
|
}
|
|
type VideoLink struct {
|
ID int `gorm:"column:id" json:"id"`
|
VideoId string `gorm:"column:video_id" json:"videoId"`
|
TaskId int `gorm:"column:task_id" json:"taskId"`
|
}
|
|
func GetTasks() map[string][]interface{} {
|
|
var lists []VideoLink
|
sqlStr := `select id,video_id,task_id from mal_task_video_link`
|
if err := db.Raw(sqlStr).Scan(&lists).Error; err != nil {
|
fmt.Println(err.Error())
|
return nil
|
}
|
|
checkMap := make(map[string][]interface{})
|
for _, d2 := range lists {
|
if d2.VideoId != "" {
|
checkMap[d2.VideoId] = append(checkMap[d2.VideoId], d2)
|
}
|
}
|
return checkMap
|
}
|
