最近需要从数据库导出特定格式的数据，写代码导太麻烦了，试了几个mysql客户端都只能导固定格式的，最后发现idea的Database工具可以自定义导出的格式，所以就试了下。
导出功能在这个位置
当然这些是idea默认的，想自定义点击Go to Scripts Directory
可以直接修改原来的脚本，也可以新建，新建的也会在刚才的右键菜单里显示，很方便。
从原来的脚本可以看出，idea支持三种脚本语言:js、groovy和clojure，可以用自己熟悉的语言写，这里我选的是js。
最终导出的文件后缀名是由脚本的文件名决定的，如xxx.xml.js，导出的文件是xml格式的。
数据库的信息可以直接从上下文中取，分别有：
COLUMNS List<DataColumn> // 每一列的信息
DataColumn { columnNumber(), name() } // 列信息对象有两个方法，取列序号和列名
ROWS Iterable<DataRow> //第一行的信息
DataRow { rowNumber(); first(); last(); data(): List<Object>; value(column): Object } //行对象
OUT { append() } // 这个是用来输出到文件
FORMATTER { format(row, col); formatValue(Object, col) } //取某行某列的值
TABLE // 整个表的信息，最常用的是TABLE.getName()，获取表名
function eachWithIdx(iterable, f) { var i = iterable.iterator(); var idx = 0; while (i.hasNext()) f(i.next(), idx++); }
// 特殊字符转义
function escape(str) {
str = str.replaceAll("\t|\b|\\f|\\r|\\n|\\r\\n", "");
str = com.intellij.openapi.util.text.StringUtil.escapeXml(str);
str = str.replaceAll("\"", """);
if (str == null || str == ‘NULL‘ || str == ‘null‘) {
str = ‘‘;
}
return str;
}
// 换行
var NEWLINE = "\n";
function output() { for (var i = 0; i < arguments.length; i++) { OUT.append(arguments[i]); } }
output("<?xml version=\"1.0\" encoding=\"UTF-8\"?>", NEWLINE);
output("<dataset>", NEWLINE);
eachWithIdx(ROWS, function (row) {
output(" <", TABLE.getName());
eachWithIdx(COLUMNS, function (col) {
output(" " + col.name(), "=\"", escape(FORMATTER.format(row, col)), "\"");
});
output(" />", NEWLINE);
});
output("</dataset>")
生成的文件格式
<?xml version="1.0" encoding="UTF-8"?>
<dataset>
<sys_user id="1" username="123" password="123" mobile="123" />
<sys_user id="2" username="321" password="123" mobile="222555" />
<sys_user id="3" username="111" password="321" mobile="12313" />
</dataset>
