iOS tableView实现搜索功能

#import "ChineseToPinyin.h"#import "HCSortString.h"#import "WLCCityTVC.h"#import "WLCProvinceModel.h"#import "ZYPinYinSearch.h"#define reusedID @"cityCell"@interface WLCCityTVC () <UITableViewDelegate, UITableViewDataSource, UISearchResultsUpdating>/** * 存放城市字典的数组,后来也存放排序后的 */@property (nonatomic, strong) NSMutableArray* cityArrM;@property (nonatomic, strong) WLCUser* user;/** * 存放首字母的数组 */@property (nonatomic, strong) NSMutableArray* letterArrM;@property (nonatomic, strong) NSMutableArray* wordArr;/** * 存放城市模型的数组 */@property (nonatomic, strong) NSMutableArray* modelArrM;@property (strong, nonatomic) NSMutableArray* searchDataSource; /**<搜索结果数据源*/@property (strong, nonatomic) UISearchController* searchController;@end@implementation WLCCityTVC- (void)viewDidLoad{ [super viewDidLoad]; [self setupUI];}#pragma mark - navitionBarBackBarItem返回事件- (BOOL)navigationShouldPopOnBackButton{ self.searchController.active = NO; return YES;}- (void)didReceiveMemoryWarning{ [super didReceiveMemoryWarning]; // Dispose of any resources that can be recreated.}- (WLCUser*)user{ if (nil == _user) {  _user = [NSKeyedUnarchiver unarchiveObjectWithFile:fileUser]; } return _user;}- (NSMutableArray*)cityArrM{ if (nil == _cityArrM) {  _cityArrM = [NSMutableArray array]; } return _cityArrM;}- (NSMutableArray*)wordArr{ if (nil == _wordArr) {  _wordArr = [NSMutableArray array]; } return _wordArr;}- (NSMutableArray*)letterArrM{ if (nil == _letterArrM) {  _letterArrM = [NSMutableArray array]; } return _letterArrM;}- (UISearchController*)searchController{ if (!_searchController) {  _searchController = [[UISearchController alloc] initWithSearchResultsController:nil];  _searchController.searchResultsUpdater = self;  _searchController.dimsBackgroundDuringPresentation = NO;  _searchController.hidesNavigationBarDuringPresentation = NO;  _searchController.searchBar.placeholder = @"搜索";    _searchController.searchBar.tintColor = [UIColor whiteColor];  [_searchController.searchBar sizeToFit]; } return _searchController;}- (NSMutableArray*)modelArrM{ if (nil == _modelArrM) {  _modelArrM = [NSMutableArray array];  for (NSArray* tempArr in self.cityArrM) {   for (NSDictionary* dict in tempArr) {    [WLCProvinceModel setupReplacedKeyFromPropertyName:^NSDictionary* {     return @{      @"pID" : @"id"     };    }];    WLCProvinceModel* model = [WLCProvinceModel objectWithKeyValues:dict];    [_modelArrM addObject:model];   }  } } return _modelArrM;}- (NSMutableArray*)searchDataSource{ if (nil == _searchDataSource) {  _searchDataSource = [NSMutableArray array]; } return _searchDataSource;}- (void)setupUI{ self.tableView.backgroundColor = [UIColor rgb:234 andGreen:234 andBlue:243]; [self getAllCities]; self.tableView.delegate = self; self.tableView.dataSource = self; self.tableView.tableFooterView = [[UIView alloc] init];}- (void)getAllCities{ NSString* url = [kURL stringByAppendingString:@"promary/"]; url = [url stringByAppendingString:self.cityID]; url = [url stringByAppendingString:@"/city"]; [SVProgressHUD showWithStatus:@"获取城市中"]; [NetRequestTool requestWithParamsDict:nil image:nil name:nil token:self.user.token value:nil hearerField:nil URL:url type:GET successBlock:^(AFHTTPRequestOperation* _Nonnull operation, id _Nonnull responseObject) {  NSString* errNum = [NSString stringWithFormat:@"%@", responseObject[@"errNum"]];  if ([errNum isEqualToString:@"1"]) {   [SVProgressHUD setMinimumDismissTimeInterval:2.5];   [SVProgressHUD showInfoWithStatus:[NSString stringWithFormat:@"%@", responseObject[@"retMsg"]]];   [self.navigationController popViewControllerAnimated:YES];  }  else {   [SVProgressHUD dismiss];   @try {    self.cityArrM = responseObject[@"retData"];    NSMutableArray* cityArrM = [NSMutableArray array];    for (NSMutableDictionary* cityDic in self.cityArrM) {     [cityArrM addObject:[cityDic objectForKey:@"name"]];     NSMutableDictionary* dictM = [NSMutableDictionary dictionaryWithDictionary:cityDic];     [self prepareCityListDatasourceWithArray:cityArrM andToDictionary:dictM];    }    self.cityArrM = [self sortArray:self.wordArr];   } @catch (NSException* exception) {   } @finally {    [self.tableView setTableHeaderView:self.searchController.searchBar];    [self.tableView reloadData];   }  } }  anderrorBlock:^(AFHTTPRequestOperation* _Nonnull operation, NSError* _Nonnull error) {   WLog(@"error == %@", error);   [SVProgressHUD showErrorWithStatus:@"获取省份失败,请稍后重试"];   [self.navigationController popViewControllerAnimated:YES];  }];}#pragma mark -排序城市- (void)prepareCityListDatasourceWithArray:(NSArray*)array andToDictionary:(NSMutableDictionary*)dic{ for (NSString* city in array) {  NSString* cityPinyin = [ChineseToPinyin pinyinFromChiniseString:city];  if ([city isEqualToString:@"重庆"]) {   cityPinyin = @"CHONGQING";  }  NSString* firstLetter = [cityPinyin substringWithRange:NSMakeRange(0, 1)];  if (![dic objectForKey:firstLetter]) {   //   NSMutableArray* arr = [NSMutableArray array];   //   [dic setValue:firstLetter forKey:@"letter"];   dic[@"letter"] = firstLetter;  }  if ([[dic objectForKey:firstLetter] containsObject:city]) {   return;  } } [self.wordArr addObject:dic]; //  [self.wordArr addObjectsFromArray:[[dic allValues] sortedArrayUsingSelector:@selector(compare:)]];}/** * 排序并按首字母分组 * * @param arrayToSort <#arrayToSort description#> * * @return <#return value description#> */- (NSMutableArray*)sortArray:(NSMutableArray*)arrayToSort{ NSMutableArray* arrayForArrays = [[NSMutableArray alloc] init]; //根据拼音对数组排序 NSArray* sortDescriptors = [NSArray arrayWithObject:[NSSortDescriptor sortDescriptorWithKey:@"letter" ascending:YES]]; //排序 [arrayToSort sortUsingDescriptors:sortDescriptors]; NSMutableArray* tempArray = nil; BOOL flag = NO; //分组 for (int i = 0; i < arrayToSort.count; i++) {  NSString* pinyin = [arrayToSort[i] objectForKey:@"letter"];  NSString* firstChar = [pinyin substringToIndex:1];  //  NSLog(@"%@",firstChar);  if (![self.letterArrM containsObject:[firstChar uppercaseString]]) {   [self.letterArrM addObject:[firstChar uppercaseString]];   tempArray = [[NSMutableArray alloc] init];   flag = NO;  }  if ([self.letterArrM containsObject:[firstChar uppercaseString]]) {   [tempArray addObject:arrayToSort[i]];   if (flag == NO) {    [arrayForArrays addObject:tempArray];    flag = YES;   }  } } return arrayForArrays;}//让cell下划线左对齐- (void)tableView:(UITableView*)tableView willDisplayCell:(UITableViewCell*)cell forRowAtIndexPath:(NSIndexPath*)indexPath{ if ([cell respondsToSelector:@selector(setSeparatorInset:)]) {  [cell setSeparatorInset:UIEdgeInsetsMake(0, 0, 0, 0)]; } if ([cell respondsToSelector:@selector(setPreservesSuperviewLayoutMargins:)]) {  [cell setPreservesSuperviewLayoutMargins:NO]; } if ([cell respondsToSelector:@selector(setLayoutMargins:)]) {  [cell setLayoutMargins:UIEdgeInsetsZero]; }}#pragma mark - tableView's delegate and datasource- (NSInteger)numberOfSectionsInTableView:(UITableView*)tableView{ if (!self.searchController.active) {  return self.letterArrM.count; } else {  return 1; }}- (NSInteger)tableView:(UITableView*)tableView numberOfRowsInSection:(NSInteger)section{ if (!self.searchController.active) {  return [self.cityArrM[section] count]; } else {  return self.searchDataSource.count; }}- (UITableViewCell*)tableView:(UITableView*)tableView cellForRowAtIndexPath:(NSIndexPath*)indexPath{ UITableViewCell* cell = [tableView dequeueReusableCellWithIdentifier:reusedID]; if (!cell) {  cell = [[UITableViewCell alloc] initWithStyle:UITableViewCellStyleDefault reuseIdentifier:reusedID]; } @try {  if (!self.searchController.active) {   NSDictionary* dict = self.cityArrM[indexPath.section][indexPath.row];   cell.textLabel.text = [NSString stringWithFormat:@"%@", dict[@"name"]];  }  else {   //   NSDictionary* dict = self.searchDataSource[indexPath.row];   WLCProvinceModel* model = self.searchDataSource[indexPath.row];   cell.textLabel.text = [NSString stringWithFormat:@"%@", model.name];  } } @catch (NSException* exception) { } @finally { } return cell;}- (NSString*)tableView:(UITableView*)tableView titleForHeaderInSection:(NSInteger)section{ if (!self.searchController.active) {  return [self.letterArrM objectAtIndex:section]; } else  return nil;}- (NSArray*)sectionIndexTitlesForTableView:(UITableView*)tableView{ if (!self.searchController.active) {  return self.letterArrM; } else  return nil;}- (void)tableView:(UITableView*)tableView didSelectRowAtIndexPath:(NSIndexPath*)indexPath{ NSDictionary* dict; @try {  if (!self.searchController.active) {   dict = self.cityArrM[indexPath.section][indexPath.row];  }  else {   WLCProvinceModel* model = self.searchDataSource[indexPath.row];   dict = [NSDictionary dictionaryWithObjects:@[ model.name, model.pID ] forKeys:@[ @"name", @"id" ]];  }  [[NSNotificationCenter defaultCenter] postNotificationName:@"cityChoosed" object:self userInfo:dict]; } @catch (NSException* exception) { } @finally { } self.searchController.active = NO; [self.navigationController popToViewController:self.fatherVC animated:YES]; [tableView deselectRowAtIndexPath:indexPath animated:YES];}#pragma mark - UISearchDelegate- (void)updateSearchResultsForSearchController:(UISearchController*)searchController{ [self.searchDataSource removeAllObjects]; NSArray* ary = [NSArray new]; //对排序好的数据进行搜索 NSDictionary* allDataSource = [HCSortString sortAndGroupForArray:self.modelArrM PropertyName:@"name"]; ary = [HCSortString getAllValuesFromDict:allDataSource]; if (searchController.searchBar.text.length == 0) {  [self.searchDataSource addObjectsFromArray:ary]; } else {  ary = [ZYPinYinSearch searchWithOriginalArray:ary andSearchText:searchController.searchBar.text andSearchByPropertyName:@"name"];  [self.searchDataSource addObjectsFromArray:ary]; } [self.tableView reloadData];}@end