Allow to change (or disable) the default driver name for registration

conn.go

@@ -57,8 +57,19 @@ func (d Driver) Open(name string) (driver.Conn, error) {
 	return Open(name)
 }
 
+// This variable can be replaced with -ldflags like below:
+// go build "-ldflags=-X github.com/lib/pq.driverName=custom"
+var driverName = "postgres"
+
 func init() {
-	sql.Register("postgres", &Driver{})
+	if driverName != "" {
+		sql.Register(driverName, &Driver{})
+	}
+}
+
+// DriverName return default driver name
+func DriverName() string {
+	return driverName
 }
 
 type parameterStatus struct {
@@ -1251,7 +1262,7 @@ func (cn *conn) auth(r *readBuf, o values) {
 			token, err = cli.GetInitTokenFromSpn(spn)
 		} else {
 			// Allow the kerberos service name to be overridden
-			service := "postgres"
+			service := driverName
 			if val, ok := o["krbsrvname"]; ok {
 				service = val
 			}

conn_test.go

@@ -15,6 +15,16 @@ import (
 	"time"
 )
 
+// This variable can be replaced with -ldflags like below:
+// go test "-ldflags=-X github.com/lib/pq.driverNameTest=custom"
+var driverNameTest string
+
+func init() {
+	if driverNameTest == "" {
+		driverNameTest = driverName
+	}
+}
+
 type Fatalistic interface {
 	Fatal(args ...interface{})
 }
@@ -49,7 +59,7 @@ func testConninfo(conninfo string) string {
 }
 
 func openTestConnConninfo(conninfo string) (*sql.DB, error) {
-	return sql.Open("postgres", testConninfo(conninfo))
+	return sql.Open(driverNameTest, testConninfo(conninfo))
 }
 
 func openTestConn(t Fatalistic) *sql.DB {
@@ -1971,3 +1981,10 @@ func TestStmtExecContext(t *testing.T) {
 		})
 	}
 }
+
+func TestDriverName(t *testing.T) {
+	dn := DriverName()
+	if dn != driverName {
+		t.Fatalf("DriverName() returned driverName: %v, want: %v", dn, driverName)
+	}
+}

hstore/hstore_test.go

@@ -5,7 +5,7 @@ import (
 	"os"
 	"testing"
 
-	_ "github.com/lib/pq"
+	"github.com/lib/pq"
 )
 
 type Fatalistic interface {
@@ -24,7 +24,7 @@ func openTestConn(t Fatalistic) *sql.DB {
 		os.Setenv("PGSSLMODE", "disable")
 	}
 
-	conn, err := sql.Open("postgres", "")
+	conn, err := sql.Open(pq.DriverName(), "")
 	if err != nil {
 		t.Fatal(err)
 	}

ssl_test.go

@@ -359,7 +359,7 @@ func TestSNISupport(t *testing.T) {
 
 			// We are okay to skip this error as we are polling serverErrChan and we'll get an error
 			// or timeout from the server side in case of problems here.
-			db, _ := sql.Open("postgres", connStr)
+			db, _ := sql.Open(driverNameTest, connStr)
 			_, _ = db.Exec("SELECT 1")
 
 			// Check SNI data

url.go

@@ -35,7 +35,7 @@ func ParseURL(url string) (string, error) {
 		return "", err
 	}
 
-	if u.Scheme != "postgres" && u.Scheme != "postgresql" {
+	if u.Scheme != driverName && u.Scheme != "postgresql" {
 		return "", fmt.Errorf("invalid connection protocol: %s", u.Scheme)
 	}