完成msg.cs生成功能

Author: 杨利兵

src/com/yanglb/utilitys/codegen/core/generator/BaseGenerator.java

@@ -18,6 +18,8 @@
 import java.lang.reflect.Field;
 import java.util.ArrayList;
 import java.util.Arrays;
+import java.util.Date;
+import java.util.HashMap;
 import java.util.List;
 
 import com.yanglb.utilitys.codegen.core.model.OptionModel;
@@ -28,6 +30,7 @@
 public class BaseGenerator implements IGenerator{
 	protected ParamaModel paramaModel;
 	protected List<OptionModel> supportOptions;
+	protected HashMap<String, String> settingMap;
 
 	protected void printInfo() {
 		System.out.println("生成信息:");
@@ -68,6 +71,9 @@ protected void printInfo() {
 	public void init(ParamaModel paramaModel) {
 		this.paramaModel = paramaModel;
 		this.supportOptions = new ArrayList<OptionModel>();
+		this.settingMap = new HashMap<String, String>();
+		
+		this.settingMap.put("generationDate", new Date().toString());
 	}
 
 	@Override

src/com/yanglb/utilitys/codegen/core/generator/impl/MsgCSGeneratorImpl.java

@@ -40,7 +40,9 @@ protected void onGeneration() throws CodeGenException {
 
 		// 读取必要的配置数据
 		ISettingReader settingReader = GenFactory.createByName(SupportGen.setting_reader);
-		HashMap<String, String> settingMap = settingReader.settingReader();
+		String genKey = String.format("%s_%s", paramaModel.getType().name(), paramaModel.getLang().name());
+		HashMap<String, String> map = settingReader.settingReader(genKey);
+		settingMap.putAll(map);
 
 		// 读取DB信息表
 		ITableReader tableReader = GenFactory.createByName(SupportGen.table_reader);
@@ -64,6 +66,8 @@ protected void onGeneration() throws CodeGenException {
 					supportTrans = SupportGen.msg_java_translator;
 				} else if (this.paramaModel.getLang() == SupportLang.json) {
 					supportTrans = SupportGen.msg_json_translator;
+				} else if (this.paramaModel.getLang() == SupportLang.cs) {
+					supportTrans = SupportGen.msg_cs_translator;
 				}
 
 				// 转换为可写入的Model（单个文件）

src/com/yanglb/utilitys/codegen/core/generator/impl/MsgGeneratorImpl.java

@@ -27,7 +27,7 @@
 import com.yanglb.utilitys.codegen.support.SupportLang;
 import com.yanglb.utilitys.codegen.utility.StringUtility;
 
-public class BaseDdlTranslator extends BaseSqlTranslator<List<DdlModel>> {
+public class BaseDdlTranslator extends BaseSqlTranslator<DdlModel> {
 	// 外键关系列表
 	protected List<ForeignModel> foreignKeyList = new ArrayList<ForeignModel>();
 
@@ -193,17 +193,6 @@ protected ForeignDetailModel getForeignByDdlDetail(DdlModel model, DdlDetail det
 		return null;
 	}
 
-	/**
-	 * 取得用于替换的Model
-	 */
-	@Override
-	protected Object getReplaceModel() {
-		if(this.model == null || this.model.size() <= 0) return null;
-		
-		// 只使用第一个Sheet替换
-		return this.model.get(0);
-	}
-
 	/**
 	 * 根据前缀查找全表名
 	 * @param prefix 前缀

src/com/yanglb/utilitys/codegen/core/translator/BaseDdlTranslator.java

@@ -19,8 +19,31 @@
 import java.util.List;
 
 import com.yanglb.utilitys.codegen.core.model.TableModel;
+import com.yanglb.utilitys.codegen.exceptions.CodeGenException;
 
 public class BaseMsgTranslator extends BaseTranslator<List<TableModel>> {
+	protected String msgLang = "";
+	
+	@Override
+	protected void onBeforeTranslate() throws CodeGenException {
+		super.onBeforeTranslate();
+		
+		// 当前生成的国际化语言
+		this.msgLang = this.settingMap.get("MsgLang");
+		
+		// 文件名
+		String fileName = getFileName();
+		if (fileName.equals("")) {
+			// 空文件名
+			fileName = this.msgLang;
+		} else {
+			if(!this.isDefaultLanguage()) {
+				fileName = fileName + "." + this.msgLang;
+			}
+		}
+		
+		this.writableModel.setFileName(fileName);
+	}
 
 	/**
 	 * 文件名，优先使用--fn参数指定的文件名，如不指定使用excel名称
@@ -43,4 +66,12 @@ protected String getFileName() {
 
 		return fileName;
 	}
+	
+	/**
+	 * 获取当前语言是否为默认语言
+	 * @return
+	 */
+	protected boolean isDefaultLanguage() {
+		return this.msgLang.equals("default");
+	}
 }

src/com/yanglb/utilitys/codegen/core/translator/BaseMsgTranslator.java

@@ -15,8 +15,32 @@
  */
 package com.yanglb.utilitys.codegen.core.translator;
 
-public class BaseSqlTranslator<T> extends BaseTranslator<T> {
+import java.util.List;
+
+import com.yanglb.utilitys.codegen.exceptions.CodeGenException;
+
+public class BaseSqlTranslator<T> extends BaseTranslator<List<T>> {
 	// 列引号字符，如：SQL Server为[], MySQL为`等
 	protected String sqlColumnStart = "\"";
 	protected String sqlColumnEnd = "\"";
+	
+    /**
+     * 取得用于替换的Model
+     */
+    private Object getReplaceModel() {
+        if(this.model == null || this.model.size() <= 0) return null;
+        
+        // 只使用第一个Sheet替换
+        return this.model.get(0);
+    }
+
+	@Override
+	protected void onAfterTranslate() throws CodeGenException {
+		super.onAfterTranslate();
+		
+		// 替换Flag
+		String data = this.writableModel.getData().toString();
+		data = this.replaceFlags(data, getReplaceModel());
+		this.writableModel.setData(new StringBuilder(data));
+	}
 }

src/com/yanglb/utilitys/codegen/core/translator/BaseSqlTranslator.java

@@ -85,10 +85,6 @@ protected void onTranslate() throws CodeGenException{
 	 */
 	protected void onAfterTranslate() throws CodeGenException {
 		// 编码转换在写入器完成
-		// 标记替换
-		this.replaceFlags();
-		
-		// TODO: 代码格式调整（后期完善）
 	}
 
 	// 翻译操作
@@ -99,14 +95,15 @@ private void doTranslate() throws CodeGenException{
 	}
 
 	/**
-	 * 替换{标记}
-	 * 如果
-	 * @throws CodeGenException 
+	 * 模板替换
+	 * @param template 模板内容
+	 * @param replaceModel 
+	 * @return
+	 * @throws CodeGenException
 	 */
-	private void replaceFlags() throws CodeGenException {
-		Object replaceModel = this.getReplaceModel();
-		List<String> flags = StringUtility.findFlags(this.writableModel.getData());
-		String data = this.writableModel.getData().toString();
+	protected String replaceFlags(String template, Object replaceModel) throws CodeGenException {
+		List<String> flags = StringUtility.findFlags(template);
+		String data = new String(template);
 		for(String key:flags) {
 			// Map元素
 			String value = null;
@@ -142,14 +139,6 @@ private void replaceFlags() throws CodeGenException {
 				data = data.replaceAll(f, value);
 			}
 		}
-		this.writableModel.setData(new StringBuilder(data));
-	}
-	
-	/**
-	 * 取得用于替换的Model
-	 * @return
-	 */
-	protected Object getReplaceModel() {
-		return this.model;
+		return data;
 	}
 }

src/com/yanglb/utilitys/codegen/core/translator/BaseTranslator.java

@@ -77,18 +77,6 @@ protected void onTranslate() throws CodeGenException {
 		sb.append(this.genForeignKey());
 	}
 
-	/**
-	 * 取得用于替换的Model
-	 */
-	@Override
-	protected Object getReplaceModel() {
-		if (this.model == null || this.model.size() <= 0)
-			return null;
-
-		// 只使用第一个Sheet替换
-		return this.model.get(0);
-	}
-
 	/**
 	 * 生成MySQL的DDL
 	 * 

src/com/yanglb/utilitys/codegen/core/translator/impl/DdlMysqlTranslatorImpl.java

@@ -60,17 +60,6 @@ protected void onTranslate() throws CodeGenException {
 			sb.append(this.genDdl(itm));
 		}
 	}
-
-	/**
-	 * 取得用于替换的Model
-	 */
-	@Override
-	protected Object getReplaceModel() {
-		if(this.model == null || this.model.size() <= 0) return null;
-		
-		// 只使用第一个Sheet替换
-		return this.model.get(0);
-	}
 
 	/**
 	 * 生成MySQL的DDL

src/com/yanglb/utilitys/codegen/core/translator/impl/DdlSqliteTranslatorImpl.java

@@ -26,7 +26,7 @@
 import com.yanglb.utilitys.codegen.support.SupportLang;
 import com.yanglb.utilitys.codegen.utility.StringUtility;
 
-public class DmlTranslatorImpl extends BaseSqlTranslator<List<DmlModel>> {
+public class DmlTranslatorImpl extends BaseSqlTranslator<DmlModel> {
 	@Override
 	protected void onBeforeTranslate() throws CodeGenException {
 		super.onBeforeTranslate();
@@ -68,17 +68,6 @@ protected void onTranslate() throws CodeGenException {
 		}
 	}
 
-	/**
-	 * 取得用于替换的Model
-	 */
-	@Override
-	protected Object getReplaceModel() {
-		if(this.model == null || this.model.size() <= 0) return null;
-		
-		// 只使用第一个Sheet替换
-		return this.model.get(0);
-	}
-
 	private String genDml(DmlModel model) {
 		StringBuilder sb = new StringBuilder();