本篇文章給大家分享的是有關(guān)怎么在Spring Boot中加載初始化數(shù)據(jù)，小編覺得挺實(shí)用的，因此分享給大家學(xué)習(xí)，希望大家閱讀完這篇文章后可以有所收獲，話不多說(shuō)，跟著小編一起來(lái)看看吧。

創(chuàng)新互聯(lián)公司專注于順平網(wǎng)站建設(shè)服務(wù)及定制，我們擁有豐富的企業(yè)做網(wǎng)站經(jīng)驗(yàn)。 熱誠(chéng)為您提供順平營(yíng)銷型網(wǎng)站建設(shè)，順平網(wǎng)站制作、順平網(wǎng)頁(yè)設(shè)計(jì)、順平網(wǎng)站官網(wǎng)定制、小程序開發(fā)服務(wù)，打造順平網(wǎng)絡(luò)公司原創(chuàng)品牌,更為您提供順平網(wǎng)站排名全網(wǎng)營(yíng)銷落地服務(wù)。

依賴條件

Spring Boot的依賴我們就不將了，因?yàn)楸纠龑?huì)有數(shù)據(jù)庫(kù)的操作，我們這里使用H2內(nèi)存數(shù)據(jù)庫(kù)方便測(cè)試：

      com.h3database
      h3
      runtime
    

我們需要使用JPA來(lái)創(chuàng)建Entity：

@Entity
public class Country {
 
  @Id
  @GeneratedValue(strategy = IDENTITY)
  private Integer id;
   
  @Column(nullable = false)
  private String name;
 
  //...
}

我們這樣定義repository：

public interface CountryRepository extends CrudRepository {
  List findByName(String name);
}

如果這時(shí)候我們啟動(dòng)Spring Boot程序，將會(huì)自動(dòng)創(chuàng)建Country表。

data.sql文件

上面我們創(chuàng)建好了數(shù)據(jù)表格，我們可以使用data.sql來(lái)加載文件：

INSERT INTO country (name) VALUES ('India');
INSERT INTO country (name) VALUES ('Brazil');
INSERT INTO country (name) VALUES ('USA');
INSERT INTO country (name) VALUES ('Italy');

在data.sql文件中我們插入了4條數(shù)據(jù)，可以寫個(gè)測(cè)試?yán)訙y(cè)試一下：

@RunWith(SpringRunner.class)
@SpringBootTest(classes = LoadIniDataApp.class)
public class SpringBootInitialLoadIntegrationTest {

  @Autowired
  private CountryRepository countryRepository;

  @Test
  public void testInitDataForTestClass() {
    assertEquals(4, countryRepository.count());
  }
}

schema.sql 文件

有時(shí)候我們需要自定義數(shù)據(jù)庫(kù)的schema，這時(shí)候我們可以使用到schema.sql文件。

先看一個(gè)schema.sql的例子：

CREATE TABLE country (
  id  INTEGER   NOT NULL AUTO_INCREMENT,
  name VARCHAR(128) NOT NULL,
  PRIMARY KEY (id)
);

Spring Boot會(huì)自動(dòng)加載這個(gè)schema文件。

我們需要關(guān)閉spring boot的schema自動(dòng)創(chuàng)建功能以防沖突：

spring.jpa.hibernate.ddl-auto=none

spring.jpa.hibernate.ddl-auto 有如下幾個(gè)選項(xiàng)：

• create : 首先drop現(xiàn)有的tables，然后創(chuàng)建新的tables

• update : 這個(gè)模式不會(huì)刪除現(xiàn)有的tables，它會(huì)比較現(xiàn)有的tables和新的注解或者xml配置是否一致，然后更新。

• create-drop : 和create很類似，不同的是會(huì)在程序運(yùn)行完畢后自動(dòng)drop掉tables。通常用在單元測(cè)試中。

• validate : 只會(huì)做table是否存在的驗(yàn)證，不存在則會(huì)報(bào)錯(cuò)。

• none : 關(guān)閉ddl自動(dòng)生成功能。

如果Spring Boot沒有檢測(cè)到自定義的schema manager的話，則會(huì)自動(dòng)使用create-drop模式。否則使用none模式。

@sql注解

@Sql 是測(cè)試包中的一個(gè)注解，可以顯示的導(dǎo)入要執(zhí)行的sql文件，它可以用在class上或者方法之上，如下所示：

  @Test
  @Sql({"classpath:new_country.sql"})
  public void testLoadDataForTestCase() {
    assertEquals(6, countryRepository.count());
  }

上面的例子將會(huì)顯示的導(dǎo)入classpath中的new_country.sql文件。

@Sql有如下幾個(gè)屬性：

• config : 用來(lái)配置SQL腳本，我們?cè)谙旅娴腀SqlConfig詳細(xì)講解。

• executionPhase : 可以選擇腳本是在BEFORE_TEST_METHOD 或者 AFTER_TEST_METHOD來(lái)執(zhí)行。

• statements: 可以接受內(nèi)聯(lián)的sql語(yǔ)句

• scripts: 可以指明要執(zhí)行腳本的路徑

@SqlConfig 注解

@SqlConfig主要用在class級(jí)別或者用在@Sql注解的config屬性中：

  @Sql(scripts = {"classpath:new_country2.sql"},
      config = @SqlConfig(encoding = "utf-8", transactionMode = SqlConfig.TransactionMode.ISOLATED))

@SqlConfig有如下幾個(gè)屬性：

• blockCommentStartDelimiter: 指明了SQL腳本的開始標(biāo)記。

• blockCommentEndDelimiter: SQL腳本的結(jié)束標(biāo)記。

• commentPrefix: SQL 腳本的注釋標(biāo)記

• dataSource : javax.sql.DataSource的名字，指定該腳本將會(huì)在什么datasource下執(zhí)行

• encoding: SQL 文件的編碼

• errorMode: 腳本遇到錯(cuò)誤的處理模式

• separator: 分隔符

• transactionManager: 指定的PlatformTransactionManager

• transactionMode: 事務(wù)模式

@Sql是可以多個(gè)同時(shí)使用的，如下所示：

  @Test
  @Sql({"classpath:new_country.sql"})
  @Sql(scripts = {"classpath:new_country2.sql"},
      config = @SqlConfig(encoding = "utf-8", transactionMode = SqlConfig.TransactionMode.ISOLATED))
  public void testLoadDataForTestCase() {
    assertEquals(6, countryRepository.count());
  }

以上就是怎么在Spring Boot中加載初始化數(shù)據(jù)，小編相信有部分知識(shí)點(diǎn)可能是我們?nèi)粘９ぷ鲿?huì)見到或用到的。希望你能通過這篇文章學(xué)到更多知識(shí)。更多詳情敬請(qǐng)關(guān)注創(chuàng)新互聯(lián)行業(yè)資訊頻道。