Add “limit” option to CopyAll, CleanAll samplers

src/BaseSampler.php

@@ -40,6 +40,13 @@ abstract class BaseSampler implements SamplerInterface
      * @var array
      */
     protected $referenceFields = [];
+
+    /**
+     * Max number to match (default Db order)
+     *
+     * @var integer
+     */
+    protected $limit;
 
     /**
      * Table name
@@ -147,6 +154,7 @@ public function setReferenceStore(ReferenceStore $referenceStore)
     public function loadConfig($config)
     {
         $this->referenceFields = isset($config->remember) ? $config->remember : [];
+        $this->limit = isset($config->limit) ? (int)$config->limit : false;
     }
 
     /**

src/Sampler/CleanAll.php

@@ -56,9 +56,13 @@ public function loadConfig($config)
      */
     public function getRows()
     {
-        $query = $this->sourceConnection->createQueryBuilder()->select('*')->from($this->tableName)->execute();
+        $query = $this->sourceConnection->createQueryBuilder()->select('*')->from($this->tableName);
 
-        $rows = $query->fetchAll();
+        if ($this->limit) {
+            $query->setMaxResults($this->limit);
+        }
+
+        $rows = $query->execute()->fetchAll();
 
         foreach ($rows as &$row) {
             $this->rowCleaner->cleanRow($row);

src/Sampler/CopyAll.php

@@ -13,8 +13,12 @@ public function getName()
 
     public function getRows()
     {
-        $query = $this->sourceConnection->createQueryBuilder()->select('*')->from($this->tableName)->execute();
+        $query = $this->sourceConnection->createQueryBuilder()->select('*')->from($this->tableName);
 
-        return $query->fetchAll();
+        if ($this->limit) {
+            $query->setMaxResults($this->limit);
+        }
+
+        return $query->execute()->fetchAll();
     }
 }

src/Sampler/Matched.php

@@ -29,13 +29,6 @@ class Matched extends BaseSampler
      */
     protected $constraints;
 
-    /**
-     * Max number to match (default Db order)
-     *
-     * @var integer
-     */
-    protected $limit;
-
     /**
      * Return a unique name for this sampler for informational purposes
      *
@@ -59,7 +52,6 @@ public function loadConfig($config)
     {
         parent::loadConfig($config);
         $this->constraints = (array)$this->demandParameterValue($config, 'constraints');
-        $this->limit = isset($config->limit) ? (int)$config->limit : false;
     }
 
     /**