产品中心
package com.example.demo;import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;public class MysqlERGenerate {
public static void main(String[] args) throws IOException {
StringBuilder builder = new StringBuilder();
// 输入从Navicat导出的表结构sql文件 将文件读取出来 放入字符串中
InputStream is = Files.newInputStream(Paths.get("C:\\Users\\admin\\Desktop\\ddl.sql"));
String line;
BufferedReader reader = new BufferedReader(new InputStreamReader(is));
line = reader.readLine();
while (line != null) {
builder.append(line);
builder.append("\n");
line = reader.readLine();
}
reader.close();
is.close();
String sql = builder.toString();// 按照规律 使用CREATE TABLE进行分割 并删掉一个文件注释部分
String[] split = sql.split("CREATE TABLE");
List<String> list = new ArrayList<>(Arrays.asList(split));
list.remove(0);
// 使用开头两个小撇号进行截取 得到表名 转为Map<表名， SQL>
Map<String, String> collect = list.stream().collect(Collectors.toMap(k -> {
int firstIndex = k.indexOf("`");
return k.substring(++firstIndex, k.indexOf("`", firstIndex));
}, v -> v));// 需要创建外键的字段与对应的主表名称 Map<外键名, 外键主表名>
Map<String, String> foreignKey = new HashMap<>();
foreignKey.put("ticket_no", "ticket");
foreignKey.put("ticket_define_no", "ticket_define");
foreignKey.put("pro_no", "pro_main");
// 循环判断，生成外键SQL
Set<String> foreignKeyFields = foreignKey.keySet();
for (String mainTableName : collect.keySet()) {
String val = collect.get(mainTableName);
for (String field : foreignKeyFields) {
if (!mainTableName.equals(foreignKey.get(field)) && val.indexOf("`" + field + "`") > 0) {
String createForeignKeySql = String.format("alter table %s add foreign key %s(%s) references %s(%s);", mainTableName, mainTableName + field + System.currentTimeMillis(), field, foreignKey.get(field), field);
System.out.println(createForeignKeySql);
}
}
}
}
}运行效果：只复制出部分，实际远比这个多
alter table pro_param add foreign key pro_parampro_no1650765563395(pro_no) references pro_main(pro_no);
alter table pro_shop_priority_his_20200805 add foreign key pro_shop_priority_his_20200805pro_no1650765563423(pro_no) references pro_main(pro_no);alter table ticket_define_shop add foreign key ticket_define_shopticket_define_no1650765563423(ticket_define_no) references ticket_define(ticket_define_no);
alter table ticket_define_item add foreign key ticket_define_itemticket_define_no1650765563425(ticket_define_no) references ticket_define(ticket_define_no);alter table ticket_his_2019 add foreign key ticket_his_2019ticket_no1650765563432(ticket_no) references ticket(ticket_no);
alter table ticket_his_2018 add foreign key ticket_his_2018ticket_no1650765563433(ticket_no) references ticket(ticket_no);
