java通過實體類生成serivce文件

原創 inrtyx 2018-09-03 19:16 
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：testttt     
* @author：tangwzh   
* @createTime：2014-5-19 上午11:21:03     
* @function：   利用反射自動生成service.java和serviceImpl.java
* @version:
 */
public class GenerateService {
	private static final Logger LOGGER = Logger.getLogger(Thread
			.currentThread().getStackTrace()[1].getClassName());
	/**
	 * 
	 * @param ServiceFilePath
	 * @param tableName
	 * @param modelQualityName
	 * @throws ClassNotFoundException 
	 */
	public static void generateServiceFile(String serviceFilePath, String tableName, String modelQualifiedName) throws ClassNotFoundException{
		String lowerCaseName = tableName.toLowerCase();
		String fistLetter = lowerCaseName.substring(0,1).toUpperCase();
		String name = fistLetter + lowerCaseName.substring(1);
		String fileName = serviceFilePath + "/" + name + "Service.java";
		File file = new File(fileName);
		if(!file.exists()){
			try {
				boolean isSuccess = file.createNewFile();
				if(!isSuccess){
					LOGGER.error("創建Service文件失敗");
					return;
				}
			} catch (IOException e) {
				e.printStackTrace();
				LOGGER.error("create Service file error:", e);
			}
		}
		FileWriter fileWriter = null;
		String commonFields = getFields(modelQualifiedName);
		try {
			fileWriter = new FileWriter(file);
			fileWriter.write("\n");
			fileWriter.write("public interface " + name + "Service {");
			//get方法,支持排序
			printTab(fileWriter, 1);
			fileWriter.write("public List<" + name + "> get"+ name +"s(" + commonFields +
					", String sort, String order);");
			fileWriter.write("\n");
			//get方法,支持排序和分頁
			printTab(fileWriter, 1);
			fileWriter.write("public List<" + name + "> get"+ name +"s(" + commonFields + 
					", String sort, String order, int startrow, int pagesize);");
			fileWriter.write("\n");
			//getCount方法
			printTab(fileWriter, 1);
			fileWriter.write("public int get" + name + "sCount(" + commonFields +");");
			fileWriter.write("\n");
			//add方法
			printTab(fileWriter, 1);
			fileWriter.write("public void add" + name + "(" + commonFields + ");");
			fileWriter.write("\n");
			//mod方法
			printTab(fileWriter, 1);
			fileWriter.write("public void mod" + name + "(" + commonFields + ");");
			fileWriter.write("\n");
			//mod方法
			printTab(fileWriter, 1);
			fileWriter.write("public void del" + name + "(String " + name + "Id);");
			fileWriter.write("\n");
			//結束
			fileWriter.write("}");
			
		} catch (Exception e) {
			e.printStackTrace();
			LOGGER.error(e);
		}finally{
			try {
				if(fileWriter != null){
					fileWriter.close();
				}
			} catch (Exception e2) {
				e2.printStackTrace();
			}
		}
		
	}
	public static void generateServiceImplFile(String serviceFilePath, String tableName, String modelQualifiedName) throws ClassNotFoundException{
		String lowerCaseName = tableName.toLowerCase();
		String fistLetter = lowerCaseName.substring(0,1).toUpperCase();
		String name = fistLetter + lowerCaseName.substring(1);
		String fileName = serviceFilePath + "/" + name + "ServiceImpl.java";
		File file = new File(fileName);
		if(!file.exists()){
			try {
				boolean isSuccess = file.createNewFile();
				if(!isSuccess){
					LOGGER.error("創建ServiceImpl文件失敗");
					return;
				}
			} catch (IOException e) {
				e.printStackTrace();
				LOGGER.error("create ServiceImpl file error:", e);
			}
		}
		FileWriter fileWriter = null;
		String commonFields = getFields(modelQualifiedName);
		try {
			fileWriter = new FileWriter(file);
			//開始
			implStart(name, fileWriter);
			
			//get方法,支持排序
			selectOrder(modelQualifiedName, name, fileWriter, commonFields);
			
			//get方法,支持排序和分頁
			selectOrderAndPage(modelQualifiedName, name, fileWriter, commonFields);
			
			//getCount方法
			selectCount(modelQualifiedName, name, fileWriter, commonFields);
			
			//add方法
			insert(modelQualifiedName, name, fileWriter, commonFields);
			
			//mod方法
			update(modelQualifiedName, name, fileWriter, commonFields);
			
			//delete方法
			delete(name, fileWriter);
			
			//結束
			fileWriter.write("}");
			
		} catch (Exception e) {
			e.printStackTrace();
			LOGGER.error(e);
		}finally{
			try {
				if(fileWriter != null){
					fileWriter.close();
				}
			} catch (Exception e2) {
				e2.printStackTrace();
			}
		}
		
	}
	private static void delete(String name, FileWriter fileWriter)
			throws IOException {
		printTab(fileWriter, 1);
		fileWriter.write("@Override");
		printTab(fileWriter, 1);
		fileWriter.write("public void del" + name + "(String " + name.toLowerCase() + "Id){");
		printTab(fileWriter, 2);
		fileWriter.write("Map<String, Object> index = new HashMap<String, Object>();");
		printTab(fileWriter, 2);
		fileWriter.write("if(Constants.DEF_STRING_NULL != " + name.toLowerCase() + "Id){");
		printTab(fileWriter, 3);
		fileWriter.write("index.put(\"" + name.toLowerCase() + "Id\", " + name.toLowerCase() + "Id);");
		printTab(fileWriter, 2);
		fileWriter.write("}");
		printTab(fileWriter, 1);
		printTab(fileWriter, 2);
		fileWriter.write(name.toLowerCase() + "Dao.delete" + name + "(index);");
		printTab(fileWriter, 1);
		fileWriter.write("}");
		fileWriter.write("\n");
	}
	private static void update(String modelQualifiedName, String name,
			FileWriter fileWriter, String commonFields) throws IOException,
			ClassNotFoundException {
		printTab(fileWriter, 1);
		fileWriter.write("@Override");
		printTab(fileWriter, 1);
		fileWriter.write("public void mod" + name + "(" + commonFields + "){");
		printTab(fileWriter, 0);
		fileWriter.write(generateValid(modelQualifiedName));
		printTab(fileWriter, 1);
		printTab(fileWriter, 2);
		fileWriter.write(name.toLowerCase() + "Dao.update" + name + "(index);");
		printTab(fileWriter, 1);
		fileWriter.write("}");
		fileWriter.write("\n");
	}
	private static void insert(String modelQualifiedName, String name,
			FileWriter fileWriter, String commonFields) throws IOException,
			ClassNotFoundException {
		printTab(fileWriter, 1);
		fileWriter.write("@Override");
		printTab(fileWriter, 1);
		fileWriter.write("public void add" + name + "(" + commonFields + "){");
		printTab(fileWriter, 0);
		fileWriter.write(generateInstance(name, modelQualifiedName));
		fileWriter.write("\n");
		printTab(fileWriter, 2);
		fileWriter.write(name.toLowerCase() + "Dao.insert" + name + "(instance);");
		//結束
		printTab(fileWriter, 1);
		fileWriter.write("}");
		fileWriter.write("\n");
	}
	private static String generateInstance(String name, String modelQualifiedName) throws ClassNotFoundException {
		Class<?> c = Class.forName(modelQualifiedName);    
		Field[] fields = c.getDeclaredFields();  
		String fieldName = null;
		StringBuilder sb = new StringBuilder();
		sb.append("\t").append("\t").append( name + " instance = new " + name + "();");
		for(Field field: fields){   
			fieldName = field.getName();
			sb.append("\n").append("\t").append("\t").append("instance.set" + firstLetterUpper(fieldName) + "(" + fieldName + ");");
		}
		return sb.toString();
	}
	private static void selectCount(String modelQualifiedName, String name,
			FileWriter fileWriter, String commonFields) throws IOException,
			ClassNotFoundException {
		printTab(fileWriter, 1);
		fileWriter.write("@Override");
		printTab(fileWriter, 1);
		fileWriter.write("public int get" + name + "sCount(" + commonFields +"){");
		printTab(fileWriter, 0);
		fileWriter.write(generateValid(modelQualifiedName));
		//return語句
		fileWriter.write("\n");
		printTab(fileWriter, 2);
		fileWriter.write("return " + name.toLowerCase() + "Dao.select" + name + "sCount(index);");
		//結束
		printTab(fileWriter, 1);
		fileWriter.write("}");
		fileWriter.write("\n");
	}
	private static void selectOrderAndPage(String modelQualifiedName,
			String name, FileWriter fileWriter, String commonFields)
			throws IOException, ClassNotFoundException {
		printTab(fileWriter, 1);
		fileWriter.write("@Override");
		printTab(fileWriter, 1);
		fileWriter.write("public List<" + name + "> get"+ name +"s(" + commonFields + 
				", String sort, String order, int startRow, int pageSize){");
		printTab(fileWriter, 0);
		fileWriter.write(generateValid(modelQualifiedName));
		//sort
		printTab(fileWriter, 2);
		fileWriter.write("if(Constants.DEF_STRING_NULL != sort){");
		printTab(fileWriter, 3);
		fileWriter.write("index.put(\"sort\", sort);");
		printTab(fileWriter, 2);
		fileWriter.write("}");
		//order
		printTab(fileWriter, 2);
		fileWriter.write("if(Constants.DEF_STRING_NULL != order){");
		printTab(fileWriter, 3);
		fileWriter.write("index.put(\"order\", order);");
		printTab(fileWriter, 2);
		fileWriter.write("}");
		//startRow
		printTab(fileWriter, 2);
		fileWriter.write("if(Constants.DEF_INT_NULL != startRow){");
		printTab(fileWriter, 3);
		fileWriter.write("index.put(\"startRow\", startRow);");
		printTab(fileWriter, 2);
		fileWriter.write("}");
		//pageSize
		printTab(fileWriter, 2);
		fileWriter.write("if(Constants.DEF_INT_NULL != pageSize){");
		printTab(fileWriter, 3);
		fileWriter.write("index.put(\"pageSize\", pageSize);");
		printTab(fileWriter, 2);
		fileWriter.write("}");
		//return語句
		fileWriter.write("\n");
		printTab(fileWriter, 2);
		fileWriter.write("return " + name.toLowerCase() + "Dao.select" + name + "s(index);");
		printTab(fileWriter, 1);
		//結束
		fileWriter.write("}");
		fileWriter.write("\n");
	}
	private static void selectOrder(String modelQualifiedName,
			String name, FileWriter fileWriter, String commonFields)
			throws IOException, ClassNotFoundException {
		printTab(fileWriter, 1);
		fileWriter.write("@Override");
		printTab(fileWriter, 1);
		fileWriter.write("public List<" + name + "> get"+ name +"s(" + commonFields + 
				", String sort, String order){");
		printTab(fileWriter, 0);
		fileWriter.write(generateValid(modelQualifiedName));
		//sort
		printTab(fileWriter, 2);
		fileWriter.write("if(Constants.DEF_STRING_NULL != sort){");
		printTab(fileWriter, 3);
		fileWriter.write("index.put(\"sort\", sort);");
		printTab(fileWriter, 2);
		fileWriter.write("}");
		//order
		printTab(fileWriter, 2);
		fileWriter.write("if(Constants.DEF_STRING_NULL != order){");
		printTab(fileWriter, 3);
		fileWriter.write("index.put(\"order\", order);");
		printTab(fileWriter, 2);
		fileWriter.write("}");
		//return語句
		fileWriter.write("\n");
		printTab(fileWriter, 2);
		fileWriter.write("return " + name.toLowerCase() + "Dao.select" + name + "s(index);");
		printTab(fileWriter, 1);
		//結束
		fileWriter.write("}");
		fileWriter.write("\n");
	}
	private static void implStart(String name, FileWriter fileWriter)
			throws IOException {
		//註解
		fileWriter.write("\n");
		fileWriter.write("@Service(\"" + name + "ServiceImpl\")");
		//開始
		fileWriter.write("\n");
		fileWriter.write("public class " + name + "ServiceImpl implements " + name + "Service {");
		fileWriter.write("\n");
		printTab(fileWriter, 1);
		fileWriter.write("@Resource(name = \"" + name + "DaoImpl\")");
		printTab(fileWriter, 1);
		fileWriter.write("private " + name + "Dao " + name.toLowerCase() + "Dao;");
		fileWriter.write("\n");
	}
	
