java代码实现自动生成数据库表er图

最近有同事看到字节跳动产品设计文档里有数据库表er图。就想问问又没有现成的工具也给直接生成一个er图，经查找验证发现并没有。因为现在表关系都是用的逻辑外键而非物理外键约束的，所以像navicat等工具就算生成了也没有描述关系的连接线。那么为了满足需求，这边就略微出手写了个代码实现er关系的工具。（注意：这里只有后端部分，如需更直观使用请配合前端进行页面渲染）
一：首先获取数据库的所有表名和表的字段sql代码如下：

1.    
2.     <select id="showTables" resultType="java.lang.String">
3.         SHOW TABLES;
4.     </select>
6.    
7.     <select id="showColumns" resultType="com.alibaba.fastjson.JSONObject">
8.         SHOW COLUMNS  FROM ${tableName};
9.     </select>

复制代码

二：mapper接口层代码如下：

1.     /**
2.      * 获取所有表名
3.      * @return  该数据库下的所有表名
4.      */
5.     List<String> showTables();
7.     /**
8.      * 根据表名获取该表下的所有字段
9.      * @param tableName 要获取字段的表名
10.      * @return          该表下的所有字段
11.      */
12.     List<JSONObject> showColumns(@Param("tableName") String tableName);

复制代码

三：业务代码如下：（为了方便看这里省略业务层直接在controller实现）

1. 　  @Autowired
2.     private ERMapper erMapper;
3.     @GetMapping("/getEr")
4.     @ApiOperation(value = "获取er关系")
5.     public Map getEr(){
6.         List<String> list = erMapper.showTables();
7.         Map<String, List<String>> map = new HashMap<>();
8.         Map<String, String> tableMap = new HashMap<>();
9.         List<String> ts = new ArrayList<>();
10.         for (String s : list) {
11.             String table = s.replace("erp_", "");
12.             ts.add(table);
13.             tableMap.put(table, s);
14.         }
15.         for (String s : list) {
16.             List<JSONObject> jsonList = erMapper.showColumns(s);
17.             String erp_ = s.replace("erp_", "");
18.             ArrayList<String> tList = new ArrayList<>();
19.             for (JSONObject jsonObject : jsonList) {
20.                 String field = jsonObject.getString("Field");
21.                 if (field.endsWith("_id")){
22.                     String replaceId = field.replace("_id", "");
23.                     if (ts.contains(replaceId)){
24.                         tList.add(tableMap.get(replaceId));
25.                     }
26.                 }
27.             }
28.             map.put(s, tList);
29.             log.info("s:{}", erp_);
30.         }
32.         return map;
33.     }

复制代码

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！