实现在心跳中采集各设备信息

internal/storage/repository/device.go

@@ -34,6 +34,9 @@ type DeviceRepo interface {
 
 	// ListAll 获取所有设备列表
 	ListAll() ([]model.Device, error)
+
+	// FindRelayDevices 获取所有中继设备
+	FindRelayDevices() ([]*model.Device, error)
 }
 
 // DeviceControlRepo 设备控制仓库接口
@@ -135,6 +138,16 @@ func (r *deviceRepo) FindByType(deviceType model.DeviceType) ([]*model.Device, e
 	return devices, nil
 }
 
+// FindRelayDevices 获取所有中继设备
+func (r *deviceRepo) FindRelayDevices() ([]*model.Device, error) {
+	var devices []*model.Device
+	result := r.db.Where("type = ?", model.DeviceTypeRelay).Find(&devices)
+	if result.Error != nil {
+		return nil, result.Error
+	}
+	return devices, nil
+}
+
 // Update 更新设备信息
 func (r *deviceRepo) Update(device *model.Device) error {
 	result := r.db.Save(device)