6

我有以下代码用于连接到 Postgres 数据库：

func connectToPostgres(ctx context.Context, url string) (*pgxpool.Pool, error) {
    var err error
    for i := 0; i < 5; i++ {
        p, err := pgxpool.Connect(ctx, url)
        if err != nil || p == nil {
            time.Sleep(3 * time.Second)
            continue
        }
        log.Printf("pool returned from connect: %s", p)
        return p, nil
    }
    return nil, errors.Wrap(err, "timed out waiting to connect postgres")
}

用例是在使用 docker-compose 启动我的服务器时等待 Postgres 可用。即使代码休眠 if p == nil，第一次返回之前的日志也会打印出来：pool returned from connect: %!s(*pgxpool.Pool=<nil>)

有什么方法pgxpool可以让后台进程实现p == nil吗？

关于为什么会发生这种情况的任何想法？

编辑：这似乎只发生在通过 docker-compose 运行我的应用程序和 Postgres 时。我正在使用以下撰写文件：

services:
    app:
        build: .
        ports:
            - "8080:8080"
        depends_on:
            - "db"

    db:
        image: postgres
        restart: always
        environment:
            - POSTGRES_DB=demo_db
            - POSTGRES_PASSWORD=${POSTGRES_PASSWORD}
        ports:
            - "8081:5432"

和我的应用程序的 Dockerfile：

FROM golang:1.17

WORKDIR /

COPY go.mod .
COPY go.sum .
COPY *.go .

RUN go mod download
RUN go build

EXPOSE 8080

CMD [ "./app" ]

还有一个最小可重现的示例 go 文件：

package main

import (
    "context"
    "fmt"
    "log"
    "net/http"
    "time"

    "github.com/jackc/pgx/v4/pgxpool"
    "github.com/pkg/errors"
)

func main() {
    log.Printf("connecting to postgres...")
    pgpool, err := connectToPostgres(context.Background(), "postgresql://localhost:5432/demo_db")
    log.Printf("pool: %s", pgpool)
    if err != nil {
        log.Fatalln(err)
    }
    log.Printf("successfully connected to postgres")

    if err := http.ListenAndServe(":8080", nil); err != nil {
        log.Fatal(err)
    }
    log.Println("stopped")
}

func connectToPostgres(ctx context.Context, url string) (*pgxpool.Pool, error) {
    var err error
    for i := 0; i < 5; i++ {
        p, err := pgxpool.Connect(ctx, url)
        if err != nil || p == nil {
            time.Sleep(3 * time.Second)
            continue
        }
        log.Printf("pool returned from connect: %s", p)
        return p, nil
    }
    return nil, errors.Wrap(err, "timed out waiting to connect postgres")
}
4

1 回答 1

4

问题是在docker-compose网络中连接时，您必须连接到容器的主机名，在这种情况下是db.

您也可以使用其他容器IP，但需要额外的工作量，仅使用主机名更简单。

换句话说，你有错误的连接字符串，我在连接时也得到了这个localhost

app_1  | 2021/12/21 18:53:28 pool: %!s(*pgxpool.Pool=<nil>)
app_1  | 2021/12/21 18:53:28 successfully connected to postgres

使用正确的连接字符串连接时：

 "postgres://postgres:mysecretpassword@db:5432/postgres"

它完美地工作。

其余的日志

db_1   | 2021-12-21 18:56:04.122 UTC [1] LOG:  database system is ready to accept connections
app_1  | 2021/12/21 18:56:06 pool returned from connect: &{%!s(*puddle.Pool=&{0xc00007c040 0xc0000280b0 [0xc00007c0c0] [0xc00007c0c0] 0x65cb60 0x65dc80 16 1 9872796 1 0 false}) %!s(*pgxpool.Config=&{0xc0000a2000 <nil> <nil> <nil> <nil> 3600000000000 1800000000000 16 0 60000000000 false true}) %!s(func(context.Context, *pgx.ConnConfig) error=<nil>) %!s(func(context.Context, *pgx.Conn) error=<nil>) %!s(func(context.Context, *pgx.Conn) bool=<nil>) %!s(func(*pgx.Conn) bool=<nil>) %!s(int32=0) %!s(time.Duration=3600000000000) %!s(time.Duration=1800000000000) %!s(time.Duration=60000000000) {%!s(uint32=0) {%!s(int32=0) %!s(uint32=0)}} %!s(chan struct {}=0xc000024060)}
app_1  | 2021/12/21 18:56:06 pool: &{%!s(*puddle.Pool=&{0xc00007c040 0xc0000280b0 [0xc00007c0c0] [0xc00007c0c0] 0x65cb60 0x65dc80 16 1 9872796 1 0 false}) %!s(*pgxpool.Config=&{0xc0000a2000 <nil> <nil> <nil> <nil> 3600000000000 1800000000000 16 0 60000000000 false true}) %!s(func(context.Context, *pgx.ConnConfig) error=<nil>) %!s(func(context.Context, *pgx.Conn) error=<nil>) %!s(func(context.Context, *pgx.Conn) bool=<nil>) %!s(func(*pgx.Conn) bool=<nil>) %!s(int32=0) %!s(time.Duration=3600000000000) %!s(time.Duration=1800000000000) %!s(time.Duration=60000000000) {%!s(uint32=0) {%!s(int32=0) %!s(uint32=0)}} %!s(chan struct {}=0xc000024060)}
app_1  | 2021/12/21 18:56:06 successfully connected to postgres
于 2021-12-21T18:54:32.923 回答