confrontation-training/service/device/device.go

package device

import (
	"confrontation-training/global"
	deviceModel "confrontation-training/models/gateway"
	"gorm.io/gorm"
)

type DeviceService struct {
}

func (d *DeviceService) FindDeviceByMac(mac string) (deviceModel.DeviceInfo, int64) {
	info := deviceModel.DeviceInfo{}
	count := global.Db.Where(" mac = ?", mac).First(&info).RowsAffected
	return info, count
}
func (d *DeviceService) FindDeviceByType(deviceType string) ([]deviceModel.DeviceInfo, int64) {
	var deviceList = make([]deviceModel.DeviceInfo, 0)
	count := global.Db.Where("type = ?", deviceType).Find(&deviceList).RowsAffected
	return deviceList, count
}
func (d *DeviceService) CreateDevice(param deviceModel.DeviceAddParam) *gorm.DB {
	addParam := deviceModel.DeviceAddParam{
		Type: param.Type,
		Mac:  param.Mac,
	}
	return global.Db.Create(addParam)
}