Stream.h

Go to the documentation of this file.

/*
 * Copyright (C) 1996-2023 The Squid Software Foundation and contributors
 *
 * Squid software is distributed under GPLv2+ license and includes
 * contributions from numerous individuals and organizations.
 * Please see the COPYING and CONTRIBUTORS files for details.
 */
 
#ifndef SQUID_SRC_SBUF_STREAM_H
#define SQUID_SRC_SBUF_STREAM_H
 
#include "base/PackableStream.h"
#include "sbuf/SBuf.h"
 
class SBufStream : public std::ostream
{
public:
    SBufStream(const SBuf &aBuf):
        std::ostream(nullptr), // initialize the parent; no stream buffer yet
        sink_(aBuf),
        streamBuffer_(sink_) // initialize the stream buffer
    {
        rdbuf(&streamBuffer_); // supply the now-initialized stream buffer
        clear(); // clear the badbit that a nullptr stream buffer has triggered
    }
 
    SBufStream(): SBufStream(SBuf()) {}
 
    SBuf buf() {
        flush();
        return sink_;
    }
 
    SBufStream& clearBuf() {
        flush();
        sink_.clear();
        return *this;
    }
 
private:
    SBuf sink_;
    AppendingStreamBuf<SBuf> streamBuffer_;
};
 
template <typename... Args>
inline
SBuf ToSBuf(Args&&... args)
{
    SBufStream out;
    (out << ... << args);
    return out.buf();
}
 
#endif /* SQUID_SRC_SBUF_STREAM_H */