怎么在SpringBoot中加载初始化数据

本篇文章给大家分享的是有关怎么在Spring Boot中加载初始化数据,小编觉得挺实用的,因此分享给大家学习,希望大家阅读完这篇文章后可以有所收获,话不多说,跟着小编一起来看看吧。

创新互联公司专注于顺平网站建设服务及定制,我们拥有丰富的企业做网站经验。 热诚为您提供顺平营销型网站建设,顺平网站制作、顺平网页设计、顺平网站官网定制、小程序开发服务,打造顺平网络公司原创品牌,更为您提供顺平网站排名全网营销落地服务。

依赖条件

Spring Boot的依赖我们就不将了,因为本例将会有数据库的操作,我们这里使用H2内存数据库方便测试:


      com.h3database
      h3
      runtime
    

我们需要使用JPA来创建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);
}

如果这时候我们启动Spring Boot程序,将会自动创建Country表。

data.sql文件

上面我们创建好了数据表格,我们可以使用data.sql来加载文件:

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条数据,可以写个测试例子测试一下:

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

  @Autowired
  private CountryRepository countryRepository;

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

schema.sql 文件

有时候我们需要自定义数据库的schema,这时候我们可以使用到schema.sql文件。

先看一个schema.sql的例子:

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

Spring Boot会自动加载这个schema文件。

我们需要关闭spring boot的schema自动创建功能以防冲突:

spring.jpa.hibernate.ddl-auto=none

spring.jpa.hibernate.ddl-auto 有如下几个选项:

  • create : 首先drop现有的tables,然后创建新的tables

  • update : 这个模式不会删除现有的tables,它会比较现有的tables和新的注解或者xml配置是否一致,然后更新。

  • create-drop : 和create很类似,不同的是会在程序运行完毕后自动drop掉tables。通常用在单元测试中。

  • validate : 只会做table是否存在的验证,不存在则会报错。

  • none : 关闭ddl自动生成功能。

如果Spring Boot没有检测到自定义的schema manager的话,则会自动使用create-drop模式。否则使用none模式。

@sql注解

@Sql 是测试包中的一个注解,可以显示的导入要执行的sql文件,它可以用在class上或者方法之上,如下所示:

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

上面的例子将会显示的导入classpath中的new_country.sql文件。

@Sql有如下几个属性:

  • config : 用来配置SQL脚本,我们在下面的@SqlConfig详细讲解。

  • executionPhase : 可以选择脚本是在BEFORE_TEST_METHOD 或者 AFTER_TEST_METHOD来执行。

  • statements: 可以接受内联的sql语句

  • scripts: 可以指明要执行脚本的路径

@SqlConfig 注解

@SqlConfig主要用在class级别或者用在@Sql注解的config属性中:

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

@SqlConfig有如下几个属性:

  • blockCommentStartDelimiter: 指明了SQL脚本的开始标记。

  • blockCommentEndDelimiter: SQL脚本的结束标记。

  • commentPrefix: SQL 脚本的注释标记

  • dataSource : javax.sql.DataSource的名字,指定该脚本将会在什么datasource下执行

  • encoding: SQL 文件的编码

  • errorMode: 脚本遇到错误的处理模式

  • separator: 分隔符

  • transactionManager: 指定的PlatformTransactionManager

  • transactionMode: 事务模式

@Sql是可以多个同时使用的,如下所示:

  @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中加载初始化数据,小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注创新互联行业资讯频道。


本文名称:怎么在SpringBoot中加载初始化数据
本文链接:http://scyanting.com/article/jdjghj.html