沫沫金-Mybatis工具类，生成dao层xml、mapper文件和实体类entity层-阿里云开发者社区

Mybatis Generator工具使用起来，总感觉不太灵活加上初次环境配置麻烦，特编写java文件单文件不依赖，直接生成。源码如下（此为Oracle数据库版）
 
        package 
        net.icarefx.booking.util; 
       
        import 
        java.io.BufferedWriter; 
       
        import 
        java.io.File; 
       
        import 
        java.io.FileOutputStream; 
       
        import 
        java.io.IOException; 
       
        import 
        java.io.OutputStreamWriter; 
       
        import 
        java.sql.Connection; 
       
        import 
        java.sql.DriverManager; 
       
        import 
        java.sql.PreparedStatement; 
       
        import 
        java.sql.ResultSet; 
       
        import 
        java.sql.SQLException; 
       
        import 
        java.util.ArrayList; 
       
        import 
        java.util.List; 
       
        import 
        org.apache.commons.lang.StringUtils; 
       
        /**
       
        * 代码生成MyBatis的实体类、dao层下实体映射(sqlmap)XML文件和Mapper类 
       
        * @author 沫沫金 
       
        * @email zl0828@yeah.net 
       
        * @WeiXin zl4828 
       
        */ 
       
        public 
        class 
        MybatisGenerator { 
       
        /** 
       
        **********************************使用前必读******************* 
       
        ** 
       
        ** 使用前请将moduleName更改为自己数据库表名，全库生成请留空。这是Oracle版本 
       
        ** 
       
        *********************************************************** 
       
        */ 
       
        private 
        final 
        String type_str = 
        "VARCHAR2"
        ; 
       
        private 
        final 
        String type_date = 
        "DATE"
        ; 
       
        private 
        final 
        String type_decimal = 
        "NUMBER"
        ; 
       
        private 
        final 
        String moduleName = 
        "ORDERWEB_BED_REC"
        ; 
        // 表名称（所有请留空，指定表名只生成单表mapper等） 
       
        private 
        static 
        String fileName = 
        "MybatisGenerator"
        ; 
        // 生成文件存储根目录文件夹名 
       
        private 
        final 
        static 
        String BASE_PATH = 
        "D:/"
        +fileName+
        "/"
        ; 
       
        private 
        final 
        static 
        String OPEN_PATH = 
        "D:\\"
        +fileName; 
       
        private 
        final 
        String bean_path = BASE_PATH + 
        "entity"
        ; 
       
        private 
        final 
        String mapper_path = BASE_PATH + 
        "dao"
        ; 
       
        private 
        final 
        String xml_path = BASE_PATH + 
        "dao/sqlmap"
        ; 
       
        private 
        final 
        String bean_package = 
        "net.icarefx.booking.pojo"
        ; 
       
        private 
        final 
        String mapper_package = 
        "net.icarefx.booking.dao"
        ; 
       
        /** 
       
        * 数据库配置 
       
        */ 
       
        private 
        final 
        String driverName = 
        "oracle.jdbc.driver.OracleDriver"
        ; 
       
        private 
        final 
        String url = 
        "jdbc:oracle:thin:@127.0.0.1:1521:orcl"
        ; 
       
        private 
        final 
        String user = 
        "orderweb"
        ; 
       
        private 
        final 
        String password = 
        "carefx"
        ; 
       
        private 
        String tableName = 
        null
        ; 
       
        private 
        String beanName = 
        null
        ; 
       
        private 
        String mapperName = 
        null
        ; 
       
        private 
        Connection conn = 
        null
        ; 
       
        private 
        void 
        init() 
        throws 
        ClassNotFoundException, SQLException { 
       
        Class.forName(driverName); 
       
        conn = DriverManager.getConnection(url, user, password); 
       
        } 
       
        /** 
       
        *  获取所有的表 
       
        * 
       
        * @return 
       
        * @throws SQLException  
       
        */ 
       
        private 
        List<String> getTables() 
        throws 
        SQLException { 
       
        List<String> tables = 
        new 
        ArrayList<String>(); 
       
        PreparedStatement pstate = 
        null
        ; 
       
        if
        (StringUtils.isNotBlank(moduleName)){ 
       
        pstate = conn.prepareStatement(
        "select table_name from user_tables where table_name ='"
        +moduleName+
        "'"
        ); 
       
        }
        else
        { 
       
        pstate = conn.prepareStatement(
        "select table_name from user_tables"
        ); 
       
        } 
       
        ResultSet results = pstate.executeQuery(); 
       
        while 
        ( results.next() ) { 
       
        String tableName = results.getString(
        1
        ); 
       
        tables.add(tableName); 
       
        } 
       
        return 
        tables; 
       
        } 
       
        private 
        void 
        processTable( String table ) { 
       
        StringBuffer sb = 
        new 
        StringBuffer(table.length()); 
       
        String tableNew = table.toLowerCase(); 
       
        String[] tables = tableNew.split(
        "_"
        ); 
       
        String temp = 
        null
        ; 
       
        if
        (tables.length>
        1
        ){ 
       
        for 
        ( 
        int 
        i = 
        0 
        ; i < tables.length ; i++ ) { 
       
        temp = tables[i].trim(); 
       
        sb.append(temp.substring(
        0
        , 
        1
        ).toUpperCase()).append(temp.substring(
        1
        )); 
       
        } 
       
        }
        else
        { 
       
        sb.append(tableNew.substring(
        0
        , 
        1
        ).toUpperCase()).append(tableNew.substring(
        1
        )); 
       
        } 
       
        beanName = sb.toString(); 
       
        mapperName = beanName + 
        "Mapper"
        ; 
       
        } 
       
        private 
        String processType( String type ) { 
       
        if 
        ( type.indexOf(type_str) > -
        1 
        ) { 
       
        return 
        "String"
        ; 
       
        }
        else 
        if 
        ( type.indexOf(type_date) > -
        1 
        ) { 
       
        return 
        "java.util.Date"
        ; 
       
        }
        else 
        if 
        ( type.indexOf(type_decimal) > -
        1 
        ) { 
       
        return 
        "java.math.BigDecimal"
        ; 
       
        } 
       
        return 
        null
        ; 
       
        } 
       
        private 
        String processField( String field ) { 
       
        StringBuffer sb = 
        new 
        StringBuffer(field.length()); 
       
        field = field.toLowerCase(); 
       
        String[] fields = field.split(
        "_"
        ); 
       
        String temp = 
        null
        ; 
       
        sb.append(fields[
        0
        ]); 
       
        if
        (fields.length>
        1
        ){ 
       
        for 
        ( 
        int 
        i = 
        1 
        ; i < fields.length ; i++ ) { 
       
        temp = fields[i].trim(); 
       
        sb.append(temp.substring(
        0
        , 
        1
        ).toUpperCase()).append(temp.substring(
        1
        )); 
       
        } 
       
        } 
       
        return 
        sb.toString(); 
       
        } 
       
        /** 
       
        *  将实体类名首字母改为小写 
       
        * 
       
        * @param beanName 
       
        * @return  
       
        */ 
       
        private 
        String processResultMapId( String beanName ) { 
       
        return 
        beanName.substring(
        0
        , 
        1
        ).toLowerCase() + beanName.substring(
        1
        ); 
       
        } 
       
        /** 
       
        *  构建类上面的注释 
       
        * 
       
        * @param bw 
       
        * @param text 
       
        * @return 
       
        * @throws IOException  
       
        */ 
       
        private 
        BufferedWriter buildClassComment( BufferedWriter bw, String text ) 
        throws 
        IOException { 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        bw.write(
        "/**"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        " * "
        ); 
       
        bw.newLine(); 
       
        bw.write(
        " * " 
        + text); 
       
        bw.newLine(); 
       
        bw.write(
        " * "
        ); 
       
        bw.newLine(); 
       
        bw.write(
        " **/"
        ); 
       
        return 
        bw; 
       
        } 
       
        /** 
       
        *  构建方法上面的注释 
       
        * 
       
        * @param bw 
       
        * @param text 
       
        * @return 
       
        * @throws IOException  
       
        */ 
       
        private 
        BufferedWriter buildMethodComment( BufferedWriter bw, String text ) 
        throws 
        IOException { 
       
        bw.newLine(); 
       
        bw.write(
        "\t/**"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t * "
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t * " 
        + text); 
       
        bw.newLine(); 
       
        bw.write(
        "\t * "
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t **/"
        ); 
       
        return 
        bw; 
       
        } 
       
        /** 
       
        *  生成实体类 
       
        * 
       
        * @param columns 
       
        * @param types 
       
        * @param comments 
       
        * @throws IOException  
       
        */ 
       
        private 
        void 
        buildEntityBean( List<String> columns, List<String> types, List<String> comments, String tableComment ) 
       
        throws 
        IOException { 
       
        File folder = 
        new 
        File(bean_path); 
       
        if 
        ( !folder.exists() ) { 
       
        folder.mkdir(); 
       
        } 
       
        File beanFile = 
        new 
        File(bean_path, beanName + 
        ".java"
        ); 
       
        BufferedWriter bw = 
        new 
        BufferedWriter(
        new 
        OutputStreamWriter(
        new 
        FileOutputStream(beanFile))); 
       
        bw.write(
        "package " 
        + bean_package + 
        ";"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "import java.io.Serializable;"
        ); 
       
        bw.newLine(); 
       
        //bw.write("import lombok.Data;"); 
       
        //      bw.write("import javax.persistence.Entity;"); 
       
        bw = buildClassComment(bw, tableComment); 
       
        bw.newLine(); 
       
        bw.write(
        "@SuppressWarnings(\"serial\")"
        ); 
       
        bw.newLine(); 
       
        //      bw.write("@Entity"); 
       
        //bw.write("@Data"); 
       
        //bw.newLine(); 
       
        bw.write(
        "public class " 
        + beanName + 
        " implements Serializable {"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        int 
        size = columns.size(); 
       
        for 
        ( 
        int 
        i = 
        0 
        ; i < size ; i++ ) { 
       
        bw.write(
        "\t/**" 
        + comments.get(i) + 
        "**/"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\tprivate " 
        + processType(types.get(i)) + 
        " " 
        + processField(columns.get(i)) + 
        ";"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        } 
       
        bw.newLine(); 
       
        // 生成get 和 set方法 
       
        String tempField = 
        null
        ; 
       
        String _tempField = 
        null
        ; 
       
        String tempType = 
        null
        ; 
       
        for 
        ( 
        int 
        i = 
        0 
        ; i < size ; i++ ) { 
       
        tempType = processType(types.get(i)); 
       
        _tempField = processField(columns.get(i)); 
       
        tempField = _tempField.substring(
        0
        , 
        1
        ).toUpperCase() + _tempField.substring(
        1
        ); 
       
        bw.newLine(); 
       
        //          bw.write("\tpublic void set" + tempField + "(" + tempType + " _" + _tempField + "){"); 
       
        bw.write(
        "\tpublic void set" 
        + tempField + 
        "(" 
        + tempType + 
        " " 
        + _tempField + 
        "){"
        ); 
       
        bw.newLine(); 
       
        //          bw.write("\t\tthis." + _tempField + "=_" + _tempField + ";"); 
       
        bw.write(
        "\t\tthis." 
        + _tempField + 
        " = " 
        + _tempField + 
        ";"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t}"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        bw.write(
        "\tpublic " 
        + tempType + 
        " get" 
        + tempField + 
        "(){"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\treturn this." 
        + _tempField + 
        ";"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t}"
        ); 
       
        bw.newLine(); 
       
        } 
       
        bw.newLine(); 
       
        bw.write(
        "}"
        ); 
       
        bw.newLine(); 
       
        bw.flush(); 
       
        bw.close(); 
       
        } 
       
        /** 
       
        *  构建Mapper文件 
       
        * 
       
        * @throws IOException  
       
        */ 
       
        private 
        void 
        buildMapper(List<String> columns, List<String> types) 
        throws 
        IOException { 
       
        File folder = 
        new 
        File(mapper_path); 
       
        if 
        ( !folder.exists() ) { 
       
        folder.mkdirs(); 
       
        } 
       
        File mapperFile = 
        new 
        File(mapper_path, mapperName + 
        ".java"
        ); 
       
        BufferedWriter bw = 
        new 
        BufferedWriter(
        new 
        OutputStreamWriter(
        new 
        FileOutputStream(mapperFile), 
        "utf-8"
        )); 
       
        bw.write(
        "package " 
        + mapper_package + 
        ";"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        bw.write(
        "import " 
        + bean_package + 
        "." 
        + beanName + 
        ";"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "import org.apache.ibatis.annotations.Param;"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "import java.util.List;"
        ); 
       
        bw = buildClassComment(bw, mapperName + 
        "数据库操作接口类"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        //      bw.write("public interface " + mapperName + " extends " + mapper_extends + "<" + beanName + "> {"); 
       
        bw.write(
        "public interface " 
        + mapperName + 
        "{"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        // 主键名称 
       
        String pkName = processField(columns.get(
        0
        )); 
       
        String pkType = processType(types.get(
        0
        )); 
       
        // ----------定义Mapper中的方法Begin---------- 
       
        bw = buildMethodComment(bw, 
        "查询（根据主键ID查询）"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\tpublic " 
        + beanName + 
        "  selectByPrimaryKey ( @Param(\""
        +pkName+
        "\") "
        +pkType+
        " "
        +pkName+
        " );"
        ); 
       
        bw.newLine(); 
       
        bw = buildMethodComment(bw, 
        "删除（根据主键ID删除）"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\tpublic " 
        + 
        "int deleteByPrimaryKey ( @Param(\""
        +pkName+
        "\") "
        +pkType+
        " "
        +pkName+
        " );"
        ); 
       
        bw.newLine(); 
       
        bw = buildMethodComment(bw, 
        "添加"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\tpublic " 
        + 
        "int insert( " 
        + beanName + 
        " record );"
        ); 
       
        bw.newLine(); 
       
        bw = buildMethodComment(bw, 
        "添加 （匹配有值的字段）"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\tpublic " 
        + 
        "int insertSelective( " 
        + beanName + 
        " record );"
        ); 
       
        bw.newLine(); 
       
        bw = buildMethodComment(bw, 
        "修改 （匹配有值的字段）"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\tpublic " 
        + 
        "int updateByPrimaryKeySelective( " 
        + beanName + 
        " record );"
        ); 
       
        bw.newLine(); 
       
        bw = buildMethodComment(bw, 
        "修改（根据主键ID修改）"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\tpublic " 
        + 
        "int updateByPrimaryKey ( " 
        + beanName + 
        " record );"
        ); 
       
        bw.newLine(); 
       
        bw = buildMethodComment(bw, 
        "集合（匹配有值的字段）"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\tpublic List<" 
        + beanName + 
        ">  findByList ( " 
        + beanName + 
        " record );"
        ); 
       
        bw.newLine(); 
       
        // ----------定义Mapper中的方法End---------- 
       
        bw.newLine(); 
       
        bw.write(
        "}"
        ); 
       
        bw.flush(); 
       
        bw.close(); 
       
        } 
       
        /** 
       
        *  构建实体类映射XML文件 
       
        * 
       
        * @param columns 
       
        * @param types 
       
        * @param comments 
       
        * @throws IOException  
       
        */ 
       
        private 
        void 
        buildMapperXml( List<String> columns, List<String> types, List<String> comments ) 
        throws 
        IOException { 
       
        File folder = 
        new 
        File(xml_path); 
       
        if 
        ( !folder.exists() ) { 
       
        folder.mkdirs(); 
       
        } 
       
        File mapperXmlFile = 
        new 
        File(xml_path, beanName + 
        ".sqlmap.xml"
        ); 
       
        BufferedWriter bw = 
        new 
        BufferedWriter(
        new 
        OutputStreamWriter(
        new 
        FileOutputStream(mapperXmlFile))); 
       
        bw.write(
        "<?xml version=\"1.0\" encoding=\"UTF-8\"?>"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "<!DOCTYPE mapper PUBLIC \"-//mybatis.org//DTD Mapper 3.0//EN\" "
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "    \"http://mybatis.org/dtd/mybatis-3-mapper.dtd\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "<mapper namespace=\"" 
        + mapper_package + 
        "." 
        + mapperName + 
        "\">"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        buildSQL(bw, columns, types); 
       
        bw.write(
        "</mapper>"
        ); 
       
        bw.flush(); 
       
        bw.close(); 
       
        } 
       
        private 
        void 
        buildSQL( BufferedWriter bw, List<String> columns, List<String> types ) 
        throws 
        IOException { 
       
        int 
        size = columns.size(); 
       
        // 通用结果列 
       
        bw.write(
        "\t<!-- 通用查询结果列-->"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t<sql id=\"Base_Column_List\">"
        ); 
       
        bw.newLine(); 
       
        for 
        ( 
        int 
        i = 
        0 
        ; i < size ; i++ ) { 
       
        bw.write(
        "\t" 
        + columns.get(i)); 
       
        if 
        ( i != size - 
        1 
        ) { 
       
        bw.write(
        ","
        ); 
       
        } 
       
        } 
       
        bw.newLine(); 
       
        bw.write(
        "\t</sql>"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        // 查询（根据主键ID查询） 
       
        bw.write(
        "\t<!-- 查询（根据主键ID查询） -->"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t<select id=\"selectByPrimaryKey\" resultType=\"" 
       
        + processResultMapId(beanName) + 
        "\" parameterType=\"java.lang." 
        + processType(types.get(
        0
        )) + 
        "\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t SELECT"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t <include refid=\"Base_Column_List\" />"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t FROM " 
        + tableName); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t WHERE " 
        + columns.get(
        0
        ) + 
        " = #{" 
        + processField(columns.get(
        0
        )) + 
        "}"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t</select>"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        // 查询完 
       
        // 删除（根据主键ID删除） 
       
        bw.write(
        "\t<!--删除：根据主键ID删除-->"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t<delete id=\"deleteByPrimaryKey\" parameterType=\"java.lang." 
        + processType(types.get(
        0
        )) + 
        "\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t DELETE FROM " 
        + tableName); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t WHERE " 
        + columns.get(
        0
        ) + 
        " = #{" 
        + processField(columns.get(
        0
        )) + 
        "}"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t</delete>"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        // 删除完 
       
        // 添加insert方法 
       
        bw.write(
        "\t<!-- 添加 -->"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t<insert id=\"insert\" parameterType=\"" 
        + processResultMapId(beanName) + 
        "\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t INSERT INTO " 
        + tableName); 
       
        bw.newLine(); 
       
        bw.write(
        " \t\t("
        ); 
       
        for 
        ( 
        int 
        i = 
        0 
        ; i < size ; i++ ) { 
       
        bw.write(columns.get(i)); 
       
        if 
        ( i != size - 
        1 
        ) { 
       
        bw.write(
        ","
        ); 
       
        } 
       
        } 
       
        bw.write(
        ") "
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t VALUES "
        ); 
       
        bw.newLine(); 
       
        bw.write(
        " \t\t("
        ); 
       
        for 
        ( 
        int 
        i = 
        0 
        ; i < size ; i++ ) { 
       
        bw.write(
        "#{" 
        + processField(columns.get(i)) + 
        "}"
        ); 
       
        if 
        ( i != size - 
        1 
        ) { 
       
        bw.write(
        ","
        ); 
       
        } 
       
        } 
       
        bw.write(
        ") "
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t</insert>"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        // 添加insert完 
       
        //---------------  insert方法（匹配有值的字段） 
       
        bw.write(
        "\t<!-- 添加 （匹配有值的字段）-->"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t<insert id=\"insertSelective\" parameterType=\"" 
        + processResultMapId(beanName) + 
        "\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t INSERT INTO " 
        + tableName); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t <trim prefix=\"(\" suffix=\")\" suffixOverrides=\",\" >"
        ); 
       
        bw.newLine(); 
       
        String tempField = 
        null
        ; 
       
        for 
        ( 
        int 
        i = 
        0 
        ; i < size ; i++ ) { 
       
        tempField = processField(columns.get(i)); 
       
        bw.write(
        "\t\t\t<if test=\"" 
        + tempField + 
        " != null\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t\t\t " 
        + columns.get(i) + 
        ","
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t\t</if>"
        ); 
       
        bw.newLine(); 
       
        } 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t </trim>"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t <trim prefix=\"values (\" suffix=\")\" suffixOverrides=\",\" >"
        ); 
       
        bw.newLine(); 
       
        tempField = 
        null
        ; 
       
        for 
        ( 
        int 
        i = 
        0 
        ; i < size ; i++ ) { 
       
        tempField = processField(columns.get(i)); 
       
        bw.write(
        "\t\t\t<if test=\"" 
        + tempField + 
        "!=null\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t\t\t #{" 
        + tempField + 
        "},"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t\t</if>"
        ); 
       
        bw.newLine(); 
       
        } 
       
        bw.write(
        "\t\t </trim>"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t</insert>"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        //---------------  完毕 
       
        // 修改update方法 
       
        bw.write(
        "\t<!-- 修 改-->"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t<update id=\"updateByPrimaryKeySelective\" parameterType=\"" 
        + processResultMapId(beanName) + 
        "\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t UPDATE " 
        + tableName); 
       
        bw.newLine(); 
       
        bw.write(
        " \t\t <set> "
        ); 
       
        bw.newLine(); 
       
        tempField = 
        null
        ; 
       
        for 
        ( 
        int 
        i = 
        1 
        ; i < size ; i++ ) { 
       
        tempField = processField(columns.get(i)); 
       
        bw.write(
        "\t\t\t<if test=\"" 
        + tempField + 
        " != null\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t\t\t " 
        + columns.get(i) + 
        " = #{" 
        + tempField + 
        "},"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t\t</if>"
        ); 
       
        bw.newLine(); 
       
        } 
       
        bw.newLine(); 
       
        bw.write(
        " \t\t </set>"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t WHERE " 
        + columns.get(
        0
        ) + 
        " = #{" 
        + processField(columns.get(
        0
        )) + 
        "}"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t</update>"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        // update方法完毕 
       
        // ----- 修改（匹配有值的字段） 
       
        bw.write(
        "\t<!-- 修 改（匹配有值的字段）-->"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t<update id=\"updateByPrimaryKey\" parameterType=\"" 
        + processResultMapId(beanName) + 
        "\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t UPDATE " 
        + tableName); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t SET "
        ); 
       
        bw.newLine(); 
       
        tempField = 
        null
        ; 
       
        for 
        ( 
        int 
        i = 
        1 
        ; i < size ; i++ ) { 
       
        tempField = processField(columns.get(i)); 
       
        bw.write(
        "\t\t\t " 
        + columns.get(i) + 
        " = #{" 
        + tempField + 
        "}"
        ); 
       
        if 
        ( i != size - 
        1 
        ) { 
       
        bw.write(
        ","
        ); 
       
        } 
       
        bw.newLine(); 
       
        } 
       
        bw.write(
        "\t\t WHERE " 
        + columns.get(
        0
        ) + 
        " = #{" 
        + processField(columns.get(
        0
        )) + 
        "}"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t</update>"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        //---------------  完毕 
       
        // ----- 集合（匹配有值的字段） 
       
        bw.write(
        "\t<!-- 集合（匹配有值的字段）-->"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t<select id=\"findByList\" resultType=\"" 
        + processResultMapId(beanName) + 
        "\" parameterType=\"" 
        + processResultMapId(beanName) + 
        "\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t SELECT"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t <include refid=\"Base_Column_List\" />"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t FROM " 
        + tableName); 
       
        bw.newLine(); 
       
        bw.write(
        " \t\t <where> "
        ); 
       
        bw.newLine(); 
       
        tempField = 
        null
        ; 
       
        for 
        ( 
        int 
        i = 
        0 
        ; i < size ; i++ ) { 
       
        tempField = processField(columns.get(i)); 
       
        bw.write(
        "\t\t\t<if test=\"" 
        + tempField + 
        " != null\">"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t\t\t " 
        + columns.get(i) + 
        " = #{" 
        + tempField + 
        "},"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t\t\t</if>"
        ); 
       
        bw.newLine(); 
       
        } 
       
        bw.newLine(); 
       
        bw.write(
        " \t\t </where>"
        ); 
       
        bw.newLine(); 
       
        bw.write(
        "\t</select>"
        ); 
       
        bw.newLine(); 
       
        bw.newLine(); 
       
        } 
       
        public 
        void 
        generate() 
        throws 
        ClassNotFoundException, SQLException, IOException { 
       
        init(); 
       
        String prefix = 
        "select column_name,data_type from user_tab_columns where  table_name="
        ; 
       
        List<String> columns = 
        null
        ; 
       
        List<String> types = 
        null
        ; 
       
        List<String> comments = 
        null
        ; 
       
        PreparedStatement pstate = 
        null
        ; 
       
        List<String> tables = getTables(); 
       
        for 
        ( String table : tables ) { 
       
        columns = 
        new 
        ArrayList<String>(); 
       
        types = 
        new 
        ArrayList<String>(); 
       
        comments = 
        new 
        ArrayList<String>(); 
       
        pstate = conn.prepareStatement(prefix +
        "'"
        + table +
        "'"
        ); 
       
        ResultSet results = pstate.executeQuery(); 
       
        while 
        ( results.next() ) { 
       
        columns.add(results.getString(
        "column_name"
        )); 
       
        types.add(results.getString(
        "data_type"
        )); 
       
        comments.add(
        ""
        ); 
       
        } 
       
        tableName = table; 
       
        processTable(table); 
       
        buildEntityBean(columns, types, comments, 
        "数据库表名称:"
        +tableName); 
       
        buildMapper(columns, types); 
       
        buildMapperXml(columns, types, comments); 
       
        } 
       
        conn.close(); 
       
        } 
       
        /** 
       
        * 删除单个文件 
       
        * @param   sPath    被删除文件的文件名 
       
        * @return 单个文件删除成功返回true，否则返回false 
       
        */ 
       
        public 
        static 
        boolean 
        deleteFile(String sPath) { 
       
        boolean 
        flag = 
        false
        ; 
       
        File file = 
        new 
        File(sPath); 
       
        // 路径为文件且不为空则进行删除 
       
        if 
        (file.isFile() && file.exists()) { 
       
        file.delete(); 
       
        flag = 
        true
        ; 
       
        } 
       
        return 
        flag; 
       
        } 
       
        /** 
       
        * 删除目录（文件夹）以及目录下的文件 
       
        * @param   sPath 被删除目录的文件路径 
       
        * @return  目录删除成功返回true，否则返回false 
       
        */ 
       
        public 
        static 
        boolean 
        deleteDirectory(String sPath) { 
       
        //如果sPath不以文件分隔符结尾，自动添加文件分隔符 
       
        if 
        (!sPath.endsWith(File.separator)) { 
       
        sPath = sPath + File.separator; 
       
        } 
       
        File dirFile = 
        new 
        File(sPath); 
       
        //如果dir对应的文件不存在，或者不是一个目录，则退出 
       
        if 
        (!dirFile.exists() || !dirFile.isDirectory()) { 
       
        return 
        false
        ; 
       
        } 
       
        boolean 
        flag = 
        true
        ; 
       
        //删除文件夹下的所有文件(包括子目录) 
       
        File[] files = dirFile.listFiles(); 
       
        for 
        (
        int 
        i = 
        0
        ; i < files.length; i++) { 
       
        //删除子文件 
       
        if 
        (files[i].isFile()) { 
       
        flag = deleteFile(files[i].getAbsolutePath()); 
       
        if 
        (!flag) 
        break
        ; 
       
        } 
        //删除子目录 
       
        else 
        { 
       
        flag = deleteDirectory(files[i].getAbsolutePath()); 
       
        if 
        (!flag) 
        break
        ; 
       
        } 
       
        } 
       
        if 
        (!flag) 
        return 
        false
        ; 
       
        //删除当前目录 
       
        if 
        (dirFile.delete()) { 
       
        return 
        true
        ; 
       
        } 
        else 
        { 
       
        return 
        false
        ; 
       
        } 
       
        } 
       
        /** 
       
        * 生成入口 
       
        * @param args 
       
        */ 
       
        public 
        static 
        void 
        main( String[] args ) { 
       
        try 
        { 
       
        // 自动打开生成文件的目录 
       
        File folder = 
        new 
        File(BASE_PATH); 
       
        // 判断是否为文件   
       
        if 
        (folder.isFile()) {  
        // 为文件时调用删除文件方法   
       
        deleteFile(BASE_PATH);   
       
        } 
        else 
        {  
        // 为目录时调用删除目录方法   
       
        deleteDirectory(BASE_PATH);   
       
        }   
       
        folder.mkdir(); 
       
        new 
        MybatisGenerator().generate(); 
       
        Runtime.getRuntime().exec(
        "cmd /c start explorer "
        +OPEN_PATH); 
       
        } 
        catch 
        ( ClassNotFoundException e ) { 
       
        e.printStackTrace(); 
       
        } 
        catch 
        ( SQLException e ) { 
       
        e.printStackTrace(); 
       
        } 
        catch 
        ( IOException e ) { 
       
        e.printStackTrace(); 
       
        } 
       
        } 
       
        }
以上，自由复制，更灵活的生成。
本文转自沫沫金 51CTO博客，原文链接：http://blog.51cto.com/zl0828/1719291，如需转载请自行联系原作者
沫沫金-Mybatis工具类，生成dao层xml、mapper文件和实体类entity层

热门文章

最新文章

相关课程

相关电子书
