Add missing generics in yii\db namespace

framework/CHANGELOG.md

@@ -73,6 +73,7 @@ Yii Framework 2 Change Log
 - Bug #20639: Add missing generics in `yii\web` namespace (mspirkov)
 - Bug #20645: Add missing generics in `yii\helpers` and `yii\test` namespaces. Fix PHPDoc annotations in `ArrayAccessTrait` (mspirkov)
 - Bug #20640: Fix `@param` annotation for `$block` in `yii\console\Markdown::renderParagraph()` (mspirkov)
+- Bug #20654: Add missing generics in `yii\db` namespace. Fix PHPDoc annotations in `yii\db\ArrayExpression` (mspirkov)
 - Bug #20651: Add missing generics in `yii\filters` namespace (mspirkov)
 
 

framework/db/ActiveQuery.php

@@ -191,7 +191,10 @@ public function prepare($builder)
                 $this->filterByModels($viaModels);
             } elseif (is_array($this->via)) {
                 // via relation
-                /** @var self $viaQuery */
+                /**
+                 * @var self $viaQuery
+                 * @phpstan-var self<ActiveRecord|array<string, mixed>> $viaQuery
+                 */
                 list($viaName, $viaQuery, $viaCallableUsed) = $this->via;
                 if ($viaQuery->multiple) {
                     if ($viaCallableUsed) {
@@ -458,7 +461,10 @@ public function joinWith($with, $eagerLoading = true, $joinType = 'LEFT JOIN')
                 list(, $relation, $alias) = $matches;
                 $name = $relation;
                 $callback = function ($query) use ($callback, $alias) {
-                    /** @var self $query */
+                    /**
+                     * @var self $query
+                     * @phpstan-var self<ActiveRecord|array<string, mixed>> $query
+                     */
                     $query->alias($alias);
                     if ($callback !== null) {
                         call_user_func($callback, $query);
@@ -648,6 +654,9 @@ protected function getTableNameAndAlias()
      * @param ActiveQuery $parent
      * @param ActiveQuery $child
      * @param string $joinType
+     *
+     * @phpstan-param ActiveQuery<ActiveRecord|array<string, mixed>> $parent
+     * @phpstan-param ActiveQuery<ActiveRecord|array<string, mixed>> $child
      */
     private function joinWithRelation($parent, $child, $joinType)
     {

framework/db/ActiveQueryTrait.php

@@ -149,7 +149,10 @@ public function findWith($with, &$models)
             $primaryModel = $modelClass::instance();
         }
         $relations = $this->normalizeRelations($primaryModel, $with);
-        /** @var ActiveQuery $relation */
+        /**
+         * @var ActiveQuery $relation
+         * @phpstan-var ActiveQuery<ActiveRecord|array<string, mixed>> $relation
+         */
         foreach ($relations as $name => $relation) {
             if ($relation->asArray === null) {
                 // inherit asArray from primary query

framework/db/ActiveRelationTrait.php

@@ -233,13 +233,19 @@ public function populateRelation($name, &$primaryModels)
 
         if ($this->via instanceof self) {
             // via junction table
-            /** @var self $viaQuery */
+            /**
+             * @var self $viaQuery
+             * @phpstan-var self<ActiveRecord|array<string, mixed>> $viaQuery
+             */
             $viaQuery = $this->via;
             $viaModels = $viaQuery->findJunctionRows($primaryModels);
             $this->filterByModels($viaModels);
         } elseif (is_array($this->via)) {
             // via relation
-            /** @var self|ActiveQueryTrait $viaQuery */
+            /**
+             * @var self|ActiveQueryTrait $viaQuery
+             * @phpstan-var self<ActiveRecord|array<string, mixed>>|ActiveQueryTrait $viaQuery
+             */
             list($viaName, $viaQuery) = $this->via;
             if ($viaQuery->asArray === null) {
                 // inherit asArray from primary query
@@ -401,6 +407,8 @@ private function populateInverseRelation(&$primaryModels, $models, $primaryName,
      * @param self|null $viaQuery
      * @param bool $checkMultiple
      * @return array
+     *
+     * @phpstan-param self<ActiveRecord|array<string, mixed>> $viaQuery
      */
     private function buildBuckets($models, $link, $viaModels = null, $viaQuery = null, $checkMultiple = true)
     {

framework/db/ArrayExpression.php

@@ -25,6 +25,9 @@
  * @author Dmytro Naumenko <d.naumenko.a@gmail.com>
  * @since 2.0.14
  * @phpcs:disable Squiz.NamingConventions.ValidVariableName.PrivateNoUnderscore
+ *
+ * @implements \ArrayAccess<array-key, mixed>
+ * @implements \IteratorAggregate<array-key, mixed>
  */
 class ArrayExpression implements ExpressionInterface, \ArrayAccess, \Countable, \IteratorAggregate
 {
@@ -96,7 +99,7 @@ public function getDimension()
      * Whether a offset exists
      *
      * @link https://www.php.net/manual/en/arrayaccess.offsetexists.php
-     * @param mixed $offset <p>
+     * @param int|string $offset <p>
      * An offset to check for.
      * </p>
      * @return bool true on success or false on failure.
@@ -115,7 +118,7 @@ public function offsetExists($offset)
      * Offset to retrieve
      *
      * @link https://www.php.net/manual/en/arrayaccess.offsetget.php
-     * @param mixed $offset <p>
+     * @param int|string $offset <p>
      * The offset to retrieve.
      * </p>
      * @return mixed Can return all value types.
@@ -131,7 +134,7 @@ public function offsetGet($offset)
      * Offset to set
      *
      * @link https://www.php.net/manual/en/arrayaccess.offsetset.php
-     * @param mixed $offset <p>
+     * @param int|string $offset <p>
      * The offset to assign the value to.
      * </p>
      * @param mixed $value <p>
@@ -150,7 +153,7 @@ public function offsetSet($offset, $value)
      * Offset to unset
      *
      * @link https://www.php.net/manual/en/arrayaccess.offsetunset.php
-     * @param mixed $offset <p>
+     * @param int|string $offset <p>
      * The offset to unset.
      * </p>
      * @return void

framework/db/BaseActiveRecord.php

@@ -454,8 +454,13 @@ public function hasMany($class, $link)
      */
     protected function createRelationQuery($class, $link, $multiple)
     {
-        /** @var ActiveRecordInterface $class */
-        /** @var ActiveQuery $query */
+        /**
+         * @var ActiveRecordInterface $class
+         * @var ActiveQuery $query
+         *
+         * @phpstan-var ActiveQuery<ActiveRecord> $query
+         */
+
         $query = $class::find();
         $query->primaryModel = $this;
         $query->link = $link;
@@ -1321,7 +1326,10 @@ public function link($name, $model, $extraColumns = [])
                 throw new InvalidCallException('Unable to link models: the models being linked cannot be newly created.');
             }
             if (is_array($relation->via)) {
-                /** @var ActiveQuery $viaRelation */
+                /**
+                 * @var ActiveQuery $viaRelation
+                 * @phpstan-var ActiveQuery<ActiveRecord|array<string, mixed>> $viaRelation
+                 */
                 list($viaName, $viaRelation) = $relation->via;
                 $viaClass = $viaRelation->modelClass;
                 // unset $viaName so that it can be reloaded to reflect the change
@@ -1414,7 +1422,10 @@ public function unlink($name, $model, $delete = false)
 
         if ($relation->via !== null) {
             if (is_array($relation->via)) {
-                /** @var ActiveQuery $viaRelation */
+                /**
+                 * @var ActiveQuery $viaRelation
+                 * @phpstan-var ActiveQuery<ActiveRecord> $viaRelation
+                 */
                 list($viaName, $viaRelation) = $relation->via;
                 $viaClass = $viaRelation->modelClass;
                 unset($this->_related[$viaName]);
@@ -1517,7 +1528,10 @@ public function unlinkAll($name, $delete = false)
 
         if ($relation->via !== null) {
             if (is_array($relation->via)) {
-                /** @var ActiveQuery $viaRelation */
+                /**
+                 * @var ActiveQuery $viaRelation
+                 * @phpstan-var ActiveQuery<ActiveRecord|array<string, mixed>> $viaRelation
+                 */
                 list($viaName, $viaRelation) = $relation->via;
                 $viaClass = $viaRelation->modelClass;
                 unset($this->_related[$viaName]);

framework/db/BatchQueryResult.php

@@ -27,6 +27,8 @@
  *
  * @author Qiang Xue <qiang.xue@gmail.com>
  * @since 2.0
+ *
+ * @implements \Iterator<int, mixed>
  */
 class BatchQueryResult extends Component implements \Iterator
 {

framework/db/Connection.php

@@ -131,6 +131,9 @@
  *
  * @author Qiang Xue <qiang.xue@gmail.com>
  * @since 2.0
+ *
+ * @phpstan-property-read Schema<ColumnSchema> $schema
+ * @psalm-property-read Schema<ColumnSchema> $schema
  */
 class Connection extends Component
 {
@@ -436,6 +439,8 @@ class Connection extends Component
     private $_transaction;
     /**
      * @var Schema|null the database schema
+     *
+     * @phpstan-var Schema<ColumnSchema>|null
      */
     private $_schema;
     /**
@@ -854,6 +859,9 @@ private function rollbackTransactionOnLevel($transaction, $level)
      * Returns the schema information for the database opened by this connection.
      * @return Schema the schema information for the database opened by this connection.
      * @throws NotSupportedException if there is no support for the current driver type
+     *
+     * @phpstan-return Schema<ColumnSchema>
+     * @psalm-return Schema<ColumnSchema>
      */
     public function getSchema()
     {

framework/db/DataReader.php

@@ -47,6 +47,8 @@
  *
  * @author Qiang Xue <qiang.xue@gmail.com>
  * @since 2.0
+ *
+ * @implements \Iterator<int, mixed>
  */
 class DataReader extends \yii\base\BaseObject implements \Iterator, \Countable
 {

framework/db/QueryBuilder.php

@@ -400,12 +400,15 @@ protected function prepareInsertValues($table, $columns, $params = [])
      * Prepare select-subquery and field names for INSERT INTO ... SELECT SQL statement.
      *
      * @param Query $columns Object, which represents select query.
-     * @param \yii\db\Schema $schema Schema object to quote column name.
+     * @param Schema $schema Schema object to quote column name.
      * @param array $params the parameters to be bound to the generated SQL statement. These parameters will
      * be included in the result with the additional parameters generated during the query building process.
      * @return array array of column names, values and params.
      * @throws InvalidArgumentException if query's select does not contain named parameters only.
      * @since 2.0.11
+     *
+     * @phpstan-param Schema<ColumnSchema> $schema
+     * @psalm-param Schema<ColumnSchema> $schema
      */
     protected function prepareInsertSelectSubQuery($columns, $schema, $params = [])
     {

framework/db/SqlToken.php

@@ -19,6 +19,8 @@
  *
  * @author Sergey Makinen <sergey@makinen.ru>
  * @since 2.0.13
+ *
+ * @implements \ArrayAccess<int, SqlToken>
  */
 class SqlToken extends BaseObject implements \ArrayAccess
 {

framework/db/SqlTokenizer.php

@@ -46,6 +46,8 @@ abstract class SqlTokenizer extends Component
 
     /**
      * @var \SplStack stack of active tokens.
+     *
+     * @phpstan-var \SplStack<SqlToken>
      */
     private $_tokenStack;
     /**

framework/db/cubrid/QueryBuilder.php

@@ -9,6 +9,7 @@
 
 use yii\base\InvalidArgumentException;
 use yii\base\NotSupportedException;
+use yii\db\ColumnSchema;
 use yii\db\Exception;
 use yii\db\Expression;
 
@@ -183,7 +184,10 @@ public function selectExists($rawSql)
      */
     public function dropIndex($name, $table)
     {
-        /** @var Schema $schema */
+        /**
+         * @var Schema $schema
+         * @phpstan-var Schema<ColumnSchema>
+         */
         $schema = $this->db->getSchema();
         foreach ($schema->getTableUniques($table) as $unique) {
             if ($unique->name === $name) {

framework/db/mssql/QueryBuilder.php

@@ -280,10 +280,15 @@ public function resetSequence($tableName, $value = null)
      */
     public function checkIntegrity($check = true, $schema = '', $table = '')
     {
+        /**
+         * @var Schema
+         * @phpstan-var Schema<ColumnSchema>
+         */
+        $dbSchema = $this->db->getSchema();
         $enable = $check ? 'CHECK' : 'NOCHECK';
-        $schema = $schema ?: $this->db->getSchema()->defaultSchema;
-        $tableNames = $this->db->getTableSchema($table) ? [$table] : $this->db->getSchema()->getTableNames($schema);
-        $viewNames = $this->db->getSchema()->getViewNames($schema);
+        $schema = $schema ?: $dbSchema->defaultSchema;
+        $tableNames = $this->db->getTableSchema($table) ? [$table] : $dbSchema->getTableNames($schema);
+        $viewNames = $dbSchema->getViewNames($schema);
         $tableNames = array_diff($tableNames, $viewNames);
         $command = '';
 

framework/db/pgsql/QueryBuilder.php

@@ -208,10 +208,15 @@ public function resetSequence($tableName, $value = null)
      */
     public function checkIntegrity($check = true, $schema = '', $table = '')
     {
+        /**
+         * @var Schema
+         * @phpstan-var Schema<ColumnSchema>
+         */
+        $dbSchema = $this->db->getSchema();
         $enable = $check ? 'ENABLE' : 'DISABLE';
-        $schema = $schema ?: $this->db->getSchema()->defaultSchema;
-        $tableNames = $table ? [$table] : $this->db->getSchema()->getTableNames($schema);
-        $viewNames = $this->db->getSchema()->getViewNames($schema);
+        $schema = $schema ?: $dbSchema->defaultSchema;
+        $tableNames = $table ? [$table] : $dbSchema->getTableNames($schema);
+        $viewNames = $dbSchema->getViewNames($schema);
         $tableNames = array_diff($tableNames, $viewNames);
         $command = '';
 
@@ -376,7 +381,10 @@ private function oldUpsert($table, $insertColumns, $updateColumns, &$params)
             $updateColumns = false;
         }
 
-        /** @var Schema $schema */
+        /**
+         * @var Schema $schema
+         * @phpstan-var Schema<ColumnSchema>
+         */
         $schema = $this->db->getSchema();
         if (!$insertColumns instanceof Query) {
             $tableSchema = $schema->getTableSchema($table);

phpstan-baseline.neon

@@ -55,11 +55,6 @@ parameters:
 			count: 1
 			path: framework/grid/DataColumn.php
 
-		-
-			message: "#^Call to an undefined method yii\\\\db\\\\Schema\\:\\:getViewNames\\(\\)\\.$#"
-			count: 1
-			path: framework/db/pgsql/QueryBuilder.php
-
 		-
 			message: "#^Call to an undefined method yii\\\\db\\\\ExpressionInterface\\:\\:getValue\\(\\)\\.$#"
 			count: 1
@@ -85,11 +80,6 @@ parameters:
 			count: 1
 			path: framework/db/mysql/JsonExpressionBuilder.php
 
-		-
-			message: "#^Call to an undefined method yii\\\\db\\\\Schema\\:\\:getViewNames\\(\\)\\.$#"
-			count: 1
-			path: framework/db/mssql/QueryBuilder.php
-
 		-
 			message: "#^Access to an undefined property yii\\\\db\\\\ColumnSchema\\:\\:\\$isComputed\\.$#"
 			count: 1
@@ -401,7 +391,7 @@ parameters:
 			path: framework/db/BaseActiveRecord.php
 
 		-
-			message: "#^Property yii\\\\db\\\\ActiveQuery\\<array\\|yii\\\\db\\\\ActiveRecord\\>\\:\\:\\$primaryModel \\(yii\\\\db\\\\ActiveRecord\\) does not accept \\$this\\(yii\\\\db\\\\BaseActiveRecord\\)\\.$#"
+			message: "#^Property yii\\\\db\\\\ActiveQuery\\<yii\\\\db\\\\ActiveRecord\\>\\:\\:\\$primaryModel \\(yii\\\\db\\\\ActiveRecord\\) does not accept \\$this\\(yii\\\\db\\\\BaseActiveRecord\\)\\.$#"
 			count: 1
 			path: framework/db/BaseActiveRecord.php