意圖
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
