ClickHouse/dbms/include/DB/Columns/ColumnString.h

98 lines
2.7 KiB
C
Raw Normal View History

2011-09-04 00:22:19 +00:00
#pragma once
2010-03-12 18:25:35 +00:00
2011-09-04 00:22:19 +00:00
#include <string.h>
2010-05-20 19:29:04 +00:00
2010-05-13 16:13:38 +00:00
#include <DB/Columns/ColumnArray.h>
#include <DB/Columns/ColumnsNumber.h>
2010-03-12 18:25:35 +00:00
namespace DB
{
2010-05-13 16:13:38 +00:00
/** Cтолбeц значений типа "строка".
* Отличается от массива UInt8 только получением элемента (в виде String, а не Array)
*/
class ColumnString : public ColumnArray
{
2010-05-20 19:29:04 +00:00
private:
ColumnUInt8::Container_t & char_data;
2010-05-13 16:13:38 +00:00
public:
/** Создать пустой столбец строк, с типом значений */
ColumnString()
2010-05-20 19:29:04 +00:00
: ColumnArray(new ColumnUInt8()),
char_data(dynamic_cast<ColumnUInt8 &>(*data).getData())
2010-05-13 16:13:38 +00:00
{
}
2010-05-21 19:52:50 +00:00
2011-08-28 00:31:30 +00:00
std::string getName() const { return "ColumnString"; }
2011-08-09 19:19:00 +00:00
ColumnPtr cloneEmpty() const
2010-05-21 19:52:50 +00:00
{
return new ColumnString;
}
2010-05-13 16:13:38 +00:00
Field operator[](size_t n) const
{
size_t offset = n == 0 ? 0 : offsets[n - 1];
2010-05-20 19:29:04 +00:00
size_t size = offsets[n] - offset - 1;
2010-05-13 16:13:38 +00:00
const char * s = reinterpret_cast<const char *>(&dynamic_cast<const ColumnUInt8 &>(*data).getData()[offset]);
return String(s, size);
}
2010-05-20 19:29:04 +00:00
void insert(const Field & x)
{
2010-05-24 16:52:58 +00:00
const String & s = boost::get<const String &>(x);
2010-05-20 19:29:04 +00:00
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));
}
2011-09-04 00:22:19 +00:00
int compareAt(size_t n, size_t m, const IColumn & rhs_) const
{
const ColumnString & rhs = static_cast<const ColumnString &>(rhs_);
2011-09-26 11:05:38 +00:00
/** Для производительности, строки сравниваются до первого нулевого байта.
* (если нулевой байт в середине строки, то то, что после него - игнорируется)
* Замечу, что завершающий нулевой байт всегда есть.
*/
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;
2011-09-04 00:22:19 +00:00
}
2010-05-13 16:13:38 +00:00
};
2010-03-12 18:25:35 +00:00
}