MySql之自动生成CRUD代码

MySql之自动生成CRUD代码

MyBatis能够通过获取MySql中的information_schema从而获取表的字段等信息,最后通过这些信息生成代码。 笔者受此启发,将MyBatis-Generator中的核心结构体剥离出来,写成了能自动生成简单CRUD的工具。

自动生成代码原理图

information_schema

mysql本身存在一个information_schema,记录了所有的元数据信息,主要的几个有: schema表:当前mysql实例中所有数据库的信息。 COLUMNS表:关联了所有表和其中列的信息。 TABLES表:提供了关于数据库中的表的信息。 ......

jdbc中的MetaData

jdbc提供了非常方便的工具帮助我们获取这些元数据信息,就是MetaData。获取MetaData的代码如下:

 Connection connection = getConnection();   
 DatabaseMetaData metaData = connection.getMetaData();

metaData获取一张表中的所有字段

通过metaData.getColumns方法在指定了schema和table后可以很方便的获取一张表中的所有字段,代码如下:

    private void caculateColumns(DatabaseMetaData metaData) {

        ResultSet rs = null;
        try {
            rs = metaData.getColumns("", introspectedTable.getIntrospectedSchema(),
                    introspectedTable.getIntrospectedTableName(), null);

            while (rs.next()) {
                IntrospectedColumn introspectedColumn = new IntrospectedColumn();
                introspectedColumn.setJdbcType(rs.getInt("DATA_TYPE")); //$NON-NLS-1$
                introspectedColumn.setLength(rs.getInt("COLUMN_SIZE")); //$NON-NLS-1$
                introspectedColumn.setActualColumnName(rs.getString("COLUMN_NAME")); //$NON-NLS-1$
                introspectedColumn.setNullable(rs.getInt("NULLABLE") == DatabaseMetaData.columnNullable); //$NON-NLS-1$
                introspectedColumn.setScale(rs.getInt("DECIMAL_DIGITS")); //$NON-NLS-1$
                introspectedColumn.setRemarks(rs.getString("REMARKS")); //$NON-NLS-1$
                introspectedColumn.setDefaultValue(rs.getString("COLUMN_DEF")); //$NON-NLS-1$

        } catch (SQLException e) {
            e.printStackTrace();
        }

    }

metaData获取表中的主键

同样的通过getPrimaryKeys可以很轻松的计算其主键,注意主键可能是联合主键: 对应的代码如下:

       ResultSet rs = null;
        String cataLog;

        try {
            rs = metaData.getPrimaryKeys("", introspectedTable.getIntrospectedSchema(),
                    introspectedTable.getIntrospectedTableName());
            Map<Short, String> keyColumns = new TreeMap<Short, String>();
            while (rs.next()) {
                String columnName = rs.getString("COLUMN_NAME");
                short keySeq = rs.getShort("KEY_SEQ");
                keyColumns.put(keySeq, columnName);
            }
            for (String columnName : keyColumns.values()) {
                introspectedTable.addPrimaryKeyColumn(columnName);
            }
        } catch (SQLException e) {
            e.printStackTrace();
            System.exit(1);
        } finally {
            closeResultSet(rs);
        }

将meta信息组织成结构体

有了上面的信息我们就可以将一张表中的所有信息用java结构体表现出来: Table表Java结构体:

public class IntrospectedTable {

    protected List<IntrospectedColumn> primaryKeyColumns;
    protected List<IntrospectedColumn> baseColumns;
    protected List<IntrospectedColumn> blobColumns;

    protected String introspectedSchema;
    private String introspectedCatalog;
    private String introspectedTableName;

字段表Java结构体属性过多,在此不一一赘述,详情请见github

Velocity渲染:

有了上述的元数据结构体后,就可以用Velocity渲染,例如如下的Velocity模板:

    public ${name} get${name}(${primaryType} ${primaryKey}) {

        return mapper.selectByPrimaryKey(${primaryKey});

    }

就会被渲染成(假设primaryKey是id,tableName是Archer,appname是codegen):

package com.alchemystar.codegen.dal.dao.auto;
import com.alchemystar.codegen.dal.mapper.auto.ArcherMapper;

public class AutoArcherDao {
    public Archer getArcher(Long id) {

        return mapper.selectByPrimaryKey(id);
     }
}

如法炮制,就可以通过元数据信息生成不同的方法。从而生成想要的代码。

github链接

https://github.com/alchemystar/codegen

原文链接

https://my.oschina.net/alchemystar/blog/853550

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏进击的程序猿

doctrine缘来 之 造轮子

本系列是读php data persistence with doctrine2 orm的笔记,本文是第一篇:自己造轮子。

851
来自专栏有趣的django

Django REST framework+Vue 打造生鲜超市(四)

五、商品列表页 5.1.django的view实现商品列表页 (1)goods/view_base.py 在goods文件夹下面新建view_base.py,为...

2.2K9
来自专栏芋道源码1024

MyBastis 三种批量插入方式的性能比较

数据库使用的是sqlserver,JDK版本1.8,运行在SpringBoot环境下

4753
来自专栏二进制文集

Struts1 增删改查

本篇文章介绍如何使用Struts1进行增删改查,仅是对自己学习的记录,并没有深入分析每个模块(不重复造轮子O(∩_∩)O~)。

4433
来自专栏.NET开发者社区

一步一步创建ASP.NET MVC5程序[Repository+Autofac+Automapper+SqlSugar](二)

在本系列第一篇《一步一步创建ASP.NET MVC5程序[Repository+Autofac+Automapper+SqlSugar](一)》中,我为大家介绍...

30210
来自专栏程序员的酒和故事

Go实战--golang中使用RethinkDB(gorethink/gorethink.v3)

生命不止,继续go go go !!! 关于golang中操作数据库,曾经介绍了不少: Go实战–go语言操作sqlite数据库(The way to go) ...

5818
来自专栏后台开发+音视频+ffmpeg

dpvs源码分析

dpvs是爱奇艺开源的,它是一款基于dpdk的高性能4层负载均衡器。源自于LVS和改版后的alibaba/LVS. dpvs即dpdk-lvs. 等多关于dpv...

1.2K2
来自专栏MasiMaro 的技术博文

windows 下文件的高级操作

本文主要说明在Windows下操作文件的高级方法,比如直接读写磁盘,文件的异步操作,而文件普通的读写方式在网上可以找到一大堆资料,在这也就不再进行专门的说明。

2713
来自专栏Spark学习技巧

重要 | mr使用hcatalog读写hive表

企业中,由于领导们的要求,hive中有数据存储格式很多时候是会变的,比如为了优化将tsv,csv格式改为了parquet或者orcfile。那么这个时候假如是m...

2112
来自专栏一“技”之长

使用iOS原生sqlite3框架对sqlite数据库进行操作

      sqlite数据库是一种小型数据库,由于其小巧与简洁,在移动开发领域应用深广,sqlite数据库有一套完备的sqlite语句进行管理操作,一些常用的...

921

扫码关注云+社区

领取腾讯云代金券