diff --git a/pkg/query-service/app/clickhouseReader/reader.go b/pkg/query-service/app/clickhouseReader/reader.go index 8dce05250d..2fc7383d5b 100644 --- a/pkg/query-service/app/clickhouseReader/reader.go +++ b/pkg/query-service/app/clickhouseReader/reader.go @@ -184,6 +184,7 @@ func NewReaderFromClickhouseConnection( wrap := clickhouseConnWrapper{ conn: db, settings: ClickhouseQuerySettings{ + MaxExecutionTime: os.Getenv("ClickHouseMaxExecutionTime"), MaxExecutionTimeLeaf: os.Getenv("ClickHouseMaxExecutionTimeLeaf"), TimeoutBeforeCheckingExecutionSpeed: os.Getenv("ClickHouseTimeoutBeforeCheckingExecutionSpeed"), MaxBytesToRead: os.Getenv("ClickHouseMaxBytesToRead"), diff --git a/pkg/query-service/app/clickhouseReader/wrapper.go b/pkg/query-service/app/clickhouseReader/wrapper.go index c575aa7226..fc0a71dc8d 100644 --- a/pkg/query-service/app/clickhouseReader/wrapper.go +++ b/pkg/query-service/app/clickhouseReader/wrapper.go @@ -11,6 +11,7 @@ import ( ) type ClickhouseQuerySettings struct { + MaxExecutionTime string MaxExecutionTimeLeaf string TimeoutBeforeCheckingExecutionSpeed string MaxBytesToRead string @@ -47,6 +48,10 @@ func (c clickhouseConnWrapper) addClickHouseSettings(ctx context.Context, query settings["max_bytes_to_read"] = c.settings.MaxBytesToRead } + if c.settings.MaxExecutionTime != "" { + settings["max_execution_time"] = c.settings.MaxExecutionTime + } + if c.settings.MaxExecutionTimeLeaf != "" { settings["max_execution_time_leaf"] = c.settings.MaxExecutionTimeLeaf }