Свифт - более коротким способом написать прикрепленный код?

Question

У меня есть следующий код:

class Device{
    var customerDeviceId:Int!
    var attribute:DeviceAttribute!
}

class DeviceAttribute{
    var customerDeviceId:Int!
}

class MainClass{
   var devices:[Device]!

   private func handleDeviceAttributes(_ attributes:[DeviceAttribute])  {
    for attribute in attributes {
        for device in devices {
            if device.customerDeviceId == attribute.customerDeviceId {
                device.attribute = attribute
            }
         }
      }
    }
 }

есть ли более короткий способ написать это? может быть, даже пропустить вложенные циклы?

Answer 1

Вы можете использовать функциональный подход.

customerDevices.forEach { device in
    device.deviceAttribute = attributes.last(where: { $0.customerDeviceId == device.customerDeviceId })
}

Answer 2

Как-то так должно работать

customerDevices.forEach { (customerDevice) in
        customerDevice.deviceAttribute = attributes.compactMap({ (attribute) -> DeviceAttribute? in
            return attribute.customerDeviceId == customerDevice.customerDeviceId ? attribute : nil
        }).first
    }