2f7465c294
this will allow us to inspect the type of `db.Driver()` on *sql.DB connections to correctly differentiate between pg and crdb conns. as a bonus, this moves all concerns about when to replace "cockroach://" with "postgres://" out of view, letting the thin shim "driver" take care of that. Change-Id: Ib24103ab7c508231e681f89a7321b623e4e125e9
27 lines
795 B
Go
27 lines
795 B
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package dbutil
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// SplitConnStr returns the driver and DSN portions of a URL, along with the db implementation.
|
|
func SplitConnStr(s string) (driver string, source string, implementation Implementation, err error) {
|
|
// consider https://github.com/xo/dburl if this ends up lacking
|
|
parts := strings.SplitN(s, "://", 2)
|
|
if len(parts) != 2 {
|
|
return "", "", Unknown, fmt.Errorf("could not parse DB URL %s", s)
|
|
}
|
|
driver = parts[0]
|
|
source = parts[1]
|
|
implementation = ImplementationForScheme(parts[0])
|
|
|
|
if implementation == Postgres || implementation == Cockroach {
|
|
source = s // postgres and cockroach want full URLS for their DSNs
|
|
}
|
|
return driver, source, implementation, nil
|
|
}
|