From eedfbb900ed071566784a3859d7c8b47c34a4f13 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=90=91=E6=9E=97?= <as2252258@163.com>
Date: Thu, 20 Apr 2023 23:16:36 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8F=98=E6=9B=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Base/Model.php     | 1 +
 Traits/HasBase.php | 7 +++++--
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/Base/Model.php b/Base/Model.php
index e7173db..f4f6038 100644
--- a/Base/Model.php
+++ b/Base/Model.php
@@ -17,6 +17,7 @@ use ArrayAccess;
 use Database\ActiveQuery;
 use Database\Collection;
 use Database\Connection;
+use Database\ISqlBuilder;
 use Database\ModelInterface;
 use Database\Mysql\Columns;
 use Database\Relation;
diff --git a/Traits/HasBase.php b/Traits/HasBase.php
index 5166e9c..ddffd9a 100644
--- a/Traits/HasBase.php
+++ b/Traits/HasBase.php
@@ -46,16 +46,19 @@ abstract class HasBase implements \Database\Traits\Relation
 	public function __construct(public string $name)
 	{
 	}
-	
+
 	/**
 	 * @param $name
 	 * @param $arguments
 	 * @return static
+	 * @throws \ReflectionException
 	 */
 	public function __call($name, $arguments)
 	{
 		if ($name !== 'get') {
-			return di(Relation::class)->getQuery($this->name)->$name(...$arguments);
+			$relation = Kiri::getDi()->get(Relation::class);
+			$relation->getQuery($this->name)->$name(...$arguments);
+			return $this;
 		} else {
 			return $this->get();
 		}