idea自定义数据导出格式
最近需要从数据库导出特定格式的数据,写代码导太麻烦了,试了几个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>