手把手带你实现IOC,看完还不懂来打我!

前言

Spring ioc 相信很多人都知道这是Spring框架中一个非常核心的组件,IoC(控制反转),对于初学Spring的人来说,对其的设计思想理解可能非常表面,要理解好Ioc的关键是要明确 以下几点,谁控制谁,控制什么,为何是反转(那正转?),哪些方面反转了

谁控制谁,控制什么 还记得你们当时学习servlet的时候在每一层内部通过new关键字进行创建对象吗,要清楚这是程序主动去创建依赖对象; 而当你们接触了Spring之后IoC是有专门一个容器来创建这些对象,控制对象。

为何是反转(那正转?),这个简单理解当我们主动控制去获取依赖对象,这就是正转? 而反转则是由容器来帮忙创建及注入依赖对象; 为何是反转 ? 因为由容器帮我们查找及注入依赖对象,对象只是被动的接受依赖对象,所以是反转; 哪些方面反转了?依赖对象的获取被反转了。

手撕最简易版IOC容器

可能你还不明白,觉得很抽象?那么跟着小编手把手带你来实现IOC!

ioc不是一个容器嘛,ok,我们先来定义一个容器,容器的特性肯定拥有存、取对象嘛~

Spring的对象都是一个个bean,但是我们不知道bean类型是什么,那么就泛型体现,还有bean什么时候存放进容器的啊?spring是在启动的时候会进行初始化扫描,我们就定义一个initAutoWired方法来模拟。

先定义一个橘松容器接口JsContainer。

/**
 * @创建人 : 头条账号 "深夜敲代码"
 * @创建时间 2021/7/13
 */
public interface JsContainer {

    /**
     * 根据Class获取Bean
     * @param clazz
     * @return
     */
    <T> T getBean(Class<T> clazz);

    /**
     * 注册一个Class到容器中
     *
     * @param clazz
     */
    Object registerBean(Class<?> clazz);

    /**
     * 初始化装配
     */
    void initAutoWired();

我们再写一个自定义注解标识JsAutowired,来为了后面通过反射获取实例化bean,自定义注解可以指定要注入的类型,以及注入的bean名称。

package com.orangesongjava.ioc;

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

/**
 * @创建人 : 头条账号 "深夜敲代码"
 * @创建时间 2021/7/13
 */
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface JsAutowired {

    /**
     * @return 要注入的类类型
     */
    Class<?> value() default Class.class;

    /**
     * @return bean的名称
     */
    String name() default "";
}

容器接口和注解都有了,接下来我们思考来实现它!我们定义一个类 JsSampleContainer 来实现容器接口JsContainer 。

package com.orangesongjava.ioc;

import java.lang.reflect.Field;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @创建人 : 头条账号 "深夜敲代码"
 * @创建时间 2021/7/13
 */
@SuppressWarnings("unchecked")
public class JsSampleContainer implements JsContainer {
    /**
     * 保存所有bean对象,格式为 com.xxx.xxx.XxxClass : @56x2ya
     */
    private Map<String, Object> beanNameMap;

    /**
     * 存储bean和name的关系
     */
    private Map<String, String> beanKeys;

    public JsSampleContainer() {
        this.beanNameMap = new ConcurrentHashMap<>();
        this.beanKeys = new ConcurrentHashMap<>();
    }

    @Override
    public <T> T getBean(Class<T> clazz) {
        String name = clazz.getName();
        Object object = beanNameMap.get(name);
        if(null != object){
            return (T) object;
        }
        return null;
    }

    @Override
    public Object registerBean(Class<?> clazz) {
        String name = clazz.getName();
        beanKeys.put(name, name);
        Object bean = newInstance(clazz);
        beanNameMap.put(name, bean);
        return bean;
    }

