mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-22 09:40:49 +00:00
80 lines
2.6 KiB
C++
80 lines
2.6 KiB
C++
#pragma once
|
||
|
||
#include <DB/Columns/ColumnConst.h>
|
||
#include <DB/Interpreters/Set.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
/** Столбец, содержащий множество значений в секции IN.
|
||
* Ведёт себя как столбец-константа (так как множество одно, а не своё на каждую строку).
|
||
* Значение у этого столбца нестандартное, поэтому его невозможно получить через обычный интерфейс.
|
||
*/
|
||
class ColumnSet : public IColumn
|
||
{
|
||
public:
|
||
ColumnSet(size_t s_, SetPtr data_) : s(s_), data(data_) {}
|
||
|
||
std::string getName() const { return "ColumnSet"; }
|
||
ColumnPtr cloneEmpty() const { return new ColumnSet(0, NULL); }
|
||
size_t size() const { return s; }
|
||
Field operator[](size_t n) const { throw Exception("Cannot get value from ColumnSet", ErrorCodes::NOT_IMPLEMENTED); }
|
||
void get(size_t n, Field & res) const { throw Exception("Cannot get value from ColumnSet", ErrorCodes::NOT_IMPLEMENTED); };
|
||
void insert(const Field & x) { throw Exception("Cannot insert element into ColumnSet", ErrorCodes::NOT_IMPLEMENTED); }
|
||
void insertDefault() { ++s; }
|
||
size_t byteSize() const { return 0; }
|
||
int compareAt(size_t n, size_t m, const IColumn & rhs_) const { return 0; }
|
||
StringRef getDataAt(size_t n) const { throw Exception("Method getDataAt is not supported for " + getName(), ErrorCodes::NOT_IMPLEMENTED); }
|
||
void insertData(const char * pos, size_t length) { throw Exception("Method insertData is not supported for " + getName(), ErrorCodes::NOT_IMPLEMENTED); }
|
||
|
||
ColumnPtr cut(size_t start, size_t length) const
|
||
{
|
||
return new ColumnSet(length, data);
|
||
}
|
||
|
||
ColumnPtr filter(const Filter & filt) const
|
||
{
|
||
size_t new_size = 0;
|
||
for (Filter::const_iterator it = filt.begin(); it != filt.end(); ++it)
|
||
if (*it)
|
||
++new_size;
|
||
|
||
return new ColumnSet(new_size, data);
|
||
}
|
||
|
||
ColumnPtr permute(const Permutation & perm) const
|
||
{
|
||
if (s != perm.size())
|
||
throw Exception("Size of permutation doesn't match size of column.", ErrorCodes::SIZES_OF_COLUMNS_DOESNT_MATCH);
|
||
|
||
return new ColumnSet(s, data);
|
||
}
|
||
|
||
Permutation getPermutation() const
|
||
{
|
||
Permutation res(s);
|
||
for (size_t i = 0; i < s; ++i)
|
||
res[i] = i;
|
||
return res;
|
||
}
|
||
|
||
ColumnPtr replicate(const Offsets_t & offsets) const
|
||
{
|
||
if (s != offsets.size())
|
||
throw Exception("Size of offsets doesn't match size of column.", ErrorCodes::SIZES_OF_COLUMNS_DOESNT_MATCH);
|
||
|
||
return new ColumnSet(offsets.back(), data);
|
||
}
|
||
|
||
|
||
SetPtr & getData() { return data; }
|
||
const SetPtr & getData() const { return data; }
|
||
|
||
private:
|
||
size_t s;
|
||
SetPtr data;
|
||
};
|
||
|
||
}
|