更新 cell 获取方法

bestswifter · bestswifter · commit 55d1fe12bdec · 2016-02-28T20:03:42.000+08:00
diff --git a/CornerRadius/CornerRadius/ViewController.swift b/CornerRadius/CornerRadius/ViewController.swift
@@ -27,12 +27,10 @@ class ViewController: UIViewController, UITableViewDelegate, UITableViewDataSour
 
     func tableView(tableView: UITableView, cellForRowAtIndexPath indexPath: NSIndexPath) -> UITableViewCell {
         let identifier = "tableviewcell"
-        var cell = table.dequeueReusableCellWithIdentifier(identifier) as? CustomTableViewCell
-        if cell == nil {
-            cell = CustomTableViewCell(style: .Default, reuseIdentifier: identifier)
-            cell?.selectionStyle = .None
-        }
-        
+        table.registerClass(CustomTableViewCell.self, forCellReuseIdentifier: identifier)
+        let cell = table.dequeueReusableCellWithIdentifier(identifier, forIndexPath: indexPath) as? CustomTableViewCell
+
+        cell?.selectionStyle = .None
         cell?.setupContent(imgName: "photo")
         
         return cell!
