Function NewDB

server-source-code/internal/database/db.go:30–56

NewDB creates a pgx pool and sqlc Queries with retry logic.

(ctx context.Context, cfg *config.Config)

// NewDB creates a pgx pool and sqlc Queries with retry logic.
func NewDB(ctx context.Context, cfg *config.Config) (*DB, error) {
	poolCfg, err := pgxpool.ParseConfig(cfg.DatabaseURL)
	if err != nil {
		return nil, fmt.Errorf("parse pool config: %w", err)
	}
	poolCfg.MaxConns = safeconv.ClampToInt32(cfg.DBConnectionLimit)
	poolCfg.ConnConfig.ConnectTimeout = time.Duration(cfg.DBConnectTimeout) * time.Second
	// Force UTC session time zone so NOW()/CURRENT_TIMESTAMP stored into tz-naive
	// TIMESTAMP columns are UTC wall-clock, matching pgx's UTC assumption on read.
	setUTCTimeZone(poolCfg)
	pool, err := pgxpool.NewWithConfig(ctx, poolCfg)
	if err != nil {
		return nil, fmt.Errorf("create pgx pool: %w", err)
	}

	d := &DB{
		pool:    pool,
		Queries: db.New(pool),
		cfg:     cfg,
	}
	if err := d.waitForDB(ctx); err != nil {
		pool.Close()
		return nil, err
	}

	return d, nil
}

// NewFromURL creates a pgx pool and sqlc Queries from a raw database URL.
// Used for per-host pools; no retry logic. Caller must ensure the database is reachable.