50 lines
1.9 KiB
Groovy
50 lines
1.9 KiB
Groovy
/*
|
|
* Available context bindings:
|
|
* COLUMNS List<DataColumn>
|
|
* ROWS Iterable<DataRow>
|
|
* OUT { append() }
|
|
* FORMATTER { format(row, col); formatValue(Object, col); getTypeName(Object, col); isStringLiteral(Object, col); }
|
|
* TRANSPOSED Boolean
|
|
* plus ALL_COLUMNS, TABLE, DIALECT
|
|
*
|
|
* where:
|
|
* DataRow { rowNumber(); first(); last(); data(): List<Object>; value(column): Object }
|
|
* DataColumn { columnNumber(), name() }
|
|
*/
|
|
|
|
SEP = ", "
|
|
QUOTE = "\'"
|
|
STRING_PREFIX = DIALECT.getDbms().isMicrosoft() ? "N" : ""
|
|
NEWLINE = System.getProperty("line.separator")
|
|
|
|
KEYWORDS_LOWERCASE = com.intellij.database.util.DbSqlUtil.areKeywordsLowerCase(PROJECT)
|
|
KW_INSERT_INTO = KEYWORDS_LOWERCASE ? "insert into " : "INSERT INTO "
|
|
KW_VALUES = KEYWORDS_LOWERCASE ? ") values (" : ") VALUES ("
|
|
KW_NULL = KEYWORDS_LOWERCASE ? "null" : "NULL"
|
|
|
|
def record(columns, dataRow) {
|
|
OUT.append(KW_INSERT_INTO)
|
|
if (TABLE == null) OUT.append("MY_TABLE")
|
|
else OUT.append(TABLE.getParent().getName()).append(".").append(TABLE.getName())
|
|
OUT.append(" (")
|
|
|
|
columns.eachWithIndex { column, idx ->
|
|
OUT.append(column.name()).append(idx != columns.size() - 1 ? SEP : "")
|
|
}
|
|
|
|
OUT.append(KW_VALUES)
|
|
columns.eachWithIndex { column, idx ->
|
|
def value = dataRow.value(column)
|
|
def stringValue = value == null ? KW_NULL : FORMATTER.formatValue(value, column)
|
|
def isStringLiteral = value != null && FORMATTER.isStringLiteral(value, column)
|
|
if (isStringLiteral && DIALECT.getDbms().isMysql()) stringValue = stringValue.replace("\\", "\\\\")
|
|
OUT.append(isStringLiteral ? (STRING_PREFIX + QUOTE) : "")
|
|
.append(isStringLiteral ? stringValue.replace(QUOTE, QUOTE + QUOTE) : stringValue)
|
|
.append(isStringLiteral ? QUOTE : "")
|
|
.append(idx != columns.size() - 1 ? SEP : "")
|
|
}
|
|
OUT.append(");").append(NEWLINE)
|
|
}
|
|
|
|
ROWS.each { row -> record(COLUMNS, row) }
|