脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 软件编程 → java → SpringBoot 加载初始化数据

在Spring Boot中加载初始化数据的实现

 更新时间：2020年02月13日 09:44:39   作者：flydean  
这篇文章主要介绍了在Spring Boot中加载初始化数据的实现,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

在Spring Boot中，Spring Boot会自动搜索映射的Entity，并且创建相应的table，但是有时候我们希望自定义某些内容，这时候我们就需要使用到data.sql和schema.sql。

依赖条件

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

    <dependency>
      <groupId>com.h2database</groupId>
      <artifactId>h2</artifactId>
      <scope>runtime</scope>
    </dependency>

我们需要使用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<Country, Integer> {
  List<Country> 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());
  }

本文的例子可以参考 : https://github.com/ddean2009/learn-springboot2/tree/master/springboot-load-ini-data

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • 新版SpringSecurity5.x使用与配置详解

    新版SpringSecurity5.x使用与配置详解

    Spring Security是一个强大且高度可定制的身份验证和访问控制框架,本文主要介绍了新版SpringSecurity5.x使用与配置详解,具有一定的参考价值,感兴趣的可以了解一下
    2024-08-08
  • JAVA中AES加密方法实例分析

    JAVA中AES加密方法实例分析

    这篇文章主要介绍了JAVA中AES加密方法,实例分析了java中AES加密与解密的实现技巧,具有一定参考借鉴价值,需要的朋友可以参考下
    2015-07-07
  • SpringMVC源码解析之消息转换器HttpMessageConverter

    SpringMVC源码解析之消息转换器HttpMessageConverter

    本篇文章主要介绍了SpringMVC源码解析之消息转换器HttpMessageConverter ，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
    2017-11-11
  • Spring中的set注入方法

    Spring中的set注入方法

    这篇文章主要介绍了Spring中的set注入方法,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
    2023-12-12
  • Spring中@RefreshScope注解的处理方法详解

    Spring中@RefreshScope注解的处理方法详解

    这篇文章主要介绍了Spring中@RefreshScope注解的处理方法详解,spring启动时会调用ClassPathBeanDefinitionScanner.java类中的doScan()对包路径下的所有class进行扫描,获取bean的定义,同时对bean的@RefreshScope(@Scope的父类)进行处理,需要的朋友可以参考下
    2023-10-10
  • Spring Security实现禁止用户重复登陆的配置原理

    Spring Security实现禁止用户重复登陆的配置原理

    这篇文章主要介绍了Spring Security实现禁止用户重复登陆的配置原理,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-12-12
  • SpringBoot整合Druid实现SQL监控和数据库密码加密

    SpringBoot整合Druid实现SQL监控和数据库密码加密

    Druid连接池是阿里巴巴开源的数据库连接池项目,Druid连接池为监控而生,内置强大的监控功能,监控特性不影响性能,本文给大家介绍了SpringBoot整合Druid实现SQL监控和数据库密码加密,文中有相关的代码示例供大家参考,需要的朋友可以参考下
    2024-06-06
  • Spring MVC的项目准备和连接建立方法

    Spring MVC的项目准备和连接建立方法

    SpringWebMVC是基于Servlet API的Web框架,属于Spring框架的一部分,主要用于简化Web应用程序的开发,SpringMVC通过控制器接收请求,使用模型处理数据,并通过视图展示结果,感兴趣的朋友跟随小编一起看看吧
    2024-10-10
  • Java Collections.shuffle()方法案例详解

    Java Collections.shuffle()方法案例详解

    这篇文章主要介绍了Java Collections.shuffle()方法案例详解,本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下
    2021-08-08
  • Java流处理stream使用详解

    Java流处理stream使用详解

    Java8的另一大亮点Stream,它与java.io包里的InputStream和OutputStream是完全不同的概念,下面这篇文章主要给大家介绍了关于Java8中Stream详细使用方法的相关资料,需要的朋友可以参考下
    2022-10-10

最新评论