Swift - 根据其子订单数组过滤对象数组？

Question

我想根据 children 数组过滤 parent 机车数组；

即：查找所有具有orders 的locomotives；仅当订单的枚举状态为“1”或“2”时；我正在使用功能代码尝试过滤数组。

我有一系列机车

var locomotives = [Locomotive]

每个机车都有一系列订单

class Locomotive {
    var order: [Order] = [Order]()
    var state: Int
}

订单是一个结构

enum OrderState: Int {
   case initial, existing, sold
}

struct Order {
 var state: OrderState = .initial
 var value: Int
}

我希望基于locomotives 过滤数组，并仅返回那些订单具有order.state 1 或2 的机车。

但是我的功能代码返回orders 的数组而不是locomotives 的数组

let filter = locos
    .filter { (locomotive: Locomotive) -> Bool in
        return (locomotive.orders.count > 0)
    }
    .flatMap { (locomotive: Locomotive) -> [Order] in
        locomotive.orders
            .filter { (o: Order) -> Bool in
                return o.state == .existing || o.state == .sold
            }
    }

这将返回一个 [Order] 数组，而不是预期的 [Locomotive] 数组

我想知道如何使上述功能代码给我一个机车列表：

• 有订单（orders.count > 0）
• 根据orders 的子数组过滤它

---

进一步编辑：

我能够做一些小技巧，在我的机车类中添加 lazy var

class Locomotive {
    lazy var existingOrders: [Order] = {
        return self.orders.filter { (o: Order) -> Bool in
            return o.state == .existingOrder
        }
    }()
    lazy var completedOrders: [Order] = {
        return self.orders.filter { (o: Order) -> Bool in
            return o.state == .completedOrder
        }
    }()
// ...
}

然后在我的过滤器中我可以检查惰性变量

 let filter = locos
            .filter { (locomotive: Locomotive) -> Bool in
                return (
                    ((locomotive.existingOrders.count > 0) ||
                    (locomotive.completedOrders.count > 0))
                )
            }
// ....

但我会回顾一下更好的方法。

Answer 1

您不应该尝试映射到 Order，这就是您得到错误类型结果的原因。相反，您可以直接在机车上进行过滤，例如

locomotives.filter { !$0.order.isEmpty && $0.order.contains(where: { order in order.state > 0 }) }