擴展C++ ostream

Question

我想通過擴展std::streambuf來了解更多關於C++ I/O流庫的工作原理。作爲一個學習實驗，我的目標是簡單地創建一個自定義流，將所有輸出指向std::cerr。它似乎很簡單：

#include <iostream> 
using namespace std; 

class my_ostreambuf : public std::streambuf 
{ 
    public: 

    protected: 

    std::streamsize xsputn(const char * s, std::streamsize n) 
    { 
     std::cerr << "Redirecting to cerr: " << s << std::endl; 
     return n; 
    } 

}; 

int main() 
{ 
    my_ostreambuf buf; 
    std::ostream os(&buf); 
    os << "TEST"; 
} 

這似乎工作，因爲它打印Redirecting to cerr: TEST。問題是它不起作用當一個單個字符（而不是一個字符串）通過std::ostream::sputc插入到流中時。例如：

int main() 
{ 
    my_ostreambuf buf; 
    std::ostream os(&buf); 
    os << "ABC"; // works 
    std::string s("TEST"); 
    std::copy(s.begin(), s.end(), std::ostreambuf_iterator<char>(os)); // DOESN'T WORK 
} 

我想這個問題是xsputn不處理單個字符插入。 （我猜sputc內部不會調用xsputn？）但是，看看list of virtual protected functionsstd::streambuf，我沒有看到任何我應該重寫的函數處理單個字符插入。

那麼，我該如何做到這一點？

Answer 1

單字符輸出由overflow處理。下面是你可能在xsputn如果xsputn沒有實際輸出方面實現overflow：

int_type overflow(int_type c = traits_type::eof()) 
{ 
    if (c == traits_type::eof()) 
     return traits_type::eof(); 
    else 
    { 
     char_type ch = traits_type::to_char_type(c); 
     return xsputn(&ch, 1) == 1 ? c : traits_type::eof(); 
    } 
}