storj/private/dbutil/split.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 = setImplementation(parts[0])

	if driver == "postgres" {
		source = s // postgres wants full URLS for its DSN
	}
	if driver == "cockroach" {
		driver = "postgres" // cockroach's driver is actually postgres
		source = "postgres://" + source
	}
	return driver, source, implementation, nil
}