Adding snowflake URL support

xo · Jun 16, 2018 · 689b5cf · 689b5cf
1 parent 1420779
commit 689b5cf
Show file tree

Hide file tree

Showing 5 changed files with 42 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -103,6 +103,7 @@ supported out of the box:
 | OLE ODBC (oleodbc)           | oo, ole, oleodbc [adodb]              |
 | Presto (presto)              | pr, prestodb, prestos, prs, prestodbs |
 | SAP HANA (hdb)               | sa, saphana, sap, hana                |
+| Snowflake (snowflake)        | sf                                    |
 | VoltDB (voltdb)              | vo, volt, vdb                         |
 
 Any protocol scheme `alias://` can be used in place of `protocol://`, and will work
@@ -154,6 +155,7 @@ to be imported:
 | OLE ODBC (oleodbc)           | [github.com/mattn/go-adodb](https://github.com/mattn/go-adodb)                              |
 | Presto (presto)              | [github.com/prestodb/presto-go-client/presto](https://github.com/prestodb/presto-go-client) |
 | SAP HANA (hdb)               | [github.com/SAP/go-hdb/driver](https://github.com/SAP/go-hdb)                               |
+| Snowflake (snowflake)        | [github.com/snowflakedb/gosnowflake](https://github.com/snowflakedb/gosnowflake)            |
 | VoltDB (voltdb)              | [github.com/VoltDB/voltdb-client-go/voltdbclient](github.com/VoltDB/voltdb-client-go])      |
 
 Please see [the GoDoc API page](http://godoc.org/github.com/xo/dburl) for a

diff --git a/dburl.go b/dburl.go
@@ -88,6 +88,7 @@
 //   OLE ODBC (oleodbc)           | oo, ole, oleodbc [adodb]
 //   Presto (presto)              | pr, prestodb, prestos, prs, prestodbs
 //   SAP HANA (hdb)               | sa, saphana, sap, hana
+//   Snowflake (snowflake)        | sf
 //   VoltDB (voltdb)              | vo, volt, vdb
 //
 // Any protocol scheme alias:// can be used in place of protocol://, and will
@@ -131,6 +132,7 @@
 //   OLE ODBC (oleodbc)*          | github.com/mattn/go-adodb
 //   Presto (presto)              | github.com/prestodb/presto-go-client
 //   SAP HANA (hdb)               | github.com/SAP/go-hdb/driver
+//   Snowflake (snowflake)        | github.com/snowflakedb/gosnowflake
 //   VoltDB (voltdb)              | github.com/VoltDB/voltdb-client-go/voltdbclient
 //
 // * OLE ODBC is not an actual protocol, but instead is an alias for using the
@@ -179,6 +181,9 @@ const (
 	// ErrRelativePathNotSupported is the relative paths not supported error.
 	ErrRelativePathNotSupported Error = "relative path not supported"
 
+	// ErrMissingHost is the missing host error.
+	ErrMissingHost Error = "missing host"
+
 	// ErrMissingPath is the missing path error.
 	ErrMissingPath Error = "missing path"
 )

diff --git a/dburl_test.go b/dburl_test.go
@@ -36,6 +36,10 @@ func TestBadParse(t *testing.T) {
 		{`cockroach+unix:./path`, ErrInvalidTransportProtocol},
 		{`pg:./path/to/socket`, ErrRelativePathNotSupported}, // relative paths are not possible for postgres sockets
 		{`pg+unix:./path/to/socket`, ErrRelativePathNotSupported},
+		{`snowflake://`, ErrMissingHost},
+		{`sf://`, ErrMissingHost},
+		{`snowflake://account`, ErrMissingPath},
+		{`sf://account`, ErrMissingPath},
 	}
 
 	for i, test := range tests {
@@ -126,6 +130,10 @@ func TestParse(t *testing.T) {
 		{`ig://user@host:9999?timeout=1000`, `ignite`, `tcp://host:9999?timeout=1000&username=user`},
 		{`ig://user:pass@localhost:9999/?timeout=1000`, `ignite`, `tcp://localhost:9999?password=pass&timeout=1000&username=user`},
 		{`ig://user:pass@localhost:9999/dbname?timeout=1000`, `ignite`, `tcp://localhost:9999/dbname?password=pass&timeout=1000&username=user`},
+
+		{`snowflake://host/dbname/schema`, `snowflake`, `host/dbname/schema`},
+		{`sf://user@host:9999/dbname/schema?timeout=1000`, `snowflake`, `user@host:9999/dbname/schema?timeout=1000`},
+		{`sf://user:pass@localhost:9999/dbname/schema?timeout=1000`, `snowflake`, `user:pass@localhost:9999/dbname/schema?timeout=1000`},
 	}
 
 	for i, test := range tests {

diff --git a/dsn.go b/dsn.go
@@ -567,3 +567,29 @@ func GenIgnite(u *URL) (string, error) {
 	}
 	return "tcp://" + host + ":" + port + dbname + genQueryOptions(q), nil
 }
+
+// GenSnowflake generates a snowflake DSN from the passed URL.
+func GenSnowflake(u *URL) (string, error) {
+	host, port, dbname := hostname(u.Host), hostport(u.Host), strings.TrimPrefix(u.Path, "/")
+	if host == "" {
+		return "", ErrMissingHost
+	}
+	if dbname == "" {
+		return "", ErrMissingPath
+	}
+	if port != "" {
+		port = ":" + port
+	}
+
+	// add user/pass
+	var user string
+	if u.User != nil {
+		user = u.User.Username()
+		if pass, _ := u.User.Password(); pass != "" {
+			user += ":" + pass
+		}
+		user += "@"
+	}
+
+	return user + host + port + "/" + dbname + genQueryOptions(u.Query()), nil
+}
diff --git a/scheme.go b/scheme.go
@@ -84,6 +84,7 @@ func BaseSchemes() []Scheme {
 		{"oleodbc", GenOLEODBC, ProtoAny, false, []string{"oo", "ole"}, "adodb"},
 		{"presto", GenPresto, 0, false, []string{"prestodb", "prestos", "prs", "prestodbs"}, ""},
 		{"ql", GenOpaque, 0, true, []string{"ql", "cznic", "cznicql"}, ""},
+		{"snowflake", GenSnowflake, 0, false, []string{"sf"}, ""},
 		{"voltdb", GenVoltDB, 0, false, []string{"volt", "vdb"}, ""},
 	}
 }