Function PostgresDB

pkg/testinstance/postgres.go:58–135

(postgresVersion string)

}

func PostgresDB(postgresVersion string) *PostgresInstance {
	ctx := context.Background()

	image := fmt.Sprintf("postgres:%s-alpine", postgresVersion)

	postgres, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
		ContainerRequest: testcontainers.ContainerRequest{
			Image:        image,
			ExposedPorts: []string{"5432/tcp"},
			Env:          map[string]string{"POSTGRES_USER": "postgres", "POSTGRES_PASSWORD": "postgres", "POSTGRES_DB": "permify"},
			Cmd:          []string{"postgres", "-c", "track_commit_timestamp=on"},
			WaitingFor: wait.ForAll(
				wait.ForLog("database system is ready to accept connections"),
				wait.ForListeningPort("5432/tcp"),
			),
		},
		Started: true,
	})
	gomega.Expect(err).ShouldNot(gomega.HaveOccurred())

	var db *PQDatabase.Postgres

	cleanup := func() {
		if db != nil {
			_ = db.Close()
		}
		if postgres != nil {
			terminateCtx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
			defer cancel()
			_ = postgres.Terminate(terminateCtx)
		}
	}

	expectNoError := func(err error) {
		if err != nil {
			cleanup()
		}
		gomega.Expect(err).ShouldNot(gomega.HaveOccurred())
	}

	host, err := postgres.Host(ctx)
	expectNoError(err)

	port, err := postgres.MappedPort(ctx, "5432")
	expectNoError(err)

	dbAddr := fmt.Sprintf("%s:%s", host, port.Port())
	postgresDSN := fmt.Sprintf("postgres://%s:%s@%s/%s?sslmode=disable", "postgres", "postgres", dbAddr, "permify")

	cfg := config.Database{
		Engine:                "postgres",
		URI:                   postgresDSN,
		AutoMigrate:           true,
		MaxOpenConnections:    20,
		MaxIdleConnections:    1,
		MaxConnectionLifetime: 300,
		MaxConnectionIdleTime: 60,
	}