Jfinal 使用ModelGenerator生成实体类,排除表操作

分类: JAVA 评论(10) 395 阅读 2017-09-07 14:42 ZooM查看

摘要 项目上已经存在很多表结构,但是后面需要新增一个功能需要用到新表,在使用ModelGenerator生成时需要排除,特此记录
jfinal


Jfinal 生成器

public class Generator {
	protected MetaBuilder metaBuilder;
	protected BaseModelGenerator baseModelGenerator;
	protected ModelGenerator modelGenerator;
	protected MappingKitGenerator mappingKitGenerator;
	protected DataDictionaryGenerator dataDictionaryGenerator;
	protected boolean generateDataDictionary;
	
	public Generator(DataSource dataSource, String baseModelPackageName, String baseModelOutputDir, String modelPackageName, String modelOutputDir) {
		this(dataSource, new BaseModelGenerator(baseModelPackageName, baseModelOutputDir), new ModelGenerator(modelPackageName, baseModelPackageName, modelOutputDir));
	}
	// 
}


里面提供了:

// 排除表名
gernerator.addExcludedTable("adv");

但是要排除多张表时,这个方法就力不从心了,Jfinal提供了 MetaBuilder

/**
 * 生成实体跳过表
 * @author YCM
 * 2017年8月30日 下午5:10:41
 */
static class _MetaBuilder extends MetaBuilder {
    public _MetaBuilder(DataSource dataSource) {
	 super(dataSource);
    }
	    
    @Override
    protected boolean isSkipTable(String tableName) {
          // 不包含该字符
	  return !tableName.contains("h_disease");
    }
}


附上完整代码:

import javax.sql.DataSource;

import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.activerecord.generator.MetaBuilder;
import com.jfinal.plugin.druid.DruidPlugin;

public class ModelGenerator {

	public static DataSource getDataSource() {
		
		DruidPlugin druidPlugin = new DruidPlugin("jdbc:mysql://127.0.0.1/uftest1?characterEncoding=utf8&zeroDateTimeBehavior=convertToNull",
				"root", "******");
		druidPlugin.start();
		return druidPlugin.getDataSource();
	}
	
	public static void main(String[] args) {
		
		//String modelPath = "D:\\other\\ufhealth_v3\\admin\\src\\cn\\ufhealth\\v3\\admin\\model";
		String modelPath = "D:\\YangChaoMin\\Workspace\\pubbk\\pubbk-core\\src\\main\\java\\cn\\ufhealth\\pubbk\\model";
		// base model 所使用的包名
		String baseModelPackageName = "cn.ufhealth.pubbk.model.base";
		// base model 文件保存路径
		String baseModelOutputDir = modelPath+"\\base";
		
		// model 所使用的包名 (MappingKit 默认使用的包名)
		String modelPackageName = "cn.ufhealth.pubbk.model";
		// model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
		String modelOutputDir = modelPath;
		
		// 创建生成器
		Generator gernerator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);
		
		gernerator.setMetaBuilder(new _MetaBuilder(getDataSource()));
		
		// 添加不需要生成的表名
		// gernerator.addExcludedTable("adv");
		
		// 设置是否在 Model 中生成 dao 对象
		gernerator.setGenerateDaoInModel(true);
		// 设置是否生成字典文件
		gernerator.setGenerateDataDictionary(false);
		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user",移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
		gernerator.setRemovedTableNamePrefixes("h_","m_","o_","p_","s_","u_");
		// 生成
		gernerator.generate();
	}
	
	/**
	 * 跳过表
	 * @author YCM
	 * 2017年8月30日 下午5:10:41
	 */
	static class _MetaBuilder extends MetaBuilder {
	    public _MetaBuilder(DataSource dataSource) {
	        super(dataSource);
	    }
	    
	    @Override
	    protected boolean isSkipTable(String tableName) {
	        return !tableName.contains("h_disease");
	    }
	} 
	
}


评论10
评论已关闭