Function DBPool

rivershared/riversharedtest/riversharedtest.go:51–81

DBPool gets a lazily initialized database pool for `TEST_DATABASE_URL` or `river_test` if the former isn't specified.

(ctx context.Context, tb testing.TB)

// DBPool gets a lazily initialized database pool for `TEST_DATABASE_URL` or
// `river_test` if the former isn't specified.
func DBPool(ctx context.Context, tb testing.TB) *pgxpool.Pool {
	tb.Helper()

	dbPoolOnce.Do(func() {
		config, err := pgxpool.ParseConfig(TestDatabaseURL())
		require.NoError(tb, err)

		config.AfterConnect = func(ctx context.Context, conn *pgx.Conn) error {
			// Empty the search path so that tests using riverdbtest are
			// forced to pass a schema to clients and any other database
			// operations they invoke. Calls do not accidentally fall back to a
			// default schema, which would potentially hide bugs where we
			// weren't properly referencing a schema explicitly.
			_, err := conn.Exec(ctx, "SET search_path TO ''")

			// This should not be a `require` because the callback may run long
			// after the original test has completed.
			if err != nil && !errors.Is(err, context.Canceled) {
				panic(err)
			}

			return nil
		}

		dbPool, err = pgxpool.NewWithConfig(ctx, config)
		require.NoError(tb, err)
	})
	require.NotNil(tb, dbPool) // die in case initial connect from another test failed

	return dbPool
}

// DBPoolClone returns a disposable clone of DBPool. Share resources by using
// DBPool when possible, but this is useless for areas like stress tests where