意图
PickerView上显示指定时间区间(yyyy/MM)之间的时间
探索之道
- 系统自带的控件
UIDatePicker，无法只显示年、月 - gitbub上有第三方，自己实现的DatePicker，需要额外引入许多文件，计算复杂，而且想修改的话，需要读懂大量源码
- 系统自带的控件
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: - <UIPickerViewDelegate>
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
