提交 0e39d805 编写于 作者: L LonwoLonwo

#9696 keyword case simplify

上级 98e637ab
......@@ -38,7 +38,6 @@ import java.io.File;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.Reader;
import java.util.Locale;
import java.util.Map;
/**
......@@ -68,19 +67,12 @@ public class DataExporterSQL extends StreamExporterAbstract {
private DBDAttributeBinding[] columns;
private final String KEYWORD_INSERT_INTO = "INSERT INTO";
private final String KEYWORD_VALUES = "VALUES";
private DBPIdentifierCase identifierCase = DBPIdentifierCase.LOWER;
private DBPIdentifierCase identifierCase;
private transient StringBuilder sqlBuffer = new StringBuilder(100);
private transient long rowCount;
private SQLDialect dialect;
enum KeywordCase {
upper,
lower
}
private KeywordCase keywordCase;
private boolean isSkipColumn(DBDAttributeBinding attr) {
return attr.isPseudoAttribute() || (!includeAutoGenerated && attr.isAutoGenerated()) ||
attr instanceof DBDAttributeBindingCustom;
......@@ -109,7 +101,12 @@ public class DataExporterSQL extends StreamExporterAbstract {
rowDelimiter = GeneralUtils.getDefaultLineSeparator();
dialect = SQLUtils.getDialectFromObject(site.getSource());
keywordCase = CommonUtils.valueOf(KeywordCase.class, CommonUtils.toString(properties.get(PROP_KEYWORD_CASE)), KeywordCase.upper);
String keywordCase = CommonUtils.toString(properties.get(PROP_KEYWORD_CASE));
if (keywordCase.equalsIgnoreCase("lower")) {
identifierCase = DBPIdentifierCase.LOWER;
} else {
identifierCase = DBPIdentifierCase.UPPER;
}
}
@Override
......@@ -134,7 +131,6 @@ public class DataExporterSQL extends StreamExporterAbstract {
@Override
public void exportRow(DBCSession session, DBCResultSet resultSet, Object[] row) throws DBException, IOException {
PrintWriter out = getWriter();
boolean upperCase = keywordCase == KeywordCase.upper;
SQLDialect.MultiValueInsertMode insertMode = rowsInStatement == 1 ? SQLDialect.MultiValueInsertMode.NOT_SUPPORTED : getMultiValueInsertMode();
if (insertMode == SQLDialect.MultiValueInsertMode.NOT_SUPPORTED) {
rowsInStatement = 1;
......@@ -153,8 +149,7 @@ public class DataExporterSQL extends StreamExporterAbstract {
sqlBuffer.append(rowDelimiter);
}
}
sqlBuffer.append(upperCase ? KEYWORD_INSERT_INTO : identifierCase.transform(KEYWORD_INSERT_INTO));
sqlBuffer.append(" ").append(tableName).append(" (");
sqlBuffer.append(identifierCase.transform(KEYWORD_INSERT_INTO)).append(" ").append(tableName).append(" (");
boolean hasColumn = false;
for (int i = 0; i < columnsSize; i++) {
DBDAttributeBinding column = columns[i];
......@@ -171,7 +166,7 @@ public class DataExporterSQL extends StreamExporterAbstract {
if (lineBeforeValues && insertMode != SQLDialect.MultiValueInsertMode.GROUP_ROWS) {
sqlBuffer.append(rowDelimiter).append("\t");
}
sqlBuffer.append(upperCase ? KEYWORD_VALUES : identifierCase.transform(KEYWORD_VALUES));
sqlBuffer.append(identifierCase.transform(KEYWORD_VALUES));
if (insertMode != SQLDialect.MultiValueInsertMode.GROUP_ROWS) {
sqlBuffer.append(" (");
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册