fix: support dsn dd:// (#19)

databendlabs · Dec 6, 2022 · fbf4a39 · fbf4a39
1 parent 2aaa622
commit fbf4a39
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/dsn.go b/dsn.go
@@ -144,11 +144,13 @@ func ParseDSN(dsn string) (*Config, error) {
 		cfg.Scheme = u.Scheme
 	case "db+http", "db+https":
 		cfg.Scheme = u.Scheme[len("db+"):]
+	case "dd+http", "dd+https":
+		cfg.Scheme = u.Scheme[len("dd+"):]
 	case "bend+http", "bend+https":
 		cfg.Scheme = u.Scheme[len("bend+"):]
 	case "databend+http", "databend+https":
 		cfg.Scheme = u.Scheme[len("databend+"):]
-	case "databend", "db", "bend":
+	case "databend", "db", "dd", "bend":
 		if u.Query().Get("sslmode") == "disable" {
 			cfg.Scheme = "http"
 		} else {

diff --git a/dsn_test.go b/dsn_test.go
@@ -50,6 +50,7 @@ func TestParseDSN(t *testing.T) {
 		"http://username:password@app.databend.com:8000/test?tenant=tn&warehouse=wh&timeout=1s&idle_timeout=2s&tls_config=tls-settings",
 		"databend://username:password@app.databend.com:8000/test?tenant=tn&warehouse=wh&timeout=1s&idle_timeout=2s&tls_config=tls-settings&sslmode=disable",
 		"db://username:password@app.databend.com:8000/test?tenant=tn&warehouse=wh&timeout=1s&idle_timeout=2s&tls_config=tls-settings&sslmode=disable",
+		"dd://username:password@app.databend.com:8000/test?tenant=tn&warehouse=wh&timeout=1s&idle_timeout=2s&tls_config=tls-settings&sslmode=disable",
 		"bend://username:password@app.databend.com:8000/test?tenant=tn&warehouse=wh&timeout=1s&idle_timeout=2s&tls_config=tls-settings&sslmode=disable",
 	}