gb-cms/dao_platform.go

package main

// PlatformModel 数据库表结构
type PlatformModel struct {
	GBModel
	SIPUAOptions
}

func (g *PlatformModel) TableName() string {
	return "lkm_platform"
}

// DaoVirtualDevice 保存级联和1078设备的sipua参数项
type DaoVirtualDevice interface {
	LoadPlatforms() ([]*PlatformModel, error)

	QueryPlatform(addr string) (*PlatformModel, error)

	SavePlatform(platform *PlatformModel) error

	DeletePlatform(addr string) error

	UpdatePlatform(platform *PlatformModel) error

	BindChannels(addr string, channels [][2]string) ([][2]string, error)

	UnbindChannels(addr string, channels [][2]string) ([][2]string, error)

	// QueryPlatformChannel 查询级联设备的某个通道, 返回通道所属设备ID、通道.
	QueryPlatformChannel(addr string, channelId string) (string, *Channel, error)

	QueryPlatformChannels(addr string) ([]*Channel, error)
}

type daoPlatform struct {
}

func (d *daoPlatform) LoadPlatforms() ([]*PlatformModel, error) {
	var platforms []*PlatformModel
	tx := db.Find(&platforms)
	if tx.Error != nil {
		return nil, tx.Error
	}

	return platforms, nil
}

func (d *daoPlatform) QueryUAByAddr(addr string) (*PlatformModel, error) {
	var platform PlatformModel
	tx := db.Where("server_addr =?", addr).First(&platform)
	if tx.Error != nil {
		return nil, tx.Error
	}

	return &platform, nil
}

func (d *daoPlatform) SavePlatform(platform *PlatformModel) error {
	var old PlatformModel
	tx := db.Where("server_addr =?", platform.ServerAddr).First(&old)
	if tx.Error == nil {
		platform.ID = old.ID
	}
	return db.Save(platform).Error
}

func (d *daoPlatform) DeleteUAByAddr(addr string) error {
	return db.Where("server_addr =?", addr).Unscoped().Delete(&PlatformModel{}).Error
}

func (d *daoPlatform) UpdatePlatform(platform *PlatformModel) error {
	//TODO implement me
	panic("implement me")
}

func (d *daoPlatform) UpdateOnlineStatus(status OnlineStatus, addr string) error {
	return db.Model(&PlatformModel{}).Where("server_addr =?", addr).Update("status", status).Error
}

type PlatformChannelModel struct {
	GBModel
	DeviceID   string `json:"device_id"`
	Channel    string `json:"channel_id"`
	ServerAddr string `json:"server_addr"`
}

func (d *PlatformChannelModel) TableName() string {
	return "lkm_platform_channel"
}

func (d *daoPlatform) BindChannels(addr string, channels [][2]string) ([][2]string, error) {
	var res [][2]string
	for _, channel := range channels {

		var old PlatformChannelModel
		_ = db.Where("device_id =? and channel_id =? and server_addr =?", channel[0], channel[1], addr).First(&old)
		if old.ID == 0 {
			_ = db.Create(&PlatformChannelModel{
				DeviceID: channel[0],
				Channel:  channel[1],
			})
		}
		res = append(res, channel)
	}

	return res, nil
}

func (d *daoPlatform) UnbindChannels(addr string, channels [][2]string) ([][2]string, error) {
	var res [][2]string
	for _, channel := range channels {
		tx := db.Unscoped().Delete(&PlatformChannelModel{}, "device_id =? and channel_id =? and server_addr =?", channel[0], channel[1], addr)
		if tx.Error == nil {
			res = append(res, channel)
		} else {
			Sugar.Errorf("解绑级联设备通道失败. device_id: %s, channel_id: %s err: %s", channel[0], channel[1], tx.Error)
		}
	}

	return res, nil
}

func (d *daoPlatform) QueryPlatformChannel(addr string, channelId string) (string, *Channel, error) {
	var platformChannel PlatformChannelModel
	tx := db.Model(&PlatformChannelModel{}).Where("channel_id =? and server_addr =?", channelId, addr).First(&platformChannel)
	if tx.Error != nil {
		return "", nil, tx.Error
	}

	var channel Channel
	tx = db.Where("device_id =? and channel_id =?", platformChannel.DeviceID, platformChannel.Channel).First(&channel)
	if tx.Error != nil {
		return "", nil, tx.Error
	}

	return platformChannel.DeviceID, &channel, nil
}

func (d *daoPlatform) QueryPlatformChannels(addr string) ([]*Channel, error) {
	var platformChannels []*PlatformChannelModel
	tx := db.Where("server_addr =?", addr).Find(&platformChannels)
	if tx.Error != nil {
		return nil, tx.Error
	}

	var channels []*Channel
	for _, platformChannel := range platformChannels {
		var channel Channel
		tx = db.Where("device_id =? and channel_id =?", platformChannel.DeviceID, platformChannel.Channel).First(&channel)
		if tx.Error == nil {
			channels = append(channels, &channel)
		} else {
			Sugar.Errorf("查询级联设备通道失败. device_id: %s, channel_id: %s err: %s", platformChannel.DeviceID, platformChannel.Channel, tx.Error)
		}
	}

	return channels, nil
}