packagecom.generateDocs;importjava.awt.Color;importjava.io.FileOutputStream;importjava.sql.Connection;importjava.sql.DriverManager;importjava.sql.ResultSet;importjava.sql.SQLException;importjava.sql.Statement;importjava.util.ArrayList;importjava.util.HashMap;importjava.util.Iterator;importjava.util.List;importjava.util.Map;importcom.lowagie.text.Cell;importcom.lowagie.text.Document;importcom.lowagie.text.PageSize;importcom.lowagie.text.Paragraph;importcom.lowagie.text.Table;importcom.lowagie.text.rtf.RtfWriter2;
public classMysqlDocsGenerate {//键类型字典
private static Map keyType = new HashMap();//初始化jdbc
static{try{
keyType.put("PRI", "主键");
keyType.put("UNI", "唯一键");
Class.forName("com.mysql.jdbc.Driver");
}catch(ClassNotFoundException e) {
e.printStackTrace();
}
}private static String url = "jdbc:mysql://127.0.0.1:3306/cmb";//链接url
private static String username = "root"; //用户名
private static String password = "****"; //密码
private static String schema = "****"; //目标数据库 名//查询所有表的sql语句
private static String sql_get_all_tables = "select table_name,TABLE_COMMENT from INFORMATION_SCHEMA.tables where TABLE_SCHEMA='" + schema + "' and TABLE_TYPE='BASE TABLE'";//查询所有字段的sql语句
private static String sql_get_all_columns = "select column_name,data_type,CHARACTER_MAXIMUM_LENGTH,COLUMN_COMMENT,is_nullable,COLUMN_key from information_schema.`COLUMNS` where TABLE_NAME='{table_name}' and TABLE_SCHEMA='" + schema + "'";public static void main(String[] args) throwsException {//初始化word文档
Document document = newDocument(PageSize.A4);
RtfWriter2.getInstance(document,new FileOutputStream("D:/word.doc"));
document.open();//查询开始
Connection conn =getConnection();//获取所有表
List tables =getDataBySQL(sql_get_all_tables, conn);int i = 1;for (Iterator iterator =tables.iterator(); iterator.hasNext(); ) {
String[] arr=(String[]) iterator.next();//循环获取字段信息
System.out.print(i + ".正在处理数据表-----------" + arr[0]);
addTableMetaData(document, arr, i);
List columns= getDataBySQL(sql_get_all_columns.replace("{table_name}", arr[0]), conn);
addTableDetail(document, columns);
addBlank(document);
System.out.println("...done");
i++;
}
document.close();
conn.close();
}
public static void addBlank(Document document) throwsException {
Paragraph ph= new Paragraph("");
ph.setAlignment(Paragraph.ALIGN_LEFT);
document.add(ph);
}
public static void addTableDetail(Document document, List columns) throwsException {
Table table= new Table(6);
table.setWidth(100f);//表格 宽度100%
table.setBorderWidth(1);
table.setBorderColor(Color.BLACK);
table.setPadding(0);
table.setSpacing(0);
Cell cell1= new Cell("序号");//单元格
cell1.setHeader(true);
Cell cell2= new Cell("列名");//单元格
cell2.setHeader(true);
Cell cell3= new Cell("类型");//单元格
cell3.setHeader(true);
Cell cell4= new Cell("长度");//单元格
cell4.setHeader(true);
Cell cell5= new Cell("键");//单元格
cell5.setHeader(true);
Cell cell6= new Cell("说明");//单元格
cell6.setHeader(true);//设置表头格式
table.setWidths(new float[]{8f, 30f, 15f, 8f, 10f, 29f});
cell1.setHorizontalAlignment(Cell.ALIGN_CENTER);
cell1.setBackgroundColor(Color.gray);
cell2.setHorizontalAlignment(Cell.ALIGN_CENTER);
cell2.setBackgroundColor(Color.gray);
cell3.setHorizontalAlignment(Cell.ALIGN_CENTER);
cell3.setBackgroundColor(Color.gray);
cell4.setHorizontalAlignment(Cell.ALIGN_CENTER);
cell4.setBackgroundColor(Color.gray);
cell5.setHorizontalAlignment(Cell.ALIGN_CENTER);
cell5.setBackgroundColor(Color.gray);
cell6.setHorizontalAlignment(Cell.ALIGN_CENTER);
cell6.setBackgroundColor(Color.gray);
table.addCell(cell1);
table.addCell(cell2);
table.addCell(cell3);
table.addCell(cell4);
table.addCell(cell5);
table.addCell(cell6);
table.endHeaders();//表头结束
int x = 1;for (Iterator iterator =columns.iterator(); iterator.hasNext(); ) {
String[] arr2=(String[]) iterator.next();
Cell c1= new Cell(x + "");
Cell c2= new Cell(arr2[0]);
Cell c3= new Cell(arr2[1]);
Cell c4= new Cell(arr2[2]);
String key= keyType.get(arr2[5]);if (key == null) key = "";
Cell c5= newCell(key);
Cell c6= new Cell(arr2[3]);
c1.setHorizontalAlignment(Cell.ALIGN_CENTER);
c2.setHorizontalAlignment(Cell.ALIGN_CENTER);
c3.setHorizontalAlignment(Cell.ALIGN_CENTER);
c4.setHorizontalAlignment(Cell.ALIGN_CENTER);
c5.setHorizontalAlignment(Cell.ALIGN_CENTER);
c6.setHorizontalAlignment(Cell.ALIGN_CENTER);
table.addCell(c1);
table.addCell(c2);
table.addCell(c3);
table.addCell(c4);
table.addCell(c5);
table.addCell(c6);
x++;
}
document.add(table);
}
public static void addTableMetaData(Document dcument, String[] arr, int i) throwsException {
Paragraph ph= new Paragraph(i + ". 表名: " + arr[0] + " 说明: " + (arr[1] == null ? "" : arr[1]));
ph.setAlignment(Paragraph.ALIGN_LEFT);
dcument.add(ph);
}
public staticList getDataBySQL(String sql, Connection conn) {
Statement stmt= null;
ResultSet rs= null;
List list= newArrayList();try{
stmt=conn.createStatement();
rs=stmt.executeQuery(sql);while(rs.next()) {
String[] arr= newString[rs.getMetaData().getColumnCount()];for (int i = 0; i < arr.length; i++) {
arr[i]= rs.getString(i + 1);
}
list.add(arr);
}
}catch(SQLException e) {
e.printStackTrace();
}finally{try{if (rs != null) rs.close();if (stmt != null) stmt.close();
}catch(SQLException e) {
e.printStackTrace();
}
}returnlist;
}
public staticConnection getConnection() {try{returnDriverManager.getConnection(url, username, password);
}catch(SQLException e) {
e.printStackTrace();
}return null;
}
}