fix WithLazyTx

Author: asmyasnikov

CHANGELOG.md

@@ -1,4 +1,4 @@
-* Added `ydb.WithLazyTx()` option for create lazy transactions on `query.Session.Begin` call
+* Added `ydb.WithLazyTx(bool)` option for create lazy transactions on `query.Session.Begin` call
 * Added initial experimental topic and cdc-helpers, see examples in [tests/integration/topic_helpers_test.go](https://github.com/ydb-platform/ydb-go-sdk/blob/master/tests/integration/topic_helpers_test.go)
 * Added experimental `sugar.UnmarshalRows` for user unmarshaller structs in own code in go 1.23, change example for use the iterator.
 * Added `ydb_go_sdk_ydb_query_pool_size_index` metrics

internal/query/config/options.go

@@ -66,8 +66,8 @@ func WithSessionIdleTimeToLive(idleTimeToLive time.Duration) Option {
 	}
 }
 
-func WithLazyTx() Option {
+func WithLazyTx(lazyTx bool) Option {
 	return func(c *Config) {
-		c.lazyTx = true
+		c.lazyTx = lazyTx
 	}
 }

options.go

@@ -488,9 +488,9 @@ func WithSessionPoolSizeLimit(sizeLimit int) Option {
 	}
 }
 
-func WithLazyTx() Option {
+func WithLazyTx(lazyTx bool) Option {
 	return func(ctx context.Context, d *Driver) error {
-		d.queryOptions = append(d.queryOptions, queryConfig.WithLazyTx())
+		d.queryOptions = append(d.queryOptions, queryConfig.WithLazyTx(lazyTx))
 
 		return nil
 	}

query/example_test.go

@@ -415,7 +415,7 @@ func Example_retryWithTx() {
 func Example_retryWithLazyTx() {
 	ctx := context.TODO()
 	db, err := ydb.Open(ctx, "grpc://localhost:2136/local",
-		ydb.WithLazyTx(),
+		ydb.WithLazyTx(true),
 	)
 	if err != nil {
 		fmt.Printf("failed connect: %v", err)

tests/integration/query_tx_execute_test.go

@@ -71,7 +71,7 @@ func TestQueryTxExecute(t *testing.T) {
 			columnNames []string
 			columnTypes []string
 		)
-		err := scope.Driver(ydb.WithLazyTx()).Query().DoTx(scope.Ctx, func(ctx context.Context, tx query.TxActor) (err error) {
+		err := scope.Driver(ydb.WithLazyTx(true)).Query().DoTx(scope.Ctx, func(ctx context.Context, tx query.TxActor) (err error) {
 			if tx.ID() != baseTx.LazyTxID {
 				return errors.New("transaction is not lazy")
 			}
@@ -262,7 +262,7 @@ func TestQueryLazyTxExecute(t *testing.T) {
 		columnTypes []string
 	)
 	t.Run("Default", func(t *testing.T) {
-		err := scope.DriverWithLogs(ydb.WithLazyTx()).Query().DoTx(
+		err := scope.DriverWithLogs(ydb.WithLazyTx(true)).Query().DoTx(
 			scope.Ctx, func(ctx context.Context, tx query.TxActor) (err error) {
 				if tx.ID() != baseTx.LazyTxID {
 					return errors.New("transaction is not lazy")