创建注入框架（2）

这一步，我们改变的不多，主要是针对provider接口进行修改。

首先我们对Container类进行整顿，不再使用class to class这种map，将值类型换成Provider接口类型。 这样修改的好处是我们不在局限于只用反射去创造实例。

新的Container.java:

public class Container {

    //利用一个容器来存储类
    private Map<Class<?>,Provider<?>> mapper;

    public Container() {
        mapper = new HashMap<>();
    }

    public <T,K extends T> void registerClass(Class<T> inter, Class<K> impl) {
        mapper.put(inter,new ParseClass<>(inter,impl,this));
    }

    @SuppressWarnings("unchecked")
    public <T> T getImpl(Class<T> inter) {
        return (T) mapper.get(inter).get();
    }

    public boolean hasRegisted(Class inter) {
        return mapper.containsKey(inter);
    }

}

这里我们队getImpl也进行改动，现在返回的是实例，而不是class，最终取实例都从这里取。 因为它的改动，ParseClass也需要调整，我将此类添加对Provider的实现，让他变成provider，并将构造函数和类的权限变成package，这样可以不让用户直接对它进行操作。 改变后的ParseClass.java如下:

class ParseClass<T,K extends T> implements Provider<T> {

    private Class<T> tClass;
    private Class<K> implClass;

    private Container container;

    ParseClass(Class<T> tClass,Class<K> implClass,Container container) {
        this.tClass = tClass;
        this.container = container;
        this.implClass = implClass;
    }

    @SuppressWarnings("unchecked")
    private   T newInstance()
            throws IllegalAccessException, InvocationTargetException, InstantiationException {
        if (!container.hasRegisted(tClass)) {
            return null;
        }
        Class<K> implement = implClass;
        Constructor[] constructors = implement.getConstructors();
        if (constructors.length == 0){
            //没有构造器
            throw new NoConstructorsError();
        }
        T instance = null;
        for (Constructor<K> constructor:constructors) {
            if (constructor.getParameterCount() == 0) {
                instance = constructor.newInstance();
                break;
            }
            if (constructor.isAnnotationPresent(Inject.class)) {
                //获得此构造函数的参数
                Object[] params = new Object[constructor.getParameterCount()];
                int j = 0;
                for (Class z: constructor.getParameterTypes()){
                    Object param = container.getImpl(z);
                    params[j++] = param;
                }
                instance = constructor.newInstance(params);
                break;
            }
        }
        if (instance != null) {
            initFields(instance,implement);
            initMethod(instance,implement);
            return instance;
        }
        return null;
    }

    private void initFields(T instance, Class<K> kClass)
            throws IllegalAccessException, InstantiationException, InvocationTargetException {
        for (Field field : kClass.getDeclaredFields()) {
            if (field.isAnnotationPresent(Inject.class)) {
//                System.out.println(field.getDeclaringClass());
                Object value = container.getImpl(field.getType());
//                System.out.println(field.getType().isInstance(value));
                field.setAccessible(true);
                field.set(instance,value);
                field.setAccessible(false);
            }
        }
    }

    @SuppressWarnings("unchecked")
    private void initMethod(T instance, Class<K> kClass) throws IllegalAccessException, InstantiationException, InvocationTargetException {
        for (Method method : kClass.getMethods()) {
            if (method.isAnnotationPresent(Inject.class)) {
                Object[] params = new Object[method.getParameterCount()];
                int j = 0;
                for (Class z: method.getParameterTypes()) {
                    params[j++] =  container.getImpl(z);
                }
                method.invoke(instance,params);
            }
        }
    }

    @Override
    public T get() {
        try {
            return newInstance();
        } catch (IllegalAccessException | InvocationTargetException | InstantiationException e) {
            e.printStackTrace();
            throw new RuntimeException("无法获取该类实例");
        }
    }
}

 

这些文章也同步到了我自己的公众号平台：互联网科技杂谈

转载于:https://my.oschina.net/mylgb634335272/blog/777580