-
Notifications
You must be signed in to change notification settings - Fork 1.8k
/
DropAllForeignKeyConstraintsChange.java
125 lines (101 loc) · 5.04 KB
/
DropAllForeignKeyConstraintsChange.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package liquibase.change.core;
import liquibase.change.AbstractChange;
import liquibase.change.ChangeClass;
import liquibase.change.ChangeMetaData;
import liquibase.change.ChangeProperty;
import liquibase.database.Database;
import liquibase.database.structure.Table;
import liquibase.exception.DatabaseException;
import liquibase.exception.UnexpectedLiquibaseException;
import liquibase.executor.Executor;
import liquibase.executor.ExecutorService;
import liquibase.statement.SqlStatement;
import liquibase.statement.core.FindForeignKeyConstraintsStatement;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
@ChangeClass(name="dropAllForeignKeyConstraints", description = "Drop All Foreign Key Constraints", priority = ChangeMetaData.PRIORITY_DEFAULT, appliesTo = "table")
public class DropAllForeignKeyConstraintsChange extends AbstractChange {
private String baseTableCatalogName;
private String baseTableSchemaName;
private String baseTableName;
@ChangeProperty(includeInSerialization = false)
private List<DropForeignKeyConstraintChange> childDropChanges;
@ChangeProperty(mustApplyTo ="table.catalog")
public String getBaseTableCatalogName() {
return baseTableCatalogName;
}
public void setBaseTableCatalogName(String baseTableCatalogName) {
this.baseTableCatalogName = baseTableCatalogName;
}
@ChangeProperty(mustApplyTo ="table.schema")
public String getBaseTableSchemaName() {
return baseTableSchemaName;
}
public void setBaseTableSchemaName(String baseTableSchemaName) {
this.baseTableSchemaName = baseTableSchemaName;
}
@ChangeProperty(requiredForDatabase = "all", mustApplyTo = "table")
public String getBaseTableName() {
return baseTableName;
}
public void setBaseTableName(String baseTableName) {
this.baseTableName = baseTableName;
}
public SqlStatement[] generateStatements(Database database) {
List<SqlStatement> sqlStatements = new ArrayList<SqlStatement>();
if (childDropChanges == null) {
generateChildren(database);
}
if (childDropChanges != null) {
for (DropForeignKeyConstraintChange change : childDropChanges) {
sqlStatements.addAll(Arrays.asList(change.generateStatements(database)));
}
}
return sqlStatements.toArray(new SqlStatement[sqlStatements.size()]);
}
public String getConfirmationMessage() {
return "Foreign keys on base table " + getBaseTableName() + " dropped";
}
private void generateChildren(Database database) {
// Make a new list
childDropChanges = new ArrayList<DropForeignKeyConstraintChange>();
Executor executor = ExecutorService.getInstance().getExecutor(database);
FindForeignKeyConstraintsStatement sql = new FindForeignKeyConstraintsStatement(getBaseTableCatalogName(), getBaseTableSchemaName(), getBaseTableName());
try {
List<Map> results = executor.queryForList(sql);
Set<String> handledConstraints = new HashSet<String>();
if (results != null && results.size() > 0) {
for (Map result : results) {
String baseTableName =
(String) result.get(FindForeignKeyConstraintsStatement.RESULT_COLUMN_BASE_TABLE_NAME);
String constraintName =
(String) result.get(FindForeignKeyConstraintsStatement.RESULT_COLUMN_CONSTRAINT_NAME);
if (new Table(getBaseTableName()).equals(new Table(baseTableName), database)) {
if( !handledConstraints.contains(constraintName)) {
DropForeignKeyConstraintChange dropForeignKeyConstraintChange =
new DropForeignKeyConstraintChange();
dropForeignKeyConstraintChange.setBaseTableSchemaName(getBaseTableSchemaName());
dropForeignKeyConstraintChange.setBaseTableName(baseTableName);
dropForeignKeyConstraintChange.setConstraintName(constraintName);
childDropChanges.add(dropForeignKeyConstraintChange);
handledConstraints.add(constraintName);
}
} else {
throw new IllegalStateException("Expected to return only foreign keys for base table name: " +
getBaseTableName() + " and got results for table: " + baseTableName);
}
}
}
} catch (DatabaseException e) {
throw new UnexpectedLiquibaseException("Failed to find foreign keys for table: " + getBaseTableName(), e);
}
}
@Override
public boolean requiresUpdatedDatabaseMetadata(Database database) {
return true;
}
}