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();
}
}
}更多版本：
1、MySql版本http://www.oschina.net/code/snippet_123050_45716
以上，自由复制，更灵活的生成。
作者：沫沫金，WX(zl4828)欢迎来骚.精通DIV+CSS+JS、PS淘宝美工、网页设计。+++
本文出自 “沫沫金的IT心得与技巧” 博客，谢绝转载！
沫沫金-Mybatis工具类，生成dao层xml、mapper文件和实体类entity层
原文地址：http://zl0828.blog.51cto.com/2242565/1719291
