Update MySqlConnection.php

octobercms · May 13, 2024 · 2cc360d · 2cc360d
1 parent 52e67e1
commit 2cc360d
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/Database/Connections/MySqlConnection.php b/src/Database/Connections/MySqlConnection.php
@@ -31,7 +31,12 @@ public function isMaria()
      */
     protected function getDefaultQueryGrammar()
     {
-        return $this->withTablePrefix(new QueryGrammar);
+        $grammar = new QueryGrammar;
+        if (method_exists($grammar, 'setConnection')) {
+            $grammar->setConnection($this);
+        }
+
+        return $this->withTablePrefix($grammar);
     }
 
     /**
@@ -55,7 +60,12 @@ public function getSchemaBuilder()
      */
     protected function getDefaultSchemaGrammar()
     {
-        return $this->withTablePrefix(new SchemaGrammar);
+        $grammar = new SchemaGrammar;
+        if (method_exists($grammar, 'setConnection')) {
+            $grammar->setConnection($this);
+        }
+
+        return $this->withTablePrefix($grammar);
     }
 
     /**