封装下拉刷新与上拉加载

bestswifter · bestswifter · commit 61ba9745f826 · 2016-05-15T23:44:52.000+08:00
diff --git a/KtTableView/KtTableView/KTMainViewController.m b/KtTableView/KtTableView/KTMainViewController.m
@@ -27,16 +27,9 @@ @implementation KTMainViewController
 - (void)viewDidLoad {
     [super viewDidLoad];
     [self createModel];
-//    [self getFirstPage];
-    __weak typeof(self) wSelf = self;
-    self.tableView.mj_header = [MJRefreshNormalHeader headerWithRefreshingBlock:^{
-        [wSelf getFirstPage];
-    }];
-    
-    self.tableView.mj_footer = [MJRefreshAutoNormalFooter footerWithRefreshingBlock:^{
-        [wSelf getFirstPage];
-    }];
     
+    self.tableView.isNeedPullUpToRefreshAction = YES;
+    self.tableView.isNeedPullDownToRefreshAction = YES;
     // Do any additional setup after loading the view, typically from a nib.
 }
 
@@ -58,6 +51,14 @@ - (void)didReceiveMemoryWarning {
     // Dispose of any resources that can be recreated.
 }
 
+- (void)pullUpToRefreshAction {
+    [self getFirstPage];
+}
+
+- (void)pullDownToRefreshAction {
+    [self getFirstPage];
+}
+
 - (void)getFirstPage {
     self.model.params = @{@"nextPage": @0};
     [self.model loadWithShortConnection];
@@ -70,12 +71,7 @@ - (void)requestBooksSuccess {
         [self.dataSource appendItem:item];
     }
     [self.tableView reloadData];
-    if ([self.tableView.mj_header isRefreshing]) {
-        [self.tableView.mj_header endRefreshing];
-    }
-    if ([self.tableView.mj_footer isRefreshing]) {
-        [self.tableView.mj_footer endRefreshing];
-    }
+    [self.tableView stopRefreshingAnimation];
 }
 
 @end
diff --git a/KtTableView/KtTableView/KtBaseTableView.h b/KtTableView/KtTableView/KtBaseTableView.h
@@ -21,6 +21,14 @@
 
 - (UIView *)headerViewForSectionObject:(KtTableViewSectionObject *)sectionObject atSection:(NSInteger)section;
 
+// 下拉刷新触发的方法
+
+- (void)pullDownToRefreshAction;
+
+// 上拉加载触发的方法
+ 
+- (void)pullUpToRefreshAction;
+
 // 将来可以有 cell 的编辑，交换，左滑等回调
 
 // 这个协议继承了UITableViewDelegate ，所以自己做一层中转，VC 依然需要实现某些代理方法。
@@ -33,4 +41,11 @@
 
 @property (nonatomic, assign) id<KtTableViewDelegate> ktDelegate;
 
+// 是否需要下拉刷新和上拉加载
+@property (nonatomic, assign) BOOL isNeedPullDownToRefreshAction;
+@property (nonatomic, assign) BOOL isNeedPullUpToRefreshAction;
+
+- (void)stopRefreshingAnimation;
+- (void)triggerRefreshing;
+
 @end
diff --git a/KtTableView/KtTableView/KtBaseTableView.m b/KtTableView/KtTableView/KtBaseTableView.m
@@ -10,6 +10,7 @@
 #import "KtBaseTableViewCell.h"
 #import "KtTableViewSectionObject.h"
 #import "KtTableViewBaseItem.h"
+#import "MJRefresh.h"
 
 @implementation KtBaseTableView
 
@@ -23,6 +24,8 @@ - (instancetype)initWithFrame:(CGRect)frame style:(UITableViewStyle)style {
         self.sectionHeaderHeight = 0;
         self.sectionFooterHeight = 0;
         self.delegate = self;
+        self.isNeedPullDownToRefreshAction = NO;
+        self.isNeedPullUpToRefreshAction = NO;
     }
     return self;
 }
@@ -34,6 +37,52 @@ - (void)setKtDataSource:(id<KtTableViewDataSource>)ktDataSource {
     }
 }
 
+#pragma mark - 上拉加载和下拉刷新
+- (void)setIsNeedPullDownToRefreshAction:(BOOL)isEnable {
+    if (_isNeedPullDownToRefreshAction == isEnable) {
+        return;
+    }
+    _isNeedPullDownToRefreshAction = isEnable;
+    __block typeof(self) weakSelf = self;
+    if (_isNeedPullDownToRefreshAction) {
+        self.mj_header = [MJRefreshNormalHeader headerWithRefreshingBlock:^{
+            if ([weakSelf.ktDelegate respondsToSelector:@selector(pullDownToRefreshAction)]) {
+                [weakSelf.ktDelegate pullDownToRefreshAction];
+            }
+        }];
+        
+    }
+}
+
+- (void)setIsNeedPullUpToRefreshAction:(BOOL)isEnable
+{
+    if (_isNeedPullUpToRefreshAction == isEnable) {
+        return;
+    }
+    _isNeedPullUpToRefreshAction = isEnable;
+    __block typeof(self) weakSelf = self;
+    if (_isNeedPullUpToRefreshAction) {
+        self.mj_footer = [MJRefreshAutoNormalFooter footerWithRefreshingBlock:^{
+            if ([weakSelf.ktDelegate respondsToSelector:@selector(pullUpToRefreshAction)]) {
+                [weakSelf.ktDelegate pullUpToRefreshAction];
+            }
+        }];
+    }
+}
+
+- (void)stopRefreshingAnimation {
+    if ([self.mj_header isRefreshing]) {
+        [self.mj_header endRefreshing];
+    }
+    if ([self.mj_footer isRefreshing]) {
+        [self.mj_footer endRefreshing];
+    }
+}
+
+- (void)triggerRefreshing {
+    [self.mj_header beginRefreshing];
+}
+
 #pragma mark - UITableViewDelegate
 
 - (CGFloat)tableView:(UITableView*)tableView heightForRowAtIndexPath:(NSIndexPath*)indexPath {
