动手实现MVC: 2. bean加载, IoC依赖注入

背景

前面实现了java包的扫描,扫描之后自然就到了bean的加载,以及spring mvc的一大特性 IoC依赖注入的实现;

这里则将在之前的基础上,实现bean的加载和依赖注入的实现

设计

我们模仿的轮子就是spring mvc,简化一些复杂的场景,这里只实现注解的形式

1. 依赖spring-mvc的使用姿势,我们需要先定义几个注解

  • 类上注解 Service, Component, Repository, Bean

所有类上有上面注解的,都表示需要实例的bean

  • 属性注解 Autowired

表示该属性用一个bean对象来实例化

2. 实例化bean

通过Class创建Bean对象

3. 加载依赖

扫描每个Bean的属性,若包含 @Autowired 注解,则用bean进行赋值

4. 提供获取bean的各种方式

最常见的根据beanName,bean类型来获取Bean

5. 提供动态注册bean

比如业务方依赖第三方的jar包中的某个类,想将它也注册为一个bean,因为不能修改第三方类,所以可以用动态注册的方式来加载bean


实现

1. 注解定义

这个比较简单,直接贴一下几个相关的注解

几个声明类为Bean的注解

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface Bean {
    String value() default "";
}

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Bean
public @interface Component {
    String value() default "";
}

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Bean
public @interface Service {
    String value() default "";
}


@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Bean
public @interface Repository {
    String value() default "";
}

Autowired.java, value对应的是业务方定义的beanName

@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Autowired {
    String value() default "";
}

2. 扫描包,类加载,Bean创建

我们定义一个 BeanFactory 对象,来管理class的加载,bean的创建

扫描包基本上就是之前一篇博文的内容,不做多说,直接看Bean的实例化

实现思路比较清晰,大致流程如下;

  • 遍历Class集合
  • 判断class上是否有几个定义为Bean的注解
  • 确定为bean,则实例化

为了避免每次使用时都扫描一遍,所以这个扫描的结果会保存下来,放在内存中

/**
 * 所有自动实例化的bean的映射表,
 * key为bean name
 * - (如果注解中有指定value值,则bean name就是value值;若没有指定,则是首字母小写的简单类名)
 * - bean name 区分大小写
 * <p>
 * 为了避免bean name相同的问题,将value也保存为一个Map映射表
 */
private Map<String, Map<Class, Object>> nameBeanMap;


/**
 * class到bean的映射表
 */
private Map<Class, Object> clzBeanMap;

实际加载过程如下

/**
 * 实例化自动加载的bean
 *
 * @return
 */
private Map<String, Map<Class, Object>> instanceBean() throws IllegalAccessException, InstantiationException, NoSuchMethodException, InvocationTargetException {
    nameBeanMap = new ConcurrentHashMap<>();
    clzBeanMap = new ConcurrentHashMap<>();


    Annotation[] typeAnos;
    String tmpBeanName;
    Method tmpMethod;
    Object tmpBean;
    Map<Class, Object> tmpClzMap;
    for (Class clz : beanClasses) {
        if (clz.isInterface()) {
            continue;
        }


        // 获取类上注解
        typeAnos = clz.getAnnotations();
        if (typeAnos.length == 0) {
            continue;
        }


        for (Annotation ano : typeAnos) {
            if (ano instanceof Bean || ano.annotationType().isAnnotationPresent(Bean.class)) { // 需要加载bean
                tmpMethod = ano.annotationType().getMethod("value", null);
                if (tmpMethod != null) {
                    tmpBeanName = (String) tmpMethod.invoke(ano, null);
                } else {
                    tmpBeanName = null;
                }

                if (StringUtils.isEmpty(tmpBeanName)) {
                    tmpBeanName = StrUtil.lowerFirstChar(clz.getSimpleName());
                }


                if (nameBeanMap.containsKey(tmpBeanName)) {
                    tmpClzMap = nameBeanMap.get(tmpBeanName);
                } else {
                    tmpClzMap = new ConcurrentHashMap<>();
                }

                if (tmpClzMap.containsKey(clz)) {
                    throw new BeanAlreadyDefinedException("bean " + tmpBeanName + " class: " + clz.getName() + " has already defined!");
                }


                tmpBean = clz.newInstance();
                tmpClzMap.put(clz, tmpBean);
                clzBeanMap.put(clz, tmpBean);
                nameBeanMap.put(tmpBeanName, tmpClzMap);
                break;
            }
        }
    }

    return nameBeanMap;
}

上面的实现比较简单,唯一需要注意下的是判断是否包含我们期待的几个注解, 不知道是否有更优雅的写法,下面这种对于自定义一个注解,上面加上 @Service的情况时,将不太适用

if(ano instanceof Bean || ano.annotationType().isAnnotationPresent(Bean.class)) {
  // xxx
}

其次就是BeanName的生成规则

  • 当注解的value属性有被指定,则beanName即为指定的值;
  • 否则,根据class名,首字母小写即可

3. IoC依赖注入

这个也比较简单,扫描每个bean的属性,将拥有 @Autowired 注解的拎出来, 然后查对应的Bean,赋值即可

/**
 * 依赖注入
 */
private void ioc() throws IllegalAccessException {

    Field[] fields;
    String beanName;
    Object bean;
    for (Object obj : nameBeanMap.values()) {
        fields = obj.getClass().getDeclaredFields();
        for (Field field : fields) {
            if (!field.isAnnotationPresent(Autowired.class)) {
                continue;
            }

            Autowired autowired = field.getAnnotation(Autowired.class);
            beanName = StringUtils.isBlank(autowired.value()) ?
                    StrUtil.lowerFirstChar(field.getName()) : autowired.value();
            bean = nameBeanMap.get(beanName);

            if (bean == null) {
                throw new BeanNotFoundException("bean: " + beanName + " not found! bean class: " + field.getClass().getName());
            }

            field.setAccessible(true);
            field.set(obj, nameBeanMap.get(beanName));
        }
    }
}

属性赋值,关注下两行代码即可

// 强制设置可访问,这样私有的变量也可以修改其内容了
field.setAccessible(true);
field.set(obj, nameBeanMap.get(beanName));

4. 查询bean & 动态注册

查询的几个接口就比较简单了,单纯的从Map中获取对象; 注册也就是向Map中塞对象

其他

源码地址: https://github.com/liuyueyi/quick-mvc

相关博文:

个人博客:一灰的个人博客

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏happyJared

Spring Boot中读取配置属性的几种方式

  本文介绍Spring Boot中读取配置属性的几种方式,项目示例中用到的application.yml和application.properties定义如下...

1.1K2
来自专栏Java技术栈

从 0 开始手写一个 Spring MVC 框架,向高手进阶!

Spring框架对于Java后端程序员来说再熟悉不过了,以前只知道它用的反射实现的,但了解之后才知道有很多巧妙的设计在里面。如果不看Spring的源码,你将会失...

1862
来自专栏java 成神之路

从零手写IOC

36010
来自专栏Spring相关

第11章—使用对象关系映射持久化数据—SpringBoot+SpringData+Jpa进行查询修改数据库

JPA由EJB 3.0软件专家组开发,作为JSR-220实现的一部分。但它又不限于EJB 3.0,你可以在Web应用、甚至桌面应用中使用。JPA的宗旨是为POJ...

1033
来自专栏Jed的技术阶梯

Kafka 中使用 Avro 序列化框架(二):使用 Twitter 的 Bijection 类库实现 avro 的序列化与反序列化

使用传统的 avro API 自定义序列化类和反序列化类比较麻烦,需要根据 schema 生成实体类,需要调用 avro 的 API 实现 对象到 byte[]...

2814
来自专栏菩提树下的杨过

java 利用JAX-RS快速开发RESTful 服务

JAX-RS(Java API for RESTful Web Services)同样也是JSR的一部分,详细规范定义见 https://jcp.org/en/...

2777
来自专栏精讲JAVA

接口方法上的注解无法被 @Aspect 声明的切面拦截的原因分析

在Spring中使用MyBatis的Mapper接口自动生成时,用一个自定义的注解标记在Mapper接口的方法中,再利用@Aspect定义一个切面,拦截这个注解...

3234
来自专栏程序猿DD

基于Consul的分布式信号量实现

在之前《基于Consul的分布式锁实现》一文中我们介绍如何基于Consul的KV存储来实现分布式互斥锁。本文将继续讨论基于Consul的分布式锁实现。信号量是我...

2797
来自专栏互联网研发闲思录

java简单框架设计

     设计框架包可以作为一个工具给大家用,需要有完全不同设计思路给出来,不同于我们去做一个web服务、网站、

1312
来自专栏世界第一语言是java

java根据ip地址获取城市地域信息

2955

扫码关注云+社区