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

Author: asmyasnikov

CHANGELOG.md

@@ -1,3 +1,5 @@
+* Allowed the use of DSN without specifying the protocol/scheme
+
 ## v3.74.0
 * Added experimental range functions to the `query.Result` and `query.ResultSet` types, available as for-range loops starting with Go version 1.22. These features can be enabled by setting the environment variable `GOEXPERIMENT=rangefunc`.
 * Added public types for `tx.Option`, `options.DoOption` and `options.DoTxOption`

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)