grant role if not granted

keboola · Nov 5, 2024 · e6248b7 · e6248b7
1 parent 6c0f2d3
commit e6248b7
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/src/Cleanup.php b/src/Cleanup.php
@@ -157,13 +157,20 @@ public function preMigration(string $mainRoleName): void
                 }
                 $this->destinationConnection->useRole($mainRoleName);
                 if ($currentRole === $mainRoleName && !$mainRoleExistsOnTargetUser) {
-                    $this->destinationConnection->query(sprintf(
-                        'GRANT ROLE %s TO USER %s;',
-                        Helper::quoteIdentifier($currentRole),
-                        Helper::quoteIdentifier($this->config->getTargetSnowflakeUser())
-                    ));
+                    $this->destinationConnection->grantRoleToUser(
+                        $this->config->getTargetSnowflakeUser(),
+                        $currentRole,
+                    );
+                }
+                try {
+                    $this->destinationConnection->useRole($role['granted_by']);
+                } catch (RuntimeException $e) {
+                    $this->destinationConnection->grantRoleToUser(
+                        $this->config->getTargetSnowflakeUser(),
+                        $role['granted_by'],
+                    );
+                    $this->destinationConnection->useRole($role['granted_by']);
                 }
-                $this->destinationConnection->useRole($role['granted_by']);
 
                 /** @var FutureGrantToRole[] $futureGrants */
                 $futureGrants = array_map(