AC-9607::Filtering Company Grid & Then Attempting Grid CSV Export Will Fail & Throw Exception

Author: AshishKumarPundeer

app/code/Magento/Ui/Model/Export/ConvertToCsv.php

@@ -81,19 +81,30 @@ public function getCsvFile()
         $searchCriteria = $dataProvider->getSearchCriteria()
             ->setCurrentPage($i)
             ->setPageSize($this->pageSize);
-        $totalCount = (int) $dataProvider->getSearchResult()->getTotalCount();
-        $totalPagesCount = (int) ceil($totalCount / $this->pageSize);
-        while ($i <= $totalPagesCount) {
-            // setTotalCount to prevent total count from being calculated in loop
+
+        $totalCount = null;
+        $totalPagesCount = null;
+
+        do {
             $searchResult = $dataProvider->getSearchResult();
-            $searchResult->setTotalCount($totalCount);
             $items = $searchResult->getItems();
+
+            if ($totalCount === null) { // get total count only once
+                $totalCount = $searchResult->getTotalCount();
+                $totalPagesCount = (int) ceil($totalCount / $this->pageSize);
+            }
+
+            // call setTotalCount to prevent total count from being calculate in subsequent iterations of this loop
+            $searchResult->setTotalCount($totalCount);
+
             foreach ($items as $item) {
                 $this->metadataProvider->convertDate($item, $component->getName());
                 $stream->writeCsv($this->metadataProvider->getRowData($item, $fields, $options));
             }
+
             $searchCriteria->setCurrentPage(++$i);
-        }
+        } while ($i <= $totalPagesCount);
+
         $stream->unlock();
         $stream->close();