set delegate and datasource to nil when deinit

nghialv · Jan 22, 2016 · ea46712 · ea46712
1 parent 0a4333f
commit ea46712
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Source/Hakuba.swift b/Source/Hakuba.swift
@@ -56,6 +56,11 @@ public class Hakuba : NSObject {
         tableView.dataSource = self
     }
 
+    deinit {
+        tableView?.delegate = nil
+        tableView?.dataSource = nil
+    }
+
     public func deselectAllCells() {
         for cell in selectedCells {
             if let cellmodel = (cell as? MYTableViewCell)?.cellModel {