fixed TestIssue798TruncatedError

CHANGELOG.md

@@ -1,4 +1,4 @@
-* Added environment variable `YDB_EXECUTE_DATA_QUERY_OVER_QUERY_SERVICE` for execute data queries from table service client using query client API 
+* Added `ydb.WithExecuteDataQueryOverQueryClient(bool)` option for execute data queries from table service client using query client API 
 
 ## v3.98.0
 * Supported pool of encoders, which implement ResetableWriter interface

internal/table/config/config.go

@@ -1,7 +1,6 @@
 package config
 
 import (
-	"os"
 	"time"
 
 	"github.com/jonboulle/clockwork"
@@ -290,16 +289,12 @@ func (c *Config) DeleteTimeout() time.Duration {
 }
 
 func defaults() *Config {
-	executeDataQueryOverQueryService := os.Getenv("YDB_EXECUTE_DATA_QUERY_OVER_QUERY_SERVICE") != ""
-
 	return &Config{
-		sizeLimit:                        DefaultSessionPoolSizeLimit,
-		createSessionTimeout:             DefaultSessionPoolCreateSessionTimeout,
-		deleteTimeout:                    DefaultSessionPoolDeleteTimeout,
-		idleThreshold:                    DefaultSessionPoolIdleThreshold,
-		clock:                            clockwork.NewRealClock(),
-		trace:                            &trace.Table{},
-		useQuerySession:                  executeDataQueryOverQueryService,
-		executeDataQueryOverQueryService: executeDataQueryOverQueryService,
+		sizeLimit:            DefaultSessionPoolSizeLimit,
+		createSessionTimeout: DefaultSessionPoolCreateSessionTimeout,
+		deleteTimeout:        DefaultSessionPoolDeleteTimeout,
+		idleThreshold:        DefaultSessionPoolIdleThreshold,
+		clock:                clockwork.NewRealClock(),
+		trace:                &trace.Table{},
 	}
 }

internal/table/session.go

@@ -378,7 +378,7 @@ func newTableSession(ctx context.Context, cc grpc.ClientConnInterface, config *c
 	)
 	s.executor = tableExecutor{
 		client:          s.client,
-		ignoreTruncated: false,
+		ignoreTruncated: s.config.IgnoreTruncated(),
 	}
 
 	return s, nil

options.go

@@ -558,9 +558,9 @@ func WithSessionPoolIdleThreshold(idleThreshold time.Duration) Option {
 
 // WithExecuteDataQueryOverQueryClient overrides table.Session.Execute with query service
 // execute with materialized result
-func WithExecuteDataQueryOverQueryClient() Option {
+func WithExecuteDataQueryOverQueryClient(b bool) Option {
 	return func(ctx context.Context, d *Driver) error {
-		d.tableOptions = append(d.tableOptions, tableConfig.ExecuteDataQueryOverQueryService(true))
+		d.tableOptions = append(d.tableOptions, tableConfig.ExecuteDataQueryOverQueryService(b))
 
 		return nil
 	}