Jim
8d6dee09a9
* refact: Changes required for gorm v2 * refact: Add db.DB wrapper and refact all test fixtures to use it Stop leaking the gorm abstraction within our unit tests. * refact: Migrate direct calls to pq to pgx driver The refactor will remove the remaining direct dependencies on the pq driver which has been deprecated in favor of the pgx driver. It includes the addition of common.SqlOpen which is a wrapper around sql.Open, which is needed to deal with the driver name changing from "postgres" to "pgx" but in our code base the dialect and driver are synonyms.
126 lines
3.6 KiB
Go
126 lines
3.6 KiB
Go
//go:build linux || darwin || windows
|
|
// +build linux darwin windows
|
|
|
|
package docker
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/hashicorp/boundary/internal/db/common"
|
|
_ "github.com/jackc/pgx/v4/stdlib"
|
|
"github.com/ory/dockertest/v3"
|
|
)
|
|
|
|
func init() {
|
|
StartDbInDocker = startDbInDockerSupported
|
|
}
|
|
|
|
func startDbInDockerSupported(dialect string, opt ...Option) (cleanup func() error, retURL, container string, err error) {
|
|
// TODO: Debug what part of this method is actually causing race condition issues with our test and fix.
|
|
mx.Lock()
|
|
defer mx.Unlock()
|
|
pool, err := dockertest.NewPool("")
|
|
if err != nil {
|
|
return func() error { return nil }, "", "", fmt.Errorf("could not connect to docker: %w", err)
|
|
}
|
|
|
|
var resource *dockertest.Resource
|
|
var url, tag, repository string
|
|
|
|
opts := GetOpts(opt...)
|
|
if opts.withContainerImage != "" {
|
|
repository, tag, err = splitImage(opts)
|
|
if err != nil {
|
|
return func() error { return nil }, "", "", fmt.Errorf("error parsing reference: %w", err)
|
|
}
|
|
}
|
|
|
|
switch dialect {
|
|
case "postgres", "pgx":
|
|
switch {
|
|
case os.Getenv("BOUNDARY_TESTING_PG_URL") != "":
|
|
url = os.Getenv("BOUNDARY_TESTING_PG_URL")
|
|
return func() error { return nil }, url, "", nil
|
|
case repository != "":
|
|
resource, err = pool.Run(repository, tag, []string{"POSTGRES_PASSWORD=password", "POSTGRES_DB=boundary"})
|
|
url = "postgres://postgres:password@localhost:%s?sslmode=disable"
|
|
if err == nil {
|
|
url = fmt.Sprintf("postgres://postgres:password@%s/boundary?sslmode=disable", resource.GetHostPort("5432/tcp"))
|
|
}
|
|
default:
|
|
resource, err = pool.Run(dialect, tag, []string{"POSTGRES_PASSWORD=password", "POSTGRES_DB=boundary"})
|
|
url = "postgres://postgres:password@localhost:%s?sslmode=disable"
|
|
if err == nil {
|
|
url = fmt.Sprintf("postgres://postgres:password@%s/boundary?sslmode=disable", resource.GetHostPort("5432/tcp"))
|
|
}
|
|
}
|
|
|
|
default:
|
|
panic(fmt.Sprintf("unknown dialect %q", dialect))
|
|
}
|
|
if err != nil {
|
|
return func() error { return nil }, "", "", fmt.Errorf("could not start resource: %w", err)
|
|
}
|
|
|
|
cleanup = func() error {
|
|
return cleanupDockerResource(pool, resource)
|
|
}
|
|
|
|
if err := pool.Retry(func() error {
|
|
db, err := common.SqlOpen(dialect, url)
|
|
if err != nil {
|
|
return fmt.Errorf("error opening %s dev container: %w", dialect, err)
|
|
}
|
|
|
|
if err := db.Ping(); err != nil {
|
|
return err
|
|
}
|
|
defer db.Close()
|
|
return nil
|
|
}); err != nil {
|
|
return cleanup, "", "", fmt.Errorf("could not ping postgres on startup: %w", err)
|
|
}
|
|
|
|
return cleanup, url, resource.Container.Name, nil
|
|
}
|
|
|
|
// cleanupDockerResource will clean up the dockertest resources (postgres)
|
|
func cleanupDockerResource(pool *dockertest.Pool, resource *dockertest.Resource) error {
|
|
var err error
|
|
for i := 0; i < 10; i++ {
|
|
err = pool.Purge(resource)
|
|
if err == nil {
|
|
return nil
|
|
}
|
|
}
|
|
if strings.Contains(err.Error(), "No such container") {
|
|
return nil
|
|
}
|
|
return fmt.Errorf("failed to cleanup local container: %s", err)
|
|
}
|
|
|
|
// splitImage takes the WithDatabaseImage option and separates
|
|
// it into repo + tag. If a tag is not found, it sets the tag to latest
|
|
func splitImage(opts Options) (string, string, error) {
|
|
separated := strings.Split(opts.withContainerImage, ":")
|
|
separatedlen := len(separated)
|
|
|
|
switch separatedlen {
|
|
case 1:
|
|
if separated[0] == "postgres" {
|
|
return separated[0], "11", nil
|
|
}
|
|
return "", "", fmt.Errorf("valid reference format is repo:tag, if"+
|
|
" no tag provided then repo must be postgres, got: %s", opts.withContainerImage)
|
|
|
|
case 2:
|
|
return separated[0], separated[1], nil
|
|
|
|
default:
|
|
return "", "", fmt.Errorf("valid reference format is repo:tag, got: %s", opts.withContainerImage)
|
|
|
|
}
|
|
}
|