diff --git a/be/src/runtime/runtime_state.h b/be/src/runtime/runtime_state.h index 2496249d24..02b9f0f5c3 100644 --- a/be/src/runtime/runtime_state.h +++ b/be/src/runtime/runtime_state.h @@ -89,7 +89,10 @@ public: } int query_parallel_instance_num() const { return _query_options.parallel_instance; } int max_errors() const { return _query_options.max_errors; } - int execution_timeout() const { return _query_options.execution_timeout; } + int execution_timeout() const { + return _query_options.__isset.execution_timeout ? _query_options.execution_timeout + : _query_options.query_timeout; + } int max_io_buffers() const { return _query_options.max_io_buffers; } int num_scanner_threads() const { return _query_options.num_scanner_threads; } TQueryType::type query_type() const { return _query_options.query_type; } diff --git a/fe/fe-core/src/main/java/org/apache/doris/planner/StreamLoadPlanner.java b/fe/fe-core/src/main/java/org/apache/doris/planner/StreamLoadPlanner.java index 3972e27105..552e99b83f 100644 --- a/fe/fe-core/src/main/java/org/apache/doris/planner/StreamLoadPlanner.java +++ b/fe/fe-core/src/main/java/org/apache/doris/planner/StreamLoadPlanner.java @@ -257,6 +257,7 @@ public class StreamLoadPlanner { params.setParams(execParams); TQueryOptions queryOptions = new TQueryOptions(); queryOptions.setQueryType(TQueryType.LOAD); + queryOptions.setQueryTimeout(timeout); queryOptions.setExecutionTimeout(timeout); queryOptions.setMemLimit(taskInfo.getMemLimit()); // for stream load, we use exec_mem_limit to limit the memory usage of load channel. diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java b/fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java index bef25659b0..64c26dd753 100644 --- a/fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java +++ b/fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java @@ -400,6 +400,7 @@ public class Coordinator { this.queryOptions = context.getSessionVariable().toThrift(); this.queryOptions.setEnablePipelineEngine(VectorizedUtil.isPipeline()); this.queryOptions.setBeExecVersion(Config.be_exec_version); + this.queryOptions.setQueryTimeout(context.getExecTimeout()); this.queryOptions.setExecutionTimeout(context.getExecTimeout()); } @@ -461,6 +462,7 @@ public class Coordinator { public void setTimeout(int timeout) { this.queryOptions.setQueryTimeout(timeout); + this.queryOptions.setExecutionTimeout(timeout); } public void setLoadZeroTolerance(boolean loadZeroTolerance) {