前言
本文是为了学习Spring IOC容器的执行过程而写,不能完全代表Spring IOC容器,只是简单实现了容器的依赖注入和控制反转功能,无法用于生产,只能说对理解Spring容器能够起到一定的作用。
开始
创建项目
创建Gradle项目,并修改build.gradle
plugins {
id 'java'
id "io.franzbecker.gradle-lombok" version "3.1.0"
}
group 'io.github.gcdd1993'
version '1.0-SNAPSHOT'
sourceCompatibility = 1.8
repositories {
mavenCentral()
}
dependencies {
testCompile group: 'junit', name: 'junit', version: '4.12'
}
创建BeanFactory
BeanFactory是IOC中用于存放bean实例以及获取bean的核心接口,它的核心方法是getBean以及getBean的重载方法,这里简单实现两个getBean的方法。
package io.github.gcdd1993.ioc.bean;
/**
* bean factory interface
*
* @author gaochen
* @date 2019/6/2
*/
public interface BeanFactory {
/**
* 通过bean名称获取bean
*
* @param name bean名称
* @return bean
*/
Object getBean(String name);
/**
* 通过bean类型获取bean
*
* @param tClass bean类型
* @param 泛型T
* @return bean
*/
T getBean(Class tClass);
}
创建ApplicationContext上下文
ApplicationContext,即我们常说的应用上下文,实际就是Spring容器本身了。
我们创建ApplicationContext类,并实现BeanFactory接口。
public class ApplicationContext implements BeanFactory {
}
getBean方法
既然说是容器,那肯定要有地方装我们的bean实例吧,使用两个Map作为容器。
/**
* 按照beanName分组
*/
private final Map beanByNameMap = new ConcurrentHashMap<>(256);
/**
* 按照beanClass分组
*/
private final Map, Object> beanByClassMap = new ConcurrentHashMap<>(256);
然后,我们可以先完成我们的getBean方法。
@Override
public Object getBean(String name) {
return beanByNameMap.get(name);
}
@Override
public T getBean(Class tClass) {
return tClass.cast(beanByClassMap.get(tClass));
}
直接从Map中获取bean实例,是不是很简单?当然了,在真实的Spring容器中,是不会这么简单啦,不过我们这次是要化繁为简,理解IOC容器。
构造器
Spring提供了@ComponentScan来扫描包下的Component,我们为了简便,直接在构造器中指定要扫描的包。
private final Set basePackages;
/**
* 默认构造器,默认扫描当前所在包
*/
public ApplicationContext() {
this(new HashSet<>(Collections.singletonList(ApplicationContext.class.getPackage().getName())));
}
/**
* 全参构造器
* @param basePackages 扫描的包名列表
*/
public ApplicationContext(Set basePackages) {
this.basePackages = basePackages;
}
refresh方法
refresh的过程基本按照以下流程来走
扫描指定的包下所有带@Bean注解(Spring中是@Component注解)的类。
List beanClasses = PackageScanner.findClassesWithAnnotation(packageName, Bean.class);
System.out.println("scan classes with Bean annotation : " + beanClasses.toString());
for (Class beanClass : beanClasses) {
try {
createBean(beanClass);
} catch (ClassNotFoundException | NoSuchMethodException | IllegalAccessException | InvocationTargetException | InstantiationException e) {
e.printStackTrace();
}
}
遍历类,获取类的构造器以及所有字段。
Constructor constructor = beanClass.getDeclaredConstructor();
Object object = constructor.newInstance();
Field[] fields = beanClass.getDeclaredFields();
判断字段是依赖注入的还是普通字段。
如果是普通字段,通过字段类型初始化该字段,并尝试从@Value注解获取值塞给字段。
Value value = field.getAnnotation(Value.class);
if (value != null) {
// 注入
field.setAccessible(true);
// 需要做一些类型转换,从String转为对应的类型
field.set(object, value.value());
}
如果是依赖注入的字段,尝试从beanByClassMap中获取对应的实例,如果没有,就先要去实例化该字段对应的类型。
Autowired autowired = field.getAnnotation(Autowired.class);
if (autowired != null) {
// 依赖注入
String name = autowired.name();
// 按照名称注入
Object diObj;
if (!name.isEmpty()) {
diObj = beanByNameMap.get(name) == null ?
createBean(name) :
beanByNameMap.get(name);
} else {
// 按照类型注入
Class> aClass = field.getType();
diObj = beanByClassMap.get(aClass) == null ?
createBean(aClass) :
beanByClassMap.get(aClass);
}
// 注入
field.setAccessible(true);
field.set(object, diObj);
}
测试我们的IOC容器
创建Address
@Data
@Bean
public class Address {
@Value("2222")
private String longitude;
@Value("1111")
private String latitude;
}
创建Person并注入Address
@Data
@Bean
public class Person {
@Autowired
private Address address;
@Value("gaochen")
private String name;
@Value("27")
private String age;
}
创建测试类ApplicationContextTest
public class ApplicationContextTest {
@Test
public void refresh() {
Set basePackages = new HashSet<>(1);
basePackages.add("io.github.gcdd1993.ioc");
ApplicationContext ctx = new ApplicationContext(basePackages);
ctx.refresh();
Person person = ctx.getBean(Person.class);
System.out.println(person);
Object person1 = ctx.getBean("Person");
System.out.println(person1);
}
}
控制台将会输出:
scan classes with Bean annotation : [class io.github.gcdd1993.ioc.util.Address, class io.github.gcdd1993.ioc.util.Person]
scan classes with Bean annotation : [class io.github.gcdd1993.ioc.util.Address, class io.github.gcdd1993.ioc.util.Person]
Person(address=Address(longitude=2222, latitude=1111), name=gaochen, age=27)
Person(address=Address(longitude=2222, latitude=1111), name=gaochen, age=27)
可以看到,我们成功将Address实例注入到了Person实例中,并且将它们存储在了我们自己的IOC容器中。其实,Spring容器的原理大致就是如此,只不过为了应对企业级开发,提供了很多便捷的功能,例如bean的作用域、bean的自定义方法等等。
获取源码
完整源码可以在我的github仓库获取