Update DLPhotoTableViewController.m

darling0825 · Jun 17, 2017 · e2afc70 · e2afc70
1 parent 8e48bcf
commit e2afc70
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/DLPhotoPicker/Controller/DLPhotoTableViewController.m b/DLPhotoPicker/Controller/DLPhotoTableViewController.m
@@ -48,6 +48,29 @@ - (void)viewDidLoad {
     [self fetchPhotoCollectionAndReload];
 }
 
+- (void)viewWillAppear:(BOOL)animated {
+    [super viewWillAppear:animated];
+
+    [[NSNotificationCenter defaultCenter] addObserver:self
+                                             selector:@selector(applicationWillEnterForeground:)
+                                                 name:UIApplicationWillEnterForegroundNotification
+                                               object:nil];
+
+    [self setupPhotoCollection];
+    [self.tableView reloadData];
+}
+
+- (void)viewWillDisappear:(BOOL)animated {
+    [super viewWillDisappear:animated];
+
+    [[NSNotificationCenter defaultCenter] removeObserver:self];
+}
+
+- (void)applicationWillEnterForeground:(NSNotification *)notification {
+    [self fetchPhotoCollectionAndReload];
+}
+
+
 - (void)didReceiveMemoryWarning {
     [super didReceiveMemoryWarning];
     self.photoCollections = nil;