This is an automated email from the ASF dual-hosted git repository.
yiguolei 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 175d3f15343 [Feature](executor)Stream load support workload group
(#30763)
175d3f15343 is described below
commit 175d3f15343a9399a92840368542821d75a0d3a2
Author: wangbo <[email protected]>
AuthorDate: Sun Feb 4 19:52:19 2024 +0800
[Feature](executor)Stream load support workload group (#30763)
* Stream load support workload group
* skip mysql load
---
.../org/apache/doris/service/FrontendServiceImpl.java | 16 ++++++++++++++++
1 file changed, 16 insertions(+)
diff --git
a/fe/fe-core/src/main/java/org/apache/doris/service/FrontendServiceImpl.java
b/fe/fe-core/src/main/java/org/apache/doris/service/FrontendServiceImpl.java
index 2c5345cbaac..ca57b35c088 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/service/FrontendServiceImpl.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/service/FrontendServiceImpl.java
@@ -185,6 +185,7 @@ import org.apache.doris.thrift.TNodeInfo;
import org.apache.doris.thrift.TOlapTableIndexTablets;
import org.apache.doris.thrift.TOlapTablePartition;
import org.apache.doris.thrift.TPipelineFragmentParams;
+import org.apache.doris.thrift.TPipelineWorkloadGroup;
import org.apache.doris.thrift.TPrivilegeCtrl;
import org.apache.doris.thrift.TPrivilegeHier;
import org.apache.doris.thrift.TPrivilegeStatus;
@@ -1801,14 +1802,29 @@ public class FrontendServiceImpl implements
FrontendService.Iface {
ctx.setThreadLocalInfo();
try {
+ List<TPipelineWorkloadGroup> tWorkloadGroupList = null;
+ // mysql load request not carry user info, need fix it later.
+ boolean hasUserName = !StringUtils.isEmpty(ctx.getQualifiedUser());
+ if (Config.enable_workload_group && hasUserName) {
+ tWorkloadGroupList =
Env.getCurrentEnv().getWorkloadGroupMgr().getWorkloadGroup(ctx);
+ }
if (!Strings.isNullOrEmpty(request.getLoadSql())) {
httpStreamPutImpl(request, result, ctx);
+ if (tWorkloadGroupList != null && tWorkloadGroupList.size() >
0) {
+ result.params.setWorkloadGroups(tWorkloadGroupList);
+ }
return result;
} else {
if (Config.enable_pipeline_load) {
result.setPipelineParams(pipelineStreamLoadPutImpl(request));
+ if (tWorkloadGroupList != null &&
tWorkloadGroupList.size() > 0) {
+
result.pipeline_params.setWorkloadGroups(tWorkloadGroupList);
+ }
} else {
result.setParams(streamLoadPutImpl(request, result));
+ if (tWorkloadGroupList != null &&
tWorkloadGroupList.size() > 0) {
+ result.params.setWorkloadGroups(tWorkloadGroupList);
+ }
}
}
} catch (UserException e) {
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
