java通过实体类自动生成ibatis文件

package generate;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.lang.reflect.Field;

import org.apache.log4j.Logger;
/**
 * 
* @ProjectName：GenerateFile     
* @Author：tangwzh   
* @Description: TODO   
* @Version:V1.0   
* @Date：2014-5-20 下午2:54:24
 */
public class GenerateSql {
	private static final Logger LOGGER = Logger.getLogger(Thread
			.currentThread().getStackTrace()[1].getClassName());
	/*public static void main(String[] args) {
		String filePath = "C:\\Users\\inrtyx\\Desktop";
		generateSqlFile(filePath, "RESOURCE", "generate.Resource");
	}*/
	/**
	 * 
	 * @param sqlFilePath 生成的sql路径
	 * @param tableName 表名
	 * @param modelQualityName 实体类全名
	 */
	public static void generateSqlFile(String sqlFilePath, String tableName, String modelQualityName){
		String lowerCaseName = tableName.toLowerCase();
		String fileName = sqlFilePath + "/" + lowerCaseName + ".sql";
		File file = new File(fileName);
		if(!file.exists()){
			try {
				boolean isSuccess = file.createNewFile();
				if(!isSuccess){
					LOGGER.error("创建sql文件失败");
					return;
				}
			} catch (IOException e) {
				e.printStackTrace();
				LOGGER.error("create sql file error:", e);
			}
		}
		FileWriter fileWriter = null;
		try {
			fileWriter = new FileWriter(file);
			
			//文件开始
			generateHeaderSql(modelQualityName, lowerCaseName, fileWriter);
			
			//select语句
			generateSelectSql("select", modelQualityName, lowerCaseName, fileWriter);
			
			//select count语句
			generateSelectSql("selectCount", modelQualityName, lowerCaseName, fileWriter);
			
			//update语句
			generateUpdateSql(modelQualityName, lowerCaseName, fileWriter);
			
			//insert语句
			generateInsertSql(modelQualityName, lowerCaseName, fileWriter);
			
			//delete语句
			generateDeleteSql(lowerCaseName, fileWriter);
			
			//文件结束
			generateEndSql(fileWriter);
		} catch (Exception e) {
			e.printStackTrace();
			LOGGER.error(e);
		}finally{
			try {
				if(fileWriter != null){
					fileWriter.close();
				}
			} catch (Exception e2) {
				e2.printStackTrace();
			}
		}
		
	}
	private static void generateDeleteSql(String lowerCaseName,
			FileWriter fileWriter) throws IOException {
		String upperCaseName = lowerCaseName.toUpperCase();
		String fistLetter = lowerCaseName.substring(0,1).toUpperCase();
		String deleteId = "delete" + fistLetter + lowerCaseName.substring(1);
		//delete头
		printTab(fileWriter, 1);
		fileWriter.write("<insert id=\"" + deleteId + "\" parameterClass=\"java.util.Map\">");
		
		//delete语句
		printTab(fileWriter, 2);
		fileWriter.write("DELETE FROM " + upperCaseName);
		
		//生成where语句
		generateWhereSql(lowerCaseName, fileWriter);
		
		//结束
		printTab(fileWriter, 1);
		fileWriter.write("</delete>");
		
	}
	private static void generateInsertSql(String modelQualityName,
			String lowerCaseName, FileWriter fileWriter) throws ClassNotFoundException, IOException {
		String upperCaseName = lowerCaseName.toUpperCase();
		String fistLetter = lowerCaseName.substring(0,1).toUpperCase();
		String insertId = "insert" + fistLetter + lowerCaseName.substring(1);
		//insert头
		printTab(fileWriter, 1);
		fileWriter.write("<insert id=\"" + insertId + "\" parameterClass=\""+ lowerCaseName +"\">");
		
		//insert语句
		printTab(fileWriter, 2);
		fileWriter.write("<![CDATA[");
		
		Class<?> c = Class.forName(modelQualityName);    
	    Field[] fields = c.getDeclaredFields();    
	    StringBuilder sb = new StringBuilder("INSERT INTO ");    
	    sb.append(upperCaseName).append("(");    
	    String fieldName = "";    
	    for(Field field: fields){    
	    	sb.append("\n").append("\t").append("\t").append("\t").append("\t");
	        fieldName = field.getName();    
	        sb.append(fieldName).append(",");    
	    }    
	    sb.deleteCharAt(sb.length() - 1);    
	    sb.append("\n").append("\t").append("\t").append("\t");
	    sb.append(")").append("VALUES(");    
	    for(Field field: fields){    
	    	sb.append("\n").append("\t").append("\t").append("\t").append("\t");
	        fieldName = field.getName();    
	        if("createTime".equals(fieldName) || "updateTime".equals(fieldName)){
	        	sb.append("NOW()").append(","); 
	        	continue;
	        }
	        sb.append("#").append(fieldName).append("#").append(",");    
	    }    
	    sb.deleteCharAt(sb.length() - 1);  
	    sb.append("\n").append("\t").append("\t").append("\t");
	    sb.append(")");    
	    
	    printTab(fileWriter, 3);
		fileWriter.write(sb.toString());
		
	    printTab(fileWriter, 2);
		fileWriter.write("]]>");
		
		//结束
		printTab(fileWriter, 1);
		fileWriter.write("</insert>");
	}
	private static void generateUpdateSql(String modelQualityName,
			String lowerCaseName, FileWriter fileWriter) throws IOException, ClassNotFoundException {
		String upperCaseName = lowerCaseName.toUpperCase();
		String fistLetter = lowerCaseName.substring(0,1).toUpperCase();
		String updateId = "update" + fistLetter + lowerCaseName.substring(1);
		
		//update头
		printTab(fileWriter, 1);
		fileWriter.write("<update id=\"" + updateId + "\" parameterClass=\"java.util.Map\">");
		
		//update语句
		printTab(fileWriter, 2);
		fileWriter.write("UPDATE " + upperCaseName);
		
		//set语句
		generateWhereOrSetSql("set", lowerCaseName, modelQualityName, fileWriter);
		
		//生成where语句
		generateWhereSql(lowerCaseName, fileWriter);
	
		//结束
		printTab(fileWriter, 1);
		fileWriter.write("</update>");
	}
	private static void generateWhereSql(String lowerCaseName,
			FileWriter fileWriter) throws IOException {
		printTab(fileWriter, 2);
		fileWriter.write("<dynamic prepend=\"WHERE\">");
		String id = lowerCaseName + "Id";
		printTab(fileWriter, 3);
		fileWriter.write("<isNotEmpty prepend=\"AND\" property=\"" + id + "\">");
		printTab(fileWriter, 4);
		fileWriter.write(id.toUpperCase() + "= #" + id + "#");
		printTab(fileWriter, 3);
		fileWriter.write("</isNotEmpty>");
		printTab(fileWriter, 2);
		fileWriter.write("</dynamic>");
	}
	private static void generateEndSql(FileWriter fileWriter) throws IOException {
		printTab(fileWriter, 0);
		fileWriter.write("</sqlMap>");
	}
	private static void generateSelectSql(String type, String modelQualityName,
			String lowerCaseName, FileWriter fileWriter) throws IOException, ClassNotFoundException {
		String upperCaseName = lowerCaseName.toUpperCase();
		String fistLetter = lowerCaseName.substring(0,1).toUpperCase();
		//例如，selecterResources
		String selectedId = "select" + fistLetter + lowerCaseName.substring(1) + "s";
		printTab(fileWriter, 1);
		
		
		//select语句
		if("select".equals(type)){//select语句
			fileWriter.write("<select id=\"" + selectedId + "\" parameterClass=\"java.util.Map\" resultClass=\""+ lowerCaseName +"\">");
			printTab(fileWriter, 2);
			fileWriter.write("<![CDATA[ SELECT * FROM " + upperCaseName + "]]>");
		}else{
			fileWriter.write("<select id=\"" + selectedId + "Count" + "\" parameterClass=\"java.util.Map\" resultClass=\"int\">");
			printTab(fileWriter, 2);
			fileWriter.write("<![CDATA[ SELECT COUNT(*) FROM " + upperCaseName + "]]>");
		}
		//where语句
		generateWhereOrSetSql("where", lowerCaseName, modelQualityName, fileWriter);
		 
		//结束标签
		printTab(fileWriter, 1);
		fileWriter.write("</select>");
	}
	private static void generateWhereOrSetSql(String type, String lowerCaseName, String modelQualityName,
			FileWriter fileWriter) throws IOException, ClassNotFoundException {
		printTab(fileWriter, 2);
		if("where".equals(type)){
			fileWriter.write("<dynamic prepend=\"WHERE\">");
		}else{
			fileWriter.write("<dynamic prepend=\"SET\">");
		}
		
		Class<?> c = Class.forName(modelQualityName);    
		Field[] fields = c.getDeclaredFields();  
		String fieldName = null;
		for(Field field: fields){    
	        fieldName = field.getName();  
	        if("where".equals(type)){
	        	printTab(fileWriter, 3);
		        fileWriter.write("<isNotEmpty prepend=\"AND\" property=\"" + fieldName + "\">");
	        }else{
	        	String id = lowerCaseName + "Id";
	        	if(id.equals(fieldName)){
	        		continue;
	        	}
	        	printTab(fileWriter, 3);
		        fileWriter.write("<isNotEmpty prepend=\",\" property=\"" + fieldName + "\">");
	        }
	        printTab(fileWriter, 4);
	        fileWriter.write(fieldName.toUpperCase() + "=#" + fieldName + "#");
	        printTab(fileWriter, 3);
	        fileWriter.write("</isNotEmpty>");
		}
		 printTab(fileWriter, 2);
		 fileWriter.write("</dynamic>");
	}
	private static void printTab(FileWriter fileWriter, int count) throws IOException {
		fileWriter.write("\n");
		for(int i = 0 ; i < count; i++){
			fileWriter.write("\t");
		}
	}
	private static void generateHeaderSql(String modelQualityName,
			String lowerCaseName, FileWriter fileWriter) throws IOException {
		fileWriter.write("<?xml version=\"1.0\" encoding=\"UTF-8\" ?>");
		fileWriter.write("\n");
		fileWriter.write("<!DOCTYPE sqlMap PUBLIC \"-//ibatis.apache.org//DTD SQL Map 2.0//EN\" \"http://ibatis.apache.org/dtd/sql-map-2.dtd\">");
		fileWriter.write("\n");
		fileWriter.write("<sqlMap namespace=\"" + lowerCaseName + "\">");
		fileWriter.write("\n");
		fileWriter.write(" ");
		fileWriter.write("	<typeAlias alias=\"" + lowerCaseName + "\" type=\""+ modelQualityName  +"\" />");
	}
}