1. 程式人生 > 其它 >當spring 容器初始化完成後執行某個方法

當spring 容器初始化完成後執行某個方法

在做web專案開發中,尤其是企業級應用開發的時候,往往會在工程啟動的時候做許多的前置檢查。

比如檢查是否使用了我們組禁止使用的Mysql的group_concat函式,如果使用了專案就不能啟動,並指出哪個檔案的xml檔案使用了這個函式。

而在Spring的web專案中,我們可以介入Spring的啟動過程。我們希望在Spring容器將所有的Bean都初始化完成之後,做一些操作,這個時候我們就可以實現一個介面:

package com.yk.test.executor.processor
public class InstantiationTracingBeanPostProcessor implements ApplicationListener<ContextRefreshedEvent> {
@Override
public void onApplicationEvent(ContextRefreshedEvent event) {
//需要執行的邏輯程式碼,當spring容器初始化完成後就會執行該方法。
}
}

同時在Spring的配置檔案中,添加註入:

<!-- 當Spring容器啟動完成後執行下面的這個Bean -->
<bean class="com.yk.test.executor.processor.InstantiationTracingBeanPostProcessor"/>

但是這個時候,會存在一個問題,在web 專案中(spring mvc),系統會存在兩個容器,一個是root application context ,另一個就是我們自己的 projectName-servlet context(作為root application context的子容器)。

這種情況下,就會造成onApplicationEvent方法被執行兩次。為了避免上面提到的問題,我們可以只在root application context初始化完成後呼叫邏輯程式碼,其他的容器的初始化完成,則不做任何處理,修改後程式碼

如下:

@Override
public void onApplicationEvent(ContextRefreshedEvent event) {
if(event.getApplicationContext().getParent() == null){//root application context 沒有parent,他就是老大.
//需要執行的邏輯程式碼,當spring容器初始化完成後就會執行該方法。
}
}

其實更簡單的方法是使用註解:`@PostConstruct`,只需要在需要啟動的時候執行的方法上標註這個註解就搞定了。

例子:

    /**
     * 服務啟動時就執行--新增超級管理員
     */
    @PostConstruct
    public void addDefaultAdmin() {
        try {
            User user = new User();
            user.setCreateTime(new Date());
            try {
                user.setPassword(Md5.md5Encode("admin"));
            } catch (Exception e) {
                e.printStackTrace();
            }
            user.setUserName("admin");
            user.setRole(FrameConstant.USER_SUPER_ADMIN);
            userDao.save(user);
            log.debug("初始化完畢!");
        } catch (Exception e) {
            log.debug("初始化完畢!");
        }
    }