生成mybatis中所需的javabean和基础配置信息

 如果在项目中使用到了mybatis，那么就需要编写与数据库表对应的javabean对象，还有数据库表的数据列和javabean对象中字段的映射配置，以及一些通用的插入更新数据的配置，这些都可以通过工具自动生成。

 

import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Types;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Properties;

import com.cogcn.framework.util.CommonUtil;

/**
 * 把数据库中的表转化为java对象
 * 
 * @author tuozixuan
 * 
 */
public class TableToJavaTool
{
    private static final String DRIVER_NAME_ORACLE = "oracle.jdbc.driver.OracleDriver";

    // 数据库连接-用户名
    private String user;

    // 数据库连接-密码
    private String password;

    // 数据库连接-URL
    private String dbUrl;

    // 表名
    private String tableName;
    
    // 表类别名称
    private String catalog;
    
    private final List<String> primaryKeyList = new ArrayList<String>();
    
    private List<Map<String, Object>> dataList = null;

    public TableToJavaTool()
    {

    }

    public TableToJavaTool(String user, String password, String dbUrl, String tableName)
    {
        this.user = user;
        this.password = password;
        this.dbUrl = dbUrl;
        this.tableName = tableName;
    }
    
    public void process()
    {
        dataList = readData(getTableName());
        
        createJavaBeanFile(getTableName());
        createMybatisColumnConfig(getTableName());
    }

    /**
     * 获取数据库连接
     * 
     * @return Connection 数据库连接对象
     */
    private Connection getConnection()
    {
        Connection conn = null;

        try
        {
            Properties props = new Properties();
            props.put("remarksReporting", "true");
            props.put("user", getUser());
            props.put("password", getPassword());

            Class.forName(DRIVER_NAME_ORACLE);
            conn = DriverManager.getConnection(getDbUrl(), props);
        }
        catch (Exception e)
        {
            e.printStackTrace();
        }

        return conn;
    }

    /**
     * 获取数据库指定表的列信息
     * 
     * @param tableName 表名
     * @return List<Map<String, Object>> 列信息列表
     */
    private List<Map<String, Object>> readData(String tableName)
    {
        Connection conn = getConnection();
        List<Map<String, Object>> dataList = new ArrayList<Map<String, Object>>();

        try
        {
            DatabaseMetaData dbmd = conn.getMetaData();

            ResultSet rs = dbmd.getColumns(getCatalog(), null, tableName, null);
            Map<String, Object> map = null;
            while (rs.next())
            {
                map = new HashMap<String, Object>();
                map.put("columnName", rs.getString("COLUMN_NAME"));
                map.put("dataType", rs.getInt("DATA_TYPE"));
                map.put("remarks", rs.getString("REMARKS"));
                dataList.add(map);
            }
            
            ResultSet rs1 = dbmd.getPrimaryKeys(getCatalog(), null, tableName);
            while (rs1.next())
            {
                primaryKeyList.add(rs1.getString("COLUMN_NAME"));
            }
        }
        catch (SQLException e)
        {
            e.printStackTrace();
        } finally {
            CommonUtil.closeConnection(conn);
        }
        return dataList;
    }

    public void createJavaBeanFile(String tableName)
    {
        StringBuffer jbString = new StringBuffer();
        jbString.append("public class ").append(tableName).append("\r\n");
        jbString.append("{").append("\r\n");

        if (dataList != null)
        {
            for (Map<String, Object> map : dataList)
            {
                String fieldName = getFieldName((String) map.get("columnName"));
                String javaType = getJavaType((Integer) map.get("dataType"));
                jbString.append("    // ").append(map.get("remarks")).append("\r\n");
                jbString.append("    private ").append(javaType).append(" ").append(fieldName).append(";").append("\r\n");
                jbString.append("\r\n");
            }

            for (Map<String, Object> map : dataList)
            {
                String fieldName = getFieldName((String) map.get("columnName"));
                String javaType = getJavaType((Integer) map.get("dataType"));
                jbString.append("    public ").append(javaType).append(" get").append(firstUpperCase(fieldName)).append("()").append("\r\n");
                jbString.append("    {").append("\r\n");
                jbString.append("        return ").append(fieldName).append(";").append("\r\n");
                jbString.append("    }").append("\r\n");

                jbString.append("\r\n");

                jbString.append("    public void set").append(firstUpperCase(fieldName)).append("(").append(javaType).append(" ").append(fieldName)
                        .append(")").append("\r\n");
                jbString.append("    {").append("\r\n");
                jbString.append("        this.").append(fieldName).append(" = ").append(fieldName).append(";").append("\r\n");
                jbString.append("    }").append("\r\n");

                jbString.append("\r\n");

            }
        }

        jbString.append("}");

        System.out.println(jbString.toString());
    }

    /**
     * 根据表名获取对应的JavaBean的名称<br/>
     * 
     * 
     * @param tableName 表名
     * @return String JavaBean
     */
    public static String getJavaBeanName(String tableName)
    {

        return tableName;
    }

