-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathGenerateDataDictionary4PoiController.groovy
188 lines (171 loc) · 7.58 KB
/
GenerateDataDictionary4PoiController.groovy
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
package rit.tradso.system
import groovy.sql.Sql
import org.apache.poi.hssf.usermodel.HSSFCellStyle
import org.apache.poi.hssf.usermodel.HSSFRow
import org.apache.poi.hssf.usermodel.HSSFSheet
import org.apache.poi.hssf.usermodel.HSSFWorkbook
import rit.tradso.base.BaseController
/**
* POI生成数据库的所有表的数据字典，生成到一个Excel中的多个sheet页中，每个表占用一个sheet
*/
class GenerateDataDictionary4PoiController extends BaseController {
def dataSource
def sql
/**
* 使用POI生成数据字典
*/
def generateDataDictionary4Poi = {
long startTime = System.currentTimeMillis()
sql = new Sql(dataSource)
//获取数据库连接，执行sql，查询到数据库中的所有表
String allTableSql = " select TABLE_NAME name from INFORMATION_SCHEMA.TABLES where TABLE_SCHEMA='tradso-2018-5-14'"
//得到ArrayList结果，泛型是LinkedHashMap，key是数据库名，值是每个表的名称
def allTables = sql.rows(allTableSql)
//遍历每个表，然后通过该表名获取到这个表的数据结构
eachAllTables(allTables)
long endTime = System.currentTimeMillis()
long s=(endTime-startTime)/1000
render "共耗时："+s+"秒"
}
/**
* 遍历每个表，然后通过该表明获取到这个表的数据结构
* @param allTables 数据库中所有表
*/
def eachAllTables(def allTables) {
//遍历每个数据表时开始创建Excel文件
HSSFWorkbook workbook = new HSSFWorkbook();
FileOutputStream fo=new FileOutputStream("d:\\海贸数据字典.xls")
//遍历每张表名的同时，将对应的.groovy同名文件找到，并存储在Map中，key为表名，键为该文件的绝对路径
HashMap<String, String> pathMaps = new HashMap<>();
//应提前将所有对应表的实体类文件找到
allTables.each{Map map ->
//每个表的名称
String tableName = map?.name
String newTableName = tableName.replace("_", "");
//查询每个表名对应的java文件
findJavaFile(pathMaps, newTableName)
}
println pathMaps.size()
allTables.each { Map map ->
//每个表的名称
String tableName = map?.name
//查询每个表名对应的java文件
// findJavaFile(pathMaps, tableName)
//执行查询每个表的表结构，根据表名称
eachOneTables(workbook, fo, pathMaps, tableName)//根据每个表的名称，获取表结构数据
}
workbook.write(fo)
workbook.close()
fo.close()
}
/**
* 根据每个表的名称，获取表结构数据
* @param tableName 表名称
*/
def eachOneTables(HSSFWorkbook workbook, FileOutputStream fo, Map pathMaps, String tableName) {
String oneTableSql = """
SELECT
COLUMN_NAME COLUMN_NAME,
DATA_TYPE DATA_TYPE,
IFNULL(CHARACTER_MAXIMUM_LENGTH,0) length,
case when IS_NULLABLE='no' then '' when IS_NULLABLE='YES' then '√' else 0 end isNull
FROM
INFORMATION_SCHEMA. COLUMNS
WHERE
table_schema = 'tradso-2018-5-14'
AND
table_name = '${tableName}'
"""
//得到每张表的数据结构
def oneTable = sql.rows(oneTableSql)
String newTableName = tableName.replace("_", "");
//查询到每个表的数据结构后，就可以构建POI生成excel了
generateExcel4Poi(workbook, fo, pathMaps, tableName, newTableName, oneTable)
}
/**
* 查询到每个表的数据结构后，就可以构建POI生成excel了
* @param oneTable 每张表中的所有数据结构
*/
def generateExcel4Poi(HSSFWorkbook workbook, FileOutputStream fo, Map pathMaps, String tableName, String newTableName,
def oneTable) {
//判断数据库表是否有对应的实体类文件
if(pathMaps.get(newTableName)!=null){
//字段描述，从.groovy文件中提取
String name=pathMaps.get(newTableName)
FileReader groovyFile=new FileReader(name)
//读取文件
BufferedReader fis=new BufferedReader(groovyFile)
ArrayList<String> list=new ArrayList<>()
String lines
while ((lines=fis.readLine())!=null){
list.add(lines)
}
fis.close()//关闭输入流
println "表名："+newTableName
//开始创建每个sheet页
HSSFSheet sheet = workbook.createSheet(tableName)
HSSFCellStyle setBorder = workbook.createCellStyle();
setBorder.setBorderBottom(HSSFCellStyle.BORDER_THIN); //下边框
setBorder.setBorderLeft(HSSFCellStyle.BORDER_THIN);//左边框
setBorder.setBorderTop(HSSFCellStyle.BORDER_THIN);//上边框
setBorder.setBorderRight(HSSFCellStyle.BORDER_THIN);//右边框
//设置单元格样式
sheet.setColumnWidth(5, 16*2 * 256);
HSSFRow row = sheet.createRow(0);
row.createCell(0).setCellValue("序号");
row.createCell(1).setCellValue("列名");
row.createCell(2).setCellValue("字段类型");
row.createCell(3).setCellValue("长度");
row.createCell(4).setCellValue("是否为空");
row.createCell(5).setCellValue("字段描述");
oneTable.eachWithIndex{Map map,int i->
int index=0;
//第1行
HSSFRow newRow = sheet.createRow(i+1);
//第一个单元格
newRow.createCell(index).setCellValue(i+1)
newRow.createCell(index+1).setCellValue(map?.COLUMN_NAME)
newRow.createCell(index+2).setCellValue(map?.DATA_TYPE)
newRow.createCell(index+3).setCellValue(map?.length)
newRow.createCell(index+4).setCellValue(map?.isNull)
String colName=map?.COLUMN_NAME
colName=colName.replace("_","")
if(colName.endsWith("id")){
colName=colName.substring(0,colName.length()-2)
}
println colName
for(String ll:list){
if(ll!=null&&ll.toLowerCase().contains(colName.toLowerCase())){
int num=ll.indexOf("//")
if(num!=-1) {
newRow.createCell(index + 5).setCellValue(ll.substring(num + 2, ll.length()))
}
}
}
}
}
}
/**
* 传入一个表名，遍历项目下的所有.groovy文件，找到同名文件，将文件里的所有内容保存到Map中
*/
def findJavaFile(Map map, String tableName) {
//获取项目路径：递归遍历
String basePath = grailsApplication.getMainContext().servletContext.getRealPath("/")
basePath = basePath.replace("\\web-app\\", "")
File file = new File(basePath)
getRealFile(map, tableName, file)
}
/**
* 递归遍历和数据表同名.groovy文件
*/
def getRealFile(Map map, String tableName, File path) {
File[] files = path.listFiles()
for (File ff : files) {
if (ff.isFile() && ff.getName().equalsIgnoreCase(tableName + ".groovy")) {
map.put(tableName, ff)
} else if (ff.isDirectory()) {
getRealFile(map, tableName, ff);
}
}
}
}