feat: support table prefixing

CHANGELOG.md

@@ -8,6 +8,7 @@ Added
 - Allow pretending for DDL statements (#170) 
 - Allow `spanner_emulator.disable_query_null_filtered_index_check` to be set (#180)
 - Allow default max transaction attempts to be changed (#179)
+- Table prefixing support (#172)
 
 Changed
 - `Query\Builder::lock()` no longer throw an error and will be ignored instead (#156)

src/Connection.php

@@ -181,7 +181,10 @@ public function disconnect()
      */
     protected function getDefaultQueryGrammar(): QueryGrammar
     {
-        return (new QueryGrammar())->setConnection($this);
+        $grammar = new QueryGrammar();
+        $grammar->setConnection($this);
+        $this->withTablePrefix($grammar);
+        return $grammar;
     }
 
     /**
@@ -190,7 +193,10 @@ protected function getDefaultQueryGrammar(): QueryGrammar
      */
     protected function getDefaultSchemaGrammar(): SchemaGrammar
     {
-        return new SchemaGrammar();
+        $grammar = new SchemaGrammar();
+        $grammar->setConnection($this);
+        $this->withTablePrefix($grammar);
+        return $grammar;
     }
 
     /**

tests/ConnectionTest.php

@@ -555,4 +555,26 @@ public function test_escape_nested_array(): void
         $conn = $this->getDefaultConnection();
         self::assertSame('[]', $conn->escape([[]]));
     }
+
+    public function test_getTablePrefix(): void
+    {
+        config()->set('database.connections.main.prefix', 'test_');
+        $tablePrefix = $this->getConnection('main')->getTablePrefix();
+        self::assertSame('test_', $tablePrefix);
+    }
+
+    public function test_getQueryGrammar(): void
+    {
+        config()->set('database.connections.main.prefix', 'test_');
+        $conn = $this->getConnection('main');
+        self::assertSame('test_', $conn->getQueryGrammar()->getTablePrefix());
+    }
+
+    public function test_getSchemaGrammar(): void
+    {
+        config()->set('database.connections.main.prefix', 'test_');
+        $conn = $this->getConnection('main');
+        $conn->useDefaultSchemaGrammar();
+        self::assertSame('test_', $conn->getSchemaGrammar()->getTablePrefix());
+    }
 }

tests/Query/BuilderTest.php

@@ -916,6 +916,13 @@ public function test_sharedLock(): void
         $this->assertSame('select * from `User`', $sql);
     }
 
+    public function test_prefixing(): void
+    {
+        config()->set('database.connections.main.prefix', 'test_');
+        $conn = $this->getConnection('main');
+        self::assertSame('select * from `test_User`', $conn->table('User')->toRawSql());
+    }
+
     public function test_toRawSql(): void
     {
         $table = 'RawSqlTest';

tests/Schema/BuilderTestLast.php

@@ -52,6 +52,21 @@ public function testSchemaCreate(): void
         $this->assertCount(0, collect(['id', 'name', 'age', 'created_at'])->diff($columnNames));
     }
 
+    public function test_create_with_prefix(): void
+    {
+        config()->set('database.connections.main.prefix', 'test_');
+        $conn = $this->getConnection('main');
+
+        $sb = $conn->getSchemaBuilder();
+        $table = $this->generateTableName(class_basename(__CLASS__));
+
+        $sb->create($table, function (Blueprint $table) {
+            $table->uuid('id')->primary();
+        });
+        $tables = array_map(static fn (array $row) => $row['name'], $sb->getTables());
+        self::assertContains('test_' . $table, $tables);
+    }
+
     public function testSchemaDrop(): void
     {
         $conn = $this->getDefaultConnection();