闵雇 发表于 2025-6-9 08:34:19

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 COLUMNSFROM ${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;
    }
来源:程序园用户自行投稿发布,如果侵权,请联系站长删除
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
页: [1]
查看完整版本: java代码实现自动生成数据库表er图