mysql根据数据库自动生成实体（dto,service,impl,mybatis,xml）

最新推荐文章于 2022-08-06 21:16:03 发布
最新推荐文章于 2022-08-06 21:16:03 发布
阅读量5.9k 收藏 6
点赞数 2
分类专栏： 数据库 文章标签： java 数据库 mysql
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/ppt0501/article/details/13505811
版权



 

package com.csair.test;

import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class GenerateEntity {

	private String url = "jdbc:mysql://localhost:3306/database";
	private String username = "root";
	private String password = "root";
	private String driverClass = "com.mysql.jdbc.Driver";
	private DatabaseMetaData dbMetaData = null;
	private String packageName = "com.csair.test.dto;";
	private String path = "C:\\Users\\ppt\\Desktop\\test\\";
	
	/**
	 * @Description: 获取表对应的所有列
	 * @author: ppt
	 * @date: 2015-3-16 上午10:13:17
	 * @param tableName
	 * @return: void
	 */
	public void getTableColumns(String tableName) {
		try {
			String ACCESS_DOMAIN = "private";
			ResultSet resultSet = dbMetaData.getColumns(null, null, tableName, "%");
			while(resultSet.next()) {  
	            ResultSet resultSetColumn = dbMetaData.getColumns(null, null, tableName, null);  
	            String className = getFormatString(tableName,true);
	            StringBuffer header = new StringBuffer("package " + packageName + "\n\n");
	            StringBuffer footer = new StringBuffer();
	            StringBuffer contentBuffer = new StringBuffer();
	            contentBuffer.append("public class " + className + "{\n");  
	            while(resultSetColumn.next()) {  
	                contentBuffer.append("\t"+ACCESS_DOMAIN + " ");  
	                String columnType = resultSetColumn.getString("TYPE_NAME");  
	                String COLUMN_TYPE = getColumnType(columnType);
	                if("Date".equals(COLUMN_TYPE)) {
	                	header.append("import java.util.Date;\n");
	                }
	                contentBuffer.append(COLUMN_TYPE + " ");  
	                String columnName = resultSetColumn.getString("COLUMN_NAME"); 
	                String remark = resultSetColumn.getString("REMARKS");
	                columnName = this.getFormatString(columnName, false);
	                footer.append(getSetGenerater(columnName, COLUMN_TYPE));
	                contentBuffer.append(columnName+";\t//"+remark+"\n");  
	            } 
	            contentBuffer.append("\n\n"+footer);
	            contentBuffer.append("}");  
	            header.append("\n");
	            header.append(contentBuffer);
	            this.outputToFile(className+".java", header.toString());  
	        }  
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}
	
	public GenerateEntity() {
		try {
			Class.forName(driverClass);
			Connection conn = DriverManager.getConnection(url, username, password);
			dbMetaData = conn.getMetaData();
		} catch(Exception e) {
			e.printStackTrace();
		}
	}
	  
	/**
	 * @Description: 获取所以的表
	 * @author: ppt
	 * @date: 2015-3-16 上午10:12:57
	 * @return: void
	 */
	public List<String> getAllTableList() {  
		List<String> tableList = new ArrayList<String>();
        try {  
            String[] types = { "TABLE" };  
            ResultSet rs = dbMetaData.getTables(null, null, "%", types);  
            while (rs.next()) {  
                String tableName = rs.getString("TABLE_NAME");  //表名 
                tableList.add(tableName);
            }  
        } catch (SQLException e) {  
            e.printStackTrace();  
        }
        return tableList;
    }  
	
	 /** 
     * 处理字符串，去掉下划线“_”，并且把下划线的下一个字符变大写，flag为true，表示首字母要大写 
     * @param name 
     * @param flag 
     * @return 
     */  
    private String getFormatString(String name, boolean flag) {  
        name = name.toLowerCase();  
        String[] nameTemp = name.split("_");  
        StringBuffer buffer = new StringBuffer();  
        for(String str : nameTemp) {  
            String head = str.substring(0, 1).toUpperCase();  
            String tail = str.substring(1);  
            buffer.append(head+tail);  
        }  
        StringBuffer result = null;  
        if(!flag) {  
            result = new StringBuffer();  
            String head = buffer.substring(0, 1).toLowerCase();  
            String tail = buffer.substring(1);  
            result.append(head+tail);  
            return result.toString();  
        }  
        return buffer.toString();  
    }  
	/** 
     * 把String内容写到文件 
     * @param fileName 
     * @param content 
     */  
    private void outputToFile(String fileName, String content) {  
        OutputStream os = null;  
        try {  
            os = new FileOutputStream(path+fileName);  
        } catch (FileNotFoundException e1) {  
            e1.printStackTrace();  
        }  
        byte[] b = content.getBytes();  
        try {  
            os.write(b);  
            os.flush();  
        } catch (IOException e) {  
            e.printStackTrace();  
        } finally {  
            try {  
                os.close();  
            } catch (IOException e) {  
                e.printStackTrace();  
            }  
        }  
    }  
    /** 
     * 数据库类型转为java类型 
     * @param column 
     * @return 
     */  
    private String getColumnType(String column) {  
        String COLUMN_TYPE = null;
    	if("VARCHAR".equals(column)) {  
            COLUMN_TYPE = "String";  
        } else if("BIGINT".equals(column)) {  
            COLUMN_TYPE = "Long";  
        } else if("DATETIME".equals(column)) {  
            COLUMN_TYPE = "Date";  
        } else if("INT".equals(column) || "INT UNSIGNED".equals(column)) {  
            COLUMN_TYPE = "Integer";  
        } else if("BIGINT UNSIGNED".equals(column)) {  
            COLUMN_TYPE = "Long";  
        } else if("TINYINT UNSIGNED".equals(column)) {  
            COLUMN_TYPE = "Short";  
        } else if("DECIMAL".equals(column) || "FLOAT".equals(column) || "DOUBLE".equals(column)) {  
            COLUMN_TYPE = "Double";  
        } else if("TEXT".equals(column) || "MEDIUMTEXT".equals(column) || "LONGTEXT".equals(column)) {  
            COLUMN_TYPE = "String";  
        } else if("TIMESTAMP".equals(column) || "DATE".equals(column) || "DATETIME".equals(column)) {  
            COLUMN_TYPE = "Date";  
        } else if("TINYINT".equals(column)) {  
            COLUMN_TYPE = "Short";  
        } else if("DECIMAL UNSIGNED".equals(column)) {  
            COLUMN_TYPE = "Double";  
        } else if("SMALLINT".equals(column)) {  
            COLUMN_TYPE = "Short";  
        } else if("BIT".equals(column)) {  
            COLUMN_TYPE = "Short";  
        } else if("CHAR".equals(column)) {  
            COLUMN_TYPE = "String";  
        } else if("VARBINARY".equals(column)) {  
            COLUMN_TYPE = "byte";  
        } else if("BLOB".equals(column)) {  
            COLUMN_TYPE = "byte";  
        }
        return COLUMN_TYPE;  
    }
    
    private String getSetGenerater(String columnName, String columnType) {
    	StringBuffer sb = new StringBuffer();
		sb.append("\tpublic " + columnType + " get" +columnName.substring(0, 1).toUpperCase()+columnName.substring(1, columnName.length()) + "() {\n");
		sb.append("\t\treturn " + columnName + ";\n");
		sb.append("\t}\n\n");
		sb.append("\tpublic void set" + columnName.substring(0, 1).toUpperCase()+columnName.substring(1, columnName.length()));
		sb.append("("+columnType +" " + columnName +") {\n");
		sb.append("\t\tthis." + columnName + " = " + columnName + ";\n");
		sb.append("\t}\n\n");
		return sb.toString();
    }
    
	public static void main(String[] agrs) {
		GenerateEntity aa = new GenerateEntity();
		List<String> tableList = aa.getAllTableList();
		for(String tableName : tableList) {
			aa.getTableColumns(tableName);
		}
//		aa.getTableColumns("REPORT_REQ");
	}
	
	
	

}
接口 
package com.csair.test;

import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class GenerateService {

	private String url = "jdbc:mysql://localhost:3306/database";
	private String username = "root";
	private String password = "root";
	private String driverClass = "com.mysql.jdbc.Driver";
	private DatabaseMetaData dbMetaData = null;
	private String packageName = "com.csair.test.service;";
	private String packageNameDto = "com.csair.test.dto";
	private String path = "C:\\Users\\ppt\\Desktop\\test\\";
	
	public GenerateService() {
		try {
			Class.forName(driverClass);
			Connection conn = DriverManager.getConnection(url, username, password);
			dbMetaData = conn.getMetaData();
		} catch(Exception e) {
			e.printStackTrace();
		}
	}
	  
	/**
	 * @Description: 获取所有的表
	 * @author: ppt
	 * @date: 2015-3-16 上午10:12:57
	 * @return: void
	 */
	public List<String> getAllTableList() {  
		List<String> tableList = new ArrayList<String>();
        try {  
            String[] types = { "TABLE" };  
            ResultSet rs = dbMetaData.getTables(null, null, "%", types);  
            while (rs.next()) {  
                String tableName = rs.getString("TABLE_NAME");  //表名 
                tableList.add(tableName);
            }  
        } catch (SQLException e) {  
            e.printStackTrace();  
        }
        return tableList;
    }  
	
	 /** 
     * 处理字符串，去掉下划线“_”，并且把下划线的下一个字符变大写，flag为true，表示首字母要大写 
     * @param name 
     * @param flag 
     * @return 
     */  
    private String getFormatString(String name, boolean flag) {  
        name = name.toLowerCase();  
        String[] nameTemp = name.split("_");  
        StringBuffer buffer = new StringBuffer();  
        for(String str : nameTemp) {  
            String head = str.substring(0, 1).toUpperCase();  
            String tail = str.substring(1);  
            buffer.append(head+tail);  
        }  
        StringBuffer result = null;  
        if(!flag) {  
            result = new StringBuffer();  
            String head = buffer.substring(0, 1).toLowerCase();  
            String tail = buffer.substring(1);  
            result.append(head+tail);  
            return result.toString();  
        }  
        return buffer.toString();  
    }  
	/** 
     * 把String内容写到文件 
     * @param fileName 
     * @param content 
     */  
    private void outputToFile(String fileName, String content) {  
        OutputStream os = null;  
        try {  
            os = new FileOutputStream(path+fileName);  
        } catch (FileNotFoundException e1) {  
            e1.printStackTrace();  
        }  
        byte[] b = content.getBytes();  
        try {  
            os.write(b);  
            os.flush();  
        } catch (IOException e) {  
            e.printStackTrace();  
        } finally {  
            try {  
                os.close();  
            } catch (IOException e) {  
                e.printStackTrace();  
            }  
        }  
    }  
    
    private void getService() {
    	List<String> tableList = getAllTableList();
    	for(String tableName : tableList) {
    		String table = getFormatString(tableName, true);
    		String className = table + "Service";
    		StringBuffer sb = new StringBuffer();
    		sb.append("package " + packageName + "\n\n");
    		sb.append("import " + packageNameDto + "." + table + ";\n");
    		sb.append("import java.util.List;\n");
    		sb.append("import javax.ws.rs.POST;\n");
    		sb.append("import javax.ws.rs.GET;\n");
    		sb.append("import javax.ws.rs.PathParam;\n");
    		sb.append("import javax.ws.rs.Path;\n");
    		sb.append("import javax.ws.rs.Produces;\n");
    		sb.append("import javax.ws.rs.core.MediaType;\n");
    		sb.append("\n@Path(\"/" + className.substring(0, 1).toLowerCase() + className.substring(1) + "\")\n");
    		sb.append("public interface " + className + " {\n\n");
    		
    		sb.append("\t//增加实体\n");
    		sb.append("\t@POST\n");
    		sb.append("\t@Path(\"addEntity\")\n");
    		sb.append("\t@Produces(MediaType.APPLICATION_XML)\n");
    		sb.append("\tpublic long addEntity(" + table + " dto) throws Exception;\n\n");
    		
    		sb.append("\t//删除实体\n");
    		sb.append("\t@GET\n");
    		sb.append("\t@Path(\"deleteEntity/{id}\")\n");
    		sb.append("\t@Produces(MediaType.APPLICATION_XML)\n");
    		sb.append("\tpublic boolean deleteEntity(@PathParam(\"id\") long id) throws Exception;\n\n");
    		
    		sb.append("\t//修改实体\n");
    		sb.append("\t@POST\n");
    		sb.append("\t@Path(\"updateEntity\")\n");
    		sb.append("\t@Produces(MediaType.APPLICATION_XML)\n");
    		sb.append("\tpublic long updateEntity(" + table + " dto) throws Exception;\n\n");
    		
    		sb.append("\t//查询实体\n");
    		sb.append("\t@POST\n");
    		sb.append("\t@Path(\"findEntity\")\n");
    		sb.append("\t@Produces(MediaType.APPLICATION_XML)\n");
    		sb.append("\tpublic "+table+" findEntity(" + table + " dto) throws Exception;\n\n");
    		
    		sb.append("\t//查询实体List\n");
    		sb.append("\t@POST\n");
    		sb.append("\t@Path(\"findEntityList\")\n");
    		sb.append("\t@Produces(MediaType.APPLICATION_XML)\n");
    		sb.append("\tpublic List<"+table+"> findEntityList(" + table + " dto) throws Exception;\n\n");
    		
    		sb.append("}\n");
    		outputToFile(className+".java", sb.toString());
    	}
    }
    
	public static void main(String[] agrs) {
		GenerateService aa = new GenerateService();
		aa.getService();
	}

}
mybatis接口 

package com.csair.test;

import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class GenerateMybatisRepository {

	private String url = "jdbc:mysql://localhost:3306/database";
	private String username = "root";
	private String password = "root";
	private String driverClass = "com.mysql.jdbc.Driver";
	private DatabaseMetaData dbMetaData = null;
	private String packageName = "com.csair.test.mybatisRepository;";
	private String packageNameDto = "com.csair.test.dto";
	private String path = "C:\\Users\\ppt\\Desktop\\test\\";
	
	public GenerateMybatisRepository() {
		try {
			Class.forName(driverClass);
			Connection conn = DriverManager.getConnection(url, username, password);
			dbMetaData = conn.getMetaData();
		} catch(Exception e) {
			e.printStackTrace();
		}
	}
	  
	/**
	 * @Description: 获取所有的表
	 * @author: ppt
	 * @date: 2015-3-16 上午10:12:57
	 * @return: void
	 */
	public List<String> getAllTableList() {  
		List<String> tableList = new ArrayList<String>();
        try {  
            String[] types = { "TABLE" };  
            ResultSet rs = dbMetaData.getTables(null, null, "%", types);  
            while (rs.next()) {  
                String tableName = rs.getString("TABLE_NAME");  //表名 
                tableList.add(tableName);
            }  
        } catch (SQLException e) {  
            e.printStackTrace();  
        }
        return tableList;
    }  
	
	 /** 
     * 处理字符串，去掉下划线“_”，并且把下划线的下一个字符变大写，flag为true，表示首字母要大写 
     * @param name 
     * @param flag 
     * @return 
     */  
    private String getFormatString(String name, boolean flag) {  
        name = name.toLowerCase();  
        String[] nameTemp = name.split("_");  
        StringBuffer buffer = new StringBuffer();  
        for(String str : nameTemp) {  
            String head = str.substring(0, 1).toUpperCase();  
            String tail = str.substring(1);  
            buffer.append(head+tail);  
        }  
        StringBuffer result = null;  
        if(!flag) {  
            result = new StringBuffer();  
            String head = buffer.substring(0, 1).toLowerCase();  
            String tail = buffer.substring(1);  
            result.append(head+tail);  
            return result.toString();  
        }  
        return buffer.toString();  
    }  
	/** 
     * 把String内容写到文件 
     * @param fileName 
     * @param content 
     */  
    private void outputToFile(String fileName, String content) {  
        OutputStream os = null;  
        try {  
            os = new FileOutputStream(path+fileName);  
        } catch (FileNotFoundException e1) {  
            e1.printStackTrace();  
        }  
        byte[] b = content.getBytes();  
        try {  
            os.write(b);  
            os.flush();  
        } catch (IOException e) {  
            e.printStackTrace();  
        } finally {  
            try {  
                os.close();  
            } catch (IOException e) {  
                e.printStackTrace();  
            }  
        }  
    }  
    
    private void getService() {
    	List<String> tableList = getAllTableList();
    	for(String tableName : tableList) {
    		String table = getFormatString(tableName, true);
    		String className = table + "MybatisDao";
    		String dto = table;
    		StringBuffer sb = new StringBuffer();
    		sb.append("package " + packageName + "\n\n");
    		sb.append("import " + packageNameDto + "." + dto + ";\n");
    		sb.append("import java.util.List;\n");
    		sb.append("\npublic interface " + className + " {\n\n");
    		
    		sb.append("\tpublic long addEntity(" + table + " dto);\n\n");
    		
    		sb.append("\tpublic boolean deleteEntity(long id);\n\n");
    		
    		sb.append("\tpublic long updateEntity(" + table + " dto);\n\n");
    		
    		sb.append("\tpublic "+table+" findEntity(" + table + " dto);\n\n");
    		
    		sb.append("\tpublic List<"+table+"> findEntityList(" + table + " dto);\n\n");
    		
    		sb.append("}\n");
//    		System.out.println(sb.toString());
    		outputToFile(className+".java", sb.toString());
    	}
    }
    
	public static void main(String[] agrs) {
		GenerateMybatisRepository aa = new GenerateMybatisRepository();
		aa.getService();
	}


}

实现impl 

package com.csair.test;

import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

import org.springframework.beans.factory.annotation.Autowired;

public class GenerateServiceImpl {

	private String url = "jdbc:mysql://localhost:3306/database";
	private String username = "root";
	private String password = "root";
	private String driverClass = "com.mysql.jdbc.Driver";
	private DatabaseMetaData dbMetaData = null;
	private String packageName = "com.csair.test.service.impl;";
	private String packageNameService = "com.csair.test.service";
	private String packageNameDto = "com.csair.test.dto";
	private String packageNameMybatis = "com.csair.test.mybatisRepository";
	private String path = "C:\\Users\\ppt\\Desktop\\test\\";
	
	public GenerateServiceImpl() {
		try {
			Class.forName(driverClass);
			Connection conn = DriverManager.getConnection(url, username, password);
			dbMetaData = conn.getMetaData();
		} catch(Exception e) {
			e.printStackTrace();
		}
	}
	  
	/**
	 * @Description: 获取所有的表
	 * @author: ppt
	 * @date: 2015-3-16 上午10:12:57
	 * @return: void
	 */
	public List<String> getAllTableList() {  
		List<String> tableList = new ArrayList<String>();
        try {  
            String[] types = { "TABLE" };  
            ResultSet rs = dbMetaData.getTables(null, null, "%", types);  
            while (rs.next()) {  
                String tableName = rs.getString("TABLE_NAME");  //表名 
                tableList.add(tableName);
            }  
        } catch (SQLException e) {  
            e.printStackTrace();  
        }
        return tableList;
    }  
	
	 /** 
     * 处理字符串，去掉下划线“_”，并且把下划线的下一个字符变大写，flag为true，表示首字母要大写 
     * @param name 
     * @param flag 
     * @return 
     */  
    private String getFormatString(String name, boolean flag) {  
        name = name.toLowerCase();  
        String[] nameTemp = name.split("_");  
        StringBuffer buffer = new StringBuffer();  
        for(String str : nameTemp) {  
            String head = str.substring(0, 1).toUpperCase();  
            String tail = str.substring(1);  
            buffer.append(head+tail);  
        }  
        StringBuffer result = null;  
        if(!flag) {  
            result = new StringBuffer();  
            String head = buffer.substring(0, 1).toLowerCase();  
            String tail = buffer.substring(1);  
            result.append(head+tail);  
            return result.toString();  
        }  
        return buffer.toString();  
    }  
	/** 
     * 把String内容写到文件 
     * @param fileName 
     * @param content 
     */  
    private void outputToFile(String fileName, String content) {  
        OutputStream os = null;  
        try {  
            os = new FileOutputStream(path+fileName);  
        } catch (FileNotFoundException e1) {  
            e1.printStackTrace();  
        }  
        byte[] b = content.getBytes();  
        try {  
            os.write(b);  
            os.flush();  
        } catch (IOException e) {  
            e.printStackTrace();  
        } finally {  
            try {  
                os.close();  
            } catch (IOException e) {  
                e.printStackTrace();  
            }  
        }  
    }  
    
    private void getServiceImpl() {
    	List<String> tableList = getAllTableList();
    	for(String tableName : tableList) {
    		String table = getFormatString(tableName, true);
    		String mybatis = table + "MybatisDao";
    		String dto = table;
    		String service = table + "Service";
    		String className = service + "Impl";
    		StringBuffer sb = new StringBuffer();
    		sb.append("package " + packageName + "\n\n");
    		sb.append("import " + packageNameDto + "." + dto + ";\n");
    		sb.append("import " + packageNameMybatis + "." + mybatis + ";\n");
    		sb.append("import " + packageNameService + "." + service + ";\n");
    		sb.append("import org.apache.log4j.Logger;\n");
    		sb.append("import org.springframework.beans.factory.annotation.Autowired;\n");
    		sb.append("import org.springframework.stereotype.Service;\n");
    		sb.append("import java.util.List;\n");
    		sb.append("@Service(\""+service.substring(0,1).toLowerCase()+service.substring(1)+"\")\n");
    		sb.append("public class " + className + " implements "+ service +" {\n\n");
    		
    		sb.append("\tprivate Logger log = Logger.getLogger("+className+".class);\n");
    		String instance = mybatis.substring(0,1).toLowerCase() + mybatis.substring(1);
    		sb.append("\t@Autowired\n");
    		sb.append("\tprivate "+mybatis+" " + instance +";\n");
    		
    		sb.append("\t@Override\n");
    		sb.append("\tpublic long addEntity(" + table + " dto) throws Exception {\n");
    		sb.append("\t\ttry {\n");
    		sb.append("\t\t\tlog.info(\"addEntity\");\n");
    		sb.append("\t\t\t"+instance+".addEntity(dto);\n");
    		sb.append("\t\t\treturn 0;\n");
    		sb.append("\t\t} catch(Exception e) {\n");
    		sb.append("\t\t\tlog.info(\"addEntity异常\");\n");
    		sb.append("\t\t\te.printStackTrace();\n");
    		sb.append("\t\t\tthrow e;\n");
    		sb.append("\t\t}\n");
    		sb.append("\t}\n");
    		
    		sb.append("\t@Override\n");
    		sb.append("\tpublic boolean deleteEntity(long id) throws Exception {\n");
    		sb.append("\t\ttry {\n");
    		sb.append("\t\t\tlog.info(\"deleteEntity\");\n");
    		sb.append("\t\t\t"+instance+".deleteEntity(id);\n");
    		sb.append("\t\t\treturn false;\n");
    		sb.append("\t\t} catch(Exception e) {\n");
    		sb.append("\t\t\tlog.info(\"deleteEntity异常\");\n");
    		sb.append("\t\t\te.printStackTrace();\n");
    		sb.append("\t\t\tthrow e;\n");
    		sb.append("\t\t}\n");
    		sb.append("\t}\n");
    		
    		sb.append("\t@Override\n");
    		sb.append("\tpublic long updateEntity(" + table + " dto) throws Exception {\n");
    		sb.append("\t\ttry {\n");
    		sb.append("\t\t\tlog.info(\"updateEntity\");\n");
    		sb.append("\t\t\t"+instance+".updateEntity(dto);\n");
    		sb.append("\t\t\treturn 0;\n");
    		sb.append("\t\t} catch(Exception e) {\n");
    		sb.append("\t\t\tlog.info(\"updateEntity异常\");\n");
    		sb.append("\t\t\te.printStackTrace();\n");
    		sb.append("\t\t\tthrow e;\n");
    		sb.append("\t\t}\n");
    		sb.append("\t}\n");
    		
    		sb.append("\t@Override\n");
    		sb.append("\tpublic "+table+" findEntity(" + table + " dto) throws Exception {\n");
    		sb.append("\t\ttry {\n");
    		sb.append("\t\t\tlog.info(\"findEntity\");\n");
    		sb.append("\t\t\treturn "+instance+".findEntity(dto);\n");
    		sb.append("\t\t} catch(Exception e) {\n");
    		sb.append("\t\t\tlog.info(\"findEntity异常\");\n");
    		sb.append("\t\t\te.printStackTrace();\n");
    		sb.append("\t\t\tthrow e;\n");
    		sb.append("\t\t}\n");
    		sb.append("\t}\n");
    		
    		sb.append("\t@Override\n");
    		sb.append("\tpublic List<"+table+"> findEntityList(" + table + " dto) throws Exception {\n");
    		sb.append("\t\ttry {\n");
    		sb.append("\t\t\tlog.info(\"findEntityList\");\n");
    		sb.append("\t\t\treturn "+instance+".findEntityList(dto);\n");
    		sb.append("\t\t} catch(Exception e) {\n");
    		sb.append("\t\t\tlog.info(\"findEntityList异常\");\n");
    		sb.append("\t\t\te.printStackTrace();\n");
    		sb.append("\t\t\tthrow e;\n");
    		sb.append("\t\t}\n");
    		sb.append("\t}\n");
    		
    		sb.append("}\n");
//    		System.out.println(sb.toString());
    		outputToFile(className+".java", sb.toString());
    	}
    }
    
	public static void main(String[] agrs) {
		GenerateServiceImpl aa = new GenerateServiceImpl();
		aa.getServiceImpl();
	}

}

生成mybatis的实现xml文件

package com.csair.test;

import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class GenerateMybatisXml {

	private String url = "jdbc:mysql://localhost:3306/jdz";
	private String username = "root";
	private String password = "root";
	private String driverClass = "com.mysql.jdbc.Driver";
	private DatabaseMetaData dbMetaData = null;
	private String packageNameMybatisDao = "com.csair.test.mybatisRepository";
//	private String path = "C:\\Users\\ppt\\Desktop\\test\\";
	private String path = "D:\\ppt\\workspace\\shop\\shopserver\\src\\test\\java\\com\\csair\\test\\xml\\";
	
	/**
	 * @Description: 获取表对应的所有列
	 * @author: ppt
	 * @date: 2015-3-16 上午10:13:17
	 * @param tableName
	 * @return: void
	 */
	public void getTableColumns(String tableName) {
		try {
			ResultSet resultSetColumn = dbMetaData.getColumns(null, null,
					tableName, null);
			String className = getFormatString(tableName, true);//表名
			String fileName = className + "Mapper";//文件名
			String thisFileName = className + "MybatisDao";//接口地址
			StringBuffer header = new StringBuffer(
					"<?xml version=\"1.0\" encoding=\"UTF-8\" ?>\n");
			header.append("<!DOCTYPE mapper PUBLIC \"-//mybatis.org//DTD Mapper 3.0//EN\" \"http://mybatis.org/dtd/mybatis-3-mapper.dtd\">\n");
			header.append("<mapper namespace=\"");
			header.append(packageNameMybatisDao+"."+thisFileName);
			header.append("\">\n\n");
			StringBuffer footer = new StringBuffer("\n</mapper>");
			StringBuffer add = new StringBuffer();
			//增加数据
			add.append("\t<insert id=\"addEntity\" parameterType=\""+className+"\">\n");
			String insert = "\t\tinsert into " + tableName + "(";
			String column = "";
			String values = "";
			//删除数据
			StringBuffer delete = new StringBuffer();
			delete.append("\t<update id=\"deleteEntity\" parameterType=\"long\">\n");
			delete.append("\t\tupdate " + tableName + " set status = -1 where id = #{id}\n");
			delete.append("\t</update>\n\n");
			//更新数据
			StringBuffer update = new StringBuffer();
			String updateConent = "";
			update.append("\t<update id=\"updateEntity\" parameterType=\""+className+"\">\n");
			update.append("\t\tupdate " + tableName + " set ");
			//查找数据
			StringBuffer select = new StringBuffer();
			select.append("\t<select id=\"findEntity\" parameterType=\""+className+"\" resultType=\""+className+"\">\n");
			String selectContent = "\t\tselect\n";
			//查找数据list
			StringBuffer selectList = new StringBuffer();
			selectList.append("\t<select id=\"findEntityList\" parameterType=\""+className+"\" resultType=\""+className+"\">\n");
			while (resultSetColumn.next()) {
				String columnName = resultSetColumn.getString("COLUMN_NAME");
				String oldColumnName = columnName;
				columnName = this.getFormatString(columnName, false);
				//增加数据
				if("ID".equals(columnName.toUpperCase())) {
					continue;
				}
				column += columnName +",";
				values += "#{" + columnName + "},";
				//删除数据
				
				//更新数据
				updateConent += columnName + "=#{" + columnName +"},";
				//查找数据
				selectContent += "\t\t"+oldColumnName + " AS " + columnName + ",\n";
			}
			selectContent = selectContent.substring(0, selectContent.length()-2);
			selectContent += "\n\t\tfrom " + tableName + "\n";
			select.append(selectContent);
			select.append("\t</select>\n\n");
			selectList.append(selectContent);
			selectList.append("\t</select>\n\n");
			updateConent = updateConent.substring(0,updateConent.length()-1);
			update.append(updateConent);
			update.append(" where id = #{id}\n");
			update.append("\t</update>\n\n");
			column = column.substring(0, column.length()-1);
			values = values.substring(0,values.length()-1);
			String statement = insert + column + ") values(" + values + ")\n";
			add.append(statement);
			add.append("\t</insert>\n\n");
			header.append(add);
			header.append(delete);
			header.append(update);
			header.append(select);
			header.append(selectList);
			header.append(footer);
//			System.out.println(header.toString());
			 this.outputToFile(fileName+".xml", header.toString()); 
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}
	
	public GenerateMybatisXml() {
		try {
			Class.forName(driverClass);
			Connection conn = DriverManager.getConnection(url, username, password);
			dbMetaData = conn.getMetaData();
		} catch(Exception e) {
			e.printStackTrace();
		}
	}
	  
	/**
	 * @Description: 获取所以的表
	 * @author: ppt
	 * @date: 2015-3-16 上午10:12:57
	 * @return: void
	 */
	public List<String> getAllTableList() {  
		List<String> tableList = new ArrayList<String>();
        try {  
            String[] types = { "TABLE" };  
            ResultSet rs = dbMetaData.getTables(null, null, "%", types);  
            while (rs.next()) {  
                String tableName = rs.getString("TABLE_NAME");  //表名 
                tableList.add(tableName);
            }  
        } catch (SQLException e) {  
            e.printStackTrace();  
        }
        return tableList;
    }  
	
	 /** 
     * 处理字符串，去掉下划线“_”，并且把下划线的下一个字符变大写，flag为true，表示首字母要大写 
     * @param name 
     * @param flag 
     * @return 
     */  
    private String getFormatString(String name, boolean flag) {  
        name = name.toLowerCase();  
        String[] nameTemp = name.split("_");  
        StringBuffer buffer = new StringBuffer();  
        for(String str : nameTemp) {  
            String head = str.substring(0, 1).toUpperCase();  
            String tail = str.substring(1);  
            buffer.append(head+tail);  
        }  
        StringBuffer result = null;  
        if(!flag) {  
            result = new StringBuffer();  
            String head = buffer.substring(0, 1).toLowerCase();  
            String tail = buffer.substring(1);  
            result.append(head+tail);  
            return result.toString();  
        }  
        return buffer.toString();  
    }  
	/** 
     * 把String内容写到文件 
     * @param fileName 
     * @param content 
     */  
    private void outputToFile(String fileName, String content) {  
        OutputStream os = null;  
        try {  
            os = new FileOutputStream(path+fileName);  
        } catch (FileNotFoundException e1) {  
            e1.printStackTrace();  
        }  
        byte[] b = content.getBytes();  
        try {  
            os.write(b);  
            os.flush();  
        } catch (IOException e) {  
            e.printStackTrace();  
        } finally {  
            try {  
                os.close();  
            } catch (IOException e) {  
                e.printStackTrace();  
            }  
        }  
    }  
    
    
	public static void main(String[] agrs) {
		GenerateMybatisXml aa = new GenerateMybatisXml();
		List<String> tableList = aa.getAllTableList();
		for(String tableName : tableList) {
			aa.getTableColumns(tableName);
		}
//		aa.getTableColumns("loan_info");
	}
	
	
	

}



ppt0501
关注
  • 2
    点赞
  • 踩
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
java快速创建dto_还在Java手写set调用？快速自动生成方法了解下——自动生成Java dto的set方法调用代码...
weixin_39815879的博客
02-13 1316
1. 演示 如上图所示，可以快速自动生成Java dto的set方法调用。使用时只需要两步：复制、执行，之后就可以粘贴生成好的set方法调用代码(包括import、new语句等)。2. 代码地址生成set方法调用代码的工程https://github.com/Adrninistrator/GenSetterCalls 、 https://gitee.com/adrninistrator/GenSe...
IntelliJ IDEA利用dto代码生成器根据数据库表自动生成实体类(内附dto的jar包下载地址)
weixin_42072280的博客
11-03 3331
参考链接 1.新建maven项目 2.pom.xml中添加编译器版本以及sql驱动 <properties> <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding> <maven.compiler.source>1.8</maven.compiler.source> <maven.compiler.targe
【项目管理】在IDEA中使用MyBatis_Generator生成Dto、Dao、Mapping
每一名出色的架构师，必定是一位优秀程序员
03-21 9516
一、添加配置文件 新建一个空的XML配置文件，名称可以随便取，这里以generatorConfig.xml为名，放在resources目录下generator文件中。具体内容如下： xml version="1.0" encoding="UTF-8"?>         <!DOCTYPE generatorConfiguration                 PUBLIC "-//
使用MyBatis_Generator生成Dto、Dao、Mapping
硬件幽灵
05-14 957
【有奖征文】Bluemix云上实践征集喽~     Python 创意编程活动    CSDN日报20170513 ——《腾讯2017校招实习生面试总结》 MyBatis学习4---使用MyBatis_Generator生成Dto、Dao、Mapping 标签： MyBatisJava一路博客 2013-06-04 16:30 120137人阅读 评
生成DTO的两种方式
热门推荐
漂泊小柒的专栏
03-01 1万+
假设我们现在有一个DTO，其属性包括两张表的属性，我们现在需要将sql语句查询得到的内容转为一个DTO对象，其解决方法如下：  方法一： SQLQuery query = session.createSQLQuery("select a.id as areaId,a.name as areaName,c.id as screenId,c.name as screenName from
mysql快速生成.Net用的实体类DTO的cs内容
u013899802的博客
01-24 393
最近才发现这个偷懒的思路，以前一直没想到可以这样用sql语法，浪费好多时间做重复性无意义的工作。。。话不多说了直接上代码，解决繁琐手敲实体类DTO代码问题，尤其是一些表字段很多的库表。参考以下sql其实也可以快速生成其他需要的东西（举一反三）。 ——身为程序员必须得会偷懒！...
eclipse插件，根据数据库表自动生成DTO(pojo)插件
08-19
因此，"eclipse插件，根据数据库表自动生成DTO(pojo)插件"应运而生，它极大地提高了开发效率。 该插件的核心功能是根据数据库中的表结构自动生成对应的DTO和POJO类。这样，开发者无需手动编写这些类，只需专注于...
反射生成dto service serviceimpl mapper model类 集成mybatis-plus
06-27
反射生成dto service serviceimpl mapper model类，自动生成代码集成mybats-plus 让开发更高效
扩展MyBatisPlus代码生成器实现自定义源码生成，可生成前端页面、vo对象、dto对象等代码
03-07
MyBatisPlus提供了一个内置的代码生成器`mybatis-plus-generator`，能够根据数据库中的表信息自动生成诸如Entity、Mapper、Service、Controller等核心层的代码，极大地提高了开发效率。然而，对于VO对象、DTO对象...
一款自动生成mybatis dto、dao、config的工具
08-11
标题中的“一款自动生成mybatis dto、dao、config的工具”指的是一个自动化代码生成工具，它可以帮助开发者快速生成MyBatis框架所需的Data Transfer Object (DTO)、Data Access Object (DAO)以及配置文件，大大减轻...
DTO代码自动生成器
07-26
根据数据库生成实体类，生成的mapper.xml为空，请谨慎下载。使用说明：https://blog.csdn.net/L13890/article/details/81216441
快速生成DTO
01-10
通过导入Excel模板快速生成DTO,简单易用，下载即可打开使用，加快开发效率
Mybatis的DTO代码自动生成器
04-03
在进行mybatis的操作，如果不想编写接口类，就可以用代码补全的DTOHelper https://blog.csdn.net/weixin_43780471/article/details/91338243
【我的开源】拿来即用！代码生成器：mybatis-plus-generator自定义模板生成 DTO、VO、Convertor、增删改查方法
万物皆字节
08-06 8051
在开发后台管理系统的时候，感觉基本上操作都是一样的：curd，于是借助并扩展mp的代码生成器自动生成一套api,减少重复劳动而且还统一规范，何乐而不为
mybatis-plus DTO代码自动生成
qq_34308732的博客
05-17 7099
import com.baomidou.mybatisplus.generator.AutoGenerator; import com.baomidou.mybatisplus.generator.config.*; import com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder; import com.baomidou...
java 生成mysql dto_如何根据动态SQL代码自动生成DTO
weixin_28818873的博客
02-06 469
当前的状况一般做数据库相关开发, 除非学习, 否则很少有人愿意直接使用JDBC。本来Java代码就比较啰嗦了，而直接用JDBC写代码之啰嗦简直有些令人发狂！所以在实际开发过程中，我们通常都会使用一些框架/库来帮助我们操作数据库。而且开源市场上的选择也比较多，就我个人接触到的有：Hibernate，MyBatis，JdbcTemplate，DbUtils，ActiveRecord，JavaLite等...
如何根据动态SQL代码自动生成DTO
weixin_34032779的博客
06-29 1380
2019独角兽企业重金招聘Python工程师标准>>> ...
自动生成DTO
weixin_43228497的博客
01-02 777
mybatis plus 自动生成dto
最新发布
03-17
代码生成器可以根据数据库表结构自动生成实体类、Mapper接口和XML配置文件。可以通过配置文件或者命令行参数指定生成的目录和文件名。 5. 使用DTO：生成的DTO类可以直接在业务逻辑中使用，通过DTO类可以方便地进行...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值