ydb/util/stream/holder.h

#pragma once

#include <util/generic/ptr.h>

#include <utility>
#include <type_traits>

class IInputStream;
class IOutputStream;

namespace NPrivate {
    template <class Stream, bool isInput = std::is_base_of<IInputStream, Stream>::value>
    struct TStreamBase {
        using TType = IInputStream;
    };

    template <class Stream>
    struct TStreamBase<Stream, false> {
        using TType = IOutputStream;
    };

} // namespace NPrivate

/**
 * An ownership-gaining wrapper for proxy streams.
 *
 * Example usage:
 * \code
 * TCountingInput* input = new THoldingStream<TCountingInput>(new TStringInput(s));
 * \encode
 *
 * In this example, resulting counting input also owns a string input that it
 * was constructed on top of.
 */
template <class Base, class StreamBase = typename ::NPrivate::TStreamBase<Base>::TType>
class THoldingStream: private THolder<StreamBase>, public Base {
public:
    template <class... Args>
    inline THoldingStream(THolder<StreamBase> stream, Args&&... args)
        : THolder<StreamBase>(std::move(stream))
        , Base(this->Get(), std::forward<Args>(args)...)
    {
    }
};