Swift日期分箱优化咨询：如何实现多时间维度灵活分箱

我定义了如下结构体：

struct ADataPoint: Identifiable {
    var id: UUID = UUID()
    var time: Int
    var min: Double
    var max: Double
}

以及CustomObject：

struct CustomObject {
    var date: Date
    var someValue: Int
}

我有一个包含date属性的CustomObject数组，目前按小时维度分箱到字典数组中，实现代码如下：

func processedData() -> [ADataPoint] {
    var datapoints = [ADataPoint]()
    
    var binnedDates: [Int: [CustomObject]] = [0: [CustomObject]()]
    var min: Double
    var max: Double
    
    for myObject in myObjects ?? [] {
        let hour = Calendar.current.component(.hour, from: myObject.date)
        binnedDates[hour, default: [CustomObject]()].append(myObject)
    }
    
    if(myObjects?.count ?? 0 > 0) {
        for (time, o) in binnedDates  {
            (min, max) = statsForObjects(objects: o)
            datapoints.append(ADataPoint(time: time, min: min, max: max))
        }
    }
    
    return datapoints
}

这段代码能正常运行，但健壮性和灵活性不足。未来如果需要按分钟等其他时间维度分箱，会产生大量重复代码，请问有没有更优的实现方式？

核心思路是把可变的时间维度抽离成参数，同时利用Swift标准库的高阶函数简化分箱逻辑，让函数支持任意时间维度的分箱需求。

1. 关键优化点 #

• 提取时间维度参数：将硬编码的.hour改为可传入的Calendar.Component参数，支持.minute、.day等任意时间维度。
• 简化分箱逻辑：用Dictionary(grouping:by:)替代手动循环分箱，这是Swift标准库提供的安全简洁的分组API，避免手动处理字典默认值的冗余代码。
• 增强健壮性：提前判断数组为空的情况，减少不必要的循环执行。

完整优化代码 #

// 保留原有的结构体和统计函数
struct ADataPoint: Identifiable {
    var id: UUID = UUID()
    var time: Int
    var min: Double
    var max: Double
}

struct CustomObject {
    var date: Date
    var someValue: Int
}

func statsForObjects(objects: [CustomObject]) -> (min: Double, max: Double) {
    guard let firstValue = objects.first?.someValue else {
        return (0, 0)
    }
    let minVal = Double(objects.map(\.someValue).min() ?? firstValue)
    let maxVal = Double(objects.map(\.someValue).max() ?? firstValue)
    return (minVal, maxVal)
}

// 优化后的通用分箱处理函数
func processedData(by component: Calendar.Component) -> [ADataPoint] {
    guard let objects = myObjects, !objects.isEmpty else {
        return []
    }
    
    // 按指定时间维度自动分组
    let binnedDates = Dictionary(grouping: objects) { object in
        Calendar.current.component(component, from: object.date)
    }
    
    // 转换为ADataPoint数组
    return binnedDates.map { time, objects in
        let (min, max) = statsForObjects(objects: objects)
        return ADataPoint(time: time, min: min, max: max)
    }
}

使用示例 #

• 按小时分箱：processedData(by: .hour)
• 按分钟分箱：processedData(by: .minute)
• 按天分箱：processedData(by: .day)

优化优势 #

• 灵活性：新增时间维度无需修改核心逻辑，只需传入不同的Calendar.Component参数。
• 健壮性：利用标准库API避免手动处理字典的潜在错误，空数组提前返回减少无效操作。
• 可读性：代码逻辑分层清晰，分组和数据转换的意图一目了然，维护成本更低。

内容的提问来源于stack exchange，提问作者GarethPrice