Refine shortcut extension

ishkawa · Sep 12, 2018 · 2a3aa73 · 2a3aa73
1 parent 1f72c71
commit 2a3aa73
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/DataSourceKit/CollectionViewDataSource.swift b/DataSourceKit/CollectionViewDataSource.swift
@@ -40,7 +40,7 @@ public class CollectionViewDataSource<CellDeclaration>: NSObject, UICollectionVi
 extension CollectionViewDataSource where CellDeclaration == CellBinder {
     // Add meaningless parameter to avoid error `Members of constrained extensions
     // cannot be declared @objc`, which is caused by collision to init() of NSObject.
-    public convenience init(_ declarationType: CellDeclaration.Type) {
-        self.init(binderFromDeclaration: { $0 })
+    public convenience init(binderFromBinder: @escaping (CellBinder) -> CellBinder = { $0 }) {
+        self.init(binderFromDeclaration: binderFromBinder)
     }
 }
diff --git a/DataSourceKit/TableViewDataSource.swift b/DataSourceKit/TableViewDataSource.swift
@@ -40,7 +40,7 @@ public class TableViewDataSource<CellDeclaration>: NSObject, UITableViewDataSour
 extension TableViewDataSource where CellDeclaration == CellBinder {
     // Add meaningless parameter to avoid error `Members of constrained extensions
     // cannot be declared @objc`, which is caused by collision to init() of NSObject.
-    public convenience init(_ declarationType: CellDeclaration.Type) {
-        self.init(binderFromDeclaration: { $0 })
+    public convenience init(binderFromBinder: @escaping (CellBinder) -> CellBinder = { $0 }) {
+        self.init(binderFromDeclaration: binderFromBinder)
     }
 }