如何改进这个“DateFromNextWeekDay:FromDate”方法代码？

Question

很高兴听到有关如何改进/缩短此方法的建议。总之需要：

• 查找下一个日期，它的星期几（例如星期三）与传递给方法的日期相匹配。
• 例如，给定日期（包括该给定日期）的下一个 WED

代码如下：

- (NSDate*)DateFromNextWeekDay:(NSInteger)weekDay FromDate:(NSDate*)fromDate {
    // Returns the next week day, as specified by "weekDay", from the specified "fromDate" 
    NSDate *fromDateMidday = [[NSDate date] dateBySettingHour:12 andMinute:0];
    NSDate *dateCounter  = [[fromDateMidday copy] dateByAddingTimeInterval:-86400];     // Take 1 day away, which will get incremented in the loop
    NSCalendar *gregorian = [[NSCalendar alloc] initWithCalendarIdentifier:NSGregorianCalendar];
    NSInteger day;
    do{
        dateCounter  = [dateCounter dateByAddingTimeInterval:86400];
        unsigned units = NSWeekdayCalendarUnit;
        NSDateComponents *components = [gregorian components:units fromDate:dateCounter];
        day = [components weekday];
    } while(day != weekDay);
    [gregorian release];
    return dateCounter;
}

谢谢

Answer 1

您可以只找到传入日期的星期几，从目标星期几中减去该日期，最后将该结果添加到传入的日期。无需循环访问日期。所以应该是：

daysToAdd = (targetDayOfWeek - currentDayDayOfWeek) % 7

修改减法的原因是为了处理目标日期小于当前日期的情况（例如，它是星期六，您正在寻找星期二）。