	public static String getFields(String modelQualifiedName) throws ClassNotFoundException{
		Class<?> c = Class.forName(modelQualifiedName);    
		Field[] fields = c.getDeclaredFields();  
		String fieldName = null;
		Class<?> type = null;
		String fieldType = null;
		StringBuilder sb = new StringBuilder();
		for(Field field: fields){   
			fieldName = field.getName();
			type = field.getType();
			fieldType = type.getSimpleName();
			sb.append(fieldType + " " + fieldName + ", ");
		}
		sb.deleteCharAt(sb.length() - 2);
		return sb.toString();
	}
	
	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 String generateValid(String modelQualifiedName) throws ClassNotFoundException{
		Class<?> c = Class.forName(modelQualifiedName);    
		Field[] fields = c.getDeclaredFields();  
		String fieldName = null;
		Class<?> type = null;
		String fieldType = null;
		StringBuilder sb = new StringBuilder();
		sb.append("\t").append("\t").append("Map<String, Object> index = new HashMap<String, Object>();");
		String defaultNull = null;
		for(Field field: fields){   
			fieldName = field.getName();
			type = field.getType();
			fieldType = type.getSimpleName();
			if("String".equals(fieldType)){
				defaultNull = "Constants.Constants.DEF_STRING_NULL";
			}else if("int".equals(fieldType)){
				defaultNull = "Constants.DEF_INT_NULL";
			}else{
				defaultNull = "null";
			}
			sb.append("\n");
			sb.append("\t").append("\t");
			sb.append("if(" + defaultNull + " != " + fieldName +"){");
			sb.append("\n");
			sb.append("\t").append("\t").append("\t");
			sb.append("index.put(\"" + fieldName + "\", " + fieldName + ");");
			sb.append("\n");
			sb.append("\t").append("\t").append("}");
		}
		return sb.toString();
	}
	
