All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.github.drinkjava2.jdialects.DDLDropUtils Maven / Gradle / Ivy
/*
* Copyright 2016 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
* use this file except in compliance with the License. You may obtain a copy of
* the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by
* applicable law or agreed to in writing, software distributed under the
* License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS
* OF ANY KIND, either express or implied. See the License for the specific
* language governing permissions and limitations under the License.
*/
package com.github.drinkjava2.jdialects;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import com.github.drinkjava2.jdialects.annotation.jpa.GenerationType;
import com.github.drinkjava2.jdialects.id.AutoIdGenerator;
import com.github.drinkjava2.jdialects.id.IdGenerator;
import com.github.drinkjava2.jdialects.id.SequenceIdGenerator;
import com.github.drinkjava2.jdialects.id.TableIdGenerator;
import com.github.drinkjava2.jdialects.model.ColumnModel;
import com.github.drinkjava2.jdialects.model.FKeyModel;
import com.github.drinkjava2.jdialects.model.IndexModel;
import com.github.drinkjava2.jdialects.model.TableModel;
import com.github.drinkjava2.jdialects.model.UniqueModel;
import com.github.drinkjava2.jlogs.Log;
import com.github.drinkjava2.jlogs.LogFactory;
/**
* To transfer platform-independent model to drop DDL String array
*
* @author Yong Zhu
* @since 1.0.2
*/
@SuppressWarnings("all")
public class DDLDropUtils {
private static final Log logger = LogFactory.getLog(DDLDropUtils.class);
/**
* Transfer tables to drop DDL and without format it
*/
public static String[] toDropDDL(Dialect dialect, TableModel... tables) {
// resultList store mixed drop DDL + drop Ojbects
List objectResultList = new ArrayList();
for (TableModel table : tables)
transferTableToObjectList(dialect, table, objectResultList);
boolean hasAutoIdGenerator = false;
for (TableModel table : tables) {
for (ColumnModel column : table.getColumns())
if (GenerationType.AUTO.equals(column.getIdGenerationType())) {
hasAutoIdGenerator = true;
break;
}
for (IdGenerator idGens : table.getIdGenerators())
if (hasAutoIdGenerator || idGens.dependOnAutoIdGenerator()) {
hasAutoIdGenerator = true;
break;
}
}
List stringResultList = new ArrayList();
List tbGeneratorList = new ArrayList();
List sequenceList = new ArrayList();
List fKeyConstraintList = new ArrayList();
for (Object strOrObj : objectResultList) {
if (!StrUtils.isEmpty(strOrObj)) {
if (strOrObj instanceof String)
stringResultList.add((String) strOrObj);
else if (strOrObj instanceof TableIdGenerator)
tbGeneratorList.add((TableIdGenerator) strOrObj);
else if (strOrObj instanceof SequenceIdGenerator)
sequenceList.add((SequenceIdGenerator) strOrObj);
else if (strOrObj instanceof FKeyModel)
fKeyConstraintList.add((FKeyModel) strOrObj);
}
}
if (hasAutoIdGenerator) {
IdGenerator realIdGen = AutoIdGenerator.INSTANCE.getSequenceOrTableIdGenerator(dialect);
if (realIdGen instanceof TableIdGenerator)
tbGeneratorList.add((TableIdGenerator) realIdGen);
else if (realIdGen instanceof SequenceIdGenerator)
sequenceList.add((SequenceIdGenerator) realIdGen);
else
throw new DialectException("Unknow exception happen for realIdGen, please report this bug");
}
buildDropSequenceDDL(dialect, stringResultList, sequenceList);
buildDropTableGeneratorDDL(dialect, stringResultList, tbGeneratorList);
outputDropFKeyConstraintDDL(dialect, stringResultList, fKeyConstraintList);
String[] result = stringResultList.toArray(new String[stringResultList.size()]);
if (Dialect.getGlobalAllowShowSql())
logger.info("Drop DDL:\r" + StrUtils.arrayToString(result, "\r"));
return result;
}
/**
* Transfer table to a mixed DDL String or TableGen Object list
*/
private static void transferTableToObjectList(Dialect dialect, TableModel t, List objectResultList) {
StringBuilder buf = new StringBuilder();
String tableName = t.getTableName();
List columns = t.getColumns();
// Reserved words check
dialect.checkNotEmptyReservedWords(tableName, "Table name", tableName);
List l = t.getIndexConsts();// check index names
if (l != null && !l.isEmpty())
for (IndexModel index : l)
dialect.checkReservedWords(index.getName());
List l2 = t.getUniqueConsts();// check unique names
if (l2 != null && !l2.isEmpty())
for (UniqueModel unique : l2)
dialect.checkReservedWords(unique.getName());
List fkeyChks = t.getFkeyConstraints();// check Fkey names
if (fkeyChks != null && !fkeyChks.isEmpty())
for (FKeyModel fkey : fkeyChks)
dialect.checkReservedWords(fkey.getFkeyName());
for (ColumnModel col : columns)
if (!col.getTransientable())
dialect.checkNotEmptyReservedWords(col.getColumnName(), "Column name", tableName);
// idGenerator
for (IdGenerator idGen : t.getIdGenerators())
objectResultList.add(idGen);
// Foreign key
for (FKeyModel fkey : t.getFkeyConstraints())
objectResultList.add(fkey);
// drop table
buf.append(dialect.dropTableDDL(tableName));
objectResultList.add(buf.toString());
}
private static void buildDropSequenceDDL(Dialect dialect, List stringResultList,
List sequenceList) {
Set notRepeatedSequences = new HashSet();
for (SequenceIdGenerator seq : sequenceList)
DDLCreateUtils.checkAndInsertToNotRepeatSeq(notRepeatedSequences, seq);
DDLFeatures features = dialect.ddlFeatures;
for (SequenceIdGenerator seq : notRepeatedSequences) {
if (!features.supportBasicOrPooledSequence()) {
DialectException.throwEX("Dialect \"" + dialect + "\" does not support sequence setting on sequence \""
+ seq.getName() + "\"");
}
if (!DDLFeatures.NOT_SUPPORT.equals(features.dropSequenceStrings)
&& !StrUtils.isEmpty(features.dropSequenceStrings)) {
stringResultList.add(0,
StrUtils.replace(features.dropSequenceStrings, "_SEQNAME", seq.getSequenceName()));
} else
DialectException.throwEX("Dialect \"" + dialect
+ "\" does not support drop sequence ddl, on sequence \"" + seq.getName() + "\"");
}
}
private static void buildDropTableGeneratorDDL(Dialect dialect, List stringResultList,
List tbGeneratorList) {
for (TableIdGenerator tg : tbGeneratorList) {
// @formatter:off
DialectException.assureNotEmpty(tg.getName(), "TableGen name can not be empty");
DialectException.assureNotEmpty(tg.getTable(),
"TableGen tableName can not be empty of \"" + tg.getName() + "\"");
DialectException.assureNotEmpty(tg.getPkColumnName(),
"TableGen pkColumnName can not be empty of \"" + tg.getName() + "\"");
DialectException.assureNotEmpty(tg.getPkColumnValue(),
"TableGen pkColumnValue can not be empty of \"" + tg.getName() + "\"");
DialectException.assureNotEmpty(tg.getValueColumnName(),
"TableGen valueColumnName can not be empty of \"" + tg.getName() + "\"");
// @formatter:on
}
Set tableExisted = new HashSet();
for (TableIdGenerator tg : tbGeneratorList) {
String tableName = tg.getTable().toLowerCase();
if (!tableExisted.contains(tableName)) {
stringResultList.add(0, dialect.dropTableDDL(tableName));
tableExisted.add(tableName);
}
}
}
private static void outputDropFKeyConstraintDDL(Dialect dialect, List stringResultList,
List trueList) {
if (DDLFeatures.NOT_SUPPORT.equals(dialect.ddlFeatures.addForeignKeyConstraintString))
return;
for (FKeyModel t : trueList) {
if (!t.getDdl())
continue; // if ddl is false, skip
String dropStr = dialect.ddlFeatures.dropForeignKeyString;
String constName = t.getFkeyName();
if (StrUtils.isEmpty(constName))
constName = "fk_" + t.getTableName().toLowerCase() + "_"
+ StrUtils.replace(StrUtils.listToString(t.getColumnNames()), ",", "_");
if (DDLFeatures.NOT_SUPPORT.equals(dropStr))
DialectException.throwEX("Dialect \"" + dialect
+ "\" does not support drop foreign key, for setting: \"" + "fk_" + constName + "\"");
stringResultList.add(0, "alter table " + t.getTableName() + " " + dropStr + " " + constName);
}
}
}
