Merge pull request #1281 from ydb-platform/fix-dsn

* Allowed the use of DSN without specifying the protocol/scheme

Author: asmyasnikov

CHANGELOG.md

@@ -1,3 +1,4 @@
+* Allowed the use of DSN without specifying the protocol/scheme
 * Removed public `query.Identifier` interface for exclude any external implementations for use with YDB
 
 ## v3.74.0

internal/dsn/dsn.go

@@ -3,13 +3,16 @@ package dsn
 import (
 	"fmt"
 	"net/url"
+	"regexp"
 
 	"github.com/ydb-platform/ydb-go-sdk/v3/config"
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/xerrors"
 )
 
 var (
 	insecureSchema = "grpc"
+	secureSchema   = "grpcs"
+	reScheme       = regexp.MustCompile(`^\w+://`)
 	databaseParam  = "database"
 )
 
@@ -25,7 +28,13 @@ type parsedInfo struct {
 }
 
 func Parse(dsn string) (info parsedInfo, err error) {
-	uri, err := url.Parse(dsn)
+	uri, err := url.ParseRequestURI(func() string {
+		if reScheme.MatchString(dsn) {
+			return dsn
+		}
+
+		return secureSchema + "://" + dsn
+	}())
 	if err != nil {
 		return info, xerrors.WithStackTrace(err)
 	}

internal/dsn/dsn_test.go

@@ -104,6 +104,22 @@ func TestParseConnectionString(t *testing.T) {
 			"",
 			"",
 		},
+		{
+			"localhost:3049/Root",
+			true,
+			"localhost:3049",
+			"/Root",
+			"",
+			"",
+		},
+		{
+			"grpc://localhost:3049/Root",
+			false,
+			"localhost:3049",
+			"/Root",
+			"",
+			"",
+		},
 	} {
 		t.Run(test.connectionString, func(t *testing.T) {
 			info, err := Parse(test.connectionString)