[native] Allow single zero to be passed as X-Presto-Buffer-Remaining-…

…Bytes
prestodb · Mar 6, 2024 · ec92367 · ec92367
1 parent 96187ad
commit ec92367
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/presto-native-execution/presto_cpp/main/PrestoExchangeSource.cpp b/presto-native-execution/presto_cpp/main/PrestoExchangeSource.cpp
@@ -265,6 +265,10 @@ void PrestoExchangeSource::processDataResponse(
       protocol::PRESTO_BUFFER_REMAINING_BYTES_HEADER);
   if (!remainingBytesString.empty()) {
     folly::split(',', remainingBytesString, remainingBytes);
+    if (!remainingBytes.empty() && remainingBytes[0] == 0) {
+      VELOX_CHECK_EQ(remainingBytes.size(), 1);
+      remainingBytes.clear();
+    }
   }
 
   int64_t ackSequence =