Swift中如何按年月组织HealthKit Workout数组及选型数据模型？

嘿，我来帮你搞定这个按年月分组HealthKit Workout的问题！你提到的嵌套字典思路确实不够贴合Swift的风格，我们可以用类型安全的自定义结构体作为字典键，实现更清晰、更符合Swift习惯的分组方案。

一、定义合适的数据模型：YearMonth结构体 #

首先，我们需要一个能唯一标识"年月"的类型，避免嵌套字典带来的键混乱问题。这个结构体要实现Hashable才能作为字典的键，同时实现CustomStringConvertible方便后续展示：

import Foundation
import HealthKit

struct YearMonth: Hashable, CustomStringConvertible {
    let year: Int
    let month: Int
    
    // 自定义展示格式，比如"2018年1月"
    var description: String {
        let formatter = DateFormatter()
        formatter.dateFormat = "yyyy年MM月"
        guard let date = Calendar.current.date(from: DateComponents(year: year, month: month)) else {
            return "\(year)年\(month)月"
        }
        return formatter.string(from: date)
    }
    
    // 从HKWorkout的开始日期初始化YearMonth
    init?(from date: Date) {
        let components = Calendar.current.dateComponents([.year, .month], from: date)
        guard let year = components.year, let month = components.month else {
            return nil
        }
        self.year = year
        self.month = month
    }
    
    // 手动初始化（用于直接指定年月）
    init(year: Int, month: Int) {
        self.year = year
        self.month = month
    }
}

二、实现Workout的自动分组 #

接下来我们可以用Swift的reduce或者Dictionary(grouping:)方法，自动完成"判断年月是否存在→不存在则创建数组→存在则追加记录"的逻辑，完全不用手动写判断：

方法1：用reduce手动分组（更直观） #

let workouts: [HKWorkout] = // 你的HealthKit Workout数组
var workoutsByYearMonth: [YearMonth: [HKWorkout]] = [:]

workoutsByYearMonth = workouts.reduce(into: [:]) { result, workout in
    // 从锻炼记录的开始日期提取年月
    guard let yearMonth = YearMonth(from: workout.startDate) else {
        return // 跳过日期无效的记录
    }
    // 用字典的default语法：如果key不存在则创建空数组，然后追加当前workout
    result[yearMonth, default: []].append(workout)
}

方法2：用Dictionary(grouping:)更简洁 #

// 先过滤掉日期无效的记录，避免强制解包
let validWorkoutPairs = workouts.compactMap { workout in
    guard let ym = YearMonth(from: workout.startDate) else { return nil }
    return (yearMonth: ym, workout: workout)
}

// 直接按YearMonth分组
let workoutsByYearMonth = Dictionary(grouping: validWorkoutPairs, by: { $0.yearMonth })
    .mapValues { $0.map { $0.workout } }

三、按年月顺序展示分组结果 #

如果需要按时间顺序（比如从旧到新）展示分组，我们可以把字典的键值对转成数组并排序：

// 按年排序，年相同则按月排序
let sortedWorkoutGroups = workoutsByYearMonth.sorted {
    if $0.key.year != $1.key.year {
        return $0.key.year < $1.key.year
    } else {
        return $0.key.month < $1.key.month
    }
}

// 遍历展示示例
for group in sortedWorkoutGroups {
    print("时段：\(group.key)，锻炼次数：\(group.value.count)")
    // 这里可以把group.value里的HKWorkout展示到UI上
}

为什么这个方案比嵌套字典更好？ #

• 类型安全：不会出现把"月"当成"年"传入字典的错误，所有年月组合都是明确的YearMonth实例
• 代码简洁：利用Swift的字典特性自动处理"创建/追加"逻辑，不用手动写if-else判断
• 扩展性强：后续如果需要给年月添加更多属性（比如季度），直接扩展YearMonth结构体即可

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