mysql 表結構轉excel表格

 

最近須要寫文檔，因爲開發模式是先開發後寫文檔（不想吐槽。。。），數據庫表結構什麼的都搞好了，而後寫文檔的時候須要貼表結構，什麼字段，類型，相關說明須要一一對應起來，數據庫表10多張，字段又多，手動複製粘貼太蛋疼了，因而就寫了個將表結構轉excel表格的簡單實現（醜是醜了點，畢竟實現功能了不是）。

package net.cloudkit.management.util;

import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.streaming.SXSSFSheet;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import java.io.*;
import java.sql.*;

import java.util.*;

/**
 * @author : gongtao
 * @version : 2017/9/6
 */
public class TableToExcel {

    private String tableName = "";//表名
    private String[] colNames; // 列名數組
    private String[] colComment; // 列名數組
    private String[] colTypes; //列名類型數組
    private int[] colSizes; //列名大小數組
    //數據庫鏈接
    private static final String URL ="jdbc:mysql://192.168.1.101:3306/management?useUnicode=true&characterEncoding=UTF-8";
    private static final String NAME = "root";
    private static final String PASS = "admin123";
    private static final String DRIVER ="com.mysql.jdbc.Driver";

    public void genEntitySomeTable(List<String> tableNames){
        for(int p=0;p<tableNames.size();p++){
            tableName=tableNames.get(p);
            //建立鏈接
            Connection con = null;
            //查要生成實體類的表
            String sql = "select * from " + tableName;
            String sql2 = "show full fields from " + tableName;
            PreparedStatement pStemt = null;
            PreparedStatement pStemt2 = null;
            try {
                try {
                    Class.forName(DRIVER);
                } catch (ClassNotFoundException e1) {
                    e1.printStackTrace();
                }
                con = DriverManager.getConnection(URL,NAME,PASS);
                pStemt = con.prepareStatement(sql);
                ResultSetMetaData rsmd = pStemt.getMetaData();
                pStemt2 = con.prepareStatement(sql2);
                ResultSet rsResultSet=pStemt2.executeQuery();
                int size = rsmd.getColumnCount();    //統計列
                colNames = new String[size];
                colTypes = new String[size];
                colSizes = new int[size];
                colComment = new String[size];
                int j=0;
                while (rsResultSet.next()) {
                    //System.out.println(rsResultSet.getObject(9));
                    colComment[j]=rsResultSet.getObject(9).toString();
                    j++;
                }
                for (int i = 0; i < size; i++) {
                    colNames[i] = rsmd.getColumnName(i + 1);
                    colTypes[i] = rsmd.getColumnTypeName(i + 1);
                    if (colTypes[i] .equals("INT")){
                        colTypes[i] = "INTEGER";
                    }
                    colSizes[i] = rsmd.getColumnDisplaySize(i + 1);
                }

                createExcel();

            } catch (Exception e) {
                e.printStackTrace();
            } finally{
                try {
                    if (con != null){
                        con.close();
                    }
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }
        System.out.println("生成完畢！");
    }


    public void createExcel() throws Exception{
        //獲取Excel模版文件目錄
        String path = "D:/template.xlsx";
        path = path.replaceAll("%20", " ");
        try(InputStream fileInputStream = new FileInputStream(path);
            //經過Excel模板目錄獲取Excel模版文件
            XSSFWorkbook workbook1 = new XSSFWorkbook(OPCPackage.open(fileInputStream));
            //利用POI3.8及其以上，每一個Sheet能夠存1,048,576行數據，每行能夠有16,384列數據
            Workbook workbook = new SXSSFWorkbook(workbook1, 100)){
            //重命名sheet工做表名稱：第幾個工做表
            workbook.setSheetName(0, tableName);
            //建立sheet工做表
            SXSSFSheet sheet = (SXSSFSheet) workbook.getSheetAt(0);
            //從模板sheet工做表第幾行開始插入（注意行、列、單元格都是從0開始數）
            int startRow = 1;
            for (int i = 0;i<colSizes.length;i++){
                Row row = sheet.createRow(startRow++);
                row.createCell(1).setCellValue(colNames[i]);
                row.createCell(2).setCellValue(colTypes[i] + "(" + colSizes[i] + ")");
                row.createCell(3).setCellValue(colComment[i]);
            }
            try(ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
                workbook.write(outputStream);
                //輸出目錄
                String filePath = "D:/" + tableName + ".xlsx";
                File file = new File(filePath);
                try(FileOutputStream fileOutputStream = new FileOutputStream(file);
                    ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(outputStream.toByteArray())) {
                    byte[] bytes = new byte[1024];
                    int len;
                    while ((len = byteArrayInputStream.read(bytes)) != -1){
                        fileOutputStream.write(bytes,0,len);
                    }
                }
            }
        }
    }

    public static void main(String[] args) {
        TableToExcel tableToExcel = new TableToExcel();
        List<String> dataList = new ArrayList<>();
        //添加表名
        dataList.add("operation_log");
        tableToExcel.genEntitySomeTable(dataList);
    }

}

數據庫表結構：

CREATE TABLE `operation_log` (
  `ID` bigint(20) NOT NULL AUTO_INCREMENT COMMENT 'ID',
  `DATA_ID` bigint(20) DEFAULT NULL COMMENT '操做數據ID',
  `TABLE_NAME` varchar(64) DEFAULT NULL COMMENT '對應表名',
  `MODULE_NAME` varchar(64) DEFAULT NULL COMMENT '模塊名稱',
  `OPERATION` varchar(64) DEFAULT NULL COMMENT '具體操做',
  `TYPE` int(11) DEFAULT NULL COMMENT '類型:一、INFO 二、DEBUG 三、ERROR',
  `MESSAGE` varchar(5120) DEFAULT NULL COMMENT '信息',
  `OPERATION_USER` varchar(32) DEFAULT NULL COMMENT '操做人',
  `ROW_STATUS` int(8) DEFAULT NULL COMMENT '記錄狀態',
  `MODIFY_TIME` bigint(20) DEFAULT NULL COMMENT '修改時間',
  `CREATE_TIME` bigint(20) DEFAULT NULL COMMENT '建立時間',
  PRIMARY KEY (`ID`)
) ENGINE=InnoDB  DEFAULT CHARSET=utf8 COMMENT '操做日誌表';

 

模板excel:

生成的excel: