apacheGH-35007: [C++] Fix reading stdin

resetius · Apr 10, 2023 · d308833 · d308833
1 parent 2e6c13d
commit d308833
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/cpp/src/arrow/io/stdio.cc b/cpp/src/arrow/io/stdio.cc
@@ -75,12 +75,9 @@ Result<int64_t> StdinStream::Tell() const { return pos_; }
 
 Result<int64_t> StdinStream::Read(int64_t nbytes, void* out) {
   std::cin.read(reinterpret_cast<char*>(out), nbytes);
-  if (std::cin) {
-    pos_ += nbytes;
-    return nbytes;
-  } else {
-    return 0;
-  }
+  nbytes = std::cin.gcount();
+  pos_ += nbytes;
+  return nbytes;
 }
 
 Result<std::shared_ptr<Buffer>> StdinStream::Read(int64_t nbytes) {