斯威夫特-写附件代码的捷径？

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
    }