173 lines
7.2 KiB
Go
173 lines
7.2 KiB
Go
package host
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"git.lnton.com/lnton/pkg/orm"
|
|
"github.com/lib/pq"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// Channel 通道
|
|
type Channel struct {
|
|
ID string `gorm:"primaryKey;column:id" json:"id"` // ID
|
|
CreatedAt orm.Time `gorm:"type:timestamptz;notNull;default:CURRENT_TIMESTAMP;index;comment:创建时间" json:"created_at"`
|
|
UpdatedAt orm.Time `gorm:"type:timestamptz;notNull;default:CURRENT_TIMESTAMP;comment:更新时间" json:"updated_at"`
|
|
Enabled bool `gorm:"column:enabled;notNull;default:true;comment:是否启用" json:"enabled"` // 是否启用
|
|
Name string `gorm:"column:name;notNull;default:'';comment:通道名称" json:"name"` // 通道名称
|
|
DeviceID string `gorm:"column:device_id;notNull;default:'';index;comment:设备ID" json:"device_id"` // 设备 id
|
|
Protocol string `gorm:"column:protocol;notNull;default:'';comment:通道协议" json:"protocol"` // 通道协议
|
|
PTZType int `gorm:"column:ptz_type;notNull;default:0;comment:云台类型" json:"ptz_type"` // 云台类型
|
|
Remark string `gorm:"column:remark;notNull;default:'';comment:备注" json:"remark"` // 备注描述
|
|
Transport string `gorm:"column:transport;notNull;default:'TCP';comment:传输协议" json:"transport"` // TCP/UDP
|
|
IP string `gorm:"column:ip;notNull;default:'';comment:IP" json:"ip"` // ip 地址
|
|
Port int `gorm:"column:port;notNull;default:0;comment:端口号" json:"port"` // 端口号
|
|
Username string `gorm:"column:username;notNull;default:'';comment:用户名" json:"-"` // 用户名
|
|
Password string `gorm:"column:password;notNull;default:'';comment:密码" json:"-"` // 密码
|
|
BID string `gorm:"column:bid;notNull;default:'';comment:协议专属 id" json:"bid"`
|
|
PTZ bool `gorm:"column:ptz;notNull;default:FALSE;comment:是否支持 ptz" json:"ptz"` // 是否支持 ptz
|
|
Talk bool `gorm:"column:talk;notNull;default:FALSE;comment:是否支持对讲" json:"talk"` // 是否支持语音对讲
|
|
PID string `gorm:"column:pid;notNull;index;default:'';comment:父通道 ID" json:"pid"`
|
|
Groups pq.StringArray `gorm:"column:groups;type:text[];default:'{}';comment:虚拟组织" json:"-"`
|
|
Ext ChannelExt `gorm:"column:ext;type:jsonb;notNull;default:'{}';comment:扩展字段" json:"ext"`
|
|
ChildCount int `gorm:"column:child_count;notNull;default:0" json:"child_count"` // 子通道数量(不包含子孙通道)
|
|
Status bool `gorm:"column:status;notNull;default:false;comment:通道状态" json:"status"`
|
|
LastPushedAt orm.Time `gorm:"column:last_pushed_at;notNull;index;default:'1970-01-01 00:00:00';comment:最后推送时间" json:"last_pushed_at"`
|
|
// CustomGBID string `gorm:"column:custom_gb_id;notNull;default:'';comment:自定义国标ID" json:"custom_gb_id"`
|
|
DefaultGBID string `gorm:"column:default_gb_id;notNull;default:'';comment:默认国标ID" json:"default_gb_id"` // 默认国标ID
|
|
CustomName string `gorm:"column:custom_name;notNull;default:'';comment:自定义名称 " json:"custom_name"`
|
|
|
|
URL string `gorm:"-" json:"url"` //
|
|
Playing bool `gorm:"-" json:"playing"` // 是否播放中
|
|
PlayingStatus string `gorm:"-" json:"playing_status"`
|
|
IsDir bool `gorm:"-" json:"is_dir"` // 是否目录
|
|
RecordPlanEnabled bool `gorm:"-" json:"record_plan_enabled"` // 是否启用了录像计划
|
|
CascadeShareEnabled bool `gorm:"-" json:"cascade_share_enabled"` // 是否启用了级联共享
|
|
CustomID string `gorm:"-" json:"custom_id"`
|
|
IsRecording bool `gorm:"-" json:"is_recording"` // 是否配置录像计划
|
|
PlanID int `gorm:"-" json:"plan_id"` // 绑定的录像计划
|
|
DeviceName string `gorm:"-" json:"device_name"` // 设备名称;单独给前端用的
|
|
|
|
// 长沙地铁项目
|
|
ServerID string `gorm:"column:server_id;notNull;default:'';comment:平台ID" json:"server_id" `
|
|
ServerName string `json:"column:server_name;notNull;default:'';comment:平台名称" json:"server_name"`
|
|
|
|
// Latitude float32
|
|
// Longitude float32
|
|
}
|
|
|
|
func (a ChannelExt) Value() (driver.Value, error) {
|
|
return json.Marshal(a)
|
|
}
|
|
|
|
func (c *Channel) GetTransport() string {
|
|
switch strings.ToUpper(c.Transport) {
|
|
case "UDP":
|
|
return "UDP"
|
|
case "TCPA":
|
|
return "TCPA"
|
|
default:
|
|
return "TCP"
|
|
}
|
|
}
|
|
|
|
func (d *Channel) BeforeCreate(tx *gorm.DB) error {
|
|
d.CreatedAt = orm.Now()
|
|
d.UpdatedAt = orm.Now()
|
|
return nil
|
|
}
|
|
|
|
func (d *Channel) BeforeUpdate(tx *gorm.DB) error {
|
|
d.UpdatedAt = orm.Now()
|
|
return nil
|
|
}
|
|
|
|
// type sortByChannel []*device.SyncReply_Channel
|
|
|
|
// func (a sortByChannel) Len() int { return len(a) }
|
|
// func (a sortByChannel) Swap(i, j int) { a[i], a[j] = a[j], a[i] }
|
|
// func (a sortByChannel) Less(i, j int) bool { return a[i].GetBid() < a[j].GetBid() }
|
|
|
|
// ChannelExt 通道的扩展内容，主要集中在国标上
|
|
type ChannelExt struct {
|
|
Parental int `json:"parental"` // 是否有子设备(1:有;0:没有)
|
|
ParentID string `json:"parent_id"` // 父设备/区域/系统ID
|
|
}
|
|
|
|
// SetIsDir 设置是否为目录
|
|
func (i *Channel) SetIsDir() {
|
|
i.IsDir = i.ChildCount > 0 || i.Ext.Parental == 1
|
|
}
|
|
|
|
// Scan implements orm.Scaner.
|
|
func (i *ChannelExt) Scan(input interface{}) error {
|
|
return orm.JsonUnmarshal(input, i)
|
|
}
|
|
|
|
// UnmarshalBinary implements encoding.BinaryUnmarshaler.
|
|
func (c *Channel) UnmarshalBinary(data []byte) error {
|
|
return orm.JsonUnmarshal(data, c)
|
|
}
|
|
|
|
// MarshalBinary implements encoding.BinaryMarshaler.
|
|
func (c *Channel) MarshalBinary() (data []byte, err error) {
|
|
return json.Marshal(c)
|
|
}
|
|
|
|
var (
|
|
_ encoding.BinaryMarshaler = (*Channel)(nil)
|
|
_ encoding.BinaryUnmarshaler = (*Channel)(nil)
|
|
)
|
|
|
|
// GetID 根据不同的设备类型返回 id
|
|
// func (c *Channel) GetID() string {
|
|
// switch c.Protocol {
|
|
// case DeviceTypeSD, DeviceTypeGB, DeviceTypeOnvif, DeviceTypePull:
|
|
// return c.BID
|
|
// }
|
|
// return c.ID
|
|
// }
|
|
|
|
// GetBID 获取设备BID
|
|
// BID是通道的实际ID
|
|
// GB28181: 由设备提供20位纯数字
|
|
// ONVIF: 由设备提供可能带有下划线
|
|
// PULL: 由平台提供，例如: 01,02(目前只支持单通道)
|
|
func (c *Channel) GetBID() string {
|
|
// switch c.Protocol {
|
|
// case DeviceTypeSD, DeviceTypeGB, DeviceTypeOnvif, DeviceTypePull,DeviceTypePush:
|
|
// return c.BID
|
|
// }
|
|
// return c.ID
|
|
return c.BID
|
|
}
|
|
|
|
// GetStreamName 获取流 id
|
|
func (c *Channel) GetStreamName() string {
|
|
return fmt.Sprintf("%s_%s", c.DeviceID, c.GetBID())
|
|
}
|
|
|
|
// TableName 指定表名
|
|
// func (Channel) TableName() string {
|
|
// return "channels"
|
|
// }
|
|
//
|
|
// // IsGB28181 ...
|
|
// func (c Channel) IsGB28181() bool {
|
|
// return c.Protocol == DeviceTypeGB
|
|
// }
|
|
//
|
|
// // IsSaida ...
|
|
// func (c Channel) IsSaida() bool {
|
|
// return c.Protocol == DeviceTypeSD
|
|
// }
|
|
//
|
|
// // IsSupportLocalRecord 该协议是否支持本地录像
|
|
// func (c *Channel) IsSupportLocalRecord() bool {
|
|
// return c.Protocol == DeviceTypeSD
|
|
// }
|