JAVAssist---动态改动注解

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

多么奇妙的动态改动注解，执行时改动非常多问题就迎刃而解了！