Spring Boot集成H2數據庫

作者：Grey

原文地址： Spring Boot集成H2數據庫

需求

平時學習的時候，涉及到一些連接數據庫相關的操作，經常需要初始化本地數據庫，比如裝個MySQL，初始化一些腳本，比較麻煩，H2是內存數據庫，Spring Boot可以在應用啟動的時候對H2數據庫初始化一些SQL腳本，這樣的話，在學習/測試階段，可以先使用H2數據庫進行測試和學習，不需要額外安裝MySQL數據庫了。

注意

因為SQL腳本是可以在Spring Boot啟動的時候初始化的，所以，假如你的SQL腳本中有一些刪表刪數據的操作，所以這種操作方式不適合應用在生產數據庫。

環境

• JDK 1.8
• Maven 3.6.1
• Spring Boot 2.3.0.RELEASE

依賴

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">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.3.0.RELEASE</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
    <groupId>org.hui</groupId>
    <artifactId>spring-boot-h2</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>spring-boot-h2</name>
    <description>Spring Boot for H2</description>
    <properties>
        <java.version>1.8</java.version>
    </properties>
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-jdbc</artifactId>
        </dependency>
        <dependency>
            <groupId>com.h2database</groupId>
            <artifactId>h2</artifactId>
            <scope>runtime</scope>
        </dependency>
    </dependencies>
    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>
</project>

配置文件

application.properties

spring.datasource.url=jdbc:h2:mem:testdb2
spring.datasource.driverClassName=org.h2.Driver
spring.datasource.username=sa
spring.datasource.password=


#需要初始化的SQL腳本，指定位置為resources/db目錄下，如果不指定，默認就在resources目錄下
spring.datasource.schema=classpath:db/schema.sql
spring.datasource.data=classpath:db/data.sql

# H2控制台啟用
spring.h2.console.enabled=true

# H2訪問的URL
spring.h2.console.path=/h2

SQL腳本

由於配置文件中指定了sql文件的位置，所以以下兩個SQL文件放在resources/db目錄下，你也可以指定其他目錄，見配置文件的注釋信息。

schema.sql

DROP TABLE IF EXISTS USER_INFO;
 
CREATE TABLE USER_INFO (
  id INT AUTO_INCREMENT  PRIMARY KEY,
  user_name VARCHAR(250) NOT NULL,
  email VARCHAR(250) DEFAULT NULL
);

data.sql

INSERT INTO 
	USER_INFO (user_name, email)
VALUES
  	('grey', '[e-mail]'),
  	('jack', '[e-mail]');

測試類

package org.hui.springbooth2;

import org.springframework.boot.CommandLineRunner;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.jdbc.core.JdbcTemplate;

/**
 * @author zenghui
 * @date 2020-5-20
 */
@SpringBootApplication
public class H2Application implements CommandLineRunner {
    public static void main(String[] args) {
        SpringApplication.run(H2Application.class, args);
    }

    private final JdbcTemplate template;

    public H2Application(JdbcTemplate template) {
        this.template = template;
    }
    @Override
    public void run(String... args)  {
        String sql="SELECT count(*)  FROM USER_INFO";
        int count = template.queryForObject(sql, Integer.class);
        System.out.println("user count is " + count);
    }
}

運行

運行H2Application.java這個主類，控制台打印如下信息：

user count is 2

代碼

Github

Gitee