根据excel生成create建表SQL语句

Posted 2022-12-05 符华-

我平时创建数据库表的时候，都是习惯先用记事本将要用到的字段列出来，写好字段名和注释，像这种

然后用代码读取这个文本，通过中文冒号：分隔，获取到名称和注释，再通过create建表语句，将字段名、注释拼接起来，生成sql输出。这样生成的没有考虑到字段类型、长度，需要建完表后再设置。

然后今天闲来无事，想着得规范一下，在列要用到的字段的时候，就将类型和长度也设计好，执行完sql之后就不用一个一个去改了。所以我用的是excel表格（比较好看），像这样

行数不限，列数只有四列，第一行必须是 表注释:表名，用英文冒号隔开（看自己喜欢用什么符号，代码里改一下就行），可以有多张sheet。

先看看生成的效果：

代码：

import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.poi.excel.WorkbookUtil;
import lombok.Data;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.util.*;


public class GenerateSqlUtil 

    @Data
    public static class ExcelEntity 
        /**
         * 字段名
         */
        private String filedName;
        /**
         * 字段类型
         */
        private String filedType;
        /**
         * 字段注释
         */
        private String notes;
        /**
         * 字段属性
         */
        private String attribute;
        /**
         * 字段索引
         */
        private String index;
    


    public static void main(String[] args) throws IOException 
        generateSql();
    

    /**
     * 生成sql文件
     */
    private static void generateSql() throws IOException
        System.out.println("请输入excel路径：");
        Scanner sc = new Scanner(System.in);
        String filePath = sc.next();
        System.out.println("请输入要解析的sheet名称，多个用,号隔开，输入全部为全部解析");
        String sheetName = sc.next();
        List<Map<String, Object>> result = readExcel(filePath,sheetName);
        System.out.println("================================开始生成SQL，共"+result.size()+"张表================================");
        StringBuilder sb = new StringBuilder();
        StringBuilder indexSb = null;
        for (Map<String, Object> map : result) 
            JSONObject json = new JSONObject(true);// 有序
            List<ExcelEntity> list = (List<ExcelEntity>) map.get("list");
            String tableName = (String) map.get("tableName"); //表名
            String tableNotes = (String) map.get("tableNotes"); //注释
            sb.append(String.format("-- %s：%s\\n", tableName, tableNotes));
            sb.append(String.format("DROP TABLE IF EXISTS %s;\\n", tableName));
            sb.append(String.format("CREATE TABLE %s (\\n", tableName));
            int size = list.size();
            indexSb = new StringBuilder();
            for (int i = 0; i < size; i++) 
                ExcelEntity entity = list.get(i);
                sb.append(String.format("\\t%s %s", entity.getFiledName(), entity.getFiledType()));
                String type = entity.getFiledType();
                String attribute = entity.getAttribute();
                if (attribute.contains("主键")) sb.append(" PRIMARY KEY"); //设置主键
                else if (type.contains("char") || type.contains("text")) sb.append(" CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci"); //字符串、文本设置编码
                if (attribute.contains("必填")) sb.append(" NOT NULL COMMENT ");
                else sb.append(" NULL DEFAULT NULL COMMENT ");
                sb.append(String.format("'%s'", entity.getNotes()));
                if (StrUtil.isNotBlank(entity.getIndex()))  //设置索引
                    //INDEX `index_name`(`field1`, `field2`) USING BTREE
                    indexSb.append("\\tINDEX `");
                    String[] index = entity.getIndex().split(":");
                    indexSb.append(index[0]);
                    indexSb.append("`(`");
                    indexSb.append(index[1]);
                    indexSb.append("`)");
                    indexSb.append(" USING BTREE,\\n");
                
                if (i < size - 1 || indexSb.length() > 0) sb.append(",");
                sb.append("\\n");
                json.putOpt(entity.getFiledName(),entity.getNotes());
            
            if (indexSb.length() > 0) 
                sb.append(indexSb.substring(0, indexSb.lastIndexOf(",")));
                sb.append("\\n");
            
            sb.append(String.format(") ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_0900_ai_ci COMMENT = '%s' ROW_FORMAT = Dynamic;", tableNotes));
            sb.append("\\n\\n");
            System.out.println(tableNotes+"表："+json.toString());
        
        File file = new File("C:\\\\Users\\\\Admin\\\\Desktop\\\\1.sql");
        FileOutputStream fos1 = new FileOutputStream(file);
        OutputStreamWriter dos1 = new OutputStreamWriter(fos1);
        dos1.write(sb.toString());
        dos1.close();
        System.out.println("=========================SQL生成完成，是否继续？0 退出 1 继续=========================");
        String isExit = sc.next();
        if ("0".equals(isExit)) System.exit(0);
        generateSql();
    

    /**
     * 根据文件路径解析文件
     * @param filePath   文件路径
     * @param sheetName  要解析的sheet名称
     */
    private static List<Map<String, Object>> readExcel(String filePath,String sheetName) 
        List<Map<String, Object>> result = new ArrayList<>();
        File file = new File(filePath);
        Workbook book = WorkbookUtil.createBook(file);
        if ("全部".equals(sheetName)) // 全部解析
            int sheetNum = book.getNumberOfSheets();
            for (int i = 0; i < sheetNum; i++) 
                result.add(readExcel(book.getSheetAt(i)));
            
        else  // 只解析输入的指定sheet
            String[] split = sheetName.split(",");
            for (int i = 0; i < split.length; i++) 
                result.add(readExcel(book.getSheet(split[i])));
            
        
        return result;
    

    /**
     * 读取指定sheet的数据，获取表名，字段名、字段类型、字段注释
     */
    private static Map<String, Object> readExcel(Sheet sheet) 
        Map<String, Object> map = new HashMap<>();
        List<ExcelEntity> list = new ArrayList<>();
        //第一行是标题，默认是注释:表名
        Row titleRow = sheet.getRow(0);
        String title = titleRow.getCell(0).toString();
        map.put("tableName", title.split(":")[1]);
        map.put("tableNotes", title.split(":")[0]);
        //第二行是表头，不读取。第三行开始才是数据
        int lastRowNum = sheet.getLastRowNum();
        for (int i = 2; i <= lastRowNum; i++) 
            list.add(getEntity(sheet.getRow(i)));
        
        map.put("list", list);
        return map;
    

    private static ExcelEntity getEntity(Row row) 
        ExcelEntity entity = new ExcelEntity();
        //列数：5列，分别是字段名、字段类型、字段注释、字段属性、字段索引
        for (int j = 0; j < 5; j++) 
            Cell cell = row.getCell(j);
            if (!Objects.isNull(cell)) 
                String str = cell.toString();
                if (StrUtil.isNotBlank(str)) 
                    switch (j) 
                        case 0: entity.setFiledName(str); break;
                        case 1: entity.setFiledType(str); break;
                        case 2: entity.setNotes(str); break;
                        case 3: entity.setAttribute(str); break;
                        case 4: entity.setIndex(str); break;
                    
                
            
        
        return entity;