作者：京東科技 李君

書接上文，前面在 Spring 應(yīng)用合并之路（一）：摸石頭過河 介紹了幾種不成功的經(jīng)驗，下面繼續(xù)折騰…

四、倉庫合并，獨立容器

在經(jīng)歷了上面的嘗試，在同事為啥不搞兩個獨立的容器提醒下，決定拋開 Spring Boot 內(nèi)置的父子容器方案，完全自己實現(xiàn)父子容器。

如何加載 web 項目？

現(xiàn)在的難題只有一個：如何加載 web 項目？加載完成后，如何持續(xù)持有 web 項目？經(jīng)過思考后，可以創(chuàng)建一個 boot 項目的 Spring Bean，在該 Bean 中加載并持有 web 項目的容器。由于 Spring Bean 默認(rèn)是單例的，并且會伴隨 Spring 容器長期存活，就可以保證 web 容器持久存活。結(jié)合 Spring 擴(kuò)展點概覽及實踐 中介紹的 Spring 擴(kuò)展點，有兩個地方可以利用：

1.可以利用 ApplicationContextAware 獲取 boot 容器的 ApplicationContext 實例，這樣就可以實現(xiàn)自己實現(xiàn)的父子容器；

2.可以利用 ApplicationListener 獲取 ContextRefreshedEvent 事件，該事件表示容器已經(jīng)完成初始化，可以提供服務(wù)。在監(jiān)聽到該事件后，來進(jìn)行 web 容器的加載。

思路確定后，代碼實現(xiàn)就很簡單了：

package com.diguage.demo.boot.config;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ApplicationEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import org.springframework.stereotype.Component;

/**
 * @author D瓜哥 · https://www.diguage.com
 */
@Component
public class WebLoaderListener implements ApplicationContextAware,
        ApplicationListener {
    private static final Logger logger = LoggerFactory.getLogger(WebLoaderListener.class);

    /**
     * 父容器，加載 boot 項目
     */
    private static ApplicationContext parentContext;

    /**
     * 子容器，加載 web 項目
     */
    private static ApplicationContext childContext;

    @Override
    public void setApplicationContext(ApplicationContext ctx) throws BeansException {
        WebLoaderListener.parentContext = ctx;
    }

    @Override
    public void onApplicationEvent(ApplicationEvent event) {
        logger.info("receive application event: {}", event);
        if (event instanceof ContextRefreshedEvent) {
            WebLoaderListener.childContext = new ClassPathXmlApplicationContext(
                    new String[]{"classpath:web/spring-cfg.xml"},
                    WebLoaderListener.parentContext);
        }
    }
}

容器重復(fù)加載的問題

這次自己實現(xiàn)的父子容器，如同設(shè)想的那樣，沒有同名 Bean 的檢查，省去了很多麻煩。但是，觀察日志，會發(fā)現(xiàn) com.diguage.demo.boot.config.WebLoaderListener#onApplicationEvent 方法被兩次執(zhí)行，也就是監(jiān)聽到了兩次 ContextRefreshedEvent 事件，導(dǎo)致 web 容器會被加載兩次。由于項目的 RPC 服務(wù)不能重復(fù)注冊，第二次加載拋出異常，導(dǎo)致啟動失敗。

最初，懷疑是 web 容器，加載了 WebLoaderListener，但是跟蹤代碼，沒有發(fā)現(xiàn) childContext 容器中有 WebLoaderListener 的相關(guān) Bean。

昨天做了個小實驗，又調(diào)試了一下 Spring 的源代碼，發(fā)現(xiàn)了其中的奧秘。直接貼代碼吧：

SPRING/spring-context/src/main/java/org/springframework/context/support/AbstractApplicationContext.java

/**
 * Publish the given event to all listeners.
 * This is the internal delegate that all other {@code publishEvent}
 * methods refer to. It is not meant to be called directly but rather serves
 * as a propagation mechanism between application contexts in a hierarchy,
 * potentially overridden in subclasses for a custom propagation arrangement.
 * @param event the event to publish (may be an {@link ApplicationEvent}
 * or a payload object to be turned into a {@link PayloadApplicationEvent})
 * @param typeHint the resolved event type, if known.
 * The implementation of this method also tolerates a payload type hint for
 * a payload object to be turned into a {@link PayloadApplicationEvent}.
 * However, the recommended way is to construct an actual event object via
 * {@link PayloadApplicationEvent#PayloadApplicationEvent(Object, Object, ResolvableType)}
 * instead for such scenarios.
 * @since 4.2
 * @see ApplicationEventMulticaster#multicastEvent(ApplicationEvent, ResolvableType)
 */
