快捷導(dǎo)航

Spring?Boot中自動(dòng)執(zhí)行sql腳本的方法實(shí)例

更新時(shí)間：2022年01月10日 10:38:46 作者：大哥你先走

在SpringBoot的架構(gòu)中,DataSourceInitializer類(lèi)可以在項(xiàng)目啟動(dòng)后初始化數(shù)據(jù),我們可以通過(guò)自動(dòng)執(zhí)行自定義sql腳本初始化數(shù)據(jù),下面這篇文章主要給大家介紹了關(guān)于Spring?Boot中自動(dòng)執(zhí)行sql腳本的相關(guān)資料,需要的朋友可以參考下

背景

在應(yīng)用程序啟動(dòng)后，可以自動(dòng)執(zhí)行建庫(kù)、建表等SQL腳本.下文中以要自動(dòng)化執(zhí)行people.sql腳本為例說(shuō)明，腳本在SpringBoot工程中的路徑為：classpath:people.sql，腳本的具體內(nèi)容如下：

CREATE TABLE IF NOT EXISTS people(
    persion_id BIGINT NOT NULL AUTO_INCREMENT,
    first_name VARCHAR(20),
    last_name VARCHAR(20),
    PRIMARY KEY (persion_id)
);

實(shí)現(xiàn)核心

在SpringBoot的架構(gòu)中，DataSourceInitializer類(lèi)可以實(shí)現(xiàn)自動(dòng)執(zhí)行腳本的功能。通過(guò)自定義DataSourceInitializer Bean就可以實(shí)現(xiàn)按照業(yè)務(wù)要求執(zhí)行特定的腳本。

實(shí)現(xiàn)方法

前提
已經(jīng)構(gòu)建了DataSource Bean。
方法
通過(guò)@Configuration、@Bean和@Value三個(gè)注解實(shí)現(xiàn)自定義DataSourceInitializer Bean，現(xiàn)在Bean的定義中實(shí)現(xiàn)自動(dòng)化執(zhí)行腳本的業(yè)務(wù)邏輯。
代碼

import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.Resource;
import org.springframework.jdbc.datasource.init.DataSourceInitializer;
import org.springframework.jdbc.datasource.init.DatabasePopulator;
import org.springframework.jdbc.datasource.init.ResourceDatabasePopulator;
import javax.sql.DataSource;
@Configuration
public class BatchTaskDataSourceInitializer {
    /**
     * 構(gòu)建Resource對(duì)象
     */
    @Value("classpath:schema-all.sql")
    private Resource businessScript;

    /**
     * 自定義Bean實(shí)現(xiàn)業(yè)務(wù)的特殊需求
     * @param dataSource
     * @return
     */
    @Bean
    public DataSourceInitializer dataSourceInitializer(final DataSource dataSource) {
        final DataSourceInitializer initializer = new DataSourceInitializer();
        // 設(shè)置數(shù)據(jù)源
        initializer.setDataSource(dataSource);
        initializer.setDatabasePopulator(databasePopulator());
        return initializer;
    }

    private DatabasePopulator databasePopulator() {
        final ResourceDatabasePopulator populator = new ResourceDatabasePopulator();
        populator.addScripts(businessScript);
        return populator;
    }
}

注意

有兩點(diǎn)需要說(shuō)明下：

db文件夾下的腳本可能有多個(gè)，所以，這里要使用PathMatchingResourcePatternResolver，讀取多個(gè).sql的文件；
獲取文件的路徑是classpath*:script/db/*.sql，classpath后加個(gè)星號(hào)，意思是連jar包中的符合該規(guī)則的文件，都可以獲取到，因?yàn)樵擁?xiàng)目最終會(huì)打成jar包來(lái)運(yùn)行，如果不使用這個(gè)星號(hào)，就會(huì)出現(xiàn)在開(kāi)發(fā)的時(shí)，沒(méi)有問(wèn)題，等到了測(cè)試或生產(chǎn)環(huán)境，就會(huì)出現(xiàn)找不到sql文件問(wèn)題。當(dāng)然，這里的文件規(guī)則，需要自己定義好，不要掃到了本不應(yīng)加載的其他包中的文件了。

總結(jié)

到此這篇關(guān)于Spring Boot中自動(dòng)執(zhí)行sql腳本的文章就介紹到這了,更多相關(guān)Spring Boot自動(dòng)執(zhí)行sql腳本內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: