手写web服务器:定义Autorwired注解,实现属性自动注入

前言

昨天,我们已经解决了post请求的阻塞问题,所以我们今天又可以继续搞事情了,今天我们要实现的也是spring中很核心的注解——Autowired。这个注解想必大家肯定不陌生,在spring项目中,我们经常用它来为我们的属性注入值,实现属性的自动装配。

好了,话不多说,我们来看具体如何实现.

实现过程

定义注解

这一块就很简单了,前面我们也不止一次写过,这里target指定的是属性

@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Autowired {
}

加到属性上

直接把@Autowired注解加在我们要注入的属性上

优化扫描方法

属性是在类初始化的时候,自动被赋值的,所以我们要调整初始化流程

private static void initRequestMappingMap() {
        logger.info("start to scanRequestMapping, controllerSet = {}", classSet);
        if (classSet == null) {
            return;
        }
        classSet.forEach(aClass -> {
            Annotation controller = aClass.getAnnotation(Controller.class);
            if (Objects.isNull(controller)) {
                return;
            }
            Method[] methods = aClass.getDeclaredMethods();
            for (Method method : methods) {
                RequestMapping annotation = method.getAnnotation(RequestMapping.class);
                if (Objects.nonNull(annotation)) {
                    requestMappingMap.put(annotation.value(), method);
                }
            }
            Field[] fields = aClass.getDeclaredFields();
            try {
                Object o = aClass.newInstance();
                for (Field field : fields) {
                    Autowired annotation = field.getAnnotation(Autowired.class);
                    if (Objects.nonNull(annotation)) {
                        field.setAccessible(true);
                        field.set(o, contentMap.get(field.getType().getName()));
                    }
                }
                contentMap.put(aClass.getName(), o);
            } catch (InstantiationException e) {
                logger.error("初始controller失败:", e);
            } catch (IllegalAccessException e) {
                logger.error("初始controller失败:", e);
            }
        });
        logger.info("scanRequestMapping end, requestMappingMap = {}", requestMappingMap);
    }

我们在初始化这里加了一段字段初始化的代码,上面是完整代码,字段初始化只有短短几行:

Field[] fields = aClass.getDeclaredFields();
try {
    Object o = aClass.newInstance();
    for (Field field : fields) {
        Autowired annotation = field.getAnnotation(Autowired.class);
        if (Objects.nonNull(annotation)) {
            field.setAccessible(true);
            field.set(o, contentMap.get(field.getType().getName()));
        }
    }
    contentMap.put(aClass.getName(), o);
} catch (InstantiationException e) {
    logger.error("初始controller失败:", e);
} catch (IllegalAccessException e) {
    logger.error("初始controller失败:", e);
}

这里需要注意的是,因为属性是私有的,必须通过getDeclaredFields获取属性值,getFields方法是没办法拿到私有属性的;

另外一个需要注意的点是,私有属性必须通过setAccessible设置为可访问才可以,否则会报错:

因为字段赋值是基于对象实例的,所以我们要先创建类的实例:

Object o = aClass.newInstance()

然后通过field.set给属性赋值,这里赋值要通过IOC容器拿到赋值对象的实例,所以被赋值属性的实例必须先初始化,否则会有问题。

同时,我们把带有@Autowired注解的类的实例也存进了IOC容器,这样在后面调用controller对用mapping方法的时候,我们直接从ioc容器中拿出来即可:

Object o = contentMap.get(declaringClass.getName());
Object invoke = method.invoke(o, parameters);

这是因为如果你在调用的时候再去创建实例,这时候属性也要赋值,否则会报错的,所以初始化的时候直接创建实例是比较合理的方式。

测试

浏览器调用下试下:

可以看到,我们调用的时候,service已经有值了,方法调用完成后,结果正常返回:

总结

好了,今天的内容到这里就结束了。在上面的内容中,我们展示了@Autowired注解的定义、具体的应用,以及Ioc对于Autowired注解的处理过程,最后我们经过测试,结果与预期一致,当然具体springboot是如何实现的,还需要进一步的研究和探讨,我这里分享的是自己的实现思路,感兴趣的小伙伴可以自己动手试下。

下面是项目的开源仓库,有兴趣的小伙伴可以去看看,如果有想法的小伙伴,我真心推荐你自己动个手,自己写一下,真的感觉不错:

https://github.com/Syske/syske-boot

原文地址:https://www.cnblogs.com/caoleiCoding/p/14866531.html