超级迷你的javabean容器，实现annotation DI

will_lau 2014-01-06

展开全文

这是一个解释javabean的容器原理的例子，有像spring容器的意思，实现了annotation在setter方法上注入的效果，我没有看过spring或其他ejb容器的代码，自己猜测javabean容器应该就是这样实现的，所谓的DI或者叫ioc的就是这样做到的，利用xml配置文件和用annotation方法没有太大区别，两者都是为了描述注入点和注入对象。

先写一个被注入的类：

[java] view plain copy

package com.red.beans;
public class SourceBean {
public void hello(){
System.out.println("I am injected");
}
}

没什么好说的

再写一个接受注入的类

[java] view plain copy

package com.red.beans;
import com.red.annotations.MyDI;
public class DestinationBean {
private SourceBean sb;
public SourceBean getSb() {
return sb;
}
@MyDI
public void setSb(SourceBean sb) {
this.sb = sb;
}
public void callSB(){
sb.hello();
}
}

这个类有一个SourceBean成员和对应的setter方法，spring比较提倡通过setter方法注入的，我这边也实现了setter方法注入。

注意到有一个自定义的@MyDI annotation，这里的作用就是为了标记这里是个注入点，在利用反射初始化DestinationBean时，要在这里把SourceBean注入进来。

看一下这个@MyDI：

[java] view plain copy

package com.red.annotations;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;
import java.lang.annotation.ElementType;
import java.lang.annotation.RetentionPolicy;
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface MyDI {
}

这个annotation只能用于方法上，且保留时间到runtime，这样才能在运行时利用反射获取。

好了bean都有了，看看container是怎么做注入的吧：

[java] view plain copy

package com.red.beans;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import com.red.annotations.MyDI;
public class Container {
private static Container container=new Container();
public static Container getContainer(){
return container;
}
/**
* @param args
*/
public static void main(String[] args) {
try {
DestinationBean db=(DestinationBean) container.getInstance(DestinationBean.class);
db.callSB();
} catch (InstantiationException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IllegalAccessException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (ClassNotFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (IllegalArgumentException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (InvocationTargetException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
public Object getInstance(Class<?> clazz) throws InstantiationException, IllegalAccessException, ClassNotFoundException, IllegalArgumentException, InvocationTargetException{
Object obj = clazz.newInstance();
Method[] methods=clazz.getDeclaredMethods();//get all the methods
for(int i=0;i<methods.length;i++){//iterate them
if(methods[i].isAnnotationPresent(MyDI.class)&&methods[i].getName().startsWith("set")){//if annotated and is setter
methods[i].invoke(obj, container.getInstance(methods[i].getParameterTypes()[0]));//
}
}
return obj;
}
}