Adopt for last changes

src/AbstractQueryDataReader.php

@@ -30,9 +30,9 @@
 use Yiisoft\Data\Reader\FilterInterface;
 use Yiisoft\Data\Reader\Sort;
 use Yiisoft\Db\Query\QueryInterface;
-
 use function array_key_first;
 use function is_array;
+use function is_object;
 use function sprintf;
 
 /**
@@ -108,7 +108,7 @@ public function getIterator(): Generator
             $iterator = $this->getPreparedQuery()->each($this->batchSize);
 
             foreach ($iterator as $index => $row) {
-                yield $index => $row;
+                yield $index => $this->createItem($row);
             }
         }
     }
@@ -158,8 +158,16 @@ public function getPreparedQuery(): QueryInterface
         return $query;
     }
 
-    protected function getHandlerByOperation(string $operation): QueryHandlerInterface
+    /**
+     * @param string|FilterInterface $operation
+     * @return QueryHandlerInterface
+     */
+    public function getHandlerByOperation(string|FilterInterface $operation): QueryHandlerInterface
     {
+        if (is_object($operation)) {
+            $operation = $operation::getOperator();
+        }
+
         if (!isset($this->filterHandlers[$operation])) {
             throw new RuntimeException(sprintf('Operation "%s" is not supported', $operation));
         }
@@ -170,7 +178,7 @@ protected function getHandlerByOperation(string $operation): QueryHandlerInterfa
     protected function applyFilter(QueryInterface $query): QueryInterface
     {
         if ($this->filter !== null) {
-            $query = $this->getHandlerByOperation($this->filter::getOperator())
+            $query = $this->getHandlerByOperation($this->filter)
                 ->applyFilter($query, $this->filter);
         }
 
@@ -180,7 +188,7 @@ protected function applyFilter(QueryInterface $query): QueryInterface
     protected function applyHaving(QueryInterface $query): QueryInterface
     {
         if ($this->having !== null) {
-            $query = $this->getHandlerByOperation($this->having::getOperator())
+            $query = $this->getHandlerByOperation($this->having)
                 ->applyHaving($query, $this->having);
         }
 
@@ -366,11 +374,13 @@ public function readOne(): array|object|null
             return $key === null ? null : $this->data[$key];
         }
 
-        return $this->withLimit(1)->getIterator()->current();
+        $current = $this->withLimit(1)->getIterator()->current();
+
+        return $current === null ? null : $this->createItem($current);
     }
 
     /**
      * @psalm-return TValue
      */
-    abstract protected function createItem(array $row): array|object;
+    abstract protected function createItem(array|object $row): array|object;
 }

src/QueryDataReader.php

@@ -14,7 +14,7 @@
  */
 final class QueryDataReader extends AbstractQueryDataReader
 {
-    protected function createItem(array $row): array
+    protected function createItem(array|object $row): array|object
     {
         /** @psalm-var TValue */
         return $row;

tests/DataReaderTest.php

@@ -137,4 +137,17 @@ public function testSort(Sort $sort, string $expected): void
             $dataReader->getPreparedQuery()->createCommand()->getRawSql()
         );
     }
+
+    /*
+    public function testCreateItem(): void
+    {
+        $db = $this->getConnection(true);
+
+        $query = (new Query($db))
+            ->from('customer');
+
+        $dataReader = new CustomerDataReader($query);
+
+        $this->assertInstanceOf(CustomerDTO::class, $dataReader->readOne());
+    } */
 }

tests/Support/CustomerDTO.php

@@ -0,0 +1,9 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Yiisoft\Data\Db\Tests\Support;
+
+final class CustomerDTO
+{
+}

tests/Support/CustomerDataReader.php

@@ -0,0 +1,19 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Yiisoft\Data\Db\Tests\Support;
+
+use Yiisoft\Data\Db\AbstractQueryDataReader;
+
+final class CustomerDataReader extends AbstractQueryDataReader
+{
+
+    /**
+     * @inheritDoc
+     */
+    protected function createItem(object|array $row): array|object
+    {
+        return new CustomerDTO();
+    }
+}