This is an automated email from the ASF dual-hosted git repository.
dataroaring pushed a commit to branch branch-3.0
in repository https://gitbox.apache.org/repos/asf/doris.git
The following commit(s) were added to refs/heads/branch-3.0 by this push:
new b4df6805025 [cherry-pick](branch-3.0)add SessionVariable for
enableCooldownReplicaAffinity (#42677)
b4df6805025 is described below
commit b4df68050256ca2dc40c8c41da706ea43fb00d8b
Author: kkop <[email protected]>
AuthorDate: Thu Nov 7 13:47:25 2024 +0800
[cherry-pick](branch-3.0)add SessionVariable for
enableCooldownReplicaAffinity (#42677)
pick from master :https://github.com/apache/doris/pull/41741
---
.../main/java/org/apache/doris/common/Config.java | 3 --
.../org/apache/doris/planner/OlapScanNode.java | 10 +++++-
.../java/org/apache/doris/qe/SessionVariable.java | 9 +++++
.../suites/show_p0/test_show_variables.groovy | 38 ++++++++++++++++++++++
4 files changed, 56 insertions(+), 4 deletions(-)
diff --git a/fe/fe-common/src/main/java/org/apache/doris/common/Config.java
b/fe/fe-common/src/main/java/org/apache/doris/common/Config.java
index 03d720af287..af8ce5d7c2d 100644
--- a/fe/fe-common/src/main/java/org/apache/doris/common/Config.java
+++ b/fe/fe-common/src/main/java/org/apache/doris/common/Config.java
@@ -2845,9 +2845,6 @@ public class Config extends ConfigBase {
"Should the request content be logged before each request starts,
specifically the query statements"})
public static boolean enable_print_request_before_execution = false;
- @ConfField(mutable = true)
- public static boolean enable_cooldown_replica_affinity = true;
-
@ConfField
public static String spilled_profile_storage_path =
System.getenv("LOG_DIR") + File.separator + "profile";
diff --git
a/fe/fe-core/src/main/java/org/apache/doris/planner/OlapScanNode.java
b/fe/fe-core/src/main/java/org/apache/doris/planner/OlapScanNode.java
index 45d7ec0f769..afdffc748c0 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/planner/OlapScanNode.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/planner/OlapScanNode.java
@@ -858,7 +858,7 @@ public class OlapScanNode extends ScanNode {
}
}
- if (Config.enable_cooldown_replica_affinity) {
+ if (isEnableCooldownReplicaAffinity()) {
final long coolDownReplicaId = tablet.getCooldownReplicaId();
// we prefer to query using cooldown replica to make sure the
cache is fully utilized
// for example: consider there are 3BEs(A,B,C) and each has
one replica for tablet X. and X
@@ -976,6 +976,14 @@ public class OlapScanNode extends ScanNode {
}
}
+ private boolean isEnableCooldownReplicaAffinity() {
+ ConnectContext connectContext = ConnectContext.get();
+ if (connectContext != null) {
+ return
connectContext.getSessionVariable().isEnableCooldownReplicaAffinity();
+ }
+ return true;
+ }
+
private void computePartitionInfo() throws AnalysisException {
long start = System.currentTimeMillis();
// Step1: compute partition ids
diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/SessionVariable.java
b/fe/fe-core/src/main/java/org/apache/doris/qe/SessionVariable.java
index 838a7ba2f95..24168b27bad 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/qe/SessionVariable.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/qe/SessionVariable.java
@@ -669,6 +669,8 @@ public class SessionVariable implements Serializable,
Writable {
public static final String SKIP_CHECKING_ACID_VERSION_FILE =
"skip_checking_acid_version_file";
+ public static final String ENABLE_COOLDOWN_REPLICA_AFFINITY =
+ "enable_cooldown_replica_affinity";
/**
* Inserting overwrite for auto partition table allows creating partition
for
* datas which cannot find partition to overwrite.
@@ -2222,6 +2224,9 @@ public class SessionVariable implements Serializable,
Writable {
})
public boolean skipCheckingAcidVersionFile = false;
+ @VariableMgr.VarAttr(name = ENABLE_COOLDOWN_REPLICA_AFFINITY, needForward
= true)
+ public boolean enableCooldownReplicaAffinity = true;
+
public void setEnableEsParallelScroll(boolean enableESParallelScroll) {
this.enableESParallelScroll = enableESParallelScroll;
}
@@ -4433,4 +4438,8 @@ public class SessionVariable implements Serializable,
Writable {
public boolean isUseSerialExchange() {
return useSerialExchange && getEnableLocalExchange();
}
+
+ public boolean isEnableCooldownReplicaAffinity() {
+ return enableCooldownReplicaAffinity;
+ }
}
diff --git a/regression-test/suites/show_p0/test_show_variables.groovy
b/regression-test/suites/show_p0/test_show_variables.groovy
new file mode 100644
index 00000000000..6de8075b630
--- /dev/null
+++ b/regression-test/suites/show_p0/test_show_variables.groovy
@@ -0,0 +1,38 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements. See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership. The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License. You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied. See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+import org.apache.doris.regression.util.Http
+
+suite("test_show_variables", "p0") {
+
+ def result = sql """show variables like
"enable_cooldown_replica_affinity%";"""
+ assertTrue(result[0][1]=="true")
+
+ result = sql """set enable_cooldown_replica_affinity=false;"""
+ result = sql """show variables like "enable_cooldown_replica_affinity%";"""
+ assertTrue(result[0][1]=="false")
+
+ result = sql """set enable_cooldown_replica_affinity=true;"""
+ result = sql """show variables like "enable_cooldown_replica_affinity%";"""
+ assertTrue(result[0][1]=="true")
+
+ result = sql """set GLOBAL enable_cooldown_replica_affinity=false;"""
+ result = sql """show variables like "enable_cooldown_replica_affinity%";"""
+ assertTrue(result[0][1]=="false")
+ result = sql """set GLOBAL enable_cooldown_replica_affinity=true;"""
+
+}
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
