IOT/SERVER/IotAdmin/app/iot/service/device.go

package service

import (
	"IotAdmin/common/global"
	"IotAdmin/core/logger"
	"IotAdmin/core/sdk"
	"IotAdmin/core/sdk/service"
	"IotAdmin/iot/constant"
	"errors"

	"gorm.io/gorm"

	"IotAdmin/app/iot/models"
	"IotAdmin/app/iot/service/dto"
	cDto "IotAdmin/common/dto"
	"IotAdmin/common/permission"
)

// IotDeviceService 设备服务
type IotDeviceService struct {
	service.Service
}

// GetPage 获取设备列表
func (e *IotDeviceService) GetPage(c *dto.IotDeviceGetPageReq, p *permission.DataPermission, list *[]models.IotDevice, count *int64) error {
	var err error
	var data models.IotDevice

	db := e.Orm.Model(&data).Scopes(
		cDto.MakeCondition(c.GetNeedSearch()),
		cDto.Paginate(c.GetPageSize(), c.GetPageIndex()),
		permission.Permission(data.TableName(), p),
	)
	if c.Type == constant.IotDeviceTypeGateway {
		db = db.Joins("left join iot_group as g on g.id=iot_device.group_id").Select("iot_device.id,g.name as group_name,iot_device.name,type,group_id,sn,cycle,online_status,time_online,time_offline,iot_device.created_at")
	} else {
		db = db.Select("id,name,type,sn,protocol,mode,address,bm_yz,other_config,created_at")
	}
	err = db.Find(list).Limit(-1).Offset(-1).Count(count).Error
	if err != nil {
		e.Log.Errorf("IotDeviceService GetPage error:%s \r\n", err)
		return err
	}
	return nil
}

// Get 获取设备对象
func (e *IotDeviceService) Get(d *dto.IotDeviceGetReq, p *permission.DataPermission, model *models.IotDevice) error {
	var data models.IotDevice
	err := e.Orm.Model(&data).Scopes(
		permission.Permission(data.TableName(), p),
	).First(model, d.GetId()).Error
	if err != nil && errors.Is(err, gorm.ErrRecordNotFound) {
		err = errors.New("查看对象不存在或无权查看")
		e.Log.Errorf("Service GetIotDevice error:%s \r\n", err)
		return err
	}
	if err != nil {
		e.Log.Errorf("db error:%s", err)
		return err
	}
	return nil
}

// Insert 添加设备对象
func (e *IotDeviceService) Insert(c *dto.IotDeviceInsertReq) error {
	var err error
	var data models.IotDevice
	c.Generate(&data)
	err = e.Orm.Create(&data).Error
	if err != nil {
		e.Log.Errorf("IotDeviceService Insert error:%s \r\n", err)
		return err
	}
	if data.Type == constant.IotDeviceTypeGateway {
		err = updateDeviceMap(data.Id)
	} else {
		err = updateDeviceMap(data.ParentId)
	}
	if err != nil {
		return err
	}
	return nil
}

// Update 修改设备对象
func (e *IotDeviceService) Update(c *dto.IotDeviceUpdateReq, p *permission.DataPermission) error {
	var err error
	var data = models.IotDevice{}
	e.Orm.Scopes(
		permission.Permission(data.TableName(), p),
	).First(&data, c.GetId())
	c.Generate(&data)

	db := e.Orm.Save(&data)
	if err = db.Error; err != nil {
		e.Log.Errorf("IotDeviceService Save error:%s \r\n", err)
		return err
	}
	if db.RowsAffected == 0 {
		return errors.New("无权更新该数据")
	}
	if data.Type == constant.IotDeviceTypeGateway {
		err = updateDeviceMap(data.Id)
	} else {
		err = updateDeviceMap(data.ParentId)
	}
	if err != nil {
		return err
	}
	return nil
}

// Remove 删除设备
func (e *IotDeviceService) Remove(d *dto.IotDeviceDeleteReq, p *permission.DataPermission) error {
	var data models.IotDevice
	var devices = make([]models.IotDevice, 0)
	db := e.Orm.Model(&data).Scopes(
		permission.Permission(data.TableName(), p),
	).Find(&devices, d.GetId()).Delete(&data, d.GetId())
	if err := db.Error; err != nil {
		e.Log.Errorf("Service RemoveIotDevice error:%s \r\n", err)
		return err
	}
	if db.RowsAffected == 0 {
		return errors.New("无权删除该数据")
	}
	if len(devices) > 0 {
		var err error
		for _, device := range devices {
			if device.Type == constant.IotDeviceTypeGateway {
				err = updateDeviceMap(device.Id)
			} else {
				err = updateDeviceMap(device.ParentId)
			}
			if err != nil {
				return err
			}
		}
	}
	return nil
}

func updateDeviceMap(deviceId int) error {
	q := sdk.Runtime.GetMemoryQueue("")
	mp := make(map[string]interface{})
	mp["id"] = deviceId
	message, err := sdk.Runtime.GetStreamMessage("", global.DtuDeviceChange, mp)
	if err != nil {
		logger.Errorf("构建更新设备列表 message [%s]失败: %v", global.DtuDeviceChange, err)
		return err
	}
	err = q.Append(message)
	if err != nil {
		logger.Errorf("更新设备设备列表失败: %v", err)
		return err
	}
	return nil
}