Spring自定义作用域Bean生命周期终止：如何标记Bean为垃圾避免状态复用

要让Spring正确销毁TcpRequestScope下的Bean并清理状态，核心是在作用域生命周期结束时，触发所有注册的销毁回调，并彻底清理ThreadLocal存储。以下是具体实现步骤：

1. 完善自定义Scope的销毁方法 #

你的TcpRequestScope目前只实现了registerDestructionCallback，但缺少触发这些回调的逻辑。需要调整存储结构并新增销毁方法：

@Component
public class TcpRequestScope implements Scope {
    private final ThreadLocal<ConcurrentHashMap<String, Object>> scopedObjects = new InheritableThreadLocal<>() {
        @Override
        protected ConcurrentHashMap<String, Object> initialValue(){
            return new ConcurrentHashMap<>();
        }
    };
    // 把销毁回调改为ThreadLocal存储，避免线程间混乱
    private final ThreadLocal<Map<String, Runnable>> destructionCallbacks = new InheritableThreadLocal<>() {
        @Override
        protected Map<String, Runnable> initialValue() {
            return Collections.synchronizedMap(new HashMap<>());
        }
    };

    // 保留原有的get、remove、resolveContextualObject、getConversationId方法

    @Override
    public void registerDestructionCallback(final String name, final Runnable callback) {
        destructionCallbacks.get().put(name, callback);
    }

    // 新增：作用域结束时调用此方法，销毁所有Bean并清理资源
    public void destroyScope() {
        // 触发所有注册的销毁回调（比如@PreDestroy、DisposableBean的destroy方法）
        Map<String, Runnable> callbacks = destructionCallbacks.get();
        for (Runnable callback : callbacks.values()) {
            try {
                callback.run();
            } catch (Exception e) {
                // 捕获销毁异常，避免影响响应发送
                e.printStackTrace();
            }
        }
        // 彻底清理当前线程的ThreadLocal存储
        scopedObjects.remove();
        destructionCallbacks.remove();
    }
}

2. 在请求生命周期结束时触发销毁 #

在请求处理完成（发送响应前）的节点，主动调用destroyScope()方法。比如在你的TransactionController中：

@Component("transactionController")
public class TransactionController {
    @Autowired
    private RequestWrapper requestWrapper;
    @Autowired
    private TcpRequestScope tcpRequestScope; // 注入自定义Scope实例

    @ServiceActivator
    public String handle(final Message<?> requestMessage) {
        // 处理请求、修改RequestWrapper状态的逻辑...
        
        // 请求完成后立即销毁当前线程的作用域
        tcpRequestScope.destroyScope();
        return response;
    }
}

如果你的请求生命周期是通过事件监听处理的（比如tcpRequestCompletePublisher的事件），也可以把destroyScope()调用放在事件监听器中，确保在响应发送前执行。

3. 确保Autowired的作用域Bean正确代理 #

单例类（比如TransactionController）中Autowired的作用域Bean，需要通过Spring代理来保证每次请求获取到当前作用域的实例。请给RequestWrapper添加正确的Scope注解：

@Component
@Scope(value = "tcpRequestScope", proxyMode = ScopedProxyMode.TARGET_CLASS)
public class RequestWrapper {
    // ...
}

proxyMode = ScopedProxyMode.TARGET_CLASS会让Spring生成CGLIB代理，每次调用代理方法时都会从当前线程的作用域中获取实例，而非缓存单例。

4. 验证销毁逻辑 #

可以给RequestWrapper添加销毁钩子，验证销毁是否生效：

@Component
@Scope(value = "tcpRequestScope", proxyMode = ScopedProxyMode.TARGET_CLASS)
public class RequestWrapper implements DisposableBean {
    @Override
    public void destroy() throws Exception {
        System.out.println("RequestWrapper已销毁，当前线程：" + Thread.currentThread().getName());
    }
}

请求处理完成后，控制台应该会输出上述日志，说明销毁回调被正确触发。

关键原理说明 #

• Spring的自定义Scope不会自动管理生命周期：内置的request/session作用域由Web容器触发销毁，而自定义的TcpRequestScope需要你在请求结束时主动触发销毁逻辑。
• registerDestructionCallback注册的回调是Spring执行Bean销毁逻辑的入口，必须遍历执行才能触发@PreDestroy或DisposableBean.destroy()方法。
• ThreadLocal的清理必须配合销毁回调，否则不仅Bean状态会被复用，销毁逻辑也不会执行。

内容的提问来源于stack exchange，提问作者Andrew Cotton