protected void publishEvent(Object event, @Nullable ResolvableType typeHint) {
    Assert.notNull(event, "Event must not be null");
    ResolvableType eventType = null;

    // Decorate event as an ApplicationEvent if necessary
    ApplicationEvent applicationEvent;
    if (event instanceof ApplicationEvent applEvent) {
        applicationEvent = applEvent;
        eventType = typeHint;
    }
    else {
        ResolvableType payloadType = null;
        if (typeHint != null && ApplicationEvent.class.isAssignableFrom(typeHint.toClass())) {
            eventType = typeHint;
        }
        else {
            payloadType = typeHint;
        }
        applicationEvent = new PayloadApplicationEvent(this, event, payloadType);
    }

    // Determine event type only once (for multicast and parent publish)
    if (eventType == null) {
        eventType = ResolvableType.forInstance(applicationEvent);
        if (typeHint == null) {
            typeHint = eventType;
        }
    }

    // Multicast right now if possible - or lazily once the multicaster is initialized
    if (this.earlyApplicationEvents != null) {
        this.earlyApplicationEvents.add(applicationEvent);
    }
    else if (this.applicationEventMulticaster != null) {
        this.applicationEventMulticaster.multicastEvent(applicationEvent, eventType);
    }

    // Publish event via parent context as well...
    // 如果有父容器，則也將事件發(fā)布給父容器。
    if (this.parent != null) {
        if (this.parent instanceof AbstractApplicationContext abstractApplicationContext) {
            abstractApplicationContext.publishEvent(event, typeHint);
        }
        else {
            this.parent.publishEvent(event);
        }
    }
}

在 publishEvent 方法的最后，如果父容器不為 null 的情況下，則也會向父容器廣播容器的相關(guān)事件。

看到這里就清楚了，不是 web 容器持有了 WebLoaderListener 這個 Bean，而是 web 容器主動向父容器廣播了 ContextRefreshedEvent 事件。

容器銷毀

除了上述問題，還有一個問題需要思考：如何銷毀 web 容器？如果不能銷毀容器，會有一些意想不到的問題。比如，注冊中心的 RPC 提供方不能及時銷毀等等。

這里的解決方案也比較簡單：同樣基于事件監(jiān)聽，Spring 容器銷毀會有 ContextClosedEvent 事件，在 WebLoaderListener 中監(jiān)聽該事件，然后調(diào)用 AbstractApplicationContext#close 方法就可以完成 Spring 容器的銷毀工作。

父子容器加載及銷毀

結(jié)合上面的所有論述，完整的代碼如下：

package com.diguage.demo.boot.config;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ApplicationEvent;
import org.springframework.context.ApplicationListener;
import org.springframework.context.event.ContextClosedEvent;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.context.support.AbstractApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import org.springframework.stereotype.Component;

import java.util.Objects;

/**
 * 基于事件監(jiān)聽的 web 項目加載器
 *
 * @author D瓜哥 · https://www.diguage.com
 */
@Component
public class WebLoaderListener implements ApplicationContextAware,
        ApplicationListener {
    private static final Logger logger = LoggerFactory.getLogger(WebLoaderListener.class);

    /**
     * 父容器，加載 boot 項目
     */
    private static ApplicationContext parentContext;

    /**
     * 子容器，加載 web 項目
     */
    private static ClassPathXmlApplicationContext childContext;

    @Override
    public void setApplicationContext(ApplicationContext ctx) throws BeansException {
        WebLoaderListener.parentContext = ctx;
    }

    /**
     * 事件監(jiān)聽
     *
     * @author D瓜哥 · https://www.diguage.com
     */
    @Override
    public void onApplicationEvent(ApplicationEvent event) {
        logger.info("receive application event: {}", event);
        if (event instanceof ContextRefreshedEvent refreshedEvent) {
            ApplicationContext context = refreshedEvent.getApplicationContext();
            if (Objects.equals(WebLoaderListener.parentContext, context)) {
                // 加載 web 容器
                WebLoaderListener.childContext = new ClassPathXmlApplicationContext(
                        new String[]{"classpath:web/spring-cfg.xml"},
                        WebLoaderListener.parentContext);
            }
        } else if (event instanceof ContextClosedEvent) {
            // 處理容器銷毀事件
            if (Objects.nonNull(WebLoaderListener.childContext)) {
                synchronized (WebLoaderListener.class) {
                    if (Objects.nonNull(WebLoaderListener.childContext)) {
                        AbstractApplicationContext ctx = WebLoaderListener.childContext;
                        WebLoaderListener.childContext = null;
                        ctx.close();
                    }
                }
            }
        }
    }
}

五、參考資料

1.?Spring 擴(kuò)展點概覽及實踐 - "地瓜哥"博客網(wǎng)?

2.?Context Hierarchy with the Spring Boot Fluent Builder API?

3.?How to revert initial git commit?

審核編輯 黃宇