From 314839d46ad93b251d688f38fea300564e8ed908 Mon Sep 17 00:00:00 2001
From: Ricards Z <ricards.z@nnpro.at>
Date: Thu, 18 Apr 2019 14:04:34 +0300
Subject: [PATCH] Fix "Url Rewrite Step" data source pagination

---
 src/Migration/Step/UrlRewrite/Version11300to2000.php | 1 +
 src/Migration/Step/UrlRewrite/Version11410to2000.php | 1 +
 src/Migration/Step/UrlRewrite/Version191to2000.php   | 1 +
 3 files changed, 3 insertions(+)

diff --git a/src/Migration/Step/UrlRewrite/Version11300to2000.php b/src/Migration/Step/UrlRewrite/Version11300to2000.php
index 06b873f7a..c8badcfb5 100644
--- a/src/Migration/Step/UrlRewrite/Version11300to2000.php
+++ b/src/Migration/Step/UrlRewrite/Version11300to2000.php
@@ -294,6 +294,7 @@ protected function data()
             $this->migrateRewrites($records, $destinationRecords);
             $this->destination->saveRecords($destinationDocument->getName(), $destinationRecords);
             $this->destination->saveRecords($destProductCategory->getName(), $destProductCategoryRecords);
+            $this->source->setLastLoadedRecord($sourceDocument->getName(), end($data));
         }
         $this->copyEavData('catalog_category_entity_url_key', 'catalog_category_entity_varchar', 'category');
         $this->copyEavData('catalog_product_entity_url_key', 'catalog_product_entity_varchar', 'product');
diff --git a/src/Migration/Step/UrlRewrite/Version11410to2000.php b/src/Migration/Step/UrlRewrite/Version11410to2000.php
index a7f771bdc..bc19ff9ac 100644
--- a/src/Migration/Step/UrlRewrite/Version11410to2000.php
+++ b/src/Migration/Step/UrlRewrite/Version11410to2000.php
@@ -299,6 +299,7 @@ protected function data()
             $this->migrateRewrites($records, $destinationRecords);
             $this->destination->saveRecords($destinationDocument->getName(), $destinationRecords);
             $this->destination->saveRecords($destProductCategory->getName(), $destProductCategoryRecords);
+            $this->source->setLastLoadedRecord($sourceDocument->getName(), end($data));
         }
         $this->copyEavData('catalog_category_entity_url_key', 'catalog_category_entity_varchar', 'category');
         $this->copyEavData('catalog_product_entity_url_key', 'catalog_product_entity_varchar', 'product');
diff --git a/src/Migration/Step/UrlRewrite/Version191to2000.php b/src/Migration/Step/UrlRewrite/Version191to2000.php
index 73a21633a..eaf897ac2 100644
--- a/src/Migration/Step/UrlRewrite/Version191to2000.php
+++ b/src/Migration/Step/UrlRewrite/Version191to2000.php
@@ -220,6 +220,7 @@ protected function data()
                 $destinationRecords->addRecord($destRecord);
             }
 
+            $this->source->setLastLoadedRecord(self::SOURCE, end($bulk));
             $this->progress->advance();
             $this->destination->saveRecords(self::DESTINATION, $destinationRecords);
             $this->destination->saveRecords(self::DESTINATION_PRODUCT_CATEGORY, $destProductCategoryRecords);