iOS-年月PickerView

意图

PickerView上显示指定时间区间(yyyy/MM)之间的时间

探索之道

1. 系统自带的控件UIDatePicker，无法只显示年、月
2. gitbub上有第三方，自己实现的DatePicker，需要额外引入许多文件，计算复杂，而且想修改的话，需要读懂大量源码
3. 系统自带的控件UIPickerView，可以显示数据，但是需要组装数据

综上所述，这里选择使用PickerView，组装数据源，以实现需求。

思路

• 目的：计算两个日期之间的年月
• 方法：
  • 从大到小，让月份递减，直到年月等于最小值即可
  • 获取所有的年月后，根据年份组合，得到每一年以及这一年中所有的月份
  • 将组装后的数据填充在PickerView上

数据加工

创建模型

• 为了处理数据方便，这里使用了模型，当然，用字典也是可以的，只要你想。

class YearModel: NSObject {
    var name: String?
    var months: [String]?
    
    override var description: String {
        return "name: \(name ?? "")   months: \(months ?? [])"
    }
}

※ 开始组装数据

var minDateStr = "2016/09"
var maxDateStr = "2017/04"
var fmtStr = "yyyy/MM"
private func setupDatasource() -> [YearModel] {
    print("=================== 数据处理开始 =================")
    var currentMonth = ""
    let fmt = NSDateFormatter()
    fmt.locale = NSLocale.currentLocale()
    fmt.dateFormat = fmtStr
    var date = fmt.dateFromString(maxDateStr ?? "") ?? NSDate()
    var yearMonthList: [String] = []
    var i = 0
    // 循环，月份递减，直到年月等于最小值即可
    // 注意：循环退出条件，避免死循环！
    repeat {
        let calender = NSCalendar(calendarIdentifier: NSCalendarIdentifierGregorian)
        let adcomps = NSDateComponents()
        adcomps.year = 0
        // 因为这里需要包含最大月份，所以需要这个判断
        adcomps.month = i == 0 ? 0 : -1
        i += 1
        date = calender?.dateByAddingComponents(adcomps, toDate: date, options: NSCalendarOptions(rawValue: UInt(0))) ?? NSDate()
        currentMonth = fmt.stringFromDate(date)
        print("\(date)   |   \(currentMonth)")
        if i > 100 {
                print("可能是死循环，手动跳出循环")
                break
         }
        yearMonthList.append(currentMonth)
    } while currentMonth != minDateStr
    // 存储处理后的数据集
    var resultList: [YearModel] = []
    // 存储年月数据对象(year-months)
    var yearModel = YearModel()
    // 存储该 year 中包含的所有 month
    var monthTmpList: [String] = []
    for obj in yearMonthList {
        let tmpList = obj.split("/")
        let year = tmpList.first!
        let month = tmpList.last!
        if yearModel.name == year {
            // 当前 year 在 resultList 中已存储，局部变量不需要初始化
        } else {
            // 当前 year 在 resultList 中未存储，是新的年份，所以需要对部分变量初始化！
            monthTmpList = []
            yearModel = YearModel()
            yearModel.name = year
        }
        monthTmpList.append(month)
        yearModel.months = monthTmpList
        print("\(yearModel)")
        var i = 0
        for model in resultList {
            // 去除月份不“完整”的数据，最后只会保留该 year 中，month 最完整的那条记录
            if let currentObj = model.months where currentObj.count > 0 {
                if model.name == yearModel.name {
                    resultList.removeAtIndex(i)
                }
            }
            i += 1
        }
        resultList.append(yearModel)
    }
    print("\(resultList)")
    print("=================== 数据处理完毕 =================")
    return resultList
}

数据处理过程，log截图如下：

年月PickerView-Log输出.png

数据展示

var index = 0
// MARK: - 
func numberOfComponentsInPickerView(pickerView: UIPickerView) -> Int {
    return 2
}
func pickerView(pickerView: UIPickerView, numberOfRowsInComponent component: Int) -> Int {
    if component == 0 {
        // 如果想“无限”滚动，就返回一个比较大的值，然后取值的时候，模上它本来的数量即可
        // e.g. return dataSource.count * 1000
        return dataSource.count
    } else {
        if dataSource.count == 0 {  return 0 }
        let yearModel = dataSource[index]
        return yearModel.months?.count ?? 0
    }
}
func pickerView(pickerView: UIPickerView, titleForRow row: Int, forComponent component: Int) -> String? {
    if component == 0 {
        let yearModel = dataSource[row]
        return yearModel.name
    } else {
        let yearModel = dataSource[index]
        return yearModel.months?[row]
    }
}
func pickerView(pickerView: UIPickerView, didSelectRow row: Int, inComponent component: Int) {
    if component == 0 {
        index = pickerView.selectedRowInComponent(0)
            pickerView.reloadComponent(1)
        }
    if dataSource.count == 0 {  return }
    let yearModel = dataSource[index]
    let monthIndex = pickerView.selectedRowInComponent(1)
    let month = yearModel.months?[monthIndex]
    let dateStr = "\(yearModel.name ?? "")/\(month ?? "")"
    txfDate.text = dateStr
    }

效果图

年月PickerView效果图.png