Github user jburwell commented on a diff in the pull request:
https://github.com/apache/cloudstack/pull/1489#discussion_r60732671
--- Diff: engine/schema/src/com/cloud/upgrade/dao/Upgrade481to490.java ---
@@ -53,6 +62,139 @@ public boolean supportsRollingUpgrade() {
@Override
public void performDataMigration(Connection conn) {
+ setupRolesAndPermissionsForDynamicRBAC(conn);
+ }
+
+ private void createDefaultRole(final Connection conn, final Long id,
final String name, final RoleType roleType) {
+ final String insertSql = String.format("INSERT INTO
`cloud`.`roles` (`id`, `uuid`, `name`, `role_type`, `description`) values (%d,
UUID(), '%s', '%s', 'Default %s role');",
+ id, name, roleType.name(), roleType.name().toLowerCase());
+ try ( PreparedStatement updatePstmt =
conn.prepareStatement(insertSql) ) {
+ updatePstmt.executeUpdate();
+ } catch (SQLException e) {
+ throw new CloudRuntimeException("Unable to create default role
with id: " + id + " name: " + name, e);
+ }
+ }
+
+ private void createRoleMapping(final Connection conn, final Long
roleId, final String apiName) {
+ final String insertSql = String.format("INSERT INTO
`cloud`.`role_permissions` (`uuid`, `role_id`, `rule`, `permission`) values
(UUID(), %d, '%s', 'ALLOW') ON DUPLICATE KEY UPDATE rule=rule;",
+ roleId, apiName);
+ try ( PreparedStatement updatePstmt =
conn.prepareStatement(insertSql)) {
+ updatePstmt.executeUpdate();
+ } catch (SQLException ignored) {
+ s_logger.debug("Unable to insert mapping for role id:" +
roleId + " apiName: " + apiName);
+ }
+ }
+
+ private void addRoleColumnAndMigrateAccountTable(final Connection
conn, final RoleType[] roleTypes) {
+ final String alterTableSql = "ALTER TABLE `cloud`.`account` ADD
COLUMN `role_id` bigint(20) unsigned COMMENT 'role id for this account' AFTER
`type`, " +
+ "ADD KEY `fk_account__role_id` (`role_id`), " +
+ "ADD CONSTRAINT `fk_account__role_id` FOREIGN KEY
(`role_id`) REFERENCES `roles` (`id`);";
+ try (PreparedStatement pstmt =
conn.prepareStatement(alterTableSql)) {
+ pstmt.executeUpdate();
+ s_logger.info("Altered cloud.account table and added column
role_id");
+ } catch (SQLException e) {
+ if (e.getMessage().contains("role_id")) {
+ s_logger.warn("cloud.account table already has the role_id
column, skipping altering table and migration of accounts");
+ return;
+ } else {
+ throw new CloudRuntimeException("Unable to create column
quota_calculated in table cloud_usage.cloud_usage", e);
+ }
+ }
+ migrateAccountsToDefaultRoles(conn, roleTypes);
+ }
+
+ private void migrateAccountsToDefaultRoles(final Connection conn,
final RoleType[] roleTypes) {
+ try (PreparedStatement selectStatement =
conn.prepareStatement("SELECT `id`, `type` FROM `cloud`.`account`;");
+ ResultSet selectResultSet = selectStatement.executeQuery()) {
+ while (selectResultSet.next()) {
+ Long accountId = selectResultSet.getLong(1);
+ Short accountType = selectResultSet.getShort(2);
+ Long roleId = null;
+ for (RoleType roleType : roleTypes) {
+ if (roleType.getAccountType() == accountType) {
+ roleId = roleType.getId();
+ break;
+ }
+ }
+ if (roleId == null) {
+ continue;
+ }
+ try (PreparedStatement updateStatement =
conn.prepareStatement("UPDATE `cloud`.`account` SET role_id = ? WHERE id =
?;")) {
+ updateStatement.setLong(1, roleId);
+ updateStatement.setLong(2, accountId);
+ updateStatement.executeUpdate();
+ updateStatement.close();
+
+ } catch (SQLException e) {
+ s_logger.error("Failed to update cloud.account role_id
for account id:" + accountId + " with exception: " + e.getMessage());
+ throw new CloudRuntimeException("Exception while
updating cloud.account role_id", e);
+ }
+ }
+ } catch (SQLException e) {
+ throw new CloudRuntimeException("Exception while migrating
existing account table's role_id column to a role based on account type", e);
+ }
+ s_logger.debug("Done migrating existing accounts to use one of
default roles based on account type");
+ }
+
+ private void setupRolesAndPermissionsForDynamicRBAC(final Connection
conn) {
+ try ( PreparedStatement selectStatement =
conn.prepareStatement("SELECT * FROM `cloud`.`roles`");
+ ResultSet resultSet = selectStatement.executeQuery()) {
--- End diff --
@bhaisaab if ``resultSet`` is not declared as part of the enclosing try
with resources block, it won't be automatically closed.
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at infrastruct...@apache.org or file a JIRA ticket
with INFRA.
---
