|
我平时创建数据库表的时候，都是习惯先用记事本将要用到的字段列出来，写好字段名和注释，像这种  然后用代码读取这个文本，通过中文冒号：分隔，获取到名称和注释，再通过create建表语句，将字段名、注释拼接起来，生成sql输出。这样生成的没有考虑到字段类型、长度，需要建完表后再设置。
然后今天闲来无事，想着得规范一下，在列要用到的字段的时候，就将类型和长度也设计好，执行完sql之后就不用一个一个去改了。所以我用的是excel表格（比较好看），像这样  
行数不限，列数只有四列，第一行必须是 表注释:表名，用英文冒号隔开（看自己喜欢用什么符号，代码里改一下就行），可以有多张sheet。
先看看生成的效果：
 
代码：
cn.hutool
hutool-all
5.7.22
cn.afterturn
easypoi-base
4.4.0
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 result = readExcel(filePath,sheetName);
System.out.println("================================开始生成SQL，共"+result.size()+"张表================================");
StringBuilder sb = new StringBuilder();
StringBuilder indexSb = null;
for (Map map : result) {
JSONObject json = new JSONObject(true);// 有序
List list = (List) 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 //设置索引
//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 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 readExcel(String filePath,String sheetName) {
List 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 // 只解析输入的指定sheet
String[] split = sheetName.split(",");
for (int i = 0; i
Map map = new HashMap();
List 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
ExcelEntity entity = new ExcelEntity();
//列数：5列，分别是字段名、字段类型、字段注释、字段属性、字段索引
for (int j = 0; j
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;
}
}
|