Coverage Report - liquibase.change.core.DropColumnChange
 
Classes in this File Line Coverage Branch Coverage Complexity
DropColumnChange
72%
18/25
100%
4/4
1.214
DropColumnChange$1
0%
0/5
0%
0/6
1.214
 
 1  
 package liquibase.change.core;
 2  
 
 3  
 import liquibase.change.AbstractChange;
 4  
 import liquibase.change.ChangeMetaData;
 5  
 import liquibase.change.ColumnConfig;
 6  
 import liquibase.database.Database;
 7  
 import liquibase.database.core.DB2Database;
 8  
 import liquibase.database.core.SQLiteDatabase;
 9  
 import liquibase.database.core.SQLiteDatabase.AlterTableVisitor;
 10  
 import liquibase.database.structure.Index;
 11  
 import liquibase.statement.SqlStatement;
 12  
 import liquibase.statement.core.DropColumnStatement;
 13  
 import liquibase.statement.core.ReorganizeTableStatement;
 14  
 import liquibase.util.StringUtils;
 15  
 
 16  
 import java.util.ArrayList;
 17  
 import java.util.List;
 18  
 
 19  
 /**
 20  
  * Drops an existing column from a table.
 21  
  */
 22  
 public class DropColumnChange extends AbstractChange {
 23  
 
 24  
     private String schemaName;
 25  
     private String tableName;
 26  
     private String columnName;
 27  
 
 28  
     public DropColumnChange() {
 29  63
         super("dropColumn", "Drop Column", ChangeMetaData.PRIORITY_DEFAULT);
 30  63
     }
 31  
 
 32  
     public String getColumnName() {
 33  92
         return columnName;
 34  
     }
 35  
 
 36  
     public void setColumnName(String columnName) {
 37  49
         this.columnName = columnName;
 38  49
     }
 39  
 
 40  
     public String getSchemaName() {
 41  47
         return schemaName;
 42  
     }
 43  
 
 44  
     public void setSchemaName(String schemaName) {
 45  47
         this.schemaName = StringUtils.trimToNull(schemaName);
 46  47
     }
 47  
 
 48  
     public String getTableName() {
 49  65
         return tableName;
 50  
     }
 51  
 
 52  
     public void setTableName(String tableName) {
 53  49
         this.tableName = tableName;
 54  49
     }
 55  
 
 56  
     public SqlStatement[] generateStatements(Database database) {
 57  
 
 58  
         // todo if (database instanceof SQLiteDatabase) {
 59  
         // // return special statements for SQLite databases
 60  
         // return generateStatementsForSQLiteDatabase(database);
 61  
         // }
 62  
 
 63  46
         List<SqlStatement> statements = new ArrayList<SqlStatement>();
 64  46
         String schemaName = getSchemaName() == null ? database.getDefaultSchemaName() : getSchemaName();
 65  
 
 66  46
         statements.add(new DropColumnStatement(schemaName, getTableName(), getColumnName()));
 67  46
         if (database instanceof DB2Database) {
 68  3
             statements.add(new ReorganizeTableStatement(schemaName, getTableName()));
 69  
         }
 70  
 
 71  46
         return statements.toArray(new SqlStatement[statements.size()]);
 72  
     }
 73  
 
 74  
     private SqlStatement[] generateStatementsForSQLiteDatabase(Database database) {
 75  
 
 76  
         // SQLite does not support this ALTER TABLE operation until now.
 77  
         // For more information see: http://www.sqlite.org/omitted.html.
 78  
         // This is a small work around...
 79  
 
 80  0
         List<SqlStatement> statements = new ArrayList<SqlStatement>();
 81  
 
 82  
         // define alter table logic
 83  0
         AlterTableVisitor rename_alter_visitor = new AlterTableVisitor() {
 84  
             public ColumnConfig[] getColumnsToAdd() {
 85  0
                 return new ColumnConfig[0];
 86  
             }
 87  
 
 88  
             public boolean createThisColumn(ColumnConfig column) {
 89  0
                 return !column.getName().equals(getColumnName());
 90  
             }
 91  
 
 92  
             public boolean copyThisColumn(ColumnConfig column) {
 93  0
                 return !column.getName().equals(getColumnName());
 94  
             }
 95  
 
 96  
             public boolean createThisIndex(Index index) {
 97  0
                 return !index.getColumns().contains(getColumnName());
 98  
             }
 99  
         };
 100  
 
 101  
         try {
 102  
             // alter table
 103  0
             statements.addAll(SQLiteDatabase.getAlterTableStatements(rename_alter_visitor, database, getSchemaName(),
 104  
                     getTableName()));
 105  
 
 106  0
         } catch (Exception e) {
 107  0
             e.printStackTrace();
 108  0
         }
 109  
 
 110  0
         return statements.toArray(new SqlStatement[statements.size()]);
 111  
     }
 112  
 
 113  
     public String getConfirmationMessage() {
 114  1
         return "Column " + getTableName() + "." + getColumnName() + " dropped";
 115  
     }
 116  
 }