Merge pull request #1695 from ydb-platform/wide-types

* Added binding `ydb.WithWideTimeTypes()` which interprets `time.Time` and `time.Duration` as `Timestamp64` and `Interval64` YDB types

Author: asmyasnikov

CHANGELOG.md

@@ -1,3 +1,5 @@
+* Added binding `ydb.WithWideTimeTypes()` which interprets `time.Time` and `time.Duration` as `Timestamp64` and `Interval64` YDB types
+
 ## v3.103.0
 * Supported wide `Interval64` type
 

dsn.go

@@ -114,6 +114,8 @@ func parseConnectionString(dataSourceName string) (opts []Option, _ error) {
 				binders = append(binders, xsql.WithQueryBind(bind.PositionalArgs{}))
 			case "numeric":
 				binders = append(binders, xsql.WithQueryBind(bind.NumericArgs{}))
+			case "wide_time_types":
+				binders = append(binders, xsql.WithQueryBind(bind.WideTimeTypes{}))
 			default:
 				if strings.HasPrefix(transformer, tablePathPrefixTransformer) {
 					prefix, err := extractTablePathPrefixFromBinderName(transformer)

dsn_test.go

@@ -171,6 +171,21 @@ func TestParse(t *testing.T) {
 			},
 			err: nil,
 		},
+		{
+			dsn: "grpc://localhost:2135/local?query_mode=scripting&go_query_bind=positional,declare,wide_time_types", //nolint:lll
+			opts: []config.Option{
+				config.WithSecure(false),
+				config.WithEndpoint("localhost:2135"),
+				config.WithDatabase("/local"),
+			},
+			connectorOpts: []xsql.Option{
+				xsql.WithDefaultQueryMode(xtable.ScriptingQueryMode),
+				xsql.WithQueryBind(bind.PositionalArgs{}),
+				xsql.WithQueryBind(bind.AutoDeclare{}),
+				xsql.WithQueryBind(bind.WideTimeTypes{}),
+			},
+			err: nil,
+		},
 	} {
 		t.Run("", func(t *testing.T) {
 			opts, err := parseConnectionString(tt.dsn)

internal/bind/auto_declare.go

@@ -13,8 +13,8 @@ func (m AutoDeclare) blockID() blockID {
 	return blockDeclare
 }
 
-func (m AutoDeclare) ToYdb(sql string, args ...interface{}) (
-	yql string, newArgs []interface{}, err error,
+func (m AutoDeclare) ToYdb(sql string, args ...any) (
+	yql string, newArgs []any, err error,
 ) {
 	params, err := Params(args...)
 	if err != nil {

internal/bind/bind.go

@@ -11,22 +11,24 @@ import (
 type blockID int
 
 const (
-	blockPragma = blockID(iota)
+	blockDefault = blockID(iota)
+	blockPragma
 	blockDeclare
 	blockYQL
+	blockCastArgs
 )
 
 type Bind interface {
-	ToYdb(sql string, args ...interface{}) (
-		yql string, newArgs []interface{}, _ error,
+	ToYdb(sql string, args ...any) (
+		yql string, newArgs []any, _ error,
 	)
 
 	blockID() blockID
 }
 
 type Bindings []Bind
 
-func (bindings Bindings) ToYdb(sql string, args ...interface{}) (
+func (bindings Bindings) ToYdb(sql string, args ...any) (
 	yql string, params params.Params, err error,
 ) {
 	if len(bindings) == 0 {

internal/bind/noop.go

@@ -0,0 +1,11 @@
+package bind
+
+type noopBind struct{}
+
+func (m noopBind) ToYdb(sql string, args ...any) (yql string, newArgs []any, _ error) {
+	return sql, args, nil
+}
+
+func (m noopBind) blockID() blockID {
+	return blockCastArgs
+}

internal/bind/numeric_args.go

@@ -17,7 +17,7 @@ func (m NumericArgs) blockID() blockID {
 	return blockYQL
 }
 
-func (m NumericArgs) ToYdb(sql string, args ...interface{}) (yql string, newArgs []interface{}, err error) {
+func (m NumericArgs) ToYdb(sql string, args ...any) (yql string, newArgs []any, err error) {
 	l := &sqlLexer{
 		src:        sql,
 		stateFn:    numericArgsStateFn,
@@ -36,7 +36,7 @@ func (m NumericArgs) ToYdb(sql string, args ...interface{}) (yql string, newArgs
 		if err != nil {
 			return "", nil, err
 		}
-		newArgs = make([]interface{}, len(parameters))
+		newArgs = make([]any, len(parameters))
 		for i, param := range parameters {
 			newArgs[i] = param
 		}
@@ -119,7 +119,7 @@ func numericArgsStateFn(l *sqlLexer) stateFn {
 	}
 }
 
-func parsePositionalParameters(args []interface{}) ([]*params.Parameter, error) {
+func parsePositionalParameters(args []any) ([]*params.Parameter, error) {
 	newArgs := make([]*params.Parameter, len(args))
 	for i, arg := range args {
 		paramName := fmt.Sprintf("$p%d", i)