SpringBoot集成H2數據庫的實現示例

更新時間：2024年07月01日 08:34:44 作者：可兒·四系桜

H2數據庫作為一個輕量級的內存數據庫,非常適合開發(fā)階段作為嵌入式數據庫進行單元測試和功能驗證,本文主要介紹了SpringBoot集成H2數據庫的實現示例,具有一定的參考的參考價值,感興趣的可以了解一下

1. 引言

Spring Boot 以其簡潔的配置和快速開發(fā)能力，成為現代微服務架構的首選框架之一。而H2數據庫作為一個輕量級的內存數據庫，非常適合開發(fā)階段作為嵌入式數據庫進行單元測試和功能驗證。本文將手把手教你如何在Spring Boot項目中集成H2數據庫，實現數據的快速存取與管理。

2. H2數據庫所需依賴

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

3. H2數據庫application.yaml配置文件

spring:
  datasource:
    url: jdbc:h2:~/user
    driver-class-name: org.h2.Driver
    username: root
    password: 123456
  h2:
    console:
      path: /h2-console  #h2嵌入式數據庫控制臺
      enabled: true

4. 初始化h2數據庫

注意：

schema為初始化sql路徑

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.AutoConfigureAfter;
import org.springframework.context.ApplicationContext;
import org.springframework.core.io.Resource;
import org.springframework.jdbc.datasource.init.ScriptUtils;
import org.springframework.stereotype.Service;

import javax.annotation.PostConstruct;
import javax.sql.DataSource;
import java.io.File;

/**
 * 初始化h2數據庫
 */
@Slf4j
@Service
// DataSource創(chuàng)建完后才初始化此類
@AutoConfigureAfter(DataSource.class)
public class H2DataSourceConfig {

    //初始化sql
    private static final String schema = "classpath:db/schema-h2.sql";

    @Autowired
    DataSource dataSource;

    @Autowired
    ApplicationContext applicationContext;

    @PostConstruct
    public void init() throws Exception {
        //初始化本地數據庫
        String userHome = System.getProperty("user.home");//獲取系統(tǒng)用戶目錄
        File f = new File(userHome + File.separator + "h2.lck");
        if (!f.exists()) {
            log.info("--------------初始化h2數據----------------------");
            f.createNewFile();
            Resource resource = applicationContext.getResource(schema);
            ScriptUtils.executeSqlScript(dataSource.getConnection(), resource);
        } else {
            log.info("--------------h2數據庫已經存在----------------------");
        }
    }
}

5. 初始化sql語句

在resources目錄下創(chuàng)建db/schema-h2.sql文件。

DROP TABLE IF EXISTS student;
CREATE TABLE student
(
    id BIGINT NOT NULL COMMENT 'id',
    name VARCHAR(30)  COMMENT '姓名',
    age INT COMMENT '年齡',
    PRIMARY KEY (id)
);
insert into student(id,name,age) values(1, 'zhangsan',23);
insert into student(id,name,age) values(2, 'lisi',45);
insert into student(id,name,age) values(3, 'wangwu',12);