Avoid useless round-trip when using scan(limit=...)

See #73.
python-happybase · Nov 29, 2014 · 6205177 · 6205177
1 parent ad2673e
commit 6205177
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/happybase/table.py b/happybase/table.py
@@ -387,7 +387,7 @@ def scan(self, row_start=None, row_stop=None, row_prefix=None,
                     scan_id, how_many)
 
                 if not items:
-                    break  # scan has finished
+                    return  # scan has finished
 
                 n_fetched += len(items)
 
@@ -401,7 +401,7 @@ def scan(self, row_start=None, row_stop=None, row_prefix=None,
                     yield item.row, row
 
                     if limit is not None and n_returned == limit:
-                        break  # not interested in the remainder
+                        return  # scan has finished
         finally:
             self.connection.client.scannerClose(scan_id)
             logger.debug(