sql腳本怎么在Spring Boot中執行

這篇文章將為大家詳細講解有關sql腳本怎么在Spring Boot中執行，文章內容質量較高，因此小編分享給大家做個參考，希望大家閱讀完這篇文章后對相關知識有一定的了解。

在應用程序啟動后，可以自動執行建庫、建表等SQL腳本.下文中以要自動化執行people.sql腳本為例說明，腳本在SpringBoot工程中的路徑為：classpath:people.sql，腳本的具體內容如下：

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)
);

實現核心

在SpringBoot的架構中，DataSourceInitializer類可以實現自動執行腳本的功能。通過自定義DataSourceInitializer Bean就可以實現按照業務要求執行特定的腳本。

實現方法

前提

已經構建了DataSource Bean。

方法

通過@Configuration、@Bean和@Value三個注解實現自定義DataSourceInitializer Bean，現在Bean的定義中實現自動化執行腳本的業務邏輯。

代碼

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 {
  /**
   * 構建Resource對象
   */
  @Value("classpath:schema-all.sql")
  private Resource businessScript;

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

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

關于sql腳本怎么在Spring Boot中執行就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。