-
Notifications
You must be signed in to change notification settings - Fork 85
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Added experimental
ydb.RegisterSqlOpenDsnParser for register exte…
…rnal custom DSN parser for `ydb.Open` and `sql.Open` driver constructor
- Loading branch information
1 parent
a9ac5e1
commit 009c18f
Showing
8 changed files
with
235 additions
and
147 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,116 @@ | ||
| package ydb | ||
|
|
||
| import ( | ||
| "errors" | ||
| "fmt" | ||
| "regexp" | ||
| "strings" | ||
|
|
||
| "github.com/ydb-platform/ydb-go-sdk/v3/balancers" | ||
| "github.com/ydb-platform/ydb-go-sdk/v3/credentials" | ||
| "github.com/ydb-platform/ydb-go-sdk/v3/internal/bind" | ||
| "github.com/ydb-platform/ydb-go-sdk/v3/internal/dsn" | ||
| "github.com/ydb-platform/ydb-go-sdk/v3/internal/xerrors" | ||
| "github.com/ydb-platform/ydb-go-sdk/v3/internal/xsql" | ||
| ) | ||
|
|
||
| const tablePathPrefixTransformer = "table_path_prefix" | ||
|
|
||
| var dsnParsers = []func(dsn string) (opts []Option, _ error){ | ||
| func(dsn string) ([]Option, error) { | ||
| opts, err := parseConnectionString(dsn) | ||
| if err != nil { | ||
| return nil, xerrors.WithStackTrace(err) | ||
| } | ||
|
|
||
| return opts, nil | ||
| }, | ||
| } | ||
|
|
||
| // RegisterDsnParser registers DSN parser for ydb.Open and sql.Open driver constructors | ||
| // | ||
| // Experimental: https://github.com/ydb-platform/ydb-go-sdk/blob/master/VERSIONING.md#experimental | ||
| func RegisterDsnParser(parser func(dsn string) (opts []Option, _ error)) { | ||
| dsnParsers = append(dsnParsers, parser) | ||
| } | ||
|
|
||
| func parseConnectionString(dataSourceName string) (opts []Option, _ error) { | ||
| info, err := dsn.Parse(dataSourceName) | ||
| if err != nil { | ||
| return nil, xerrors.WithStackTrace(err) | ||
| } | ||
| opts = append(opts, With(info.Options...)) | ||
| if token := info.Params.Get("token"); token != "" { | ||
| opts = append(opts, WithCredentials(credentials.NewAccessTokenCredentials(token))) | ||
| } | ||
| if balancer := info.Params.Get("go_balancer"); balancer != "" { | ||
| opts = append(opts, WithBalancer(balancers.FromConfig(balancer))) | ||
| } else if balancer := info.Params.Get("balancer"); balancer != "" { | ||
| opts = append(opts, WithBalancer(balancers.FromConfig(balancer))) | ||
| } | ||
| if queryMode := info.Params.Get("go_query_mode"); queryMode != "" { | ||
| mode := xsql.QueryModeFromString(queryMode) | ||
| if mode == xsql.UnknownQueryMode { | ||
| return nil, xerrors.WithStackTrace(fmt.Errorf("unknown query mode: %s", queryMode)) | ||
| } | ||
| opts = append(opts, withConnectorOptions(xsql.WithDefaultQueryMode(mode))) | ||
| } else if queryMode := info.Params.Get("query_mode"); queryMode != "" { | ||
| mode := xsql.QueryModeFromString(queryMode) | ||
| if mode == xsql.UnknownQueryMode { | ||
| return nil, xerrors.WithStackTrace(fmt.Errorf("unknown query mode: %s", queryMode)) | ||
| } | ||
| opts = append(opts, withConnectorOptions(xsql.WithDefaultQueryMode(mode))) | ||
| } | ||
| if fakeTx := info.Params.Get("go_fake_tx"); fakeTx != "" { | ||
| for _, queryMode := range strings.Split(fakeTx, ",") { | ||
| mode := xsql.QueryModeFromString(queryMode) | ||
| if mode == xsql.UnknownQueryMode { | ||
| return nil, xerrors.WithStackTrace(fmt.Errorf("unknown query mode: %s", queryMode)) | ||
| } | ||
| opts = append(opts, withConnectorOptions(xsql.WithFakeTx(mode))) | ||
| } | ||
| } | ||
| if info.Params.Has("go_query_bind") { | ||
| var binders []xsql.ConnectorOption | ||
| queryTransformers := strings.Split(info.Params.Get("go_query_bind"), ",") | ||
| for _, transformer := range queryTransformers { | ||
| switch transformer { | ||
| case "declare": | ||
| binders = append(binders, xsql.WithQueryBind(bind.AutoDeclare{})) | ||
| case "positional": | ||
| binders = append(binders, xsql.WithQueryBind(bind.PositionalArgs{})) | ||
| case "numeric": | ||
| binders = append(binders, xsql.WithQueryBind(bind.NumericArgs{})) | ||
| default: | ||
| if strings.HasPrefix(transformer, tablePathPrefixTransformer) { | ||
| prefix, err := extractTablePathPrefixFromBinderName(transformer) | ||
| if err != nil { | ||
| return nil, xerrors.WithStackTrace(err) | ||
| } | ||
| binders = append(binders, xsql.WithTablePathPrefix(prefix)) | ||
| } else { | ||
| return nil, xerrors.WithStackTrace( | ||
| fmt.Errorf("unknown query rewriter: %s", transformer), | ||
| ) | ||
| } | ||
| } | ||
| } | ||
| opts = append(opts, withConnectorOptions(binders...)) | ||
| } | ||
|
|
||
| return opts, nil | ||
| } | ||
|
|
||
| var ( | ||
| tablePathPrefixRe = regexp.MustCompile(tablePathPrefixTransformer + "\\((.*)\\)") | ||
| errWrongTablePathPrefix = errors.New("wrong '" + tablePathPrefixTransformer + "' query transformer") | ||
| ) | ||
|
|
||
| func extractTablePathPrefixFromBinderName(binderName string) (string, error) { | ||
| ss := tablePathPrefixRe.FindAllStringSubmatch(binderName, -1) | ||
| if len(ss) != 1 || len(ss[0]) != 2 || ss[0][1] == "" { | ||
| return "", xerrors.WithStackTrace(fmt.Errorf("%w: %s", errWrongTablePathPrefix, binderName)) | ||
| } | ||
|
|
||
| return ss[0][1], nil | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.