SpringBoot集成H2內存數據庫的方法

更新時間：2021年09月03日 15:27:07 作者：java干貨

H2是Thomas Mueller提供的一個開源的、純java實現的關系數據庫。本文主要介紹了SpringBoot集成H2內存數據庫，具有一定的參考價值，感興趣的可以了解一下

前言

本篇文章引導你使用Spring Boot，Spring Data JPA集成H2內存數據庫。更多關于H2數據參考：http://www.h2database.com/html/tutorial.html

準備

JDK 1.8 或更高版本
Maven 3 或更高版本

技術棧

Spring Data JPA
Spring Boot

目錄結構

pom.xml

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <parent>
        <artifactId>jpa-example</artifactId>
        <groupId>cn.merryyou</groupId>
        <version>1.0-SNAPSHOT</version>
    </parent>
    <modelVersion>4.0.0</modelVersion>
 
    <artifactId>h2-webconsole</artifactId>
 
    <properties>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
        <java.version>1.8</java.version>
    </properties>
 
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-jpa</artifactId>
        </dependency>
 
        <dependency>
            <groupId>com.h2database</groupId>
            <artifactId>h2</artifactId>
            <version>1.4.196</version>
        </dependency>
 
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
        </dependency>
 
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
    </dependencies>
 
    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
            <plugin>
                <groupId>org.apache.maven.plugins</groupId>
                <artifactId>maven-compiler-plugin</artifactId>
                <version>3.6.1</version>
                <configuration>
                    <source>1.8</source>
                    <target>1.8</target>
                </configuration>
            </plugin>
        </plugins>
    </build>
 
</project>

實體類 User

@Entity
@Table(name = "t_user")
@Data
public class User {
 
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
 
    private String name;
 
    private String url;
 
    @Override
    public String toString() {
        return "User{" +
                "id=" + id +
                ", name='" + name + '\'' +
                ", url='" + url + '\'' +
                '}';
    }
}

@Table聲明此對象映射到數據庫的數據表，通過它可以為實體指定表(talbe),目錄(Catalog)和schema的名字。該注釋不是必須的，如果沒有則系統使用默認值(實體的短類名)。
@Id 聲明此屬性為主鍵。該屬性值可以通過應該自身創(chuàng)建，但是Hibernate推薦通過Hibernate生成
@GeneratedValue 指定主鍵的生成策略。
- TABLE：使用表保存id值
- IDENTITY：identitycolumn
- SEQUENCR ：sequence
- AUTO：根據數據庫的不同使用上面三個

@Column 聲明該屬性與數據庫字段的映射關系。

AddressRepository

public interface UserRepository extends JpaRepository<User, Integer> {
}

Spring Data JPA包含了一些內置的Repository，實現了一些常用的方法：findone，findall，save等。

application.yml

spring:
  datasource:
    url: jdbc:h2:mem:h2test;DB_CLOSE_DELAY=-1;DB_CLOSE_ON_EXIT=FALSE
    platform: h2
    username: sa
    password:
    driverClassName: org.h2.Driver
  jpa:
    database-platform: org.hibernate.dialect.H2Dialect
    hibernate:
      ddl-auto: update
    properties:
      hibernate:
        show_sql: true
        use_sql_comments: true
        format_sql: true
  h2:
    console:
      enabled: true
      path: /console
      settings:
        trace: false
        web-allow-others: false
logging:
  level: debug

連接配置

在application.yml文件中對數據庫進行連接配置

spring.datasource.url=jdbc:h2:mem:h2test，配置h2數據庫的連接地址
spring.datasource.driver-class-name=org.h2.Driver，配置JDBC Driver
spring.datasource.username=sa，配置數據庫用戶名
spring.datasource.password=，配置數據庫密碼

當你完成依賴和連接配置這兩步之后，你就可以在程序種使用h2了。spring會自動幫你完成DataSource的注入。

數據初始化配置

如果你需要在程序啟動時對數據庫進行初始化操作，則在application.properties文件中對數據庫進接配置

spring.datasource.schema=classpath:db/schema.sql，進行該配置后，每次啟動程序，程序都會運行resources/db/schema.sql文件，對數據庫的結構進行操作。
spring.datasource.data=classpath:db/data.sql,進行該配置后，每次啟動程序，程序都會運行resources/db/data.sql文件，對數據庫的數據操作。

該配置非常適合開發(fā)環(huán)境，我會把數據庫的結構構建sql放在resources/db/schema.sql，數據sql放在resources/db/data.sql中。這樣每次運行程序我都可以得到一個新的數據庫。這樣就不需要我每次為了測試而修改數據中的內容了。

h2 web consloe配置

h2 web consloe是一個數據庫GUI管理應用，就和phpMyAdmin類似。程序運行時，會自動啟動h2 web consloe。當然你也可以進行如下的配置。

spring.h2.console.settings.web-allow-others=true，進行該配置后，h2 web consloe就可以在遠程訪問了。否則只能在本機訪問。
spring.h2.console.path=/h2-console，進行該配置，你就可以通過YOUR_URL/h2-console訪問h2 web consloe。YOUR_URL是你程序的訪問URl。
spring.h2.console.enabled=true，進行該配置，程序開啟時就會啟動h2 web consloe。當然這是默認的，如果你不想在啟動程序時啟動h2 web consloe，那么就設置為false。

UserRepositoryTest

@SpringBootTest
@RunWith(SpringRunner.class)
@Slf4j
public class UserRepositoryTest {
 
    @Autowired
    private UserRepository userRepository;
 
    @Test
    public void saveTest() throws Exception {
        User user = new User();
        user.setName("鄭龍飛");
        user.setUrl("http://merryyou.cn");
        User result = userRepository.save(user);
        log.info(result.toString());
        Assert.assertNotNull(user.getId());
    }
 
    @Test
    public void findOneTest() throws Exception{
        User user = userRepository.findOne(1l);
        log.info(user.toString());
        Assert.assertNotNull(user);
        Assert.assertTrue(1l==user.getId());
    }
}

h2 web consloe