    /**
     * 把以_分隔的列明转化为字段名
     * 
     * @param columnName 列名
     * @return String 字段名
     */
    private static String getFieldName(String columnName)
    {
        if (columnName == null)
        {
            return "";
        }

        StringBuffer fieldNameBuffer = new StringBuffer();

        boolean nextUpperCase = false;
        columnName = columnName.toLowerCase();
        for (int i = 0; i < columnName.length(); i++)
        {
            char c = columnName.charAt(i);

            if (nextUpperCase)
            {
                fieldNameBuffer.append(columnName.substring(i, i + 1).toUpperCase());
            }
            else
            {
                fieldNameBuffer.append(c);
            }

            if (c == '_')
            {
                nextUpperCase = true;
            }
            else
            {
                nextUpperCase = false;
            }
        }

        String fieldName = fieldNameBuffer.toString();
        fieldName = fieldName.replaceAll("_", "");
        return fieldName;
    }

    /**
     * 字符串的第一个字母大写
     * 
     * @param str 字符串
     * @return String 处理后的字符串
     */
    private static String firstUpperCase(String str)
    {
        if (str == null)
        {
            return "";
        }

        if (str.length() == 1)
        {
            str = str.toUpperCase();
        }
        else
        {
            str = str.substring(0, 1).toUpperCase() + str.substring(1);
        }
        return str;
    }

    /**
     * 将数据库列类型转换为java数据类型
     * 
     * @param dataType 列类型
     * @return String java数据类型
     */
    private static String getJavaType(int dataType)
    {
        String javaType = "";
        if (dataType == Types.INTEGER)
        {
            javaType = "int";
        }
        else if (dataType == Types.BIGINT)
        {
            javaType = "long";
        }
        else if (dataType == Types.CHAR || dataType == Types.VARCHAR || dataType == Types.NVARCHAR || dataType == Types.CLOB)
        {
            javaType = "String";
        }
        else if (dataType == Types.TINYINT)
        {
            javaType = "short";
        }
        else if (dataType == Types.FLOAT)
        {
            javaType = "float";
        }
        else if (dataType == Types.NUMERIC || dataType == Types.DECIMAL || dataType == Types.DOUBLE)
        {
            javaType = "double";
        }
        else if (dataType == Types.DATE || dataType == Types.TIMESTAMP)
        {
            javaType = "Date";
        }
        return javaType;
    }

    public void createMybatisColumnConfig(String tableName)
    {
        StringBuffer buffer = new StringBuffer();
        if (dataList != null)
        {
            buffer.append("<resultMap id=\"BaseResultMap\" type=\"").append(tableName).append("\"> ").append("\r\n");
            for (Map<String, Object> map : dataList)
            {
                // <result column="CI_TYP" jdbcType="CHAR" property="ciTyp" />
                String columnName = (String) map.get("columnName");
                String fieldName = getFieldName(columnName);
                String jdbcType = getMybatisJdbcType((Integer) map.get("dataType"));

                if (primaryKeyList.contains(columnName))
                {
                    buffer.append("    <id column=\"").append(columnName).append("\" ").append("jdbcType=\"").append(jdbcType)
                    .append("\" property=\"").append(fieldName).append("\" />").append("\r\n");
                } else {
                buffer.append("    <result column=\"").append(columnName).append("\" ").append("jdbcType=\"").append(jdbcType)
                        .append("\" property=\"").append(fieldName).append("\" />").append("\r\n");
                }
            }
            buffer.append("</resultMap>").append("\r\n");
        }

        buffer.append("<sql id=\"BaseColumnList\">").append("\r\n");
        int length = dataList.size();
        int count = 0;
        buffer.append("    ");
        for (Map<String, Object> map : dataList)
        {
            count++;
            buffer.append(map.get("columnName"));
            if (count != length)
            {
                buffer.append(", ");
            }
        }
        buffer.append("\r\n");
        buffer.append("</sql>").append("\r\n");
        
        // insert配置
        buffer.append("<insert id=\"insert\" parameterType=\"\">").append("\r\n");
        buffer.append("    insert into ").append(getTableName()).append("\r\n");
        buffer.append("    <trim prefix=\"(\" suffix=\")\" suffixOverrides=\",\">").append("\r\n");
        for (Map<String, Object> map : dataList)
        {
            String columnName = (String) map.get("columnName");
            String fieldName = getFieldName(columnName);
            buffer.append("        <if test=\"").append(fieldName).append(" != null\"> \r\n");
            buffer.append("            ").append(columnName).append(",").append("\r\n");
            buffer.append("        </if> \r\n");
        }
        buffer.append("    </trim>").append("\r\n");
        buffer.append("    <trim prefix=\"values (\" suffix=\")\" suffixOverrides=\",\"> \r\n");
        for (Map<String, Object> map : dataList)
        {
            String columnName = (String) map.get("columnName");
            String fieldName = getFieldName(columnName);
            String jdbcType = getMybatisJdbcType((Integer) map.get("dataType"));
            buffer.append("        <if test=\"").append(fieldName).append(" != null\"> \r\n");
            buffer.append("            #{").append(fieldName).append(",jdbcType=").append(jdbcType).append("}, \r\n");
            buffer.append("        </if> \r\n");
        }
        buffer.append("    </trim>").append("\r\n");
        
        // update配置
        buffer.append("<update id=\"update\" parameterType=\"java.util.Map\"> \r\n");
        buffer.append("    update ").append(getTableName()).append("\r\n");
        buffer.append("    <set>").append("\r\n");
        for (Map<String, Object> map : dataList)
        {
            String columnName = (String) map.get("columnName");
            String fieldName = getFieldName(columnName);
            String jdbcType = getMybatisJdbcType((Integer) map.get("dataType"));
            buffer.append("        <if test=\"").append(fieldName).append(" != null\"> \r\n");
            buffer.append("            ").append(columnName).append(" = ").append("#{").append(fieldName).append(",jdbcType=").append(jdbcType).append("}, \r\n");
            buffer.append("        </if> \r\n");
        }
        buffer.append("    </set> \r\n");
        buffer.append("    where ").append("\r\n");
        
//        for (String primaryKey : primaryKeyList)
//        {
//            buffer.append("        ").append(primaryKey).append(" = #{lnNo,jdbcType=CHAR}");
//        }
        
        buffer.append("</update>");
        

        System.out.println(buffer.toString());
    }

