172 lines
8.0 KiB
Go
172 lines
8.0 KiB
Go
|
package host
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"database/sql/driver"
|
|||
|
|
"encoding"
|
|||
|
|
"encoding/json"
|
|||
|
|
|
|||
|
|
"git.lnton.com/lnton/pkg/orm"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var (
|
|||
|
|
_ encoding.BinaryMarshaler = (*Device)(nil)
|
|||
|
|
_ encoding.BinaryUnmarshaler = (*Device)(nil)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Ability 设备能力
|
|||
|
|
type Ability struct {
|
|||
|
|
PTZ bool `json:"ptz"` // 云台控制
|
|||
|
|
WIFI bool `json:"wifi"` // WIFI配网
|
|||
|
|
OTA bool `json:"ota"` // OTA升级
|
|||
|
|
CloudBroadcast bool `json:"cloud_broadcast"` // 云广播
|
|||
|
|
AI bool `json:"ai"` // 智能检测
|
|||
|
|
GAT1400 bool `json:"gat_1400"` // GA/T1400
|
|||
|
|
Talk bool `json:"talk"` // P2P对讲
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Device 设备
|
|||
|
|
type Device struct {
|
|||
|
|
ID string `gorm:"primaryKey" json:"id"`
|
|||
|
|
CreatedAt orm.Time `gorm:"notNull;default:CURRENT_TIMESTAMP;index;comment:创建时间" json:"created_at"`
|
|||
|
|
UpdatedAt orm.Time `gorm:"notNull;default:CURRENT_TIMESTAMP;comment:更新时间" json:"updated_at"`
|
|||
|
|
Name string `gorm:"column:name;notNull;default:'';comment:设备名称" json:"name"` // 设备名称
|
|||
|
|
Protocol string `gorm:"column:protocol;notNull;default:'';comment:设备协议" json:"protocol"` // 设备协议 rtsp/rtmp
|
|||
|
|
IP string `gorm:"column:ip;notNull;default:'';comment:设备IP" json:"ip"` // ip 地址
|
|||
|
|
Port int `gorm:"column:port;notNull;default:0;comment:端口" json:"port"` // 端口号
|
|||
|
|
Addr string `gorm:"column:addr;notNull;default:'';comment:ip 物理地址" json:"addr"`
|
|||
|
|
Remark string `gorm:"column:remark;notNull;default:'';comment:备注" json:"remark"` // 备注描述
|
|||
|
|
Username string `gorm:"column:username;notNull;default:'';comment:账号" json:"username"` // 账号
|
|||
|
|
Password string `gorm:"column:password;notNull;default:'';comment:密码" json:"password"` // 密码
|
|||
|
|
MediaTransport string `json:"media_transport,omitempty"` // TCP/UDP 服务端配置，空表示 UDP
|
|||
|
|
GBCode string `gorm:"column:gb_code;notNull;default:'';comment:国标设备编码" json:"gb_code"` // 设备编码
|
|||
|
|
UID int `gorm:"column:uid;notNull;default:0;comment:创建者" json:"uid"` // 创建者
|
|||
|
|
Version string `gorm:"column:version;notNull;default:'';comment:版本" json:"version"` // 固件版本号
|
|||
|
|
Ability Ability `gorm:"column:ability;type:jsonb;notNull;default:'{}';comment:设备能力" json:"ability"` // 能力
|
|||
|
|
Status bool `gorm:"column:status;notNull;default:FALSE;comment:在线状态" json:"status"` // true:在线;false;离线
|
|||
|
|
Model string `gorm:"column:model;notNull;default:'';comment:设备型号" json:"model"` // 状态
|
|||
|
|
ChannelCount int `gorm:"column:channel_count;notNull;default:0;comment:通道数量" json:"channel_count"` // 通道数量
|
|||
|
|
Ext DeviceExt `gorm:"type:jsonb;notNull;default:'{}';comment:设备扩展信息" json:"ext"`
|
|||
|
|
GB GBExt `gorm:"type:jsonb;notNull;default:'{}';comment:国标扩展信息" json:"gb"`
|
|||
|
|
ONVIF ONVIFExt `gorm:"type:jsonb;notNull;default:'{}';comment:ONVIF 扩展信息" json:"onvif"`
|
|||
|
|
Network string `gorm:"column:network;notNull;default:'WAN';comment:网络类型" json:"network"` // LAN:局域网;WAN:公网
|
|||
|
|
// CustomRTPIP string `gorm:"column:custom_rtp_ip;notNull;default:'';comment:自定义 rtp ip" json:"custom_rtp_ip"` // 自定义 rtp ip
|
|||
|
|
RTPID string `gorm:"column:rtp_id;notNull;default:''" json:"rtp_id"`
|
|||
|
|
ISPlatform bool `gorm:"column:is_platform;notNull;default:FALSE;comment:设备或平台" json:"is_platform"` // 设备/平台
|
|||
|
|
URL string `gorm:"column:url;notNull;default:'';comment:拉流地址" json:"url"` // 拉流地址
|
|||
|
|
// LastUpdatedAt orm.Time `gorm:"column:last_updated_at;notNull;default:'1970-01-01 00:00:00';index;comment:最后更新时间" json:"last_updated_at"`
|
|||
|
|
LastPushedAt orm.Time `gorm:"column:last_pushed_at;notNull;index;default:'1970-01-01 00:00:00';comment:最后推送时间" json:"last_pushed_at"`
|
|||
|
|
FilterList string `gorm:"column:filter_list;notNull;default:'';comment:过滤列表" json:"filter_list"`
|
|||
|
|
|
|||
|
|
// 示例数据 1,2,3
|
|||
|
|
// 应用场景
|
|||
|
|
// - 播放的时候，提供相关线路连接，默认取第一个
|
|||
|
|
Routes string `gorm:"column:routes;notNull;default:'';comment:线路" json:"routes"`
|
|||
|
|
DefaultGBID string `gorm:"-" json:"default_gb_id"`
|
|||
|
|
PullMode int `gorm:"column:pull_mode;notNull;default:0;comment:拉流模式" json:"pull_mode"` // 拉流模式 默认(0) 拉流库(1)
|
|||
|
|
Audio string `gorm:"column:audio;notNull;default:'0';comment:音频控制" json:"audio"` // 音频控制 0 关闭 1 启用 *** 自定义音频地址
|
|||
|
|
|
|||
|
|
// 长沙地铁项目
|
|||
|
|
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"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (a DeviceExt) Value() (driver.Value, error) {
|
|||
|
|
return json.Marshal(a)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d *Device) BeforeCreate(tx *gorm.DB) error {
|
|||
|
|
d.CreatedAt = orm.Now()
|
|||
|
|
d.UpdatedAt = orm.Now()
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d *Device) BeforeUpdate(tx *gorm.DB) error {
|
|||
|
|
d.UpdatedAt = orm.Now()
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d *Device) NetworkIP(iip, eip string) string {
|
|||
|
|
if d.Network == "LAN" {
|
|||
|
|
return iip
|
|||
|
|
}
|
|||
|
|
return eip
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type ONVIFExt struct {
|
|||
|
|
AuthType string `gorm:"column:auth_type;notNull;default:'';comment:认证类型" json:"auth_type"` // 认证类型
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Scan implements orm.Scaner.
|
|||
|
|
func (o *ONVIFExt) Scan(input interface{}) error {
|
|||
|
|
return orm.JsonUnmarshal(input, o)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (o ONVIFExt) Value() (driver.Value, error) {
|
|||
|
|
return json.Marshal(o)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GBExt 国标扩展信息
|
|||
|
|
// 实际在使用过程中，发现部分字段存在支持其它协议的情况
|
|||
|
|
type GBExt struct {
|
|||
|
|
SIPTransport string `json:"sip_transport,omitempty"` // TCP/UDP 由设备主动推送
|
|||
|
|
SIPVersion string `json:"sip_version,omitempty"` // 国标 sip 版本号 2011/2016/2022
|
|||
|
|
|
|||
|
|
CatalogPeriod int `json:"catalog_period"` // 目录刷新周期
|
|||
|
|
CatelogSubscribe bool `json:"catelog_subscribe,omitempty"` // 目录订阅，仅支持 gb28181
|
|||
|
|
AlarmSubscribe bool `json:"alarm_subscribe,omitempty"` // 报警订阅，支持 gb28181/sdsdk
|
|||
|
|
PositionSubscribe bool `json:"position_subscribe,omitempty"` // 位置订阅，仅支持 gb28181
|
|||
|
|
NotifyURL string `json:"notify_url,omitempty"` // 通知地址
|
|||
|
|
|
|||
|
|
StorageID int `json:"storage_id,string,omitempty"`
|
|||
|
|
StrategyID int `json:"strategy_id,string,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Scan implements orm.Scaner.
|
|||
|
|
func (d *GBExt) Scan(input interface{}) error {
|
|||
|
|
return orm.JsonUnmarshal(input, d)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (d GBExt) Value() (driver.Value, error) {
|
|||
|
|
return json.Marshal(d)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// DeviceExt 设备信息
|
|||
|
|
type DeviceExt struct {
|
|||
|
|
WanIP string `json:"wan_ip"`
|
|||
|
|
LanIP string `json:"lan_ip"`
|
|||
|
|
Mac string `json:"mac,omitempty"`
|
|||
|
|
SDKVersion string `json:"sdk_version,omitempty"`
|
|||
|
|
UID string `json:"uid,omitempty"` // 没啥用，等同于 sn，甲方设计必须有这个参数
|
|||
|
|
Manufacturer string `json:"manufacturer,omitempty"` // 厂商
|
|||
|
|
RemoteIP string `json:"remote_ip,omitempty"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Scan implements orm.Scaner.
|
|||
|
|
func (d *DeviceExt) Scan(input interface{}) error {
|
|||
|
|
return orm.JsonUnmarshal(input, d)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// UnmarshalBinary implements encoding.BinaryUnmarshaler.
|
|||
|
|
func (d *Device) UnmarshalBinary(data []byte) error {
|
|||
|
|
return json.Unmarshal(data, d)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MarshalBinary implements encoding.BinaryMarshaler.
|
|||
|
|
func (d Device) MarshalBinary() ([]byte, error) {
|
|||
|
|
return json.Marshal(d)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// func (d *Device) GetRoutes() []Route {
|
|||
|
|
// routes := make([]Route, 0, 2)
|
|||
|
|
// routeIDs := strings.Split(d.Routes, ",")
|
|||
|
|
// for _, id := range routeIDs {
|
|||
|
|
// if id == "" {
|
|||
|
|
// continue
|
|||
|
|
// }
|
|||
|
|
// intID, _ := strconv.Atoi(id)
|
|||
|
|
// routes = append(routes, Route{ID: intID})
|
|||
|
|
// }
|
|||
|
|
// return routes
|
|||
|
|
// }
|