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

77 lines
1.5 KiB
C
Raw Normal View History

#pragma once
2012-06-22 16:54:51 +00:00
#include <boost/noncopyable.hpp>
#define DBMS_DEFAULT_BUFFER_SIZE 1048576ULL
namespace DB
{
2012-06-21 18:43:29 +00:00
namespace detail
{
/** Замена std::vector<char> для использования в буферах.
* Отличается тем, что не делает лишний memset. (И почти ничего не делает.)
*/
2012-06-22 16:54:51 +00:00
struct Memory : boost::noncopyable
2012-06-21 18:43:29 +00:00
{
2012-06-21 19:04:52 +00:00
size_t m_capacity;
2012-06-21 18:43:29 +00:00
size_t m_size;
char * m_data;
2012-06-21 19:04:52 +00:00
Memory() : m_capacity(0), m_size(0), m_data(NULL) {}
Memory(size_t size_) : m_capacity(size_), m_size(m_capacity), m_data(new char[m_capacity]) {}
2012-06-21 18:43:29 +00:00
~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)
{
2012-06-21 19:04:52 +00:00
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];
}
2012-06-21 18:43:29 +00:00
}
};
}
/** Буфер, который сам владеет своим куском памяти для работы.
* Аргумент шаблона - ReadBuffer или WriteBuffer
*/
template <typename Base>
class BufferWithOwnMemory : public Base
{
protected:
2012-06-21 18:43:29 +00:00
detail::Memory memory;
public:
BufferWithOwnMemory(size_t size = DBMS_DEFAULT_BUFFER_SIZE) : Base(NULL, 0), memory(size)
{
Base::set(&memory[0], size);
}
};
}