简单模拟IOC/DI

2019-05-07

|已阅： 1

IOC/DI是Spring里非常重要的概念，本文简单记录一下自己模拟的IOC/DI实现原理，这里实例化Bean使用了JDK动态代理，相当于模拟了一部分Aop的概念。

1、创建接口

业务开发大部分时候都是基于接口，这样做方便以后扩展，这里定义了一个Ai接口类：

1
2
3

public interface Ai {
    void hello(String name);
}

该接口中只有一个hello方法需要实现。

2、接口实现类

接口实现类添加@ServiceMode注解，这里为了不使用Spring依赖找了一个java自带的注解模拟@Service注解

import javax.xml.ws.ServiceMode;

@ServiceMode
public class AiImpl implements Ai{
    @Override
    public void hello(String name) {
        System.out.println("hello "+name);
    }
}

3、代理类

添加通用代理类，使用JDK动态代理，添加before和after输出（对应切面操作)

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;

public class ProxyAi implements InvocationHandler {
    Object obj;
    public ProxyAi(Object obj){
        this.obj = obj;
    }
    @Override
    public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
        System.out.println("before");
        Object o = method.invoke(obj, args);
        System.out.println("after");
        return o;
    }
}

4、模拟初始化及注入、调用

将这四个类放在同一个包下，启动main方法，执行过程如下：

自动扫描包下的类

将定义了@ServiceMode的类初始化并注册取beans容器里

将使用@Resource注解标注变量注入beans容器里的实例

调用方法

import javax.annotation.Resource;
import java.io.File;
import java.lang.annotation.Annotation;
import java.lang.reflect.AnnotatedType;
import java.lang.reflect.Field;
import java.lang.reflect.Proxy;
import java.util.HashMap;
import java.util.Map;

public class Ma {
    @Resource
    public Ai ai;
    public static void main(String args[]){
        try {
            Map<String, Object> beans = new HashMap<>();
            String pac = Ma.class.getPackage().toString();
            String packageName = pac.replaceAll("package ", "");
            // 取class文件目录
            String url = Ma.class.getResource("").getPath();
            System.out.println("package dir :"+url);
            // 加载当前包下的类
            File test = new File(url);
            String[] fs = test.list();
            for (String f : fs) {
                // 反射加载类
                Class clzz = Class.forName(packageName+"."+f.replace(".class", ""));
                Annotation[] annotations = clzz.getAnnotations();
                for (Annotation annotation : annotations) {
                    String simpleName = annotation.annotationType().getSimpleName();
                    // 如果有ServiceMode注解(对应Spring 里的Service注解，不想加依赖找了一个java自带的注解使用)
                    // 模拟Ioc 初始化Bean 放到窗口map中
                    if(simpleName.equals("ServiceMode")){
                        AnnotatedType[] annotatedInterfaces = clzz.getAnnotatedInterfaces();
                        for (AnnotatedType annotatedInterface : annotatedInterfaces) {
                            // 取接口做代理
                            String typeName = annotatedInterface.getType().getTypeName();
                            System.out.println(f+" interface :"+ typeName);
                            Class<?> aClass = Class.forName(typeName);
                            ProxyAi proxyAi = new ProxyAi(clzz.newInstance());
                            Object obj = (Proxy.newProxyInstance(aClass.getClassLoader(), new Class[]{aClass}, proxyAi));
                            // 按类型保存bean
                            beans.put(typeName, obj);
                        }
                    }
                }
            }
            // bean数量
            System.out.println("bean size:"+ beans.size());
            // 模拟DI
            Ma ma = new Ma();
            Field[] fields = Ma.class.getFields();
            for (Field field : fields) {
                // 字段名
                System.out.println("field name :"+field.getName());
                Annotation[] annotations = field.getAnnotations();
                for (Annotation annotation : annotations) {
                    // 如果有Resource注解 则注入我们的bean 因为不知道类型 所以使用反射的field.set方法设值
                    if(annotation.annotationType().getSimpleName().equals("Resource")){
                        String type = field.getType().getTypeName();
                        System.out.println("DI type is "+type);
                        if(beans.get(type) != null){
                            field.set(ma, beans.get(type));
                        }
                    }
                }
            }
            // 调用注入方法
            ma.ai.hello("ixx");
        }catch (Exception e){
            e.printStackTrace();
        }
    }
}

具体的实现代码里都有注释，看一下执行的输出：

package dir :/C:/Users/admin/IdeaProjects/test/target/test-classes/test/
AiImpl.class interface :test.Ai
bean size:1
field name :ai
Di type is test.Ai
before
hello ixx
after

本文作者： ixx
本文链接： http://blog.jisuye.com/2019/05/07/simulation_ioc_di/
版权声明： 本作品采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可。转载请注明出处！