    /**
     * 根据列的类型，获取mybatis配置中的jdbcType
     * 
     * @param dataType 列的类型
     * @return String jdbcType
     */
    private static String getMybatisJdbcType(int dataType)
    {
        String jdbcType = "";
        if (dataType == Types.TINYINT)
        {
            jdbcType = "TINYINT";
        }
        else if (dataType == Types.SMALLINT)
        {
            jdbcType = "SMALLINT";
        }
        else if (dataType == Types.INTEGER)
        {
            jdbcType = "INTEGER";
        }
        else if (dataType == Types.BIGINT)
        {
            jdbcType = "BIGINT";
        }
        else if (dataType == Types.FLOAT)
        {
            jdbcType = "FLOAT";
        }
        else if (dataType == Types.DOUBLE)
        {
            jdbcType = "DOUBLE";
        }
        else if (dataType == Types.DECIMAL)
        {
            jdbcType = "DECIMAL";
        }
        else if (dataType == Types.NUMERIC)
        {
            jdbcType = "NUMERIC";
        }
        else if (dataType == Types.VARCHAR)
        {
            jdbcType = "VARCHAR";
        }
        else if (dataType == Types.NVARCHAR)
        {
            jdbcType = "NVARCHAR";
        }
        else if (dataType == Types.CHAR)
        {
            jdbcType = "CHAR";
        }
        else if (dataType == Types.NCHAR)
        {
            jdbcType = "NCHAR";
        }
        else if (dataType == Types.CLOB)
        {
            jdbcType = "CLOB";
        }
        else if (dataType == Types.BLOB)
        {
            jdbcType = "BLOB";
        }
        else if (dataType == Types.NCLOB)
        {
            jdbcType = "NCLOB";
        }
        else if (dataType == Types.DATE)
        {
            jdbcType = "DATE";
        }
        else if (dataType == Types.TIMESTAMP)
        {
            jdbcType = "TIMESTAMP";
        }
        else if (dataType == Types.ARRAY)
        {
            jdbcType = "ARRAY";
        }
        else if (dataType == Types.TIME)
        {
            jdbcType = "TIME";
        }
        else if (dataType == Types.BOOLEAN)
        {
            jdbcType = "BOOLEAN";
        }
        else if (dataType == Types.BIT)
        {
            jdbcType = "BIT";
        }
        else if (dataType == Types.BINARY)
        {
            jdbcType = "BINARY";
        }
        else if (dataType == Types.OTHER)
        {
            jdbcType = "OTHER";
        }
        else if (dataType == Types.REAL)
        {
            jdbcType = "REAL";
        }
        else if (dataType == Types.LONGVARCHAR)
        {
            jdbcType = "LONGVARCHAR";
        }
        else if (dataType == Types.VARBINARY)
        {
            jdbcType = "VARBINARY";
        }
        else if (dataType == Types.LONGVARBINARY)
        {
            jdbcType = "LONGVARBINARY";
        }

        return jdbcType;
    }

    public String getUser()
    {
        return user;
    }

    public void setUser(String user)
    {
        this.user = user;
    }

    public String getPassword()
    {
        return password;
    }

    public void setPassword(String password)
    {
        this.password = password;
    }

    public String getDbUrl()
    {
        return dbUrl;
    }

    public void setDbUrl(String dbUrl)
    {
        this.dbUrl = dbUrl;
    }

    public String getTableName()
    {
        return tableName;
    }

    public void setTableName(String tableName)
    {
        this.tableName = tableName;
    }
    
    public String getCatalog()
    {
        return catalog;
    }

    public void setCatalog(String catalog)
    {
        this.catalog = catalog;
    }
}

 

使用示例：

 

    public static void main(String[] args)
    {
         String dbUrl = "jdbc:oracle:thin:@10.10.16.80:1521/orcl";
         String userName = "user";
         String password = "password";
         String tableName = "MEMBER";
         TableToJavaTool tool = new TableToJavaTool(userName, password, dbUrl, tableName);
         // tool.setCatalog("me");
         tool.process();
    }