java代碼實現自動生成資料庫表er圖

最近有同事看到位元組跳動產品設計文檔里有資料庫表er圖。就想問問又沒有現成的工具也給直接生成一個er圖，經查找驗證發現並沒有。因為現在表關係都是用的邏輯外鍵而非物理外鍵約束的，所以像navicat等工具就算生成了也沒有描述關係的連接線。那麼為了滿足需求，這邊就略微出手寫了個代碼實現er關係的工具。（註意：這裡只有後端部分，如需更直觀使用請配合前端進行頁面渲染）

一：首先獲取資料庫的所有表名和表的欄位sql代碼如下：

    <!--  獲取所有表名  -->
    <select id="showTables" resultType="java.lang.String">
        SHOW TABLES;
    </select>

    <!--根據表名獲取該表下的所有欄位-->
    <select id="showColumns" resultType="com.alibaba.fastjson.JSONObject">
        SHOW COLUMNS  FROM ${tableName};
    </select>

二：mapper介面層代碼如下：

    /**
     * 獲取所有表名
     * @return  該資料庫下的所有表名
     */
    List<String> showTables();

    /**
     * 根據表名獲取該表下的所有欄位
     * @param tableName 要獲取欄位的表名
     * @return          該表下的所有欄位
     */
    List<JSONObject> showColumns(@Param("tableName") String tableName);

三：業務代碼如下：（為了方便看這裡省略業務層直接在controller實現）

　  @Autowired
    private ERMapper erMapper;
    @GetMapping("/getEr")
    @ApiOperation(value = "獲取er關係")
    public Map getEr(){
        List<String> list = erMapper.showTables();
        Map<String, List<String>> map = new HashMap<>();
        Map<String, String> tableMap = new HashMap<>();
        List<String> ts = new ArrayList<>();
        for (String s : list) {
            String table = s.replace("erp_", "");
            ts.add(table);
            tableMap.put(table, s);
        }
        for (String s : list) {
            List<JSONObject> jsonList = erMapper.showColumns(s);
            String erp_ = s.replace("erp_", "");
            ArrayList<String> tList = new ArrayList<>();
            for (JSONObject jsonObject : jsonList) {
                String field = jsonObject.getString("Field");
                if (field.endsWith("_id")){
                    String replaceId = field.replace("_id", "");
                    if (ts.contains(replaceId)){
                        tList.add(tableMap.get(replaceId));
                    }
                }
            }
            map.put(s, tList);
            log.info("s:{}", erp_);
        }

        return map;
    }