闵雇 发表于 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;
    }
来源:程序园用户自行投稿发布,如果侵权,请联系站长删除
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!

寥唏 发表于 2025-11-5 10:28:47

前排留名,哈哈哈

敞撬 发表于 2025-12-6 06:45:08

收藏一下   不知道什么时候能用到

巴沛若 发表于 2025-12-10 00:35:22

鼓励转贴优秀软件安全工具和文档!

副我 发表于 2025-12-10 10:49:28

谢谢楼主提供!

奸轲嫣 发表于 2025-12-14 01:28:39

热心回复!

阴昭昭 发表于 2025-12-15 06:16:59

收藏一下   不知道什么时候能用到

古修蟑 发表于 2025-12-25 13:40:44

谢谢分享,试用一下

暴灵珊 发表于 2025-12-29 06:21:33

不错,里面软件多更新就更好了

蚬蕞遂 发表于 2026-1-6 10:55:18

过来提前占个楼

米榜饴 发表于 2026-1-12 13:02:30

yyds。多谢分享

碛物 发表于 2026-1-13 21:31:19

感谢分享,学习下。

崔瑜然 发表于 2026-1-14 02:53:44

不错,里面软件多更新就更好了

梳踟希 发表于 2026-1-14 15:02:47

感谢分享

明思义 发表于 2026-1-14 22:20:03

收藏一下   不知道什么时候能用到

盖彗云 发表于 2026-1-14 22:20:26

谢谢分享,辛苦了

百杲憔 发表于 2026-1-20 08:07:31

不错,里面软件多更新就更好了

各卧唯 发表于 2026-1-20 08:56:25

鼓励转贴优秀软件安全工具和文档!

蚬蕞遂 发表于 2026-1-21 11:14:22

这个好,看起来很实用

材部 发表于 2026-1-22 22:23:14

这个有用。
页: [1] 2
查看完整版本: java代码实现自动生成数据库表er图