善用 C++ istream 的 eof() 和 fail()

當函式需要讀入資料做處理時, 讓函式接收一個 std::istream* 會比直接呼叫 std::cin 好一些, 必要時可以改傳 std::istringstream 提供更多彈性, 比方說方便作測試。

不過實測的結果, std::cin 和 std::istringstream 的行為略有不同。std::cin 讀到 EOF 後, 呼叫 eof() 才會回傳 true, 但是 std::istringstream 在讀完最後一個 byte 後, eof() 就回傳 true 了。

以下是一個小範例, 可以正確處理兩者, 關鍵是讀完資料後用 fail() 檢查剛才的輸入是否有效。

#include <iostream>
#include <sstream>

void foo(std::istream* s)
{
  int n;
  while (!s->eof()) {
    *s >> n;
    if (!s->fail()) {
      std::cout << n << std::endl;
    }
  }
}

int main(void)
{
  std::istringstream ss("1 2 3 4 5");
  foo(&ss);
  foo(&std::cin);
  return 0;
}

另外附上別人截錄的官方說明。

留言

這個網誌中的熱門文章

virtualbox 使用 USB 裝置

熟悉系統工具好處多多

如何 git merge 更改檔名的檔案