wip

Author: brendt

packages/database/src/Migrations/MigrationManager.php

@@ -109,19 +109,19 @@ public function dropAll(): void
             $tables = $this->getTableDefinitions();
 
             // Disable foreign key checks
-            new SetForeignKeyChecksStatement(enable: false)->execute($this->dialect);
+            new SetForeignKeyChecksStatement(enable: false)->execute($this->dialect, $this->onDatabase);
 
             // Drop each table
             foreach ($tables as $table) {
-                new DropTableStatement($table->name)->execute($this->dialect);
+                new DropTableStatement($table->name)->execute($this->dialect, $this->onDatabase);
 
                 event(new TableDropped($table->name));
             }
         } catch (Throwable $throwable) {
             event(new FreshMigrationFailed($throwable));
         } finally {
             // Enable foreign key checks
-            new SetForeignKeyChecksStatement(enable: true)->execute($this->dialect);
+            new SetForeignKeyChecksStatement(enable: true)->execute($this->dialect, $this->onDatabase);
         }
     }
 
@@ -249,15 +249,15 @@ public function executeDown(MigrationInterface $migration): void
             // TODO: don't just disable FK checking when executing down
 
             // Disable foreign key checks
-            new SetForeignKeyChecksStatement(enable: false)->execute($this->dialect);
+            new SetForeignKeyChecksStatement(enable: false)->execute($this->dialect, $this->onDatabase);
 
             $this->database->execute($query);
 
             // Disable foreign key checks
-            new SetForeignKeyChecksStatement(enable: true)->execute($this->dialect);
+            new SetForeignKeyChecksStatement(enable: true)->execute($this->dialect, $this->onDatabase);
         } catch (PDOException $pdoException) {
             // Disable foreign key checks
-            new SetForeignKeyChecksStatement(enable: true)->execute($this->dialect);
+            new SetForeignKeyChecksStatement(enable: true)->execute($this->dialect, $this->onDatabase);
 
             event(new MigrationFailed($migration->name, $pdoException));
 

packages/database/src/QueryStatements/CanExecuteStatement.php

@@ -7,17 +7,18 @@
 use Tempest\Database\Config\DatabaseDialect;
 use Tempest\Database\Id;
 use Tempest\Database\Query;
+use UnitEnum;
 
 trait CanExecuteStatement
 {
-    public function execute(DatabaseDialect $dialect): ?Id
+    public function execute(DatabaseDialect $dialect, null|string|UnitEnum $onDatabase): ?Id
     {
         $sql = $this->compile($dialect);
 
         if (! $sql) {
             return null;
         }
 
-        return new Query($sql)->execute();
+        return new Query($sql)->onDatabase($onDatabase)->execute();
     }
 }