Docker中PostgreSQL+Spring场景下SQL脚本执行时机问题求解

问题根源 #

你之前的两种方案失效原因：

• 挂载到docker-entrypoint-initdb.d的脚本仅在PostgreSQL首次创建数据卷时执行，且执行时机远早于Spring JPA的建表操作，此时目标表还不存在，必然报错。
• 固定延迟10秒的方式完全不可靠，Spring JPA的初始化时间受环境、数据量影响，无法通过硬编码延迟保证时机准确。

以下是几种可靠的解决方案：

方案1：利用Spring Boot内置初始化机制（最推荐） #

Spring Boot原生支持在JPA完成数据表创建后执行初始化脚本，步骤如下：

1. 将添加用户的SQL脚本放到项目的src/main/resources目录下，命名为data.sql（如果需要多环境区分，可命名为data-{profile}.sql，比如data-dev.sql）。
2. 修改Spring Boot的配置文件（application.properties或application.yml），添加以下配置：

   # 控制脚本执行时机：always表示每次启动都执行，embedded仅在嵌入式数据库时执行，按需选择
   spring.sql.init.mode=always
   # 关键配置：延迟数据源初始化，等待JPA完成建表后再执行data.sql
   spring.jpa.defer-datasource-initialization=true
   

3. 重新构建Spring服务镜像并启动，脚本会自动在JPA建表完成后执行。

方案2：自定义启动后执行逻辑（适合复杂场景） #

如果需要更灵活的控制（比如判断表是否存在再执行、动态生成SQL），可以通过CommandLineRunner或ApplicationRunner实现：

1. 在Spring项目中创建一个组件类：

   import org.springframework.boot.CommandLineRunner;
   import org.springframework.jdbc.core.JdbcTemplate;
   import org.springframework.stereotype.Component;
   
   @Component
   public class PostJpaInitRunner implements CommandLineRunner {
   
       private final JdbcTemplate jdbcTemplate;
   
       // 构造注入JdbcTemplate
       public PostJpaInitRunner(JdbcTemplate jdbcTemplate) {
           this.jdbcTemplate = jdbcTemplate;
       }
   
       @Override
       public void run(String... args) throws Exception {
           // 执行添加用户的SQL，替换为你的实际脚本
           String addUserSql = "INSERT INTO users (username, password, enabled) VALUES ('test_user', 'encrypted_password', true)";
           jdbcTemplate.execute(addUserSql);
   
           // 如果脚本内容较多，也可以读取外部文件执行
           // Resource scriptResource = new ClassPathResource("post-init.sql");
           // String scriptContent = FileCopyUtils.copyToString(new InputStreamReader(scriptResource.getInputStream()));
           // jdbcTemplate.execute(scriptContent);
       }
   }
   

2. 该组件会在Spring上下文完全加载、JPA初始化完成后自动执行，确保目标表已存在。

方案3：配合Docker Compose的健康检查（辅助优化） #

虽然无法直接保证JPA建表完成，但可以通过健康检查确保PostgreSQL完全就绪后再启动Spring服务，避免因DB未就绪导致的连接失败：
修改你的docker-compose.yaml如下：

version: '3.8'

services:

  db-postgres:
    image: "postgres:15.1"
    container_name: my-database
    volumes:
      - my-data:/var/lib/postgresql/data
    ports:
      - "5432:5432"
    environment:
      - POSTGRES_DB=postgres
      - POSTGRES_USER=postgres
      - POSTGRES_PASSWORD=postgres
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U postgres -d postgres"]
      interval: 5s
      timeout: 5s
      retries: 5

  spring:
    build: ./my-service
    container_name: my-service
    ports:
      - "8080:8080"
    depends_on:
      db-postgres:
        condition: service_healthy

注意：此方案需配合前面两种方案使用，仅解决DB就绪问题，不解决JPA建表时机问题。

内容的提问来源于stack exchange，提问作者Brave Spirit