1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192 |
- /*
- * SonarQube
- * Copyright (C) 2009-2023 SonarSource SA
- * mailto:info AT sonarsource DOT com
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 3 of the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public License
- * along with this program; if not, write to the Free Software Foundation,
- * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
- */
- package org.sonar.server.platform.db.migration.sql;
-
- import java.util.Arrays;
- import java.util.Collections;
- import java.util.Iterator;
- import java.util.List;
- import org.sonar.db.dialect.Dialect;
- import org.sonar.db.dialect.H2;
- import org.sonar.db.dialect.MsSql;
- import org.sonar.db.dialect.Oracle;
- import org.sonar.db.dialect.PostgreSql;
-
- /**
- * Generate a SQL query to drop multiple columns from a table
- */
- public class DropColumnsBuilder {
-
- private static final String ALTER_TABLE = "ALTER TABLE ";
-
- private final Dialect dialect;
- private final String tableName;
- private final String[] columns;
-
- public DropColumnsBuilder(Dialect dialect, String tableName, String... columns) {
- this.tableName = tableName;
- this.dialect = dialect;
- this.columns = columns;
- }
-
- public List<String> build() {
- switch (dialect.getId()) {
- case PostgreSql.ID:
- StringBuilder sql = new StringBuilder().append(ALTER_TABLE).append(tableName).append(" ");
- dropColumns(sql, "DROP COLUMN ", columns);
- return Collections.singletonList(sql.toString());
- case MsSql.ID:
- return Collections.singletonList(getMsSQLStatement(columns));
- case Oracle.ID:
- return Collections.singletonList(getOracleStatement());
- case H2.ID:
- return Arrays.stream(columns).map(this::getMsSQLStatement).toList();
- default:
- throw new IllegalStateException(String.format("Unsupported database '%s'", dialect.getId()));
- }
- }
-
- private String getOracleStatement() {
- StringBuilder sql = new StringBuilder().append(ALTER_TABLE).append(tableName).append(" ");
- sql.append("SET UNUSED (");
- dropColumns(sql, "", columns);
- sql.append(")");
- return sql.toString();
- }
-
- private String getMsSQLStatement(String... columnNames) {
- StringBuilder sql = new StringBuilder().append(ALTER_TABLE).append(tableName).append(" ");
- sql.append("DROP COLUMN ");
- dropColumns(sql, "", columnNames);
- return sql.toString();
- }
-
- private static void dropColumns(StringBuilder sql, String columnPrefix, String... columnNames) {
- Iterator<String> columnNamesIterator = Arrays.stream(columnNames).iterator();
- while (columnNamesIterator.hasNext()) {
- sql.append(columnPrefix);
- sql.append(columnNamesIterator.next());
- if (columnNamesIterator.hasNext()) {
- sql.append(", ");
- }
- }
- }
-
- }
|