Docker-Compose下后端连PostgreSQL：连接拒却但日志误报成功

核心问题原因 #

你看到的日志并不矛盾：

• 20:28:24的连接拒绝错误是第一次重试时的失败日志，此时PostgreSQL容器还未完全启动就绪（depends_on仅保证容器启动顺序，不保证服务内部就绪）。
• 10秒后第二次重试时PostgreSQL服务已就绪，连接成功，因此日志输出"连接已打开"。

但你的代码存在潜在漏洞：gorm.Open()方法本身不会立即建立实际数据库连接，它仅初始化GORM客户端实例，真正的连接会延迟到首次执行SQL操作时建立。这意味着即使gorm.Open()返回err==nil，也不代表数据库连接已经可用，后续业务操作仍可能触发连接失败。

修复方案 #

修改ConnectDB()函数，在初始化GORM实例后，主动获取底层sql.DB并调用Ping()验证连接有效性，确保重试逻辑真正判断连接是否可用：

package database

import (
    "fmt"
    "log"
    "os"
    "time"

    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

var GDB *gorm.DB

func ConnectDB() {
    var err error
    pgPort := os.Getenv("DATABASE_PORT")
    pgHost := os.Getenv("DATABASE_HOST")
    pgUser := os.Getenv("POSTGRES_USER")
    pgPassword := os.Getenv("POSTGRES_PASSWORD")
    pgName := os.Getenv("POSTGRES_DB")

    configData := fmt.Sprintf("postgres://%v:%v@%v:%v/%v?sslmode=disable",
        pgUser,
        pgPassword,
        pgHost,
        pgPort,
        pgName,
    )

    for i := 0; i < 5; i++ {
        // 初始化GORM实例
        GDB, err = gorm.Open(postgres.Open(configData), &gorm.Config{})
        if err != nil {
            log.Printf("Producer: 第%d次初始化GORM失败: %v", i+1, err)
            time.Sleep(10 * time.Second)
            continue
        }

        // 获取底层sql.DB实例
        sqlDB, err := GDB.DB()
        if err != nil {
            log.Printf("Producer: 第%d次获取sql.DB失败: %v", i+1, err)
            time.Sleep(10 * time.Second)
            continue
        }

        // 主动Ping数据库，确认连接可用
        if err := sqlDB.Ping(); err != nil {
            log.Printf("Producer: 第%d次连接数据库失败: %v", i+1, err)
            time.Sleep(10 * time.Second)
            continue
        }

        // 所有验证通过，跳出循环
        break
    }

    if err != nil {
        log.Fatal("Producer: 最终无法连接到数据库")
    } else {
        log.Println("Producer: 数据库连接已成功建立")
    }
}

额外优化建议 #

1. 增强容器就绪检测：在docker-compose.yml中为imgress-db添加健康检查，确保PostgreSQL服务完全就绪后再启动imgress-producer：

imgress-db:
    image: postgres
    container_name: imgress-db
    environment:
      - POSTGRES_USER=${POSTGRES_USER}
      - POSTGRES_PASSWORD=${POSTGRES_PASSWORD}
      - POSTGRES_DB=${POSTGRES_DB}
    volumes:
      - postgres-data:/var/lib/postgresql/data
    ports:
      - 5432:5432
    networks:
      - imgress-network
    restart: always
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U ${POSTGRES_USER} -d ${POSTGRES_DB}"]
      interval: 5s
      timeout: 5s
      retries: 5

同时修改imgress-producer的depends_on配置，依赖数据库的健康状态：

depends_on:
  imgress-db:
    condition: service_healthy

2. 细化日志输出：在重试过程中添加次数标识，清晰区分中间失败尝试与最终连接结果，便于排查问题。

内容的提问来源于stack exchange，提问作者Eziz Durdyyev