This is an automated email from the ASF dual-hosted git repository.
luozenglin pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/doris.git
The following commit(s) were added to refs/heads/master by this push:
new a66d5a6ae0 [fix](workload-group) fix workload group non-existence
error (#20428)
a66d5a6ae0 is described below
commit a66d5a6ae0ef8f2a132e9684e1f5e7d54ffb4395
Author: luozenglin <luozeng...@baidu.com>
AuthorDate: Mon Jun 5 15:33:26 2023 +0800
[fix](workload-group) fix workload group non-existence error (#20428)
---
.../main/java/org/apache/doris/qe/StmtExecutor.java | 6 +++---
.../resource/workloadgroup/WorkloadGroupMgr.java | 19 ++++++++++++++-----
2 files changed, 17 insertions(+), 8 deletions(-)
diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/StmtExecutor.java
b/fe/fe-core/src/main/java/org/apache/doris/qe/StmtExecutor.java
index ca315e480b..b7c4591e4d 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/qe/StmtExecutor.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/qe/StmtExecutor.java
@@ -560,9 +560,9 @@ public class StmtExecutor {
private void handleQueryWithRetry(TUniqueId queryId) throws Exception {
// queue query here
- if (!parsedStmt.isExplain() && Config.enable_workload_group &&
Config.enable_query_queue) {
- this.queryQueue = analyzer.getEnv().getWorkloadGroupMgr()
-
.getWorkloadGroupQueryQueue(context.sessionVariable.workloadGroup);
+ if (!parsedStmt.isExplain() && Config.enable_workload_group &&
Config.enable_query_queue
+ && context.getSessionVariable().enablePipelineEngine()) {
+ this.queryQueue =
analyzer.getEnv().getWorkloadGroupMgr().getWorkloadGroupQueryQueue(context);
try {
this.offerRet = queryQueue.offer();
} catch (InterruptedException e) {
diff --git
a/fe/fe-core/src/main/java/org/apache/doris/resource/workloadgroup/WorkloadGroupMgr.java
b/fe/fe-core/src/main/java/org/apache/doris/resource/workloadgroup/WorkloadGroupMgr.java
index 1813271a4d..98a8d6891d 100644
---
a/fe/fe-core/src/main/java/org/apache/doris/resource/workloadgroup/WorkloadGroupMgr.java
+++
b/fe/fe-core/src/main/java/org/apache/doris/resource/workloadgroup/WorkloadGroupMgr.java
@@ -102,10 +102,7 @@ public class WorkloadGroupMgr implements Writable,
GsonPostProcessable {
}
public List<TPipelineWorkloadGroup> getWorkloadGroup(ConnectContext
context) throws UserException {
- String groupName = context.getSessionVariable().getWorkloadGroup();
- if (Strings.isNullOrEmpty(groupName)) {
- groupName =
Env.getCurrentEnv().getAuth().getWorkloadGroup(context.getQualifiedUser());
- }
+ String groupName = getWorkloadGroupName(context);
List<TPipelineWorkloadGroup> workloadGroups = Lists.newArrayList();
readLock();
try {
@@ -120,7 +117,8 @@ public class WorkloadGroupMgr implements Writable,
GsonPostProcessable {
return workloadGroups;
}
- public QueryQueue getWorkloadGroupQueryQueue(String groupName) throws
UserException {
+ public QueryQueue getWorkloadGroupQueryQueue(ConnectContext context)
throws UserException {
+ String groupName = getWorkloadGroupName(context);
readLock();
try {
WorkloadGroup workloadGroup = nameToWorkloadGroup.get(groupName);
@@ -133,6 +131,17 @@ public class WorkloadGroupMgr implements Writable,
GsonPostProcessable {
}
}
+ private String getWorkloadGroupName(ConnectContext context) {
+ String groupName = context.getSessionVariable().getWorkloadGroup();
+ if (Strings.isNullOrEmpty(groupName)) {
+ groupName =
Env.getCurrentEnv().getAuth().getWorkloadGroup(context.getQualifiedUser());
+ }
+ if (Strings.isNullOrEmpty(groupName)) {
+ groupName = DEFAULT_GROUP_NAME;
+ }
+ return groupName;
+ }
+
private void checkAndCreateDefaultGroup() {
WorkloadGroup defaultWorkloadGroup = null;
writeLock();
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscr...@doris.apache.org
For additional commands, e-mail: commits-h...@doris.apache.org