	public static String firstLetterUpper(String str){
		String s = str.substring(0, 1).toUpperCase();
		return s + str.substring(1);
	}
}

發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章

【java工具類】字節操作

public class ByteUtil {        public static byte[] join(byte[] src,byte[] des){        byte[] result = new byte[src.le

赜 2020-07-08 00:34:12

【java工具類】IPUtil

import java.io.BufferedReader;import java.io.InputStreamReader;import org.apache.struts2.ServletActionContext;public cl

赜 2020-07-08 00:34:12

JAVA SNOWFLAKE 生成唯一ID

snowflake方案 snowflake是Twitter開源的分佈式ID生成算法，結果是一個long型的ID。 這種方案大致來說是一種以劃分命名空間（UUID也算，由於比較常見，所以單獨分析）來生成ID的一種算法，這種方案把64

HelloWorld_MHC 2020-07-06 20:10:19

線程池管理類和線程池類

一.線程池管理類 /** * 線程池管理類 */ public class ThreadPoolFactory { private static final Log logger = LogFactory.getLo

如来神掌十八式 2020-07-06 03:04:05

Java實現簡繁體互轉

package com.hongfei.util; public class Convert { String jtPy = "萬與醜專業叢東絲丟兩嚴喪個丬豐臨爲麗舉麼義烏樂喬習鄉書買亂爭於虧雲亙亞產畝親褻嚲億僅從侖倉儀們價衆優夥會

取与舍 2020-07-04 15:00:46

JAVA調用打印機

一、Java的打印簡介 在我們的實際工作中，經常需要實現打印功能。但由於歷史原因，Java 提供的打印功能一直都比較弱。實際上最初的 jdk 根本不支持打印，直到 jdk1.1 才引入了很輕量的打印支持。實際上，SUN 公司也一直致力於

谱写 2020-07-03 18:28:31

生成帶有參數的二維碼，並返回base64碼到前端進行展示

package com.cx.www.cxbl.core.utils; import com.google.zxing.BarcodeFormat; import com.google.zxing.EncodeHintType; im

慕言小鹏 2020-07-03 17:33:50

VersionUtils

/**  * 1:獲取文件後綴名  * 2：獲取版本號  * 3：獲取的字符串是否爲空  * @author Administrator  *  *

braincer 2020-07-03 14:17:02

target和source之間的轉換（目標對象和源對象之間的轉換）

public class Transfer<Source, Target> {// 定義兩個Class對象private Class sourceC

braincer 2020-07-03 14:17:02

FastDfsUtil

public class FastDFSClient {private TrackerClient trackerClient = null;pri

braincer 2020-07-03 14:16:51

在一段字符中，找出一個字的第幾次出現位置

http://blog.csdn.net/arui_email/article/details/8800553?locationNum=14 版權聲明：本文爲博主原創文章，未經博主允許不得轉載。 [java] view

qq_36026721 2020-07-03 06:06:41

cn.hutool包下DateUtil說明

<!-- https://mvnrepository.com/artifact/cn.hutool/hutool-all --> <dependency> <groupId>cn.

Jacob's blog 2020-07-02 21:08:26

RestTemplate 發送 Https 請求調用

RestTemplate 發送 Https 請求調用 個人博客:https://jacob.org.cn import org.apache.http.conn.ssl.NoopHostnameVerifier; import o

Jacob's blog 2020-07-02 21:08:24

HttpUtils 工具類

package com.vinsuan.park.platform.api; import cn.hutool.core.date.DateUtil; import com.google.gson.JsonObject; import

Jacob's blog 2020-07-02 21:08:24

代碼統計，文本查找，文本匹配，指定路徑文本查找：java版本

很簡單的一個工具類： 工具類功能：會遞歸查找所有子文件夾，代碼統計，模糊匹配等，指定文件統計，文件大小統計 1 指定文件夾，代碼行數統計，調用示例： //統計指定文件夾.java,.js文件的代碼行數，各候後綴的文本數量 How

ToBeYourBaBa 2020-06-28 17:24:42