package service
|
|
import (
|
"basic.com/gb28181api.git"
|
"basic.com/pubsub/protomsg.git"
|
"basic.com/valib/bhomeclient.git"
|
"basic.com/valib/bhomedbapi.git"
|
"basic.com/valib/logger.git"
|
"vamicro/config"
|
"vamicro/sync-service/models"
|
"encoding/json"
|
"github.com/satori/go.uuid"
|
"strings"
|
"time"
|
"vamicro/extend/util"
|
)
|
|
type Gb28181Service struct {
|
bk bhomeclient.Broker
|
}
|
|
func NewGb28181Service(broker bhomeclient.Broker) *Gb28181Service {
|
return &Gb28181Service{
|
bk: broker,
|
}
|
}
|
|
func (s Gb28181Service) Edit(gbConfig models.Gb28181Config)bool{
|
timeUnix :=time.Now().Unix()
|
fmtTimeStr := time.Unix(timeUnix,0).Format("2006-01-02 15:04:05")
|
gbConfig.UpdateTime = fmtTimeStr
|
var gbConf models.Gb28181Config
|
rows, _ := gbConf.Select()
|
if rows >0 {//已存在，编辑
|
gbConfig.Id = gbConf.Id
|
data := util.Struct2Map(gbConfig)
|
if err := models.GetDB().Table("gb28181_config").Updates(data).Error;err !=nil{
|
return false
|
}
|
} else { //新增
|
gbConfig.Id = uuid.NewV4().String()
|
if err := models.GetDB().Table("gb28181_config").Create(&gbConfig).Error;err !=nil{
|
return false
|
}
|
}
|
|
setGbIp := ""
|
var syssetApi bhomedbapi.SysSetApi
|
fb, rConf := syssetApi.GetResourceConfig()
|
if fb {
|
setGbIp = rConf.ServiceIp //如果有对外服务配置，则取设置的
|
} else {
|
setGbIp,_,_ = util.GetLocalIP(config.Server.NetworkAdapter) //没有配置，则取本机ip
|
}
|
if setGbIp != "" {
|
//接入国标的平台树
|
var gbApi gb28181api.Gb28181Api
|
gb28181api.Init(setGbIp,gbConfig.ServerPort)
|
//设置国标平台基础配置
|
gbApi.SetPlatformServerInfo(gb28181api.GbServerInfo{
|
Name: "贝思科国标服务",
|
PublicID: gbConfig.PublicId,
|
GbServerPort: gbConfig.GbServerPort,
|
RtspServrPort: 7554,
|
HTTPClientIP: "",
|
HTTPClientPort: gbConfig.ServerPort,
|
RegisterAuth: gbConfig.IsAuth,
|
GbPasswd: gbConfig.Password,
|
})
|
//先更新资源
|
gbApi.UpdateAllResource()
|
|
go s.updateGb28181Data(gbApi,gbConfig)
|
}
|
|
return true
|
}
|
|
type gbDevAndArea struct {
|
PublicId string
|
ParentId string
|
}
|
|
func (s Gb28181Service) updateGb28181Data(gbApi gb28181api.Gb28181Api,config models.Gb28181Config) bool {
|
|
time.Sleep(65 * time.Second)
|
|
var err error
|
tx := models.GetDB().Begin()
|
defer func() {
|
if err !=nil && tx !=nil{
|
tx.Rollback()
|
}
|
}()
|
|
gbCids := ""
|
uIds := ""
|
devices, flag := gbApi.GetAllDevices()
|
logger.Debug("gb28181 group devices:",devices)
|
uIdMap := make(map[string]gbDevAndArea,0)
|
if flag {
|
for _,gbu := range devices {
|
uIdMap[gbu.PublicID] = gbDevAndArea{
|
PublicId: gbu.PublicID,
|
ParentId: gbu.Parentid,
|
}
|
uIds = uIds + "'"+gbu.PublicID+"',"
|
}
|
|
for _,d := range devices {
|
gbArea :=models.Area{
|
Id:d.PublicID,
|
Name:d.Name,
|
Parentid: d.Parentid,
|
Type: models.TYPE_GB28181_TREE,
|
}
|
if _,ok :=uIdMap[d.Parentid];!ok {
|
gbArea.Parentid = "0"
|
}
|
//递归得到parentids
|
parentIds := ""
|
GetPidsByRecursion(&uIdMap, gbArea.Parentid, &parentIds)
|
gbArea.Parentids = parentIds
|
|
if d.Name == ""{
|
gbArea.Name = d.PublicID
|
}
|
var areaE models.Area
|
rows, _ := areaE.SelectbyId(gbArea.Id)
|
if rows>0{
|
gbArea.Alias = areaE.Alias
|
if err = tx.Table("area").Where("id=?",gbArea.Id).Update(&gbArea).Error;err !=nil{
|
return false
|
}
|
} else {
|
if err = tx.Table("area").Create(&gbArea).Error;err !=nil{
|
return false
|
}
|
}
|
|
groupDomains := gbApi.GetAllGroupBydevID(d.PublicID)
|
logger.Debug("gb28181 groupDomains:",groupDomains)
|
if groupDomains !=nil && len(groupDomains) >0 {
|
|
for _,gbd := range groupDomains {
|
uIdMap[gbd.PublicID] = gbDevAndArea{
|
PublicId: gbd.PublicID,
|
ParentId: gbd.ParentID,
|
}
|
uIds = uIds + "'"+gbd.PublicID+"',"
|
}
|
|
|
for _,gD :=range groupDomains {
|
|
gbArea :=models.Area{
|
Id: gD.PublicID,
|
Name: gD.Name,
|
Parentid: gD.ParentID,
|
Type: models.TYPE_GB28181_TREE,
|
}
|
if _,ok :=uIdMap[gD.ParentID];!ok {
|
gbArea.Parentid = "0"
|
}
|
//递归得到parentids
|
parentIds := ""
|
GetPidsByRecursion(&uIdMap, gbArea.Parentid, &parentIds)
|
gbArea.Parentids = parentIds
|
|
if gD.Name == ""{
|
gbArea.Name = gD.PublicID
|
}
|
var areaE models.Area
|
rows, _ := areaE.SelectbyId(gbArea.Id)
|
if rows>0{
|
gbArea.Alias = areaE.Alias
|
if err = tx.Table("area").Where("id=?",gbArea.Id).Update(&gbArea).Error;err !=nil{
|
return false
|
}
|
} else {
|
if err = tx.Table("area").Create(&gbArea).Error;err !=nil{
|
return false
|
}
|
}
|
|
groupCams, gf := gbApi.GetAllCamerasByGroupID(gD.PublicID)
|
logger.Debug("gb28181 groupCams:",groupCams)
|
if gf {
|
for _,gbCam :=range groupCams{
|
gbCids = gbCids + "'"+gbCam.PublicID+"',"
|
var cameraE models.Camera
|
rI, _ := cameraE.SelectById(gbCam.PublicID)
|
if rI>0{
|
cameraE.Name = gbCam.Name
|
cameraE.Ip = gbCam.IP
|
cameraE.Port = gbCam.Port
|
cameraE.Rtsp = gbCam.RealRtspURL
|
|
cameraE.Longitude = float32(gbCam.Longitude)
|
cameraE.Latitude = float32(gbCam.Latitude)
|
if err = tx.Table("cameras").Where("id=?",cameraE.Id).Update(&cameraE).Error;err !=nil{
|
return false
|
}
|
}else{
|
cameraE =models.Camera{
|
Id:gbCam.PublicID,
|
Name:gbCam.Name,
|
Ip:gbCam.IP,
|
Port:gbCam.Port,
|
Rtsp:gbCam.RealRtspURL,
|
|
Addr:gbCam.Name,
|
Alias: gbCam.Name,//别名默认取国标摄像机的名称
|
Longitude : float32(gbCam.Longitude),
|
Latitude : float32(gbCam.Latitude),
|
Type: models.TYPE_GB28181_CAMERA,//国标摄像机
|
Floor: models.Default_Layer,
|
RunType: models.TYPE_RUNTYPE_VIDEO,//新增无任务，默认是监控
|
}
|
if err = tx.Table("cameras").Create(&cameraE).Error;err !=nil{
|
return false
|
}
|
}
|
//挂到设备树下
|
var cameraArea []models.CameraArea
|
if err = tx.Table("camera_area").Where("cameraId=? and areaId=?",gbCam.PublicID,gbArea.Id).Find(&cameraArea).Error;err !=nil{
|
return false
|
}
|
if len(cameraArea) ==0 {
|
if err = tx.Table("camera_area").Create(&models.CameraArea{Cameraid:gbCam.PublicID,Areaid:gbArea.Id}).Error;err !=nil{
|
return false
|
}
|
}
|
}
|
}
|
}
|
}
|
}
|
}
|
|
domainUnits, b := gbApi.GetAllDomainUnit()//获取所有国标区域
|
logger.Debug("gb28181 gb units:",domainUnits)
|
if b{
|
for _,gbu := range domainUnits {
|
uIdMap[gbu.PublicID] = gbDevAndArea{
|
PublicId: gbu.PublicID,
|
ParentId: gbu.ParentID,
|
}
|
uIds = uIds + "'"+gbu.PublicID+"',"
|
}
|
|
for _,gbUnit :=range domainUnits{
|
gbArea :=models.Area{
|
Id:gbUnit.PublicID,
|
Name:gbUnit.Name,
|
Parentid: gbUnit.ParentID,
|
Type: models.TYPE_GB28181_TREE,
|
}
|
if _,ok :=uIdMap[gbUnit.ParentID];!ok {
|
gbArea.Parentid = "0"
|
}
|
//递归得到parentids
|
parentIds := ""
|
GetPidsByRecursion(&uIdMap, gbArea.Parentid, &parentIds)
|
gbArea.Parentids = parentIds
|
|
if gbUnit.Name == ""{
|
gbArea.Name = gbUnit.PublicID
|
}
|
var areaE models.Area
|
rows, _ := areaE.SelectbyId(gbArea.Id)
|
if rows>0{
|
gbArea.Alias = areaE.Alias
|
if err = tx.Table("area").Where("id=?",gbArea.Id).Update(&gbArea).Error;err !=nil{
|
return false
|
}
|
} else {
|
if err = tx.Table("area").Create(&gbArea).Error;err !=nil{
|
return false
|
}
|
}
|
|
if cameras, flag := gbApi.GetAllCamerasByDomainUnit(gbArea.Id);flag{
|
logger.Debug("gb28181 cameras by ",gbArea.Id," ",cameras)
|
for _,gbCam :=range cameras{
|
gbCids = gbCids + "'"+gbCam.PublicID+"',"
|
var cameraE models.Camera
|
rI, _ := cameraE.SelectById(gbCam.PublicID)
|
if rI>0{
|
cameraE.Name = gbCam.Name
|
cameraE.Ip = gbCam.IP
|
cameraE.Port = gbCam.Port
|
cameraE.Rtsp = gbCam.RealRtspURL
|
|
cameraE.Longitude = float32(gbCam.Longitude)
|
cameraE.Latitude = float32(gbCam.Latitude)
|
if err = tx.Table("cameras").Where("id=?",cameraE.Id).Update(&cameraE).Error;err !=nil{
|
return false
|
}
|
}else{
|
cameraE =models.Camera{
|
Id: gbCam.PublicID,
|
Name: gbCam.Name,
|
Ip: gbCam.IP,
|
Port: gbCam.Port,
|
Rtsp: gbCam.RealRtspURL,
|
|
Addr: gbCam.Name,
|
Alias: gbCam.Name,//别名默认取国标摄像机的名称
|
Longitude : float32(gbCam.Longitude),
|
Latitude : float32(gbCam.Latitude),
|
Type: models.TYPE_GB28181_CAMERA,//国标摄像机
|
Floor: models.Default_Layer,
|
RunType: models.TYPE_RUNTYPE_VIDEO,//新增无任务，默认是监控
|
}
|
if err = tx.Table("cameras").Create(&cameraE).Error;err !=nil{
|
return false
|
}
|
}
|
//挂到设备树下
|
var cameraArea []models.CameraArea
|
if err = tx.Table("camera_area").Where("cameraId=? and areaId=?",gbCam.PublicID,gbArea.Id).Find(&cameraArea).Error;err !=nil{
|
return false
|
}
|
if len(cameraArea) ==0 {
|
if err = tx.Table("camera_area").Create(&models.CameraArea{Cameraid:gbCam.PublicID,Areaid:gbArea.Id}).Error;err !=nil{
|
return false
|
}
|
}
|
}
|
}
|
}
|
|
}
|
|
uIds = strings.Trim(uIds, ",")
|
if uIds != "" {
|
var rmCamAreas []models.CameraArea
|
|
if err = tx.Raw("select * from camera_area where areaId in (select id from area where type=1) and areaId not in ("+uIds+")").Scan(&rmCamAreas).Error;err !=nil {
|
return false
|
}
|
if rmCamAreas !=nil && len(rmCamAreas) >0 {
|
for _,rmCA :=range rmCamAreas {
|
if err = tx.Exec("delete from camera_area where areaId='"+rmCA.Areaid+"'").Error;err !=nil {
|
return false
|
}
|
}
|
}
|
var rmAreas []models.Area
|
if err = tx.Raw("select * from area where type=1 and id not in ("+uIds+")").Scan(&rmAreas).Error;err !=nil {
|
return false
|
}
|
if rmAreas !=nil && len(rmAreas) >0 {
|
for _,rmArea :=range rmAreas {
|
if err = tx.Exec("delete from area where type=1 and id='"+rmArea.Id+"'").Error;err !=nil {
|
return false
|
}
|
}
|
}
|
|
} else {
|
if err = tx.Exec("delete from camera_area where areaId in (select id from area where type=1)").Error;err !=nil {
|
return false
|
}
|
if err = tx.Exec("delete from area where type=1").Error;err !=nil {
|
return false
|
}
|
}
|
|
//删除已经被删除的国标摄像机
|
gbCids = strings.Trim(gbCids, ",")
|
if gbCids != "" {
|
var rmCams []models.Camera
|
if err = tx.Raw("select * from cameras where type=1 and id not in ("+gbCids+")").Scan(&rmCams).Error;err !=nil {
|
return false
|
}
|
if rmCams !=nil && len(rmCams) >0 {
|
for _,rmCam :=range rmCams {
|
if err = tx.Exec("delete from cameras where id='"+rmCam.Id+"'").Error;err !=nil {
|
return false
|
}
|
}
|
}
|
} else {
|
if err = tx.Exec("delete from cameras where type=1").Error;err !=nil {
|
return false
|
}
|
}
|
tx.Commit()
|
s.AddDbChangeMsg(protomsg.DbAction_Update)
|
return true
|
}
|
|
func GetPidsByRecursion(uIdMap *map[string]gbDevAndArea,parentId string,pIds *string) {
|
if unit,ok :=(*uIdMap)[parentId];ok {
|
*pIds = ","+parentId +","+ *pIds
|
GetPidsByRecursion(uIdMap,unit.ParentId,pIds)
|
} else {
|
*pIds = ",0,"+*pIds
|
*pIds = strings.Replace(*pIds,",,",",",-1)
|
return
|
}
|
}
|
|
//刷新国标树和设备
|
func (s Gb28181Service) RefreshById(id string) bool {
|
var gbE models.Gb28181Config
|
rows, err := gbE.Select()
|
if err ==nil && rows >0 {
|
return s.Edit(gbE)
|
}
|
return false
|
}
|
|
//删除国标
|
func (s Gb28181Service) Delete() bool {
|
var err error
|
tx := models.GetDB().Begin()
|
defer func() {
|
if err !=nil && tx !=nil{
|
tx.Rollback()
|
}
|
}()
|
if err = tx.Exec("delete from area where type=1").Error;err !=nil {
|
return false
|
}
|
if err = tx.Exec("delete from cameras where type=1").Error;err !=nil {
|
return false
|
}
|
if err = tx.Exec("delete from camera_area where areaId in (select id from area where type=1)").Error;err !=nil {
|
return false
|
}
|
tx.Commit()
|
s.AddDbChangeMsg(protomsg.DbAction_Update)
|
return false
|
}
|
|
//摄像机表有更新
|
func (s Gb28181Service) AddDbChangeMsg(action protomsg.DbAction){
|
dbMsg := protomsg.DbChangeMessage{
|
Table:protomsg.TableChanged_T_Camera,
|
Action:action,
|
}
|
pb,_ := json.Marshal(dbMsg)
|
s.bk.Publish(ProcName,pb)
|
}
|
