mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-23 02:00:49 +00:00
84 lines
2.2 KiB
C++
84 lines
2.2 KiB
C++
#pragma once
|
||
|
||
#include <Poco/Mutex.h>
|
||
|
||
#include <DB/Core/NamesAndTypes.h>
|
||
#include <DB/Storages/IStorage.h>
|
||
#include <DB/DataStreams/IProfilingBlockInputStream.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
class StorageMemory;
|
||
|
||
class MemoryBlockInputStream : public IProfilingBlockInputStream
|
||
{
|
||
public:
|
||
MemoryBlockInputStream(const Names & column_names_, BlocksList::iterator begin_, BlocksList::iterator end_);
|
||
String getName() const { return "MemoryBlockInputStream"; }
|
||
BlockInputStreamPtr clone() { return new MemoryBlockInputStream(column_names, begin, end); }
|
||
protected:
|
||
Block readImpl();
|
||
private:
|
||
Names column_names;
|
||
BlocksList::iterator begin;
|
||
BlocksList::iterator end;
|
||
BlocksList::iterator it;
|
||
};
|
||
|
||
|
||
class MemoryBlockOutputStream : public IBlockOutputStream
|
||
{
|
||
public:
|
||
MemoryBlockOutputStream(StorageMemory & storage_);
|
||
void write(const Block & block);
|
||
BlockOutputStreamPtr clone() { return new MemoryBlockOutputStream(storage); }
|
||
private:
|
||
StorageMemory & storage;
|
||
};
|
||
|
||
|
||
/** Реализует хранилище в оперативке.
|
||
* Подходит для временных данных.
|
||
* В нём не поддерживаются ключи.
|
||
* Данные хранятся в виде набора блоков и никуда дополнительно не сохраняются.
|
||
*/
|
||
class StorageMemory : public IStorage
|
||
{
|
||
friend class MemoryBlockInputStream;
|
||
friend class MemoryBlockOutputStream;
|
||
|
||
public:
|
||
StorageMemory(const std::string & name_, NamesAndTypesListPtr columns_);
|
||
|
||
std::string getName() const { return "Memory"; }
|
||
std::string getTableName() const { return name; }
|
||
|
||
const NamesAndTypesList & getColumnsList() const { return *columns; }
|
||
|
||
BlockInputStreams read(
|
||
const Names & column_names,
|
||
ASTPtr query,
|
||
QueryProcessingStage::Enum & processed_stage,
|
||
size_t max_block_size = DEFAULT_BLOCK_SIZE,
|
||
unsigned threads = 1);
|
||
|
||
BlockOutputStreamPtr write(
|
||
ASTPtr query);
|
||
|
||
void drop();
|
||
void rename(const String & new_path_to_db, const String & new_name) { name = new_name; }
|
||
|
||
private:
|
||
String name;
|
||
NamesAndTypesListPtr columns;
|
||
|
||
/// Сами данные. list - чтобы при вставке в конец, существующие итераторы не инвалидировались.
|
||
BlocksList data;
|
||
|
||
Poco::FastMutex mutex;
|
||
};
|
||
|
||
}
|