This is an automated email from the ASF dual-hosted git repository.

yiguolei pushed a commit to branch branch-2.1
in repository https://gitbox.apache.org/repos/asf/doris.git

commit 95593f8e19268bd62e4496f6502b3f28674f4de1
Author: Kaijie Chen <[email protected]>
AuthorDate: Wed May 8 21:37:11 2024 +0800

    [config](move-memtable) support HTTP header load_stream_per_node in stream 
load (#34444)
---
 be/src/http/action/stream_load.cpp | 4 ++++
 be/src/http/http_common.h          | 1 +
 2 files changed, 5 insertions(+)

diff --git a/be/src/http/action/stream_load.cpp 
b/be/src/http/action/stream_load.cpp
index 6ecb12ab806..a8768b83860 100644
--- a/be/src/http/action/stream_load.cpp
+++ b/be/src/http/action/stream_load.cpp
@@ -620,6 +620,10 @@ Status StreamLoadAction::_process_put(HttpRequest* 
http_req,
         bool value = iequal(http_req->header(HTTP_MEMTABLE_ON_SINKNODE), 
"true");
         request.__set_memtable_on_sink_node(value);
     }
+    if (!http_req->header(HTTP_LOAD_STREAM_PER_NODE).empty()) {
+        int value = std::stoi(http_req->header(HTTP_LOAD_STREAM_PER_NODE));
+        request.__set_stream_per_node(value);
+    }
     if (ctx->group_commit) {
         if (!http_req->header(HTTP_GROUP_COMMIT).empty()) {
             
request.__set_group_commit_mode(http_req->header(HTTP_GROUP_COMMIT));
diff --git a/be/src/http/http_common.h b/be/src/http/http_common.h
index 175b7290378..a64c872c424 100644
--- a/be/src/http/http_common.h
+++ b/be/src/http/http_common.h
@@ -64,6 +64,7 @@ static const std::string HTTP_TWO_PHASE_COMMIT = 
"two_phase_commit";
 static const std::string HTTP_TXN_ID_KEY = "txn_id";
 static const std::string HTTP_TXN_OPERATION_KEY = "txn_operation";
 static const std::string HTTP_MEMTABLE_ON_SINKNODE = "memtable_on_sink_node";
+static const std::string HTTP_LOAD_STREAM_PER_NODE = "load_stream_per_node";
 static const std::string HTTP_WAL_ID_KY = "wal_id";
 static const std::string HTTP_AUTH_CODE = "auth_code";
 static const std::string HTTP_GROUP_COMMIT = "group_commit";


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to