feature: gb28181 support add group and add channels to group

2026-05-08 12:51:04 +08:00 · 2025-04-09 09:08:09 +08:00
parent d1d6b28e0a
commit 851ba4329a
2 changed files with 218 additions and 0 deletions
@@ -0,0 +1,107 @@
+package gb28181
+
+// GroupsChannelModel 表示分组与通道的关联关系
+type GroupsChannelModel struct {
+	ID        int    `gorm:"primaryKey;autoIncrement" json:"id"`       // ID表示数据库中的唯一标识符
+	GroupID   int    `gorm:"column:group_id;index" json:"groupId"`     // GroupID表示关联的分组ID
+	ChannelID string `gorm:"column:channel_id;index" json:"channelId"` // ChannelID表示关联的通道ID
+	DeviceID  string `gorm:"column:device_id;index" json:"deviceId"`   // DeviceID表示关联的设备ID
+}
+
+// TableName 指定数据库表名
+func (g *GroupsChannelModel) TableName() string {
+	return "groups_channel"
+}
+
+// NewGroupsChannel 创建并返回一个新的GroupsChannelModel实例
+func NewGroupsChannel(groupID int, channelID string, deviceID string) *GroupsChannelModel {
+	return &GroupsChannelModel{
+		GroupID:   groupID,
+		ChannelID: channelID,
+		DeviceID:  deviceID,
+	}
+}
+
+// FindGroupChannels 通过分组ID查找关联的通道
+func FindGroupChannels(db interface{}, groupID int) ([]*GroupsChannelModel, error) {
+	// db 参数应为 *gorm.DB 类型
+	type DBAPI interface {
+		Find(dest interface{}, conds ...interface{}) error
+	}
+
+	if gdb, ok := db.(DBAPI); ok {
+		var channels []*GroupsChannelModel
+		err := gdb.Find(&channels, "group_id = ?", groupID)
+		return channels, err
+	}
+
+	return nil, nil
+}
+
+// FindChannelGroups 通过通道ID查找关联的分组
+func FindChannelGroups(db interface{}, channelID string) ([]*GroupsChannelModel, error) {
+	// db 参数应为 *gorm.DB 类型
+	type DBAPI interface {
+		Find(dest interface{}, conds ...interface{}) error
+	}
+
+	if gdb, ok := db.(DBAPI); ok {
+		var groups []*GroupsChannelModel
+		err := gdb.Find(&groups, "channel_id = ?", channelID)
+		return groups, err
+	}
+
+	return nil, nil
+}
+
+// FindDeviceGroups 通过设备ID查找关联的分组
+func FindDeviceGroups(db interface{}, deviceID string) ([]*GroupsChannelModel, error) {
+	// db 参数应为 *gorm.DB 类型
+	type DBAPI interface {
+		Find(dest interface{}, conds ...interface{}) error
+	}
+
+	if gdb, ok := db.(DBAPI); ok {
+		var groups []*GroupsChannelModel
+		err := gdb.Find(&groups, "device_id = ?", deviceID)
+		return groups, err
+	}
+
+	return nil, nil
+}
+
+// FindGroupChannelsByDevice 通过分组ID和设备ID查找关联的通道
+func FindGroupChannelsByDevice(db interface{}, groupID int, deviceID string) ([]*GroupsChannelModel, error) {
+	// db 参数应为 *gorm.DB 类型
+	type DBAPI interface {
+		Find(dest interface{}, conds ...interface{}) error
+	}
+
+	if gdb, ok := db.(DBAPI); ok {
+		var channels []*GroupsChannelModel
+		err := gdb.Find(&channels, "group_id = ? AND device_id = ?", groupID, deviceID)
+		return channels, err
+	}
+
+	return nil, nil
+}
+
+// AutoMigrate 执行自动迁移
+// 此函数应在插件初始化时调用
+func AutoMigrateGroupChannel(db interface{}) error {
+	// db 参数应为 *gorm.DB 类型
+	// 使用类型断言获取 DB 实例
+	type DBAPI interface {
+		AutoMigrate(dst ...interface{}) error
+	}
+
+	if gdb, ok := db.(DBAPI); ok {
+		// 执行表结构自动迁移
+		if err := gdb.AutoMigrate(&GroupsChannelModel{}); err != nil {
+			return err
+		}
+		return nil
+	}
+
+	return nil
+}
@@ -0,0 +1,111 @@
+package gb28181
+
+import (
+	"time"
+)
+
+// GroupsModel 表示分组结构
+type GroupsModel struct {
+	ID         int       `gorm:"primaryKey;autoIncrement" json:"id"`   // ID表示数据库中的唯一标识符
+	CreateTime time.Time `gorm:"column:create_time" json:"createTime"` // CreateTime表示记录创建时间
+	UpdateTime time.Time `gorm:"column:update_time" json:"updateTime"` // UpdateTime表示记录更新时间
+	Name       string    `gorm:"column:name" json:"name"`              // Name表示分组名称
+	PID        int       `gorm:"column:pid;default:0" json:"pid"`      // PID表示父分组ID
+	Level      int       `gorm:"column:level;default:0" json:"level"`  // Level表示分组层级
+}
+
+// TableName 指定数据库表名
+func (g *GroupsModel) TableName() string {
+	return "groups"
+}
+
+// NewGroup 创建并返回一个新的GroupsModel实例
+func NewGroup(name string, pid int, level int) *GroupsModel {
+	now := time.Now()
+	return &GroupsModel{
+		Name:       name,
+		PID:        pid,
+		Level:      level,
+		CreateTime: now,
+		UpdateTime: now,
+	}
+}
+
+// NewRootGroup 创建根分组实例
+func NewRootGroup() *GroupsModel {
+	return NewGroup("根", 0, 0)
+}
+
+// InitRootGroup 初始化根分组记录
+// 如果数据库中不存在根分组，则创建一个
+func InitRootGroup(db interface{}) error {
+	// db 参数应为 *gorm.DB 类型
+	// 使用类型断言获取 DB 实例
+	// 这里使用 interface{} 是为了避免直接依赖 GORM
+	type DBAPI interface {
+		First(dest interface{}, conds ...interface{}) error
+		Create(value interface{}) error
+	}
+
+	if gdb, ok := db.(DBAPI); ok {
+		root := &GroupsModel{}
+
+		// 检查是否存在根分组
+		err := gdb.First(root, "pid = ? AND level = ?", 0, 0)
+		if err != nil {
+			// 如果不存在，则创建一个根分组
+			rootGroup := NewRootGroup()
+			return gdb.Create(rootGroup)
+		}
+
+		return nil
+	}
+
+	return nil
+}
+
+// AutoMigrateAll 执行分组及分组-通道关联的自动迁移，并初始化根组织
+// 此函数应在插件初始化时调用，一次完成所有相关表的迁移
+func AutoMigrateAll(db interface{}) error {
+	// db 参数应为 *gorm.DB 类型
+	// 使用类型断言获取 DB 实例
+	type DBAPI interface {
+		AutoMigrate(dst ...interface{}) error
+		First(dest interface{}, conds ...interface{}) error
+		Create(value interface{}) error
+	}
+
+	if gdb, ok := db.(DBAPI); ok {
+		// 执行表结构自动迁移 - 分组表和分组-通道关联表
+		if err := gdb.AutoMigrate(&GroupsModel{}, &GroupsChannelModel{}); err != nil {
+			return err
+		}
+
+		// 检查是否存在根分组
+		root := &GroupsModel{}
+		err := gdb.First(root, "pid = ? AND level = ?", 0, 0)
+		if err != nil {
+			// 如果不存在，则创建一个根分组
+			rootGroup := NewRootGroup()
+			return gdb.Create(rootGroup)
+		}
+
+		return nil
+	}
+
+	return nil
+}
+
+// BeforeCreate GORM钩子，在创建记录前设置创建时间和更新时间
+func (g *GroupsModel) BeforeCreate() error {
+	now := time.Now()
+	g.CreateTime = now
+	g.UpdateTime = now
+	return nil
+}
+
+// BeforeUpdate GORM钩子，在更新记录前设置更新时间
+func (g *GroupsModel) BeforeUpdate() error {
+	g.UpdateTime = time.Now()
+	return nil
+}