springboot 啟動(dòng)時(shí)初始化數(shù)據(jù)庫的步驟

更新時(shí)間：2021年01月06日 08:40:24 作者：鯨冬香

這篇文章主要介紹了springboot 啟動(dòng)時(shí)初始化數(shù)據(jù)庫的步驟，幫助大家更好的理解和使用springboot框架，感興趣的朋友可以了解下

問題描述

在spring-boot啟動(dòng)時(shí)，希望能執(zhí)行相應(yīng)的sql文件來初始化數(shù)據(jù)庫。

使用配置文件初始化數(shù)據(jù)庫

可以在spring-boot的配置文件application.yml中設(shè)置要初始化的sql文件。這是最簡(jiǎn)單的方法，只需要添加屬性就可以實(shí)現(xiàn)。

首先設(shè)置spring.datasource.initialization-mode=always表示任何類型數(shù)據(jù)庫都進(jìn)行數(shù)據(jù)庫初始化,默認(rèn)情況下，spring-boot會(huì)自動(dòng)加載data.sql或data-${platform}.sql文件來初始化數(shù)據(jù)庫?？梢酝ㄟ^設(shè)置不同的數(shù)據(jù)庫平臺(tái)來改變啟動(dòng)的腳本名稱。

例如設(shè)置spring.datasource.platform=mysql,就會(huì)加載data-mysql.sql的數(shù)據(jù)庫腳本。把數(shù)據(jù)庫腳本文件放在resources路徑下即可。

如果項(xiàng)目使用的是flyway管理數(shù)據(jù)庫的話，可以直接在flyway路徑下添加一個(gè)新版本的sql文件，flyway也會(huì)自動(dòng)執(zhí)行sql文件并記錄版本信息。

通過代碼初始化數(shù)據(jù)庫

如果通過配置文件不能滿足需求，可以通過代碼來初始化數(shù)據(jù)庫。
只需要提供DataSourceInitializer這個(gè)bean，spring-boot啟動(dòng)時(shí)就會(huì)根據(jù)DataSourceInitializer來初始化數(shù)據(jù)庫了。

@Bean
public DataSourceInitializer dataSourceInitializer(final DataSource dataSource) {
  ResourceDatabasePopulator resourceDatabasePopulator = new ResourceDatabasePopulator();
  resourceDatabasePopulator.addScript(new ClassPathResource("/data.sql"));
  DataSourceInitializer dataSourceInitializer = new DataSourceInitializer();
  dataSourceInitializer.setDataSource(dataSource);
  dataSourceInitializer.setDatabasePopulator(resourceDatabasePopulator);
  return dataSourceInitializer;
}

在此基礎(chǔ)上，我們可以自定義注解，通過獲取注解上的sql文件路徑，來達(dá)到通過注解初始化數(shù)據(jù)庫目的，這樣更方便簡(jiǎn)潔。
首先定義注解InitDataSource:

/** 
 * 用于補(bǔ)充：Hibernate無法自動(dòng)創(chuàng)建視圖的缺陷。 
 * 系統(tǒng)啟動(dòng)時(shí)(hibernate根據(jù)entity創(chuàng)建完基本的數(shù)據(jù)表后)，開始執(zhí)行本注解下的sql文件中的SQL語言。 
 * 使用方法： 
 * @InitDataSource("sql文件路徑(相對(duì)于resources路徑下)") ---- 注解到對(duì)應(yīng)的類上 
 * 比如：@InitDataSource("db/view/createView.sql)") 
 * 使用示例請(qǐng)參見：ResourceApplication.java 
 * 預(yù)了解詳細(xì)的實(shí)現(xiàn)過程請(qǐng)參考：WebConfig.java 的 dataSourceInitializer方法 
 * @author huangtingxiang 
 */
@Target({ElementType.TYPE})   // 該注解用于類上 
@Retention(RetentionPolicy.RUNTIME) // 在運(yùn)行時(shí)起作用 
@Component 
public @interface InitDataSource { 
  String\[\] value(); 
}

然后通過ClassPathScanningCandidateComponentProvider這個(gè)類來掃描spring組件上InitDataSource注解的值，將值取出，添加到DataSourceInitializer的初始化腳本中:

@Bean 
public DataSourceInitializer dataSourceInitializer(final DataSource dataSource) { 
  ResourceDatabasePopulator resourceDatabasePopulator = new ResourceDatabasePopulator(); 
  // 掃描com.mengyunzhi.measurement 包 找到InitDataSource注解的類(注解需使用到實(shí)現(xiàn)類上) 
 ClassPathScanningCandidateComponentProvider provider 
      = new ClassPathScanningCandidateComponentProvider(false); 
  provider.addIncludeFilter(new AnnotationTypeFilter(InitDataSource.class)); //添加包含的過濾信息 
 for (BeanDefinition beanDef : provider.findCandidateComponents("com.mengyunzhi.measurement")) { 
    Class<?> cl = null; 
    try { 
      cl = Class.forName(beanDef.getBeanClassName()); 
      InitDataSource initDataSource = cl.getAnnotation(InitDataSource.class); 
      String\[\] sqlFiles = initDataSource.value(); 
      for (String sql: sqlFiles) { 
        // 如果sql文件存在 加入數(shù)據(jù)庫初始化中 否則拋出異常終止執(zhí)行 
 ClassPathResource resource = new ClassPathResource("/" + sql); 
        if (resource.exists()) { 
          resourceDatabasePopulator.addScript(resource); 
        } else { 
          throw new DataSourceInitializerException("未找到資源文件:" + sql, cl); 
        } 
      } 
    } catch (ClassNotFoundException e) { 
      e.printStackTrace(); 
    } 
  } 
  DataSourceInitializer dataSourceInitializer = new DataSourceInitializer(); 
  dataSourceInitializer.setDataSource(dataSource); 
  dataSourceInitializer.setDatabasePopulator(resourceDatabasePopulator); 
  return dataSourceInitializer; 
}

這樣一來，只需要在spring-boot類上使用@InitDataSource({"data.sql"})注解，就可以自動(dòng)進(jìn)行數(shù)據(jù)庫的初始化操作了。

以上就是springboot 啟動(dòng)時(shí)初始化數(shù)據(jù)庫的步驟的詳細(xì)內(nèi)容，更多關(guān)于springboot 初始化數(shù)據(jù)庫的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: