框架对Hibernate的扩展

为了便于二次开发新增表字段,定义了OverrideExsitsBean,凡是标记此注解的Bean都将覆盖其父类,框架需要实现所有的HQL查询返回的都是派生类的实例。实现此功能,自然要扩展Hibernate。

首先是对Hibernate的SessionFactory——AnnotationSessionFactoryBean进行扩展:

 

public class AnnotationSessionFactoryBeanExt extends AnnotationSessionFactoryBean {
	public AnnotationSessionFactoryBeanExt() {
		setAnnotationConfigurationExtClass(AnnotationConfigurationExt.class);
	}

	public void setAnnotationConfigurationExtClass(Class configurationClass) {
		if (configurationClass == null || !AnnotationConfigurationExt.class.isAssignableFrom(configurationClass)) {
			throw new IllegalArgumentException(
					"AnnotationSessionFactoryBean only supports AnnotationConfigurationExt or subclasses");
		}
		super.setConfigurationClass(configurationClass);
	}
	
	@Override
	protected void postProcessAnnotationConfiguration(AnnotationConfiguration config)
			throws HibernateException {
		super.postProcessAnnotationConfiguration(config);
		
		AnnotationConfigurationExt configExt = (AnnotationConfigurationExt)config;
		Map<String, Object[]> mapOverrideMsg = new HashMap<String, Object[]>();
		Map<String, String> mapRedirect = new HashMap<String, String>();
		
		// 扫描
		for(Iterator iterKey = configExt.getClasses().keySet().iterator(); iterKey.hasNext();) {
			String key = (String)iterKey.next();
			PersistentClass rootClass = (PersistentClass)configExt.getClasses().get(key);
			
			if(rootClass instanceof Subclass) {
				Integer overrideValue = getOverrideValue(key);
				
				if(overrideValue != null) {					
					PersistentClass originalClass = rootClass.getSuperclass();
					
					while(getOverrideValue(originalClass.getEntityName()) != null) {
						originalClass = originalClass.getSuperclass();
					}
					
					String originalEntityName = originalClass.getEntityName();
					
					if(mapOverrideMsg.get(originalEntityName)==null 
							|| overrideValue>(Integer)mapOverrideMsg.get(originalEntityName)[1]) {
						mapOverrideMsg.put(originalEntityName, new Object[]{rootClass, overrideValue});
					}
					
					mapRedirect.put(key, originalEntityName);
				}
			}
		}
		
		// 清理
		Map<String, String> mapRedirectTmp = new HashMap<String, String>();
		for(Iterator<String> iterKey=mapRedirect.keySet().iterator(); iterKey.hasNext(); ) {
			String key = (String)iterKey.next();
			
			configExt.getClasses().remove(key);
			String originalEntityName = mapRedirect.get(key);
			String overrideEntityName = ((Subclass)mapOverrideMsg.get(originalEntityName)[0]).getEntityName();
			
			if(key.equals(overrideEntityName)) {
				iterKey.remove();
				mapRedirectTmp.put(originalEntityName, overrideEntityName);
			} else {
				mapRedirectTmp.put(key, overrideEntityName);
			}
		}
		
		//修正重定向信息
		mapRedirect.putAll(mapRedirectTmp);		
		configExt.setMapRedirect(mapRedirect);
		
		// 修改
		Map<String, String> mapFinalReserve = new HashMap<String, String>();
		for(Iterator<String> iterKey=mapOverrideMsg.keySet().iterator(); iterKey.hasNext(); ) {
			String originalEntityName = (String)iterKey.next();
			Subclass rootClass = (Subclass)mapOverrideMsg.get(originalEntityName)[0];
			PersistentClass newRootClass = CreateNewRootClass(rootClass, originalEntityName);
			configExt.getClasses().put(originalEntityName, newRootClass);
			mapFinalReserve.put(rootClass.getClassName(), originalEntityName);
		}
		configExt.setMapFinalReserve(mapFinalReserve);
	}
	
	protected PersistentClass CreateNewRootClass(Subclass templateClass, String originalEntityName) {		
		PersistentClass originalClass = templateClass;
		Stack<Iterator> lstPropertyIters = new Stack<Iterator>();
		
		while(!originalClass.getEntityName().equals(originalEntityName)) {
			lstPropertyIters.push(templateClass.getUnjoinedPropertyIterator())
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值