mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-27 18:12:02 +00:00
98 lines
2.7 KiB
C++
98 lines
2.7 KiB
C++
#pragma once
|
||
|
||
#include <string.h>
|
||
|
||
#include <DB/Columns/ColumnArray.h>
|
||
#include <DB/Columns/ColumnsNumber.h>
|
||
|
||
|
||
namespace DB
|
||
{
|
||
|
||
/** Cтолбeц значений типа "строка".
|
||
* Отличается от массива UInt8 только получением элемента (в виде String, а не Array)
|
||
*/
|
||
class ColumnString : public ColumnArray
|
||
{
|
||
private:
|
||
ColumnUInt8::Container_t & char_data;
|
||
|
||
public:
|
||
/** Создать пустой столбец строк, с типом значений */
|
||
ColumnString()
|
||
: ColumnArray(new ColumnUInt8()),
|
||
char_data(dynamic_cast<ColumnUInt8 &>(*data).getData())
|
||
{
|
||
}
|
||
|
||
std::string getName() const { return "ColumnString"; }
|
||
|
||
ColumnPtr cloneEmpty() const
|
||
{
|
||
return new ColumnString;
|
||
}
|
||
|
||
Field operator[](size_t n) const
|
||
{
|
||
size_t offset = n == 0 ? 0 : offsets[n - 1];
|
||
size_t size = offsets[n] - offset - 1;
|
||
const char * s = reinterpret_cast<const char *>(&dynamic_cast<const ColumnUInt8 &>(*data).getData()[offset]);
|
||
return String(s, size);
|
||
}
|
||
|
||
void insert(const Field & x)
|
||
{
|
||
const String & s = boost::get<const String &>(x);
|
||
size_t old_size = char_data.size();
|
||
size_t size_to_append = s.size() + 1;
|
||
char_data.resize(old_size + size_to_append);
|
||
memcpy(&char_data[old_size], s.c_str(), size_to_append);
|
||
offsets.push_back((offsets.size() == 0 ? 0 : offsets.back()) + size_to_append);
|
||
}
|
||
|
||
void insertDefault()
|
||
{
|
||
char_data.push_back(0);
|
||
offsets.push_back(offsets.size() == 0 ? 1 : (offsets.back() + 1));
|
||
}
|
||
|
||
int compareAt(size_t n, size_t m, const IColumn & rhs_) const
|
||
{
|
||
const ColumnString & rhs = static_cast<const ColumnString &>(rhs_);
|
||
|
||
/** Для производительности, строки сравниваются до первого нулевого байта.
|
||
* (если нулевой байт в середине строки, то то, что после него - игнорируется)
|
||
* Замечу, что завершающий нулевой байт всегда есть.
|
||
*/
|
||
return strcmp(
|
||
reinterpret_cast<const char *>(&char_data[offsetAt(n)]),
|
||
reinterpret_cast<const char *>(&rhs.char_data[rhs.offsetAt(m)]));
|
||
}
|
||
|
||
struct less
|
||
{
|
||
const ColumnString & parent;
|
||
less(const ColumnString & parent_) : parent(parent_) {}
|
||
bool operator()(size_t lhs, size_t rhs) const
|
||
{
|
||
return 0 > strcmp(
|
||
reinterpret_cast<const char *>(&parent.char_data[parent.offsetAt(lhs)]),
|
||
reinterpret_cast<const char *>(&parent.char_data[parent.offsetAt(rhs)]));
|
||
}
|
||
};
|
||
|
||
Permutation getPermutation() const
|
||
{
|
||
size_t s = offsets.size();
|
||
Permutation res(s);
|
||
for (size_t i = 0; i < s; ++i)
|
||
res[i] = i;
|
||
|
||
std::sort(res.begin(), res.end(), less(*this));
|
||
return res;
|
||
}
|
||
};
|
||
|
||
|
||
}
|