lc_tunnel/server/dao/device.go

package dao

import (
	"gorm.io/gorm"
	"server/global"
)

type Device struct {
	global.GVA_MODEL
	Sn           string        `json:"sn" gorm:"comment:设备sn"`
	Name         string        `json:"name" gorm:"comment:名称"`
	Address      string        `json:"address" gorm:"comment:设备地址"`
	TaskTime     uint          `json:"taskTime" gorm:"comment:采集时间"`
	WaitTime     uint          `json:"waitTime" gorm:"comment:延迟时间"`
	RadarId      int8          `json:"radarId" gorm:"comment:雷达id"`
	SerialId     int8          `json:"serialId" gorm:"comment:串口id"`
	Genre        int           `json:"genre" gorm:"index;comment:类型id"`
	DeviceGenre  DeviceGenre   `json:"deviceGenre" gorm:"foreignKey:Genre;"`
	TunnelId     int           `json:"tunnelId" gorm:"column:tunnel_id"`
	Tunnel       Tunnel        `json:"tunnel" gorm:"foreignKey:TunnelId"`
	DeviceRelays []DeviceRelay `json:"deviceRelays" gorm:"foreignKey:DeviceId"`
}

func (Device) TableName() string {
	return "device"
}

func QueryAllDevices() (devices []Device, err error) {
	err = global.GVA_DB.Model(Device{}).Preload("DeviceGenre").Preload("DeviceRelays").Preload("Tunnel").Find(&devices).Error
	return
}

func QueryDeviceList(sn, name string, genre, limit, offset int) (devices []Device, total int64, err error) {
	db := global.GVA_DB.Model(Device{})

	if sn != "" {
		db = db.Where("sn LIKE ?", "%"+sn+"%")
	}

	if name != "" {
		db = db.Where("name LIKE ?", "%"+name+"%")
	}

	if genre != 0 {
		db = db.Where("genre = ?", genre)
	}
	err = db.Count(&total).Error
	if err != nil {
		return
	}
	err = db.Limit(limit).Offset(offset).Preload("DeviceGenre").Preload("DeviceRelays").Preload("Tunnel").Find(&devices).Error
	return
}

func QueryDeviceByTunnelId(id int) (devices []Device, err error) {
	err = global.GVA_DB.Where("tunnel_id =?", id).Preload("DeviceGenre").Preload("DeviceRelays").Preload("Tunnel").Find(&devices).Error
	return
}

func QueryGenreByTunnel(id int) (devices []Device, err error) {
	err = global.GVA_DB.Model(&Device{}).
		Select("device.*").
		Joins("JOIN device_genre ON device.genre = device_genre.id").
		Where("device.tunnel_id = ? AND device_genre.type = ?", id, "开关设备").
		Find(&devices).Error
	return
}

func QueryDeviceByTunnelSnAndRadarId(sn string, id int) (device Device, err error) {
	err = global.GVA_DB.Model(&Device{}).
		Select("device.*").
		Joins("JOIN tunnel ON device.tunnel_id = tunnel.id").
		Where("device.radar_id = ? AND tunnel.tunnel_sn = ?", id, sn).
		First(&device).Error
	return
}

func (d Device) CreateDevice() error {
	return global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		err := tx.Create(&d).Error
		if err != nil {
			return err
		}
		if d.Genre != 6 {
			return nil
		}
		var deviceRelays []DeviceRelay
		for i := 1; i <= 4; i++ {
			relay := DeviceRelay{
				RelayID:  i,
				DeviceId: int(d.ID),
				State:    false,
			}
			deviceRelays = append(deviceRelays, relay)
		}
		return tx.Create(&deviceRelays).Error
	})
}

func (d Device) UpdateDevice() error {
	return global.GVA_DB.Where("id = ?", d.ID).Updates(&d).Error
}

func DeleteDevice(id int) error {
	return global.GVA_DB.Unscoped().Where("id = ?", id).Delete(&Device{}).Error
}