hcq/oceanbase
hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

abort receiving packets if idle more than 10 seconds

Browse Source
This commit is contained in:
hnwyllmm 2024-02-10 03:47:42 +00:00 committed by ob-robot
parent 0bd3590519
commit 4dab055605
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/sql/engine/cmd/ob_load_data_file_reader.cpp
View File

@ -303,8 +303,16 @@ ObPacketStreamFileReader::~ObPacketStreamFileReader()
// normal SQL processor cannot handle the packets, so we
// eat all packets with file content.
timeout_ts_ = -1;
while (!eof_ && OB_SUCC(ret)) {
// We will wait at most 10 seconds if there is no more data come in.
const int64_t wait_timeout = 10 * 1000000L; // seconds
int64_t wait_deadline = ObTimeUtility::current_time() + wait_timeout;
int64_t last_received_size = received_size_;
while (!eof_ && OB_SUCC(ret) && ObTimeUtility::current_time() <= wait_deadline) {
ret = receive_packet();
if (received_size_ > last_received_size) {
last_received_size = received_size_;
wait_deadline = ObTimeUtility::current_time() + wait_timeout;
}
}
arena_allocator_.reset();
}