source.dussan.org Git - sonarqube.git/blob

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2020 SonarSource SA
   4  * mailto:info AT sonarsource DOT com
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  */
  20 package org.sonar.server.platform.db.migration.sql;
  21
  22 import java.util.List;
  23 import org.sonar.db.dialect.Dialect;
  24 import org.sonar.db.dialect.H2;
  25 import org.sonar.db.dialect.MsSql;
  26 import org.sonar.db.dialect.Oracle;
  27 import org.sonar.db.dialect.PostgreSql;
  28
  29 import static com.google.common.base.Preconditions.checkArgument;
  30 import static java.util.Arrays.asList;
  31 import static java.util.Collections.singletonList;
  32 import static org.sonar.server.platform.db.migration.def.Validations.validateTableName;
  33
  34 public class RenameTableBuilder {
  35
  36   private final Dialect dialect;
  37   private String name;
  38   private String newName;
  39   private boolean autoGeneratedId = true;
  40
  41   public RenameTableBuilder(Dialect dialect) {
  42     this.dialect = dialect;
  43   }
  44
  45   public RenameTableBuilder setName(String s) {
  46     this.name = s;
  47     return this;
  48   }
  49
  50   public RenameTableBuilder setNewName(String s) {
  51     this.newName = s;
  52     return this;
  53   }
  54
  55   /**
  56    * When a table has no auto generated id, this parameter has to be set to false.
  57    * On Oracle, it will allow to not try to drop and recreate the trigger.
  58    * On other databases, this method is useless.
  59    *
  60    * Default value is true.
  61    */
  62   public RenameTableBuilder setAutoGeneratedId(boolean autoGeneratedId) {
  63     this.autoGeneratedId = autoGeneratedId;
  64     return this;
  65   }
  66
  67   public List<String> build() {
  68     validateTableName(name);
  69     validateTableName(newName);
  70     checkArgument(!name.equals(newName), "Names must be different");
  71     return createSqlStatement();
  72   }
  73
  74   private List<String> createSqlStatement() {
  75     switch (dialect.getId()) {
  76       case H2.ID:
  77       case PostgreSql.ID:
  78         return singletonList("ALTER TABLE " + name + " RENAME TO " + newName);
  79       case MsSql.ID:
  80         return singletonList("EXEC sp_rename '" + name + "', '" + newName + "'");
  81       case Oracle.ID:
  82         String renameSqlCommand = "RENAME " + name + " TO " + newName;
  83         return autoGeneratedId ? asList(
  84           "DROP TRIGGER " + name + "_idt",
  85             renameSqlCommand,
  86           "RENAME " + name + "_seq TO " + newName + "_seq",
  87             CreateTableBuilder.createOracleTriggerForTable(newName))
  88           : singletonList(renameSqlCommand);
  89       default:
  90         throw new IllegalArgumentException("Unsupported dialect id " + dialect.getId());
  91     }
  92   }
  93 }