Use an Iterator to generate the list of Items

Abizern · Sep 20, 2017 · b478f2c · b478f2c
1 parent 2e09a6f
commit b478f2c
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/PartialTable/DataSource.swift b/PartialTable/DataSource.swift
@@ -15,20 +15,18 @@ class DataSource: NSObject, UITableViewDataSource {
     private var numberOfItems = 10
 
     lazy private var items: [Item] = {
-        var list = [Item]()
-        for _ in 0 ..< self.maximumNumberOfItems {
-            list.append(Item(value: UUID().uuidString))
+        let iterator = AnyIterator {
+            return Item(value: UUID().uuidString)
         }
-
-        return list
+        
+        return Array(iterator.prefix(maximumNumberOfItems))
     }()
 
 
     func getMoreItems() {
         numberOfItems = min(numberOfItems + numberOfItemsToAdd, maximumNumberOfItems)
     }
 
-
     func numberOfSections(in tableView: UITableView) -> Int {
         switch (numberOfItems < maximumNumberOfItems) {
         case true:
@@ -38,7 +36,6 @@ class DataSource: NSObject, UITableViewDataSource {
         }
     }
 
-
     func tableView(_ tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
         switch section {
         case 0:
@@ -50,7 +47,6 @@ class DataSource: NSObject, UITableViewDataSource {
         }
     }
 
-
     func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
         switch indexPath.section {
         case 0: