ClickHouse/dbms/include/DB/Core/Block.h

140 lines
5.1 KiB
C
Raw Normal View History

2011-08-09 15:57:33 +00:00
#pragma once
2010-03-01 16:59:51 +00:00
#include <vector>
#include <map>
#include <list>
2014-09-12 16:05:29 +00:00
#include <initializer_list>
2010-03-01 16:59:51 +00:00
#include <DB/Core/BlockInfo.h>
2010-03-04 19:20:28 +00:00
#include <DB/Core/ColumnWithNameAndType.h>
2011-11-06 02:29:13 +00:00
#include <DB/Core/NamesAndTypes.h>
2010-03-05 15:29:17 +00:00
#include <DB/Core/Exception.h>
#include <DB/Core/ErrorCodes.h>
#include "ColumnsWithNameAndType.h"
2010-03-01 16:59:51 +00:00
namespace DB
{
/** Тип данных для представления подмножества строк и столбцов в оперативке.
* Содержит также метаданные (типы) столбцов и их имена.
* Позволяет вставлять, удалять столбцы в любом порядке, менять порядок столбцов.
*/
class Context;
2010-03-01 16:59:51 +00:00
class Block
{
public:
typedef std::list<ColumnWithNameAndType> Container_t;
typedef std::vector<Container_t::iterator> IndexByPosition_t;
typedef std::map<String, Container_t::iterator> IndexByName_t;
2010-03-01 16:59:51 +00:00
private:
2010-03-04 19:20:28 +00:00
Container_t data;
2010-03-01 16:59:51 +00:00
IndexByPosition_t index_by_position;
IndexByName_t index_by_name;
public:
BlockInfo info;
2014-09-12 16:05:29 +00:00
Block() = default;
Block(std::initializer_list<ColumnWithNameAndType> il) : data{il}
{
index_by_position.reserve(il.size());
for (auto it = std::begin(data); it != std::end(data); ++it)
{
index_by_name[it->name] = it;
index_by_position.push_back(it);
}
}
2010-03-12 20:44:25 +00:00
/// нужны, чтобы правильно скопировались индексы
Block(const Block & other);
2014-01-08 16:23:31 +00:00
Block(Block && other) = default;
2010-03-12 20:44:25 +00:00
Block & operator= (const Block & other);
2014-01-08 16:23:31 +00:00
Block & operator= (Block && other) = default;
2010-03-12 20:44:25 +00:00
2010-03-05 15:29:17 +00:00
/// вставить столбец в заданную позицию
2010-03-01 16:59:51 +00:00
void insert(size_t position, const ColumnWithNameAndType & elem);
2010-03-05 15:29:17 +00:00
/// вставить столбец в конец
void insert(const ColumnWithNameAndType & elem);
void insertDefault(const String & name, const DataTypePtr & type);
2011-09-04 05:14:52 +00:00
/// вставить столбец в конец, если столбца с таким именем ещё нет
void insertUnique(const ColumnWithNameAndType & elem);
2010-03-05 15:29:17 +00:00
/// удалить столбец в заданной позиции
2010-03-01 16:59:51 +00:00
void erase(size_t position);
/// удалить столбец с заданным именем
void erase(const String & name);
/// Добавляет в блок недостающие столбцы со значениями по-умолчанию
2014-10-16 16:55:23 +00:00
void addDefaults(const NamesAndTypesList & required_columns);
2010-03-01 16:59:51 +00:00
ColumnWithNameAndType & getByPosition(size_t position);
const ColumnWithNameAndType & getByPosition(size_t position) const;
ColumnWithNameAndType & unsafeGetByPosition(size_t position) { return *index_by_position[position]; }
const ColumnWithNameAndType & unsafeGetByPosition(size_t position) const { return *index_by_position[position]; }
2010-03-01 16:59:51 +00:00
ColumnWithNameAndType & getByName(const std::string & name);
const ColumnWithNameAndType & getByName(const std::string & name) const;
2011-12-12 06:15:34 +00:00
bool has(const std::string & name) const;
2011-08-12 20:39:42 +00:00
size_t getPositionByName(const std::string & name) const;
ColumnsWithNameAndType getColumns() const;
2011-11-06 02:29:13 +00:00
NamesAndTypesList getColumnsList() const;
2010-03-04 19:20:28 +00:00
/** Возвращает количество строк в блоке.
2013-06-08 20:19:29 +00:00
* Заодно проверяет, что все столбцы содержат одинаковое число значений.
2010-03-04 19:20:28 +00:00
*/
size_t rows() const;
2013-06-08 20:19:29 +00:00
/** То же самое, но без проверки - берёт количество строк из первого столбца, если он есть или возвращает 0.
*/
size_t rowsInFirstColumn() const;
size_t columns() const { return index_by_position.size(); }
2010-03-04 19:20:28 +00:00
2012-05-17 19:15:53 +00:00
/// Приблизительное количество байт в оперативке - для профайлинга.
size_t bytes() const;
operator bool() const { return !index_by_position.empty(); }
bool operator!() const { return index_by_position.empty(); }
2011-08-14 00:49:30 +00:00
/** Получить список имён столбцов через запятую. */
std::string dumpNames() const;
2011-09-19 03:34:23 +00:00
/** Список имен, типов и длин столбцов. Предназначен для отладки. */
std::string dumpStructure() const;
2011-09-19 03:34:23 +00:00
/** Получить такой же блок, но пустой. */
Block cloneEmpty() const;
/** Заменяет столбцы смещений внутри вложенных таблиц на один общий для таблицы.
* Кидает исключение, если эти смещения вдруг оказались неодинаковы.
*/
void optimizeNestedArraysOffsets();
/** Тоже самое, только без замены смещений. */
void checkNestedArraysOffsets() const;
void clear();
void swap(Block & other);
2010-03-01 16:59:51 +00:00
};
2012-07-17 20:04:39 +00:00
typedef std::vector<Block> Blocks;
typedef std::list<Block> BlocksList;
2012-07-17 20:04:39 +00:00
/// Сравнить типы столбцов у блоков. Порядок столбцов имеет значение. Имена не имеют значения.
bool blocksHaveEqualStructure(const Block & lhs, const Block & rhs);
2010-03-01 16:59:51 +00:00
}
2014-12-04 15:50:48 +00:00
namespace std
{
template<> inline void swap<DB::Block>(DB::Block & one, DB::Block & another)
2014-12-04 15:50:48 +00:00
{
one.swap(another);
}
}