Adding "total" to adapter queries (introduced by Magento 2.3) (#791)

Author: damcou

Adapter/Algolia.php

@@ -6,6 +6,7 @@
 use AlgoliaSearch\AlgoliaConnectionException;
 use Magento\Framework\App\ResourceConnection;
 use Magento\Framework\DB\Ddl\Table;
+use Magento\Framework\DB\Select;
 use Magento\Framework\Search\Adapter\Mysql\Aggregation\Builder as AggregationBuilder;
 use Magento\Framework\Search\Adapter\Mysql\DocumentFactory;
 use Magento\Framework\Search\Adapter\Mysql\Mapper;
@@ -40,6 +41,11 @@ class Algolia implements AdapterInterface
     /** @var DocumentFactory */
     private $documentFactory;
 
+    private $countSqlSkipParts = [
+        \Magento\Framework\DB\Select::LIMIT_COUNT => true,
+        \Magento\Framework\DB\Select::LIMIT_OFFSET => true,
+    ];
+
     /**
      * @param Mapper $mapper
      * @param ResponseFactory $responseFactory
@@ -103,6 +109,7 @@ public function query(RequestInterface $request)
         $response = [
             'documents' => $documents,
             'aggregations' => $aggregations,
+            'total' => 0,
         ];
 
         return $this->responseFactory->create($response);
@@ -120,6 +127,7 @@ private function nativeQuery(RequestInterface $request)
         $response = [
             'documents' => $documents,
             'aggregations' => $aggregations,
+            'total' => $this->getSize($query),
         ];
 
         return $this->responseFactory->create($response);
@@ -130,6 +138,29 @@ private function getApiDocument($document)
         return $this->documentFactory->create($document);
     }
 
+    private function getSize(Select $query)
+    {
+        $sql = $this->getSelectCountSql($query);
+        $parentSelect = $this->getConnection()->select();
+        $parentSelect->from(['core_select' => $sql]);
+        $parentSelect->reset(\Magento\Framework\DB\Select::COLUMNS);
+        $parentSelect->columns('COUNT(*)');
+        $totalRecords = $this->getConnection()->fetchOne($parentSelect);
+
+        return intval($totalRecords);
+    }
+
+    private function getSelectCountSql(Select $query)
+    {
+        foreach ($this->countSqlSkipParts as $part => $toSkip) {
+            if ($toSkip) {
+                $query->reset($part);
+            }
+        }
+
+        return $query;
+    }
+
     /**
      * Executes query and return raw response
      *

Model/Observer/CatalogPermissions/ProductCollectionAddPermissions.php

@@ -65,7 +65,6 @@ protected function addProductPermissionsData($additionalData, $productIds, $stor
             foreach ($permissions as $permission) {
                 list($permissionStoreId, $customerGroupId, $level) = explode('_', $permission);
                 if ($permissionStoreId == $storeId) {
-
                     $additionalData->addProductData($productId, [
                         'customer_group_permission_' . $customerGroupId => (($level == -2 || $level != -1
                         && !$catalogPermissionsHelper->isAllowedCategoryView()) ? 0 : 1),