1. 背景
要画ER图，一个个打费时费力，StarUML文件打开是json。那么就有可能自动生成。
2. 效果
把表结构生成好，自己只要维护关系即可。
3. 代码
import lombok.Data; import java.io.FileWriter; import java.io.IOException; import java.sql.*; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; /** * @author liwei * @version 1.0 * @className MySqlToStarUML * @date 2022/9/21 22:47 */ public class MySqlToStarUML { /** * 自动生成代码入口 * * @author liwei * @date 2022-09-25 00:58:45 * @param args * @return void */ public static void main(String[] args) { localTest(); } public static void localTest() { String driver = "com.mysql.cj.jdbc.Driver"; String url = "jdbc:mysql://localhost:3306/test_nacos?useUnicode=true&characterEncoding=UTF-8"; String userName = "root"; String password = "密码"; String filePath = "D:\test_nacos.mdj"; List<Table> tableList = getTableList(driver, url, userName, password); saveAsFileWriter(filePath, getProject(url, tableList)); System.out.println("===============生成成功================"); } /** * 获取项目字符串 * * @author liwei * @date 2023-03-16 18:37:01 * @param url * 地址 * @param tableList * 表集合 * @return {@link String} */ public static String getProject(String url, List<Table> tableList) { String database = getDBByUrl(url); Project project = new Project(); // 不能使用中文 project.setName(database); project.set_id("AAAAAA_Project"); List<OwnedElement> erddatamodels = new ArrayList<>(); List<OwnedElement> ownedElements = new ArrayList<>(); ERDDataModel erdDataModel = new ERDDataModel(); erdDataModel.setName("Data Model1"); erdDataModel.set_id("AAAAAA_DataModel1"); erdDataModel.set_parent(project.get_id()); erddatamodels.add(erdDataModel); ERDDiagram erdDiagram = new ERDDiagram(); erdDiagram.setName("ERDDiagram1"); erdDiagram.set_id("AAAAAA_ERDDiagram1"); erdDiagram.set_parent(erdDataModel.get_id()); ownedElements.add(erdDiagram); for (Table table : tableList) { table.set_parent(erdDataModel.get_id()); } ownedElements.addAll(tableList); erdDataModel.setOwnedElements(ownedElements); project.setOwnedElements(erddatamodels); return project.toString(); } /** * 通过url获取数据库 * * @author liwei * @date 2022-09-23 09:21:09 * @param url * 地址 * @return {@link String} */ public static String getDBByUrl(String url) { if (null == url || url.isEmpty()) { throw new RuntimeException("地址为空"); } if (url.indexOf(":") == 0 && url.length() <= 1) { throw new RuntimeException("地址有误"); } while (url.indexOf(":") > 0) { url = url.substring(url.indexOf(":") + 1); } if (url.indexOf("?") > 0) { url = url.substring(0, url.indexOf("?")); } if (url.indexOf("/") > 0) { url = url.substring(url.indexOf("/") + 1); } return url; } /** * 保存内容到文件 * * @author liwei * @date 2022-11-22 14:19:47 * @param filePath * 文件路径 * @param content * 内容 * @return void */ private static void saveAsFileWriter(String filePath, String content) { FileWriter fwriter = null; try { fwriter = new FileWriter(filePath); fwriter.write(content); } catch (IOException ex) { ex.printStackTrace(); } finally { try { if (null != fwriter) { fwriter.flush(); fwriter.close(); } } catch (IOException ex) { ex.printStackTrace(); } } } /** * 获取表集合 * * @author liwei * @date 2022-11-22 14:20:24 * @param driver * 驱动 * @param url * 连接 * @param userName * 账号 * @param password * 密码 * @return {@link List< Table>} */ private static List<Table> getTableList(String driver, String url, String userName, String password) { Connection connection; try { Class.forName(driver); connection = DriverManager.getConnection(url, userName, password); } catch (ClassNotFoundException e) { throw new RuntimeException("加载驱动失败，找不到：" + driver); } catch (SQLException e) { throw new RuntimeException("获取数据库连接失败，请检查配置和日志", e); } String database = getDBByUrl(url); String sqlTable = String.format("select * from information_schema.tables where TABLE_SCHEMA='%s'", database); List<Map<String, String>> mapList = runSql(connection, sqlTable); List<Table> tableList = new ArrayList<>(); for (int i = 0; i < mapList.size(); i++) { Map<String, String> map = mapList.get(i); Table table = new Table(); String tableId = String.valueOf(i + 1); table.set_id(tableId); table.setName(map.get("TABLE_NAME")); table.setDocumentation(map.get("TABLE_COMMENT")); String sqlColumn = String.format("select * from information_schema.columns where TABLE_SCHEMA='%s' and TABLE_NAME = '%s'", database, table.getName()); List<Map<String, String>> mapList2 = runSql(connection, sqlColumn); List<Column> columnList = new ArrayList<>(); for (Map<String, String> stringMap : mapList2) { Column column = new Column(); column.setTableId(tableId); column.setName(stringMap.get("COLUMN_NAME")); column.setType(stringMap.get("DATA_TYPE")); String columnType = stringMap.get("COLUMN_TYPE"); if (columnType.indexOf("(") > 0) { column.setLength(columnType.substring(columnType.indexOf("(") + 1, columnType.indexOf(")"))); } else { column.setLength(stringMap.get("CHARACTER_MAXIMUM_LENGTH")); } column.setOrdinalPosition(Integer.parseInt(stringMap.get("ORDINAL_POSITION"))); column.setNullable("YES".equals(stringMap.get("IS_NULLABLE"))); column.setPrimaryKey("PRI".equals(stringMap.get("COLUMN_KEY"))); column.setUnique("UNI".equals(stringMap.get("COLUMN_KEY"))); columnList.add(column); } columnList.sort((c1, c2) -> c1.ordinalPosition - c2.getOrdinalPosition()); table.setColumns(columnList); tableList.add(table); } close(null, connection, null); return tableList; } /** * 关闭连接 * * @author liwei * @date 2022-09-23 09:21:53 * @param pstmt * 预编译 * @param conn * 连接 * @param rs * 结果集 * @return void */ public static void close(PreparedStatement pstmt, Connection conn, ResultSet rs) { try { if(null != rs) { rs.close(); rs = null; } if(null != pstmt) { pstmt.close(); pstmt = null; } if(null != conn) { conn.close(); conn = null; } } catch (SQLException e) { throw new RuntimeException("关闭数据库连接异常", e); } } /** * 运行sql * * @author liwei * @date 2022-11-22 14:21:40 * @param conn * 连接 * @param sql * 执行的sql * @return {@link List< Map< String, String>>} */ public static List<Map<String, String>> runSql(Connection conn, String sql) { if (null == sql || sql.isEmpty()) { throw new RuntimeException("执行的sql不可为空"); } List<Map<String, String>> list = new ArrayList<>(); if(null == conn) { throw new RuntimeException("获取数据库连接失败"); } PreparedStatement pstmt = null; ResultSet rs = null; try { pstmt = conn.prepareStatement(sql); rs = pstmt.executeQuery(); while (rs.next()) { Map<String, String> map = new HashMap<>(); ResultSetMetaData metaData = rs.getMetaData(); int columnCount = metaData.getColumnCount(); for (int i = 1; i <= columnCount; i++) { map.put(metaData.getColumnName(i), rs.getString(i)); } list.add(map); } } catch (SQLException e) { throw new RuntimeException("执行sql异常", e); } finally { close(pstmt, null, rs); } return list; } @Data static class Table extends OwnedElement { private String _type = "ERDEntity"; private String documentation; private List<Column> columns; @Override public String toString() { return "{" + ""_type":"" + _type + '"' + ", "_id":"" + super._id + '"' + ", "_parent":{"$ref":"" + super._parent + ""}" + ", "name":"" + super.name + " " + documentation + '"' + ", "documentation":"" + documentation + '"' + ", "columns":" + columns + '}'; } } @Data static class Column { private String _type = "ERDColumn"; private String name; private String tableId; private String type; private String length; // UNI、PRI private String columnKey; private int ordinalPosition; private Boolean primaryKey; private Boolean unique; private Boolean nullable; @Override public String toString() { return "{" + ""_type":"ERDColumn"" + ", "_parent":{"$ref":"" + tableId + ""}" + ", "name":"" + name + '"' + ", "type":"" + type + '"' + (null != length ? ", "length":"" + length + '"' : "") + (primaryKey ? ", "primaryKey":"" + primaryKey + """ : "") + (unique ? ", "unique":"" + unique + """ : "") + (nullable ? ", "nullable":"" + nullable + """ : "") + '}'; } } @Data static class Project extends OwnedElement { private String _type = "Project"; @Override public String toString() { return "{" + ""_type":"Project"" + ", "_id":"" + super._id + '"' + ", "name":"" + super.name + '"' + ", "ownedElements":" + super.ownedElements + "}"; } } @Data static class OwnedElement { private String _type; private String _id; private String _parent; private String name; private List<OwnedElement> ownedElements; } @Data static class ERDDataModel extends OwnedElement { private String _type = "ERDDataModel"; @Override public String toString() { return "{" + ""_type":"ERDDataModel"" + ", "_id":"" + super._id + '"' + ", "_parent":{"$ref":"" + super._parent + ""}" + ", "name":"" + super.name + '"' + ", "ownedElements":" + super.ownedElements + "}"; } } @Data static class ERDDiagram extends OwnedElement { private String _type = "ERDDiagram"; @Override public String toString() { return "{" + ""_type":"ERDDiagram"" + ", "_id":"" + super._id + '"' + ", "_parent":{"$ref":"" + super._parent + ""}" + ", "name":"" + super.name + '"' + (null != super.ownedElements ? ", "ownedViews":" + super.ownedElements : "") + "}"; } } @Data static class ERDEntityView extends OwnedElement { private String _type = "ERDEntityView"; private String tableId; @Override public String toString() { return "{" + ""_type":"ERDEntityView"" + ", "model":{"" + tableId + ""}" + (null != super.ownedElements ? ", "subViews":" + super.ownedElements : "") + "}"; } } } 
