Implement file size aggregation services for assets

Author: markus-moser

config/services/search/search-services.yaml

@@ -22,6 +22,9 @@ services:
   Pimcore\Bundle\GenericDataIndexBundle\Service\Search\SearchService\Element\ElementSearchHelperInterface:
       class: Pimcore\Bundle\GenericDataIndexBundle\Service\Search\SearchService\Element\SearchHelper
 
+  Pimcore\Bundle\GenericDataIndexBundle\Service\Search\SearchService\Asset\Aggregation\FileSizeAggregationServiceInterface:
+      class: Pimcore\Bundle\GenericDataIndexBundle\Service\Search\SearchService\Asset\Aggregation\FileSizeAggregationService
+
   Pimcore\Bundle\GenericDataIndexBundle\Service\Search\SearchService\SearchProviderInterface:
       class: Pimcore\Bundle\GenericDataIndexBundle\Service\Search\SearchService\SearchProvider
 

doc/04_Searching_For_Data_In_Index/05_Search_Modifiers/README.md

@@ -56,10 +56,11 @@ $search->addModifier(new ParentIdFilter(1))
 ### Aggregations
 
 
-| Modifier                                                                                                                                                           | Modifier Category        | Description                                                                                           |
-|--------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------|-------------------------------------------------------------------------------------------------------|
-| [ChildrenCountAggregation](https://github.com/pimcore/generic-data-index-bundle/blob/1.x/src/Model/Search/Modifier/Aggregation/Tree/ChildrenCountAggregation.php)  | Tree related aggregation | Get children counts for given element IDs.                                                            |
-| [AssetMetaDataAggregation](https://github.com/pimcore/generic-data-index-bundle/blob/1.x/src/Model/Search/Modifier/Aggregation/Asset/AssetMetaDataAggregation.php) | Assets                   | Used for the filters in the asset grid to aggregate the filter options for supported meta data types. |
+| Modifier                                                                                                                                                           | Modifier Category        | Description                                                                                                                                                                                         |
+|--------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
+| [ChildrenCountAggregation](https://github.com/pimcore/generic-data-index-bundle/blob/1.x/src/Model/Search/Modifier/Aggregation/Tree/ChildrenCountAggregation.php)  | Tree related aggregation | Get children counts for given element IDs.                                                                                                                                                          |
+| [AssetMetaDataAggregation](https://github.com/pimcore/generic-data-index-bundle/blob/1.x/src/Model/Search/Modifier/Aggregation/Asset/AssetMetaDataAggregation.php) | Assets                   | Used for the filters in the asset grid to aggregate the filter options for supported meta data types.                                                                                               |
+| [FileSizeSumAggregation](https://github.com/pimcore/generic-data-index-bundle/blob/1.x/src/Model/Search/Modifier/Aggregation/Asset/FileSizeSumAggregation.php) | Assets                   | Aggregates the sum of file sizes for all assets for a given search. The `FileSizeAggregationServiceInterface` internally uses this aggregation and provides an easy way to use this functionality. |
 
 ## Add your own search modifier
 

src/Model/Search/Asset/SearchResult/AssetSearchResultItem.php

@@ -279,7 +279,7 @@ public function isHasChildren(): bool
             $this->lazyLoad();
         }
 
-        return $this->hasChildren;
+        return $this->hasChildren ?? false;
     }
 
     public function setHasChildren(bool $hasChildren): AssetSearchResultItem

src/Model/Search/Modifier/Aggregation/Asset/FileSizeSumAggregation.php

@@ -0,0 +1,32 @@
+<?php
+declare(strict_types=1);
+
+/**
+ * Pimcore
+ *
+ * This source file is available under two different licenses:
+ * - GNU General Public License version 3 (GPLv3)
+ * - Pimcore Commercial License (PCL)
+ * Full copyright and license information is available in
+ * LICENSE.md which is distributed with this source code.
+ *
+ *  @copyright  Copyright (c) Pimcore GmbH (http://www.pimcore.org)
+ *  @license    http://www.pimcore.org/license     GPLv3 and PCL
+ */
+
+namespace Pimcore\Bundle\GenericDataIndexBundle\Model\Search\Modifier\Aggregation\Asset;
+
+use Pimcore\Bundle\GenericDataIndexBundle\Model\Search\Modifier\SearchModifierInterface;
+
+final readonly class FileSizeSumAggregation implements SearchModifierInterface
+{
+    public function __construct(
+        private string $aggregationName,
+    ) {
+    }
+
+    public function getAggregationName(): string
+    {
+        return $this->aggregationName;
+    }
+}

src/Model/SearchIndexAdapter/SearchResultAggregation.php

@@ -24,6 +24,7 @@ public function __construct(
         private array $buckets,
         private int $otherDocCount,
         private int $docCountErrorUpperBound,
+        private array $aggregationResult = [],
     ) {
     }
 
@@ -46,4 +47,10 @@ public function getDocCountErrorUpperBound(): int
     {
         return $this->docCountErrorUpperBound;
     }
+
+    public function getAggregationResult(): array
+    {
+        return $this->aggregationResult;
+    }
+
 }

src/Model/SearchIndexAdapter/SearchResultAggregationBucket.php

@@ -19,7 +19,7 @@
 readonly class SearchResultAggregationBucket
 {
     public function __construct(
-        private string|int $key,
+        private string|int|float $key,
         private int $docCount,
     ) {
     }

src/SearchIndexAdapter/OpenSearch/Search/Modifier/Aggregation/AssetAggregations.php

@@ -18,8 +18,11 @@
 
 use Pimcore\Bundle\GenericDataIndexBundle\Attribute\OpenSearch\AsSearchModifierHandler;
 use Pimcore\Bundle\GenericDataIndexBundle\Exception\InvalidModifierException;
+use Pimcore\Bundle\GenericDataIndexBundle\Model\OpenSearch\Aggregation\Aggregation;
 use Pimcore\Bundle\GenericDataIndexBundle\Model\OpenSearch\Modifier\SearchModifierContextInterface;
+use Pimcore\Bundle\GenericDataIndexBundle\Model\Search\Asset\AssetSearch;
 use Pimcore\Bundle\GenericDataIndexBundle\Model\Search\Modifier\Aggregation\Asset\AssetMetaDataAggregation;
+use Pimcore\Bundle\GenericDataIndexBundle\Model\Search\Modifier\Aggregation\Asset\FileSizeSumAggregation;
 use Pimcore\Bundle\GenericDataIndexBundle\SearchIndexAdapter\Asset\FieldDefinitionServiceInterface;
 use Pimcore\Twig\Extension\Templating\Placeholder\Exception;
 
@@ -64,4 +67,26 @@ public function handleAssetMetaDataAggregation(
             throw new InvalidModifierException($e->getMessage(), 0, $e);
         }
     }
+
+    #[AsSearchModifierHandler]
+    public function handleFileSizeAggregation(
+        FileSizeSumAggregation         $aggregation,
+        SearchModifierContextInterface $context
+    ): void
+    {
+        if (!$context->getOriginalSearch() instanceof AssetSearch) {
+            throw new InvalidModifierException('FileSizeAggregation can only be used with AssetSearch');
+        }
+
+        $context->getSearch()->addAggregation(
+            new Aggregation(
+                name: $aggregation->getAggregationName(),
+                params: [
+                    'sum' => [
+                        'field' => 'system_fields.fileSize',
+                    ],
+                ]
+            )
+        );
+    }
 }

src/Service/Search/SearchService/Asset/Aggregation/FileSizeAggregationService.php

@@ -0,0 +1,34 @@
+<?php
+declare(strict_types=1);
+
+namespace Pimcore\Bundle\GenericDataIndexBundle\Service\Search\SearchService\Asset\Aggregation;
+
+use Pimcore\Bundle\GenericDataIndexBundle\Model\Search\Asset\AssetSearch;
+use Pimcore\Bundle\GenericDataIndexBundle\Model\Search\Modifier\Aggregation\Asset\FileSizeSumAggregation;
+use Pimcore\Bundle\GenericDataIndexBundle\Service\Search\SearchService\Asset\AssetSearchServiceInterface;
+
+/**
+ * @internal
+ */
+final readonly class FileSizeAggregationService implements FileSizeAggregationServiceInterface
+{
+    public function __construct(
+        private AssetSearchServiceInterface $assetSearchService,
+    )
+    {
+    }
+
+    public function getFileSizeSum(AssetSearch $assetSearch): int
+    {
+        $aggregation = new FileSizeSumAggregation('fileSizeSum');
+        $assetSearch
+            ->addModifier($aggregation)
+            ->setAggregationsOnly(true)
+        ;
+
+        $result = $this->assetSearchService->search($assetSearch);
+
+        $sum = $result->getAggregation($aggregation->getAggregationName())?->getAggregationResult()['value'] ?? 0;
+        return (int) $sum;
+    }
+}

src/Service/Search/SearchService/Asset/Aggregation/FileSizeAggregationServiceInterface.php

@@ -0,0 +1,14 @@
+<?php
+declare(strict_types=1);
+
+namespace Pimcore\Bundle\GenericDataIndexBundle\Service\Search\SearchService\Asset\Aggregation;
+
+use Pimcore\Bundle\GenericDataIndexBundle\Model\Search\Asset\AssetSearch;
+
+interface FileSizeAggregationServiceInterface
+{
+    /**
+     * Returns the sum of the file sizes of all assets that match the given search criteria in bytes.
+     */
+    public function getFileSizeSum(AssetSearch $assetSearch): int;
+}

src/Service/Serializer/Denormalizer/SearchIndexAdapter/SearchResultDenormalizer.php

@@ -78,6 +78,7 @@ private function hydrateAggregations(array $aggregations): array
                 buckets: $this->hydrateAggregationBuckets($aggregation['buckets'] ?? []),
                 otherDocCount: $aggregation['sum_other_doc_count'] ?? 0,
                 docCountErrorUpperBound: $aggregation['doc_count_error_upper_bound'] ?? 0,
+                aggregationResult: $aggregation,
             );
         }