add ChunkGeneratorTrait

Author: wwidergoldpimcore

src/Element/Service/ExecutionEngine/ElementReferenceService.php

@@ -14,7 +14,6 @@
 namespace Pimcore\Bundle\StudioBackendBundle\Element\Service\ExecutionEngine;
 
 use Exception;
-use Generator;
 use Pimcore\Bundle\GenericExecutionEngineBundle\Agent\JobExecutionAgentInterface;
 use Pimcore\Bundle\GenericExecutionEngineBundle\Model\Job;
 use Pimcore\Bundle\GenericExecutionEngineBundle\Model\JobStep;
@@ -28,6 +27,7 @@
 use Pimcore\Bundle\StudioBackendBundle\ExecutionEngine\Util\EnvironmentVariables;
 use Pimcore\Bundle\StudioBackendBundle\ExecutionEngine\Util\Jobs;
 use Pimcore\Bundle\StudioBackendBundle\ExecutionEngine\Util\StepConfig;
+use Pimcore\Bundle\StudioBackendBundle\ExecutionEngine\Util\Trait\ChunkGeneratorTrait;
 use Pimcore\Model\Asset;
 use Pimcore\Model\DataObject\AbstractObject;
 use Pimcore\Model\Document;
@@ -39,6 +39,8 @@
  */
 final readonly class ElementReferenceService implements ElementReferenceServiceInterface
 {
+    use ChunkGeneratorTrait;
+
     private const int REWRITE_REFERENCES_BATCH_SIZE = 500;
 
     public function __construct(
@@ -162,14 +164,4 @@ private function rewriteDataObjectReferences(
         $object->setUserModification($user->getId());
         $object->save();
     }
-
-    // TODO: replace with ChunkGeneratorTrait once https://github.com/pimcore/studio-backend-bundle/pull/1800 is merged
-    private function chunkGenerator(array $items, int $size): Generator
-    {
-        $total = count($items);
-
-        for ($i = 0; $i < $total; $i += $size) {
-            yield array_slice($items, $i, $size);
-        }
-    }
 }

src/ExecutionEngine/Util/Trait/ChunkGeneratorTrait.php

@@ -0,0 +1,33 @@
+<?php
+declare(strict_types=1);
+
+/**
+ * This source file is available under the terms of the
+ * Pimcore Open Core License (POCL)
+ * Full copyright and license information is available in
+ * LICENSE.md which is distributed with this source code.
+ *
+ *  @copyright  Copyright (c) Pimcore GmbH (https://www.pimcore.com)
+ *  @license    Pimcore Open Core License (POCL)
+ */
+
+namespace Pimcore\Bundle\StudioBackendBundle\ExecutionEngine\Util\Trait;
+
+use Generator;
+use function array_slice;
+use function count;
+
+/**
+ * @internal
+ */
+trait ChunkGeneratorTrait
+{
+    private function chunkGenerator(array $items, int $size): Generator
+    {
+        $total = count($items);
+
+        for ($i = 0; $i < $total; $i += $size) {
+            yield array_slice($items, $i, $size);
+        }
+    }
+}

src/Tag/Service/ExecutionEngine/BatchService.php

@@ -13,7 +13,6 @@
 
 namespace Pimcore\Bundle\StudioBackendBundle\Tag\Service\ExecutionEngine;
 
-use Generator;
 use Pimcore\Bundle\GenericExecutionEngineBundle\Agent\JobExecutionAgentInterface;
 use Pimcore\Bundle\GenericExecutionEngineBundle\Model\Job;
 use Pimcore\Bundle\GenericExecutionEngineBundle\Model\JobStep;
@@ -27,6 +26,7 @@
 use Pimcore\Bundle\StudioBackendBundle\ExecutionEngine\Util\EnvironmentVariables;
 use Pimcore\Bundle\StudioBackendBundle\ExecutionEngine\Util\Jobs;
 use Pimcore\Bundle\StudioBackendBundle\ExecutionEngine\Util\StepConfig;
+use Pimcore\Bundle\StudioBackendBundle\ExecutionEngine\Util\Trait\ChunkGeneratorTrait;
 use Pimcore\Bundle\StudioBackendBundle\Security\Service\SecurityServiceInterface;
 use Pimcore\Bundle\StudioBackendBundle\Tag\ExecutionEngine\AutomationAction\Messenger\Messages\BatchTagOperationMessage;
 use Pimcore\Bundle\StudioBackendBundle\Tag\MappedParameter\BatchOperationParameters;
@@ -42,6 +42,7 @@
 final readonly class BatchService implements BatchServiceInterface
 {
     use ElementProviderTrait;
+    use ChunkGeneratorTrait;
 
     private const int BATCH_TAG_SIZE = 500;
 
@@ -115,14 +116,4 @@ private function getJobStepName(string $operation): string
             JobSteps::ELEMENT_BATCH_TAG_ASSIGN->value :
             JobSteps::ELEMENT_BATCH_TAG_REPLACE->value;
     }
-
-    // TODO: replace with ChunkGeneratorTrait once https://github.com/pimcore/studio-backend-bundle/pull/1800 is merged
-    private function chunkGenerator(array $items, int $size): Generator
-    {
-        $total = count($items);
-
-        for ($i = 0; $i < $total; $i += $size) {
-            yield array_slice($items, $i, $size);
-        }
-    }
 }