优化

fanglinwei · Oct 10, 2022 · ea96f01 · ea96f01
1 parent 268fc71
commit ea96f01
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/Sources/Jenga/Core/Row/BasicRow.swift b/Sources/Jenga/Core/Row/BasicRow.swift
@@ -82,6 +82,10 @@ open class BasicRow<T: UITableViewCell>: SystemRow, RowConfigurable {
 
 public extension BasicRow {
 
+    func selectionStyle(_ value: UITableViewCell.SelectionStyle) -> Self {
+        reform { $0.selectionStyle = value }
+    }
+
     func customize(_ value: @escaping ((T) -> Void)) -> Self {
         reform { $0.customize = value }
     }

diff --git a/Sources/Jenga/Core/Row/TableRow.swift b/Sources/Jenga/Core/Row/TableRow.swift
@@ -55,6 +55,10 @@ public extension TableRow {
         reform { $0.isSelectable = value }
     }
 
+    func selectionStyle(_ value: UITableViewCell.SelectionStyle) -> Self {
+        reform { $0.selectionStyle = value }
+    }
+
     func data(_ data: Cell.CellData) -> Self {
         reform { $0.item = .constant(data) }
     }

diff --git a/Sources/Jenga/Core/TableDirector.swift b/Sources/Jenga/Core/TableDirector.swift
@@ -158,17 +158,17 @@ extension TableDirector: UITableViewDataSource {
                 let cell =
                 tableView.dequeueReusableCell(withIdentifier: row.reuseIdentifier) ??
                 row.cellType.init(style: row.cellStyle, reuseIdentifier: row.reuseIdentifier)
-                (row as? RowConfigurable)?.configure(cell)
                 cell.selectionStyle = row.selectionStyle
+                (row as? RowConfigurable)?.configure(cell)
                 return cell
 
             } else {
                 cellRegisterer?.register(cellType: row.cellType, forCellReuseIdentifier: row.reuseIdentifier)
 
                 let cell =
                 tableView.dequeueReusableCell(withIdentifier: row.reuseIdentifier, for: indexPath)
-                (row as? RowConfigurable)?.configure(cell)
                 cell.selectionStyle = row.selectionStyle
+                (row as? RowConfigurable)?.configure(cell)
                 return cell
             }
         }