在类中包装ostream并模板化

Question

我想把C流包装在一个模板类中，这样所有

你能改变下面的代码，使它编译时不会改变整体意图吗？

#include <iostream>
class Foo
{
private:

    std::ostream& os;

public:

    explicit Foo( std::ostream& os ) : os( os ) {};

    template<class T>
    Foo& operator<<( const T& t )
    {
        os << t << '!';
        return *this;
    }

};

int main()
{

    Foo( std::cout ) << "test" << '\n'; // works fine 
    Foo( std::cout ) << "test" << std::endl; // compilation error

    return 0;
}

我假设std：：endl具有某种类型，因此被模板化方法“捕获”。

Answer 1

你也许应该用SFINAE来保护它，并完美地转发这个论点。

    template<class T> auto operator<<(T &&t)
         -> decltype(os << std::forward<T>(t) << '!', *this)
    {
            return os << std::forward<T>(t) << '!', *this;
    }

现在，这应该接受每个参数 std：：ostream 可以打印并且无法与其他所有内容一起编译。

Answer 2

正如kmdreko所建议的，您需要为模板函数添加一个重载

（std::endl是模板函数，不是类或简单类型）

像这样:

#include <iostream>
class Foo
{
private:

    std::ostream& os;

public:

    explicit Foo(std::ostream& os) : os(os) {};

    // for classes
    template<class T>
    Foo& operator<<(const T& t)
    {
        os << t << '!';
        return *this;
    }

    // for endl
    Foo& operator<<(std::ostream& (*pf) (std::ostream&)) {
        os << pf;
        return *this;
    }
};