Swift中如何从CoreData按周/月维度获取日期数据？

核心思路 #

要按周或月筛选支出数据，核心是先计算目标周期的起始、结束日期区间，再通过CoreData的NSPredicate过滤该区间内的记录，还可按需对结果分组汇总。

步骤1：编写日期区间计算工具函数 #

先实现一个工具函数，根据周期类型（周/月）返回对应时间范围：

import Foundation

enum PeriodType {
    case week
    case month
}

func getPeriodRange(for type: PeriodType, referenceDate: Date = Date()) -> (start: Date, end: Date) {
    let calendar = Calendar.current
    var startDate: Date
    var endDate: Date
    
    switch type {
    case .week:
        // 取本周起始日期（默认周一，需周日起始可调整calendar.firstWeekday）
        let weekComponents = calendar.dateComponents([.yearForWeekOfYear, .weekOfYear], from: referenceDate)
        startDate = calendar.date(from: weekComponents)!
        // 本周结束日期（周六23:59:59）
        endDate = calendar.date(byAdding: .day, value: 6, to: startDate)!
        endDate = calendar.date(bySettingHour: 23, minute: 59, second: 59, of: endDate)!
        
    case .month:
        // 取当月第一天
        let monthComponents = calendar.dateComponents([.year, .month], from: referenceDate)
        startDate = calendar.date(from: monthComponents)!
        // 取下月第一天再减1秒，得到当月最后一天23:59:59
        let nextMonthStart = calendar.date(byAdding: .month, value: 1, to: startDate)!
        endDate = calendar.date(byAdding: .second, value: -1, to: nextMonthStart)!
    }
    
    return (startDate, endDate)
}

步骤2：修改查询代码，添加周期过滤 #

基于你现有的全量查询逻辑，加入NSPredicate实现周期筛选：

// 查询指定周期的支出数据
func fetchExpenses(for period: PeriodType) {
    dateArray.removeAll()
    toplamArray.removeAll()
    
    let appDelegate = UIApplication.shared.delegate as! AppDelegate
    let context = appDelegate.persistentContainer.viewContext
    
    let fetchRequest = NSFetchRequest<NSManagedObject>(entityName: "Pendings")
    fetchRequest.returnsObjectsAsFaults = false
    
    // 获取目标周期的时间范围
    let periodRange = getPeriodRange(for: period)
    // 构建日期过滤条件
    let datePredicate = NSPredicate(format: "selectDate >= %@ AND selectDate <= %@", periodRange.start as CVarArg, periodRange.end as CVarArg)
    fetchRequest.predicate = datePredicate
    
    // 可选：按日期升序排序
    let sortDescriptor = NSSortDescriptor(key: "selectDate", ascending: true)
    fetchRequest.sortDescriptors = [sortDescriptor]
    
    do {
        let results = try context.fetch(fetchRequest)
        for result in results {
            if let totalStr = result.value(forKey: "totalPend") as? String {
                toplamArray.append(totalStr)
            }
            if let date = result.value(forKey: "selectDate") as? Date {
                dateArray.append(date)
            }
        }
        tablevİEW.reloadData()
    } catch {
        print("查询失败：\(error.localizedDescription)")
    }
}

// 使用示例
fetchExpenses(for: .week) // 查询本周数据
fetchExpenses(for: .month) // 查询本月数据

额外优化建议 #

1. 存储数值而非字符串：当前totalPend存为字符串，后续计算需频繁转换，建议在CoreData模型中将其改为Integer 64类型，插入和查询时直接用数值：

// 插入数据时修改
newPendings.setValue(toplam, forKey: "totalPend") // toplam为Int类型
// 查询时直接获取
if let total = result.value(forKey: "totalPend") as? Int {
    toplamArray.append(String(total)) // 转字符串用于显示，或直接存Int数组
}

2. 按周期分组汇总：如果需要直接得到每周/每月的总支出，可查询后手动分组：

// 按月汇总总支出示例
func calculateMonthlyTotal() -> [String: Int] {
    let appDelegate = UIApplication.shared.delegate as! AppDelegate
    let context = appDelegate.persistentContainer.viewContext
    
    let fetchRequest = NSFetchRequest<NSManagedObject>(entityName: "Pendings")
    var monthlyTotals = [String: Int]()
    let dateFormatter = DateFormatter()
    dateFormatter.dateFormat = "yyyy-MM"
    
    do {
        let results = try context.fetch(fetchRequest)
        for result in results {
            guard let total = result.value(forKey: "totalPend") as? Int,
                  let date = result.value(forKey: "selectDate") as? Date else { continue }
            let monthKey = dateFormatter.string(from: date)
            monthlyTotals[monthKey, default: 0] += total
        }
    } catch {
        print("汇总失败：\(error.localizedDescription)")
    }
    return monthlyTotals
}

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