package service import ( "errors" "IotAdmin/core/sdk/service" "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 err = e.Orm.Model(&data). Scopes( cDto.MakeCondition(c.GetNeedSearch()), cDto.Paginate(c.GetPageSize(), c.GetPageIndex()), permission.Permission(data.TableName(), p), ). 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 } 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("无权更新该数据") } return nil } // Remove 删除设备 func (e *IotDeviceService) Remove(d *dto.IotDeviceDeleteReq, p * permission.DataPermission) error { var data models.IotDevice db := e.Orm.Model(&data). Scopes( permission.Permission(data.TableName(), p), ).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("无权删除该数据") } return nil }