數(shù)據(jù)源遵守協(xié)議,將數(shù)據(jù)源傳入,即可排序
import UIKit
//MARK: 排序?qū)ο髆odel
@objc protocol SortObjectModel {
var SortKey: String { get }
}
//MARK: 排序成功處理
typealias successHandler = (_ dataArray : [[SortObjectModel]], _ sectionTitlesArray : [String]) -> Void
extension UILocalizedIndexedCollation {
//MARK: 成功后 --- 得到對應(yīng)的標(biāo)題文字?jǐn)?shù)組 以及對應(yīng)分組中數(shù)據(jù)
static func getCurrentKeysAndObjectsData(needSortArray : [SortObjectModel], finishCallback : @escaping successHandler) -> Void{
// 數(shù)據(jù)源
var dataArray = [[SortObjectModel]]()
// 每個(gè)section的標(biāo)題
var sectionTitleArray = [String]()
let indexedCollation = self.current()
var sortArray = [SortObjectModel]()
for sortObj in needSortArray {
sortArray.append(sortObj)
}
// 獲得索引數(shù), 這里是27個(gè)(26個(gè)字母和1個(gè)#)
let indexCount = indexedCollation.sectionTitles.count
// 每一個(gè)一維數(shù)組可能有多個(gè)數(shù)據(jù)要添加,所以只能先創(chuàng)建一維數(shù)組哄陶,到時(shí)直接取來用
for _ in 0..<indexCount {
let array = [SortObjectModel]()
dataArray.append(array)
}
// 將數(shù)據(jù)進(jìn)行分類,存儲到對應(yīng)數(shù)組中
for sortObj in sortArray {
// 根據(jù) SortObjectModel 的 objValue 判斷應(yīng)該放入哪個(gè)數(shù)組里
// 返回值就是在 indexedCollation.sectionTitles 里對應(yīng)的下標(biāo)
let sectionNumber = indexedCollation.section(for: sortObj, collationStringSelector: #selector(getter: SortObjectModel.SortKey))
// 添加到對應(yīng)一維數(shù)組中
dataArray[sectionNumber].append(sortObj)
}
// 對每個(gè)已經(jīng)分類的一維數(shù)組里的數(shù)據(jù)進(jìn)行排序,如果僅僅只是分類可以不用這步
for i in 0..<indexCount {
// 排序結(jié)果數(shù)組
let sortedPersonArray = indexedCollation.sortedArray(from: dataArray[i], collationStringSelector: #selector(getter: SortObjectModel.SortKey))
// 替換原來數(shù)組
dataArray[i] = sortedPersonArray as! [SortObjectModel]
}
// 用來保存沒有數(shù)據(jù)的一維數(shù)組的下標(biāo)
var tempArray = [Int]()
for (i, array) in dataArray.enumerated() {
if array.count == 0 {
tempArray.append(i)
} else {
// 給標(biāo)題數(shù)組添加數(shù)據(jù)
sectionTitleArray.append(indexedCollation.sectionTitles[i])
}
}
// 刪除沒有數(shù)據(jù)的數(shù)組
for i in tempArray.reversed() {
dataArray.remove(at: i)
}
//將得到新的 分組數(shù)據(jù)以及標(biāo)題數(shù)組
finishCallback(dataArray, sectionTitleArray)
}
}
調(diào)用:
UILocalizedIndexedCollation.getCurrentKeysAndObjectsData(needSortArray: data) { (models, titles) in
// models即為按字母排好序的數(shù)組
// titles: 字母數(shù)組
}