From 581eba2725a5c651ca1238722ede9a833cc7e445 Mon Sep 17 00:00:00 2001
From: Ivan Bragin <aywan@aywan.ru>
Date: Wed, 24 Aug 2022 21:18:29 +0300
Subject: [PATCH] Adding `$db` into `one` and `all` methods of *Query stubs

---
 stubs/ActiveQuery.phpstub          | 4 ++--
 stubs/ActiveQueryInterface.phpstub | 6 +++---
 stubs/Query.phpstub                | 4 ++--
 3 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/stubs/ActiveQuery.phpstub b/stubs/ActiveQuery.phpstub
index 1f67f9b..f31a7c3 100644
--- a/stubs/ActiveQuery.phpstub
+++ b/stubs/ActiveQuery.phpstub
@@ -17,8 +17,8 @@ namespace yii\db;
  * @property bool $asArray
  * @property bool $multiple
  *
- * @method TModel|null one()
- * @method array<mixed, TModel> all()
+ * @method TModel|null one($db = null)
+ * @method array<mixed, TModel> all($db = null)
  *
  * @extends Query<TModel, TMultiple>
  */
diff --git a/stubs/ActiveQueryInterface.phpstub b/stubs/ActiveQueryInterface.phpstub
index f8f71db..e1b1138 100644
--- a/stubs/ActiveQueryInterface.phpstub
+++ b/stubs/ActiveQueryInterface.phpstub
@@ -13,9 +13,9 @@ namespace yii\db;
 /**
  * @psalm-template TModel
  *
- * @method TModel|null one()
- * @method array<mixed, TModel> all()
+ * @method TModel|null one($db = null)
+ * @method array<mixed, TModel> all($db = null)
  *
  * @extends QueryInterface<TModel>
  */
-class ActiveQueryInterface extends QueryInterface {}
\ No newline at end of file
+class ActiveQueryInterface extends QueryInterface {}
diff --git a/stubs/Query.phpstub b/stubs/Query.phpstub
index ed99b31..f3f2117 100644
--- a/stubs/Query.phpstub
+++ b/stubs/Query.phpstub
@@ -14,7 +14,7 @@ namespace yii\db;
  * @template TModel
  * @template TMultiple
  * @method iterable<mixed, TModel> each(int $batchSize = 100, ?Connection $db = null)
- * @method TModel|null one()
- * @method array<mixed, TModel> all()
+ * @method TModel|null one($db = null)
+ * @method array<mixed, TModel> all($db = null)
  */
 class Query {}