MySql之自动生成CRUD代码

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

自动生成代码原理图

information_schema

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

jdbc中的MetaData

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

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

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

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

1    private void caculateColumns(DatabaseMetaData metaData) {
2
3        ResultSet rs = null;
4        try {
5            rs = metaData.getColumns("", introspectedTable.getIntrospectedSchema(),
6                    introspectedTable.getIntrospectedTableName(), null);
7
8            while (rs.next()) {
9                IntrospectedColumn introspectedColumn = new IntrospectedColumn();
10                introspectedColumn.setJdbcType(rs.getInt("DATA_TYPE")); //$NON-NLS-1$
11                introspectedColumn.setLength(rs.getInt("COLUMN_SIZE")); //$NON-NLS-1$
12                introspectedColumn.setActualColumnName(rs.getString("COLUMN_NAME")); //$NON-NLS-1$
13                introspectedColumn.setNullable(rs.getInt("NULLABLE") == DatabaseMetaData.columnNullable); //$NON-NLS-1$
14                introspectedColumn.setScale(rs.getInt("DECIMAL_DIGITS")); //$NON-NLS-1$
15                introspectedColumn.setRemarks(rs.getString("REMARKS")); //$NON-NLS-1$
16                introspectedColumn.setDefaultValue(rs.getString("COLUMN_DEF")); //$NON-NLS-1$
17
18        } catch (SQLException e) {
19            e.printStackTrace();
20        }
21
22    }

metaData获取表中的主键

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

1       ResultSet rs = null;
2        String cataLog;
3
4        try {
5            rs = metaData.getPrimaryKeys("", introspectedTable.getIntrospectedSchema(),
6                    introspectedTable.getIntrospectedTableName());
7            Map<Short, String> keyColumns = new TreeMap<Short, String>();
8            while (rs.next()) {
9                String columnName = rs.getString("COLUMN_NAME");
10                short keySeq = rs.getShort("KEY_SEQ");
11                keyColumns.put(keySeq, columnName);
12            }
13            for (String columnName : keyColumns.values()) {
14                introspectedTable.addPrimaryKeyColumn(columnName);
15            }
16        } catch (SQLException e) {
17            e.printStackTrace();
18            System.exit(1);
19        } finally {
20            closeResultSet(rs);
21        }

将meta信息组织成结构体

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

1public class IntrospectedTable {
2
3    protected List<IntrospectedColumn> primaryKeyColumns;
4    protected List<IntrospectedColumn> baseColumns;
5    protected List<IntrospectedColumn> blobColumns;
6
7    protected String introspectedSchema;
8    private String introspectedCatalog;
9    private String introspectedTableName;

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

Velocity渲染:

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

1    public ${name} get${name}(${primaryType} ${primaryKey}) {
2
3        return mapper.selectByPrimaryKey(${primaryKey});
4
5    }

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

1package com.alchemystar.codegen.dal.dao.auto;
2import com.alchemystar.codegen.dal.mapper.auto.ArcherMapper;
3
4public class AutoArcherDao {
5    public Archer getArcher(Long id) {
6
7        return mapper.selectByPrimaryKey(id);
8     }
9}

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

公众号

关注笔者公众号，获取更多干货文章:

github链接

https://github.com/alchemystar/codegen

原文链接

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