c++ - 在 std::ostream 输出后添加一个新行而不显式调用它

Question

下面的例子最能说明我的问题:

class Stream
{
public:
    Stream(std::ostream& os)
        :
        stream(os)
    {
    }

    auto getStream()
    {
        return std::pair<std::ostream&, std::unique_ptr<StreamDelegate>>(stream, std::make_unique<StreamDelegate>(stream));
    }

private:
    std::ostream& stream;
};  

int main()
{
    Stream os(std::cout);
    os.getStream() << "input1" << "input2"; 
//execute some code without explicitly calling it 
//such os.getStream() << std::endl; 
}

我(工作)尝试实现此功能:

class StreamDelegate
{
public:
    StreamDelegate(std::ostream& os)
        :
        stream(os)
    {
    }
    ~StreamDelegate()
    {
        //some delegated functionality
        //exmpl:
        stream << std::endl;
    }

private:
    std::ostream& stream;
};

class Stream
{
public:
    Stream(std::ostream& os)
        :
        stream(os)
    {
    }

    auto getStream()
    {
        return std::pair<std::ostream&, std::unique_ptr<StreamDelegate>>(stream, std::make_unique<StreamDelegate>(stream));
    }

private:
    std::ostream& stream;
};

int main()
{
    Stream os(std::cout);
    os.getStream().first << "input1" << " input2";
    os.getStream().first << "line2: input1" << " line2: input 2";

    std::cin.get();
}

是否有其他更优雅的方法来实现此功能？

此外，我发现我的代码中可能存在缺陷。如下代码所示:

int main()
{
    Stream os(std::cout);
    auto pitfall = os.getStream();
    pitfall.first << "line1";
    pitfall.first << "should be line 2";

    std::cin.get();
}

由于 os.getStream() 的返回值被分配给一个变量，因此没有破坏 StreamDelegate，因此没有预期的行为。

我不打算将 Stream::getStream() 分配给一个变量，但如果有解决办法我想知道。

我还知道我可以通过以与 StreanDelegate 相同的方式实现 Stream 析构函数来避免使用 StreamDelegate，但我可能希望它能在单例上工作。

Answer 1

我认为您使用 RAII 的想法很好。如果你实现模板化 operator<<为你的 Stream并让它返回你的 StreamDelegate它变得更容易使用，并且更难通过分配给变量来搞乱破坏。

#include <iostream>

class StreamDelegate
{
public:
    StreamDelegate(std::ostream& os) : stream(os) {}

    ~StreamDelegate() {
        //some delegated functionality
        //exmpl:
        stream << std::endl;
    }

    template <typename T>
    StreamDelegate& operator<<(T&& val) {
        stream << std::forward<T>(val);
        return *this;
    }
    StreamDelegate& operator=(const StreamDelegate&) = delete;
    StreamDelegate(const StreamDelegate&) = delete;

private:
    std::ostream& stream;
};

class Stream
{
public:
    Stream(std::ostream& os) : stream(os) {}

    template <typename T>
    StreamDelegate operator<<(T&& val) {
        stream << std::forward<T>(val);
        return StreamDelegate(stream);
    }

private:
    std::ostream& stream;
};

int main()
{
    Stream os(std::cout);
    os << "input1" << " input2";
    os << "line2: input1" << " line2: input 2";

    std::cin.get();
}

编辑

我为 StreamDelegate 显式声明了复制构造函数和复制赋值。作为删除，所以不再可能做

auto delegate = os << "input3";