jfinal生成单个实体model

因为项目需求,需要每次生成一个model,然后进行开发,同时希望_MappingKit能够对应的进行更新或者增加。

实际使用环境:jfinal 2.2 mysql  sqlserver

下面直接放代码:

package jfinal.model;

import javax.sql.DataSource;

import jfinal.config.MappingKitBulid;
import jfinal.config.ModelBulid;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.activerecord.dialect.OracleDialect;
import com.jfinal.plugin.activerecord.dialect.SqlServerDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.c3p0.C3p0Plugin;

/**
 * 在数据库表有任何变动时,运行一下 main 方法,极速响应变化进行代码重构
 */
public class _JFinalDemoGenerator {
	
	public static DataSource getDataSource() {
		PropKit.use("db.properties");
		C3p0Plugin c3p0Plugin = new C3p0Plugin(PropKit.get("dataSource.url"), PropKit.get("dataSource.username"), PropKit.get("dataSource.password"), PropKit.get("dataSource.driverClassName"));
		c3p0Plugin.start();
		return c3p0Plugin.getDataSource();
	}
	
	public static void main(String[] args) {
		// base model 所使用的包名
		String baseModelPackageName = "jfinal.model.base";
		// base model 文件保存路径
		String baseModelOutputDir = PathKit.getWebRootPath() + "/../src/jfinal/model/base";
		
		// model 所使用的包名 (MappingKit 默认使用的包名)
		String modelPackageName = "jfinal.model";
		// model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
		String modelOutputDir = baseModelOutputDir + "/..";
		
		// 创建生成器
		Generator gernerator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);
		//System.out.println("数据库类型:"+PropKit.get("dataSource.type"));
		String dataType=PropKit.get("dataSource.type");
		
		// 添加不需要生成的表名
		//gernerator.addExcludedTable("adv");
		// 设置是否在 Model 中生成 dao 对象
		gernerator.setGenerateDaoInModel(true);
		// 设置是否生成字典文件
		gernerator.setGenerateDataDictionary(false);
		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user",移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
		//gernerator.setRemovedTableNamePrefixes("t_");
		
		ModelBulid modelBulid=new ModelBulid(getDataSource(), "basic_mainmenu",dataType);
		gernerator.setMetaBuilder(modelBulid);
		MappingKitBulid mappingKitBulid=new MappingKitBulid(modelPackageName, modelOutputDir);
		gernerator.setMappingKitGenerator(mappingKitBulid);
		// 生成
		gernerator.generate();
	}
}

package jfinal.config;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.List;

import com.jfinal.plugin.activerecord.generator.MappingKitGenerator;
import com.jfinal.plugin.activerecord.generator.TableMeta;

public class MappingKitBulid extends MappingKitGenerator{

	public MappingKitBulid(String mappingKitPackageName,
			String mappingKitOutputDir) {
		super(mappingKitPackageName, mappingKitOutputDir);
		// TODO Auto-generated constructor stub
	}
	
	
	@Override
	protected void genMappingMethod(List<TableMeta> tableMetas, StringBuilder ret) {
		ret.append(String.format(mappingMethodDefineTemplate));
		String add="";
		String addTableName="";
		for (TableMeta tableMeta : tableMetas) {
			boolean isCompositPrimaryKey = tableMeta.primaryKey.contains(",");
			if (isCompositPrimaryKey)
				ret.append(String.format(compositeKeyTemplate, tableMeta.primaryKey));
			add = String.format(mappingMethodContentTemplate, tableMeta.name, tableMeta.primaryKey, tableMeta.modelName);
			addTableName=tableMeta.name;
		}
		readOldMapping(ret,add,addTableName);
		ret.append(String.format("\t}%n"));
	}
	
	//读取之前mapping生成的数据,然后进行追加
	public void readOldMapping(StringBuilder ret,String add,String addTableName){
		String path=mappingKitOutputDir + File.separator + mappingKitClassName + ".java";
		File file=new File(path);
		BufferedReader reader = null; 
		if(file.exists()){
			try {  
	            reader = new BufferedReader(new FileReader(file));  
	            String tempString = null;  
	            boolean start=false;
	            boolean exist=false;
	            // 一次读入一行,直到读入null为文件结束  
	            while ((tempString = reader.readLine()) != null) {  
	                // 显示行号  
	                //System.out.println("line " + line + ": " + tempString);
	            	
	            	if(start && tempString.equals("	}")){
	            		start=false;
	            	}
	            	
	            	if(start && !tempString.trim().equals("")){
	            		int tempi=tempString.indexOf("\"");
	            		String tableName=tempString.substring(tempi+1,tempString.indexOf("\"",tempi+1));
	            		if(!tableName.equals("") && tableName.equals(addTableName)){
	            			if(!exist){
	            				ret.append(add);
	            			}
	            			exist=true;
	            		}else{
	            			ret.append(tempString);
	            			ret.append("\r\n");
	            		}
	            	}
	            	
	            	if(tempString.equals("	public static void mapping(ActiveRecordPlugin arp) {")){
	            		start=true;
	            	}
	            	
	            }
	            
	            if(!exist){
	            	ret.append(add);
	            }
	            
	            reader.close();  
	        } catch (IOException e) {  
	            e.printStackTrace();  
	        } finally {  
	            if (reader != null) {  
	                try {  
	                    reader.close();  
	                } catch (IOException e1) {  
	                }  
	            }  
	        }  
		}
	}
	

}


package jfinal.config;

import javax.sql.DataSource;

import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.activerecord.dialect.OracleDialect;
import com.jfinal.plugin.activerecord.dialect.SqlServerDialect;
import com.jfinal.plugin.activerecord.generator.MetaBuilder;

public class ModelBulid extends MetaBuilder{
	
	private String createTableName;

	public String getCreateTableName() {
		return createTableName;
	}

	public void setCreateTableName(String createTableName) {
		this.createTableName = createTableName;
	}

	public ModelBulid(DataSource dataSource) {
		super(dataSource);
		// TODO Auto-generated constructor stub
	}
	
	public ModelBulid(DataSource dataSource,String createTableName,String dataType) {
		super(dataSource);
		if(dataType.equals("1")){
			setDialect(new SqlServerDialect());
		}else if(dataType.equals("2")){
			setDialect(new MysqlDialect());
		}else if(dataType.equals("3")){
			setDialect(new OracleDialect());
		}
		setCreateTableName(createTableName);
		// TODO Auto-generated constructor stub
	}

	@Override
	protected boolean isSkipTable(String tableName) {
		if(!tableName.equals(createTableName)){
			return true;
		}
		return false;
	}
}


评论区

山东小木

2017-04-07 09:40

这个挺好 JFinalIDE里也在实现这个

yhxjfinal

2017-06-26 13:37

热门分享

扫码入社