    @Override
    public void initAutoWired() {
        beanNameMap.forEach((k,v) -> injection(v));
    }
    /**
     * 注入对象
     * @param object
     */
    public void injection(Object object) {
        // 所有字段
        try {
            Field[] fields = object.getClass().getDeclaredFields();
            for (Field field : fields) {
                // 需要注入的字段
                JsAutowired jsAutowired = field.getAnnotation(JsAutowired.class);
                if (null != jsAutowired) {
                    // 要注入的字段
                    Object autoAutoWiredField = null;
                    String name = jsAutowired.name();
                    // 如果说这里JsAutowired自定义注解没指定name属性 则默认值是""
                    if(!name.equals("")){
                        // 指定了特定的name
                        String className = beanKeys.get(name);
                        if(null != className && !className.equals("")){
                                            autoAutoWiredField = beanNameMap.get(className);
                        }
                        if (null == autoAutoWiredField) {
                            throw new RuntimeException("Unable to load " + name);
                        }
                    } else {
                                        // JsAutowired注解没有name属性
                                        // 判断注入的类型 是否是类Class类型 默认值也是Class
                        if(jsAutowired.value() == Class.class){
                                               autoAutoWiredField = register(field.getType());
                        } else {
                            // 指定装配的类
                                            autoAutoWiredField = this.getBean(jsAutowired.value());
                            if (null == autoAutoWiredField) {
                                                autoAutoWiredField = register(jsAutowired.value());
                            }
                    }
                    }
                    if (null == autoAutoWiredField) {
                        throw new RuntimeException("Unable to load " + field.getType().getCanonicalName());
                    }
                    boolean accessible = field.isAccessible();
                    field.setAccessible(true);
                    field.set(object, autoAutoWiredField);
                    field.setAccessible(accessible);
                }
            }
        } catch (SecurityException e) {
            e.printStackTrace();
        } catch (IllegalArgumentException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
    }

    private Object register(Class<?> clazz){
        if(null != clazz){
            return this.registerBean(clazz);
        }
        return null;
    }

    /**
     * 创建一个实例对象
     * @param clazz class对象
     * @return
     */
    public static Object newInstance(Class<?> clazz) {
        try {
            return clazz.newInstance();
        } catch (InstantiationException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
        return null;
    }
}

测试一下

我们现在来测试一下,我们先定义一个类OrangeSongJavaService,方便待会测试被调用。注意哦,这个类上面没有任何注解。也就是他并没有被Spring管理起来。

/**
 * @创建人 : 头条账号 "深夜敲代码"
 * @创建时间 2021/7/13
 */
package com.orangesongjava.ioc;

public class OrangeSongJavaService {
    public void writeArticleOnJs(String name){
        System.out.println(name + "在头条上写文章!");
    }
}

在定义本地调用的模拟调用OrangeSongJavaClient。注意这个类上面也是没有注解,另外引用的OrangeSongJavaService 被我们的自定义注解JsAutowired标识起来了。

/**
 * @创建人 : 头条账号 "深夜敲代码"
 * @创建时间 2021/7/13
 */
package com.orangesongjava.ioc;

public class OrangeSongJavaClient {

    @JsAutowired
    private OrangeSongJavaService orangeSongJavaService;

    public void work() {
        orangeSongJavaService.writeArticleOnJs("深夜敲代码");
    }
    public OrangeSongJavaService getOrangeSongJavaService() {
        return orangeSongJavaService;
    }
}

现在我们写个测试main方法测试一下。

/**
 * @创建人 : 头条账号 "深夜敲代码"
 * @创建时间 2021/7/13
 */
package com.orangesongjava.ioc;

public class IocTest {

    private static JsSampleContainer jsContainer = new JsSampleContainer();

    public static void main(String[] args) {
        // 将类注入容器
        jsContainer.registerBean(OrangeSongJavaClient.class);
        // 初始化注入-扫描引用
        jsContainer.initAutoWired();
        // 容器获取bean
        OrangeSongJavaClient client jsContainer.getBean(OrangeSongJavaClient.class);
        // 执行
        client.work();
    }
}

如果能输出执行逻辑,则表示这段程序没问题。看到这里,你对ioc理解是不是有点清晰了,有任何疑问,或者不懂的地方可以下方留言。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 218,640评论 6 507
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,254评论 3 395
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 165,011评论 0 355
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,755评论 1 294
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,774评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,610评论 1 305
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,352评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,257评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,717评论 1 315
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,894评论 3 336
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,021评论 1 350
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,735评论 5 346
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,354评论 3 330
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,936评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,054评论 1 270
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,224评论 3 371
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,974评论 2 355

推荐阅读更多精彩内容