c++ - 扩展C++ ostream

我正在尝试通过扩展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吗？)但是，在std::streambuf中查看list of virtual protected functions时，我看不到应该覆盖任何处理单个字符插入的功能。

那么，我该怎么做呢？

最佳答案

单字符输出由overflow处理。如果overflow进行实际输出，则可以按照xsputn的方式实现xsputn:

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();
    }
}