Rewrite the library

composer.json

@@ -14,6 +14,7 @@
         "doctrine/collections": "^1.6",
         "doctrine/orm": "^2.6",
         "doctrine/persistence": "^1.3 || ^2.1",
+        "setono/doctrine-object-manager-trait": "^1.0",
         "symfony/property-access": "^4.4 || ^5.2",
         "webmozart/assert": "^1.10"
     },

psalm.xml

@@ -1,7 +1,8 @@
 <?xml version="1.0"?>
 <psalm
     xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
-    errorLevel="4"
+    errorLevel="1"
+    totallyTyped="true"
     xmlns="https://getpsalm.org/schema/config"
     xsi:schemaLocation="https://getpsalm.org/schema/config vendor/vimeo/psalm/config.xsd"
 >

src/Batch/Batch.php

@@ -4,29 +4,46 @@
 
 namespace Setono\DoctrineORMBatcher\Batch;
 
+use Doctrine\ORM\Query\Parameter;
 use Doctrine\ORM\QueryBuilder;
 use InvalidArgumentException;
 
-abstract class Batch implements BatchInterface
+class Batch implements BatchInterface
 {
+    private array $data;
+
+    /** @var class-string */
     protected string $class;
 
     protected string $dql;
 
+    /**
+     * @var array<int, Parameter>
+     */
     protected array $parameters;
 
-    public function __construct(QueryBuilder $qb)
+    public function __construct(array $data, QueryBuilder $qb)
     {
+        /** @var array<array-key, class-string> $rootEntities */
         $rootEntities = $qb->getRootEntities();
         if (0 === count($rootEntities)) {
             throw new InvalidArgumentException('The number of root entities on the query builder must be one or more');
         }
 
+        $this->data = $data;
         $this->class = $rootEntities[0];
         $this->dql = $qb->getDQL();
         $this->parameters = $qb->getParameters()->toArray();
     }
 
+    public function getData(): array
+    {
+        return $this->data;
+    }
+
+    /**
+     * @return class-string
+     */
     public function getClass(): string
     {
         return $this->class;
@@ -37,6 +54,9 @@ public function getDql(): string
         return $this->dql;
     }
 
+    /**
+     * @return array<int, Parameter>
+     */
     public function getParameters(): array
     {
         return $this->parameters;

src/Batch/BatchInterface.php

@@ -6,6 +6,8 @@
 
 interface BatchInterface
 {
+    public function getData(): array;
+
     /**
      * This is the root entity that can be used to get the entity manager from the manager registry.
      */