ClickHouse/dbms/src/IO/WriteBufferFromString.h

#pragma once

#include <string>

#include <IO/WriteBuffer.h>

#define WRITE_BUFFER_FROM_STRING_INITIAL_SIZE_IF_EMPTY 32


namespace DB
{

/** Writes the data to a string.
  * Note: before using the resulting string, destroy this object.
  */
class WriteBufferFromString : public WriteBuffer
{
private:
    std::string & s;

    void nextImpl() override
    {
        size_t old_size = s.size();
        s.resize(old_size * 2);
        internal_buffer = Buffer(reinterpret_cast<Position>(&s[old_size]), reinterpret_cast<Position>(&*s.end()));
        working_buffer = internal_buffer;
    }

protected:
    void finish()
    {
        s.resize(count());
    }

public:
    WriteBufferFromString(std::string & s_)
        : WriteBuffer(reinterpret_cast<Position>(&s_[0]), s_.size()), s(s_)
    {
        if (s.empty())
        {
            s.resize(WRITE_BUFFER_FROM_STRING_INITIAL_SIZE_IF_EMPTY);
            set(reinterpret_cast<Position>(&s[0]), s.size());
        }
    }

    ~WriteBufferFromString() override
    {
        finish();
    }
};


class StringHolder {
protected:
    std::string ss;
};

/// Creates the string by itself and allows to get it.
class WriteBufferFromOwnString : public StringHolder, public WriteBufferFromString
{

public:
    WriteBufferFromOwnString() : WriteBufferFromString(ss) {}

    std::string & str()
    {
        finish();
        return ss;
    }
};

}