Fix use warehouse

keboola · Nov 25, 2024 · 3c0fcb4 · 3c0fcb4
1 parent 417961b
commit 3c0fcb4
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/MigrateStructure.php b/src/MigrateStructure.php
@@ -747,7 +747,14 @@ private function copyFunctions(string $database, ProjectRoles $projectRoles): vo
                 ->getRole($ownershipRole->getGrantedBy())
                 ->getAssignedGrants()
                 ->getWarehouseGrants();
-            $this->destinationConnection->useWarehouse(current($warehouseGrants)->getName());
+
+            $warehouseName = current($warehouseGrants)->getName();
+            try {
+                $this->destinationConnection->useWarehouse($warehouseName);
+            } catch (Throwable $e) {
+                $warehouseName = str_ireplace(['_SMALL', '_MEDIUM', '_LARGE'], '', $warehouseName);
+                $this->destinationConnection->useWarehouse($warehouseName);
+            }
 
             $this->destinationConnection->useRole($ownershipRole->getGrantedBy());