利用groovy把表中数据导出成txt或csv

Posted 2021-05-19 zhangjianying

胶水语言就是胶水,写起也挺快的.这个脚本主要是从数据库中把表中的数据导出来生成文件.而不用每次都打开数据库编辑器去手工收集.然后结合强大的脚本(shell,bat)等基本就可以实现定时生成最新数据文本了.

目前这个groovy脚本只要输入sql语句,便会自动去反射表中的字段和类型.
脚本名称:export.groovy

1. import groovy.sql.Sql;
4. try
5. {
6. if(this.args.size()<1)
7. {
8.     println "格式错误!请参照下面的正确格式"
9.     println "export [sql:可以正确执行的sql语句]"
10.     return;
11. }
12. }
13. catch (Exception e)
14. {
15.     println "格式错误!请参照下面的正确格式"
16.     println "export [sql:可以正确执行的sql语句]"
17.     return;
19. }
22. /*
23. 连接对象
24. 可以根据你自己的需要调整
25. */
26. sql = Sql.newInstance("jdbc:oracle:thin:@10.220.51.40:1521:ORA10G","用户名","密码","oracle.jdbc.driver.OracleDriver");
29. /* SQL语句 可以根据自己需要调整*/
30. strSql=this.args[0];
31. //方法_获取表名
32. def getTableName={strSql->
33.  pattern= ~"FROM .*"
34.  matcher = strSql.toUpperCase()  =~ pattern
35.  tableName="";
36.  while(matcher.find()){
37.      tableName= matcher.group()
38.  }
39.  return tableName.split(' ')[1];
40. }
42. //方法判断字符串里面是否有中文
43. def isChinaese={ str->  
44.  if(str==null)
45.      return;
46.     
47.  for (i in 0..str.size()-1)
48.  {
49.     if(str.charAt(i)>127)
50.     {
51.         return true;
52.     }
53.  } 
54.   return false;
55. }
58. def file_tableName=getTableName(strSql);//表名
59. def result_columnCount=0;//列数
60. def result_columnName=[];//列名
61. def result_columnType=[];//列类型
62. def result_rowsData=[];//行数据
63. def rows_count=0;
65. //在该脚本的同级目录下生成已表名命名的csv文件
66. file_csv= new File("${file_tableName}.txt");
68. //判断文件是否存在,如果存在写删除
69. if(file_csv.exists()){
70. println "在当前目录下发现已经存在${file_tableName},程序已经删除文件"
71.     file_csv.delete()
72. }
76. println "准备生成${file_tableName}.csv"
77. sql.eachRow(strSql,
78.     {
79.         result_columnCount= it.getColumnCount();        
80.         
81.         println "*********${file_tableName}表结构**********"
82.         println ":本次导出共生成${result_columnCount}个字段";
83.         for (i in 1..result_columnCount){  //不同与java,groovy下标从1开始          
84.             println    it.getColumnName(i) +"   |    "+ it.getColumnTypeName(i) ;   
85.             result_columnName<<it.getColumnName(i);
86.             result_columnType<<it.getColumnTypeName(i);           
87.         }
88.         println "*********${file_tableName}表结构**********"
89.         file_csv<<result_columnName.join(",")+System.getProperty("line.separator"); 
90.         
91.         println "开始生成数据,请耐心等待......"
92.     },
93.     {
94.         rows_count++;
95.         result_rowsData=[]; //清空对象  
96.         for (j in 0..result_columnCount-1)
97.         {            
98.             if(result_columnType[j]=="VARCHAR2" || result_columnType[j]=="VARCHAR"){ //针对VARCHAR类型进行处理,加上"号
99.                  String strtemp="";
100.                 //先要判断是否含有中文,有中文就不加"号
101.            /*     if(isChinaese(it[j]))
102.                 {
103.                     strtemp=it[j]
104.                 }
105.                 else
106.                 {
107.                     strtemp="/""+it[j]+"/"";
108.                 }      
109.                 */
110.                     strtemp="/""+it[j]+"/"";
111.                  result_rowsData<<strtemp;
112.             }
113.             else //其他类型不做处理
114.             {        
115.                 result_rowsData<<it[j];
116.             }
117.         }
118.  
119.          file_csv<< result_rowsData.join(",")+System.getProperty("line.separator");
120.     
121.     }    
122. );
124. println "${file_tableName}数据生成完毕(导出数据行数:${rows_count})/n"

然后再给这个groovy脚本写几个对应的shell/bat执行脚本
比如:
文件名:生成基站.bat

1. groovy export "select * from shw.chal_cell_base_info"

如图:



那么当每次都要生成最新数据的时候,直接去双击这些bat文件即可.或者你直接把这些bat配置到操作系统的执行任务当中去即可.