doctrine-dbal: implement Connection->iterateColumn() (#210)

Co-authored-by: Markus Staab <[email protected]>

Author: staabm

src/DoctrineReflection/DoctrineReflection.php

@@ -30,6 +30,7 @@ public function fetchResultType(MethodReflection $methodReflection, Type $result
                 $fetchType = QueryReflector::FETCH_TYPE_ONE;
                 break;
             case 'fetchfirstcolumn':
+            case 'iteratecolumn':
                 $fetchType = QueryReflector::FETCH_TYPE_FIRST_COL;
                 break;
             case 'fetchnumeric':
@@ -57,6 +58,10 @@ public function fetchResultType(MethodReflection $methodReflection, Type $result
                     return $valueTypes[$i];
                 }
                 if (QueryReflector::FETCH_TYPE_FIRST_COL === $fetchType) {
+                    if (\in_array($usedMethod, ['iteratecolumn'], true)) {
+                        return new GenericObjectType(Traversable::class, [new IntegerType(), $valueTypes[$i]]);
+                    }
+
                     return new ArrayType(IntegerRangeType::fromInterval(0, null), $valueTypes[$i]);
                 }
 

src/Extensions/DoctrineConnectionFetchDynamicReturnTypeExtension.php

@@ -28,7 +28,7 @@ public function getClass(): string
 
     public function isMethodSupported(MethodReflection $methodReflection): bool
     {
-        return \in_array(strtolower($methodReflection->getName()), ['fetchone', 'fetchfirstcolumn', 'fetchassociative', 'fetchallassociative', 'fetchnumeric', 'fetchallnumeric', 'iterateassociative', 'iteratenumeric', 'iterateallnumeric'], true);
+        return \in_array(strtolower($methodReflection->getName()), ['fetchone', 'fetchfirstcolumn', 'fetchassociative', 'fetchallassociative', 'fetchnumeric', 'fetchallnumeric', 'iteratecolumn', 'iterateassociative', 'iteratenumeric', 'iterateallnumeric'], true);
     }
 
     public function getTypeFromMethodCall(MethodReflection $methodReflection, MethodCall $methodCall, Scope $scope): Type

tests/data/doctrine-dbal.php

@@ -83,6 +83,13 @@ public function iterateNumeric(Connection $conn)
         assertType('Traversable<int, array{string, int<0, 4294967295>, int<-128, 127>, int<-128, 127>}>', $fetchResult);
     }
 
+    public function iterateColumn(Connection $conn)
+    {
+        $query = 'SELECT email, adaid, gesperrt, freigabe1u1 FROM ada WHERE adaid = ?';
+        $fetchResult = $conn->iterateColumn($query, [1]);
+        assertType('Traversable<int, string>', $fetchResult);
+    }
+
     public function fetchOne(Connection $conn)
     {
         $query = 'SELECT email, adaid, gesperrt, freigabe1u1 FROM ada WHERE adaid = ?';