码迷,mamicode.com
首页 > 编程语言 > 详细

JAVAssist---动态修改注解

时间:2015-03-31 10:58:29      阅读:669      评论:0      收藏:0      [点我收藏+]

标签:

       ?ITOOV3.0开始了,需要研究一些技术难点,先来说一个我觉得比较有趣的技术点吧,需求是这样的,我们需要动态的切换数据源,又由于我们是通过实体单元的注入来完成的,实体单元是通过注解的形式注入的,这样如果我们想修改数据源那么必然就要动态的修改注解(当然可能还有其他的解决方式,但是我觉得动态修改注解还是很有趣的)所以就从动态修改注解开始吧:
       先来看看我们需要修改注解的代码:
/**
 * EntityManager的实例化
 * @author 陈丽娜
 * @version 1.0.0 , 2015年3月30日 下午8:43:27
 * @param <T>
 */
public class CollectionBase<T> extends BaseEaoImpl<T> {
    /**
     * 注入实体单元
     */
    @PersistenceContext(unitName="collection-entity")
    protected EntityManager em;
    /**EntityManger
     * 实例化
     */
    @Override
    protected EntityManager getEntityManager() {        
        return this.em;
    }
}


需要修改的是unitName.

     那么该如何修改呢?开始是没有任何思路的,但当时的我总有种感觉一定可以修改,所以就查了一下,发现了JAVAssist:开源的分析。编辑和创建java字节码的类库。更深入的认识可以百度了解一下,下面一个小的demo来修改一下注解:
      首先来看是如何获得这个注解的:
	@Test
	public void ReadTest() throws NotFoundException{
		ClassPool pool = ClassPool.getDefault();
		//获取要修改的类的所有信息
		CtClass ct = pool.get("com.tgb.itoo.collection.base.CollectionBase"); 	
		//获取类中的方法
		CtMethod[] cms = ct.getDeclaredMethods();		
		 //获取第一个方法(因为只有一个方法)
		 CtMethod cm = cms[0];		
		 System.out.println("方法名称====" + cm.getName());		 
		 //获取方法信息
		 MethodInfo methodInfo = cm.getMethodInfo();		 
		 //获取类里的em属性
		 CtField cf = ct.getField("em");
		 //获取属性信息
		 FieldInfo fieldInfo = cf.getFieldInfo();  		 
		 System.out.println("属性名称===" + cf.getName());
		 
		 //获取注解属性
		 AnnotationsAttribute attribute = (AnnotationsAttribute) fieldInfo.getAttribute(AnnotationsAttribute.visibleTag);
		 System.out.println(attribute);	
		 //获取注解
		 Annotation annotation = attribute.getAnnotation("javax.persistence.PersistenceContext");		
		 System.out.println(annotation);
		 //获取注解的值
		 String text =((StringMemberValue) annotation.getMemberValue("unitName")).getValue() ;		 
		 System.out.println("注解名称===" + text);
		 
	}


运行结果:
方法名称====getEntityManager
属性名称===em
@javax.persistence.PersistenceContext(unitName="collection-entity")
@javax.persistence.PersistenceContext(unitName="collection-entity")
注解名称===collection-entity

        修改注解的方法与获取的一样,只是需要为获取的注解赋值,代码如下:
@Test
	public void UpdateTest() throws NotFoundException{
		   ClassPool pool = ClassPool.getDefault();    
		   //获取需要修改的类
		   CtClass ct = pool.get("com.tgb.itoo.collection.base.CollectionBase"); 
		   
		   //获取类里的所有方法
		   CtMethod[] cms = ct.getDeclaredMethods();
		   CtMethod cm = cms[0];    
		   System.out.println("方法名称====" + cm.getName());
		   
		   MethodInfo minInfo = cm.getMethodInfo();
		   //获取类里的em属性
		   CtField cf = ct.getField("em");
		   FieldInfo fieldInfo = cf.getFieldInfo();  
		   
		   System.out.println("属性名称===" + cf.getName());
		   
		   ConstPool cp = fieldInfo.getConstPool();
		   //获取注解信息
		   AnnotationsAttribute attribute2 = new AnnotationsAttribute(cp, AnnotationsAttribute.visibleTag);
		   Annotation annotation = new Annotation("javax.persistence.PersistenceContext", cp);
		 
		   //修改名称为unitName的注解
		   annotation.addMemberValue("unitName", new StringMemberValue("basic-entity", cp));
		   attribute2.setAnnotation(annotation);
		   minInfo.addAttribute(attribute2);
		   
		   //打印修改后方法
		   Annotation annotation2 = attribute2.getAnnotation("javax.persistence.PersistenceContext");
		   String text = ((StringMemberValue)annotation2.getMemberValue("unitName")).getValue();
		   
		   System.out.println("修改后的注解名称===" + text);
	}


运行结果:
方法名称====getEntityManager
属性名称===em
修改后的注解名称===basic-entity

多么神奇的动态修改注解,运行时修改很多问题就迎刃而解了!

 

JAVAssist---动态修改注解

标签:

原文地址:http://blog.csdn.net/jly4758/article/details/44774217

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!