ClickHouse/dbms/include/DB/IO/BufferWithOwnMemory.h

#pragma once

#include <boost/noncopyable.hpp>

#define DBMS_DEFAULT_BUFFER_SIZE 1048576ULL


namespace DB
{


namespace detail
{
	/** Замена std::vector<char> для использования в буферах.
	  * Отличается тем, что не делает лишний memset. (И почти ничего не делает.)
	  */
	struct Memory : boost::noncopyable
	{
		size_t m_capacity;
		size_t m_size;
		char * m_data;

		Memory() : m_capacity(0), m_size(0), m_data(NULL) {}
		Memory(size_t size_) : m_capacity(size_), m_size(m_capacity), m_data(size_ ? new char[m_capacity] : NULL) {}

		~Memory()
		{
			if (m_data)
			{
				delete[] m_data;
				m_data = NULL;
			}
		}

		size_t size() const { return m_size; }
		const char & operator[](size_t i) const { return m_data[i]; }
		char & operator[](size_t i) { return m_data[i]; }

		void resize(size_t new_size)
		{
			if (new_size < m_capacity)
			{
				m_size = new_size;
				return;
			}
			else
			{
				if (m_data)
					delete[] m_data;

				m_capacity = new_size;
				m_size = m_capacity;
				m_data = new char[m_capacity];
			}
		}
	};
}


/** Буфер, который может сам владеть своим куском памяти для работы.
  * Аргумент шаблона - ReadBuffer или WriteBuffer
  */
template <typename Base>
class BufferWithOwnMemory : public Base
{
protected:
	detail::Memory memory;
public:
	/// Если передать не-NULL existing_memory, то буфер не будет создавать свой кусок памяти, а будет использовать существующий (и не будет им владеть).
	BufferWithOwnMemory(size_t size = DBMS_DEFAULT_BUFFER_SIZE, char * existing_memory = NULL) : Base(NULL, 0), memory(existing_memory ? 0 : size)
	{
		Base::set(existing_memory ? existing_memory : &memory[0], size);
	}
};


}