在 Swift 中使用高阶函数更新数组值

Question

我在使用新的 map、flatMap、filter、reduce 和 zip 函数时遇到了困难。考虑以下： 您有两个数组 A 和 B，它们包含不同的对象。对于 A 中的每个对象，您需要在 B 中找到相应的对象（通过它们的 id 属性），并从 B 更新对象的一些属性。这可以用旧方法完成，使用两个 for 循环，如下所示：

private func update(statuses: [JobStatus], forJobs jobs: [JobBookPayload]) {
    for jobStatus in statuses {
        for job in jobs {
            if jobStatus.jobId == job.jobId {
                job.status = jobStatus.status!
                job.option = jobStatus.option!
            }
        }
    }
}

这是否可以使用新函数来完成，以使代码更“Swifty”并提高可读性？

Answer 1

我想既然您使用的是类，您实际上并不想从旧对象创建一个新对象。我相信这可以使用map 和first 方法来实现：

@discardableResult
func update(statuses: [JobStatus], forJobs jobs: [JobBookPayload]) -> [JobBookPayload] {
    return jobs.map({ (payload) -> JobBookPayload in
        if let status = statuses.first(where: { payload.jobId == $0.jobId }) {
            payload.status = status.status
            payload.option = status.option
        }
        return payload
    })
}

使用@discardableResult 将允许您使用此函数并忽略生成的数组 - 初始数组中的对象无论如何都会发生变异。

Answer 2

将for .. in .. { if conditional { ... } } 替换为for .. in .. where conditional { ... }

您可以通过在内部 for 循环中引入 where 子句来替换其主体的单个 if 语句来进行小规模重构：

private func update(statuses: [JobStatus], forJobs jobs: [JobBookPayload]) {
    for jobStatus in statuses {
        for job in jobs where jobStatus.jobId == job.jobId {
            job.status = jobStatus.status!
            job.option = jobStatus.option!
        }
    }
}

Answer 3

“新”（功能）方式的想法是从前一个对象创建一个新对象：

private func update(statuses: [JobStatus], for jobs: [JobBookPayload]) -> [JobBookPayload] {
    return jobs.reduce([]) { newJobs, job in
        return newJobs + statuses.flatMap { status in
            return status.jobId == job.jobId ?
                //Construct a new object
                JobBookPayload(status: status.status, option: status.option) : nil
        }
    }
}

假设有一个status对象对应一个job：

private func update(statuses: [JobStatus], for jobs: [JobBookPayload]) -> [JobBookPayload] {
    return jobs.reduce([]) { newJobs, job in
        return newJobs + statuses.lazy.filter { $0.jobId == job.jobId }.first.flatMap { 
            JobBookPayload(status: $0.status, option: $0.option) // Construct the object
        }
    }
}

避开reduce：

private func update(statuses: [JobStatus], for jobs: [JobBookPayload]) -> [JobBookPayload] {
    return jobs.flatMap { job in
        return statuses.flatMap { status in
            return status.jobId == job.jobId ?
                //Construct a new object
                JobBookPayload(status: status.status, option: status.option) : nil
        }.last
    }
}