378b863b2b
first, so that they all work the same way, because it's getting complicated, and second, so that we can do the appropriate thing instead of CREATE SCHEMA for cockroachdb. Change-Id: I27fbaeeb6223a3e06d97bcf692a2d014b31465f7
27 lines
868 B
Go
27 lines
868 B
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package tempdb
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/zeebo/errs"
|
|
|
|
"storj.io/storj/private/dbutil"
|
|
"storj.io/storj/private/dbutil/cockroachutil"
|
|
"storj.io/storj/private/dbutil/pgutil"
|
|
)
|
|
|
|
// OpenUnique opens a temporary, uniquely named database (or isolated database schema)
|
|
// for scratch work. When closed, this database or schema will be cleaned up and destroyed.
|
|
func OpenUnique(connURL string, namePrefix string) (*dbutil.TempDatabase, error) {
|
|
if strings.HasPrefix(connURL, "postgres://") || strings.HasPrefix(connURL, "postgresql://") {
|
|
return pgutil.OpenUnique(connURL, namePrefix)
|
|
}
|
|
if strings.HasPrefix(connURL, "cockroach://") {
|
|
return cockroachutil.OpenUnique(connURL, namePrefix)
|
|
}
|
|
return nil, errs.New("OpenUnique does not yet support the db type for %q", connURL)
|
|
}
|