架构のSpring扩展点（五）：如何保证在同一线程内获取的bean是同一对象-自定义Scope

Spring 的扩展点 是Spring易扩展的一个重要体现，熟悉这些扩展点的定义方式，以及其调用时机，不仅成为工作中利器，也能深度理解Spring框架的切入点。

Scope扩展点 

Scope定义

Scope定义了Bean在Spring中的作用范围到底有多大,就像用户单次的访问权限一样，它设定了Bean在应用中可用范围。

Spring中Bean的作用域

scope	description
singleton	(默认)将单个bean定义作用于每个Spring IoC容器的单个对象实例
prototype	将单个bean定义作用于任意数量的对象实例
request	将单个bean定义的范围扩大到单个HTTP请求的生命周期。也就是说，每个HTTP请求都有自己的bean实例，该实例是在单个bean定义的后面创建的。只有在感知web的Spring ApplicationContext的上下文中才有效
session	将单个bean定义作用于HTTP会话的生命周期。只有在感知web的Spring ApplicationContext的上下文中才有效
application	将一个bean定义作用域作用到ServletContext的生命周期。只有在感知web的Spring ApplicationContext的上下文中才有效
websocket	将一个bean定义定义到WebSocket的生命周期。只有在感知web的Spring ApplicationContext的上下文中才有效

自定义Scope

ThreadScope即同一个线程内多次getBean是同一个对象，不同线程之间getBean对象不一样

public class ThreadLocalScope implements Scope {
 
    private final ThreadLocal> local = ThreadLocal.withInitial(HashMap::new);
 
    @Override
    public Object get(String name, ObjectFactory objectFactory) {
        Map map = local.get();
        Object value = map.get(name);
        if(value == null) {
            value = objectFactory.getObject();
            map.put(name, value);
        }
        return value;
    }
 
    @Override
    public Object remove(String name) {
        Map map = local.get();
        return map.remove(name);
    }
 
    @Override
    public void registerDestructionCallback(String name, Runnable callback) {
        // nothing to do
    }
 
    @Override
    public Object resolveContextualObject(String key) {
        return null;
    }
 
    @Override
    public String getConversationId() {
        return Thread.currentThread().getName();
    }
}

定义一个线程Scope对象

@Target({ElementType.TYPE, ElementType.METHOD})@Retention(RetentionPolicy.RUNTIME)@Inherited@Scope("thread")public @interface ThreadLocalScopeAnn {｝

自定义注解

@Configurationpublic class ScopeConfig {     @Bean    public CustomScopeConfigurer registerThreadLocalScope() {        CustomScopeConfigurer configurer = new CustomScopeConfigurer();        configurer.addScope("thread", new ThreadLocalScope());        return configurer;    }     @Bean    @ThreadLocalScopeAnn    public User user() {        return new User();    }

注册Scope

public class Test {
 
 
    public static void main(String[] args) {
        AnnotationConfigApplicationContext applicationContext = new AnnotationConfigApplicationContext(ScopeConfig.class);
        new Thread(() -> {
            System.out.println(applicationContext.getBean(User.class));
            System.out.println(applicationContext.getBean(User.class));
        }).start();
 
        new Thread(() -> {
            System.out.println(applicationContext.getBean(User.class));
            System.out.println(applicationContext.getBean(User.class));
        }).start();
    }

测试

测试结果

工作原理

当@Scope注解作用在类或方法时，Spring容器启动后会为每个bean生成beandifinition，其中getSope()方法就会返回注解中value的值。其底层是在AbstractBeanFactory#doGetBean中进行判断的。

public abstract class AbstractBeanFactory extends FactoryBeanRegistrySupport implements ConfigurableBeanFactory {

 
    ...
 
  private final Map scopes = new LinkedHashMap<>(8);
    
    ... 
 
    protected  T doGetBean(
                ...
                //单例逻辑
            if (mbd.isSingleton()) {
          sharedInstance = getSingleton(beanName, new ObjectFactory