thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-11-21 23:21:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed function if of FixedString arguments [#CLICKHOUSE-3202].

Browse Source
This commit is contained in:
Alexey Milovidov 2017-08-05 05:13:38 +03:00 committed by alexey-milovidov
parent 03662e3d77
commit 49bdf266b0
1 changed files with 111 additions and 70 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

181
dbms/src/Functions/GatherUtils.h
View File

@ -184,39 +184,6 @@ struct ConstSource
};
template <typename T>
struct NumericArraySink
{
typename ColumnVector<T>::Container_t & elements;
typename ColumnArray::Offsets_t & offsets;
size_t row_num = 0;
ColumnArray::Offset_t current_offset = 0;
NumericArraySink(ColumnArray & arr, size_t column_size)
: elements(typeid_cast<ColumnVector<T> &>(arr.getData()).getData()), offsets(arr.getOffsets())
{
offsets.resize(column_size);
}
void next()
{
offsets[row_num] = current_offset;
++row_num;
}
bool isEnd() const
{
return row_num == offsets.size();
}
size_t rowNum() const
{
return row_num;
}
};
struct StringSource
{
using Slice = NumericArraySlice<UInt8>;
@ -249,6 +216,11 @@ struct StringSource
return row_num;
}
size_t getSizeForReserve() const
{
return elements.size();
}
Slice getWhole() const
{
return {&elements[prev_offset], offsets[row_num] - prev_offset - 1};
@ -288,40 +260,6 @@ struct StringSource
};
struct StringSink
{
typename ColumnString::Chars_t & elements;
typename ColumnString::Offsets_t & offsets;
size_t row_num = 0;
ColumnString::Offset_t current_offset = 0;
StringSink(ColumnString & col, size_t column_size)
: elements(col.getChars()), offsets(col.getOffsets())
{
offsets.resize(column_size);
}
void ALWAYS_INLINE next()
{
elements.push_back(0);
++current_offset;
offsets[row_num] = current_offset;
++row_num;
}
bool isEnd() const
{
return row_num == offsets.size();
}
size_t rowNum() const
{
return row_num;
}
};
struct FixedStringSource
{
using Slice = NumericArraySlice<UInt8>;
@ -356,6 +294,11 @@ struct FixedStringSource
return row_num;
}
size_t getSizeForReserve() const
{
return end - pos;
}
Slice getWhole() const
{
return {pos, string_size};
@ -391,6 +334,83 @@ struct FixedStringSource
};
template <typename T>
struct NumericArraySink
{
typename ColumnVector<T>::Container_t & elements;
typename ColumnArray::Offsets_t & offsets;
size_t row_num = 0;
ColumnArray::Offset_t current_offset = 0;
NumericArraySink(ColumnArray & arr, size_t column_size)
: elements(typeid_cast<ColumnVector<T> &>(arr.getData()).getData()), offsets(arr.getOffsets())
{
offsets.resize(column_size);
}
void next()
{
offsets[row_num] = current_offset;
++row_num;
}
bool isEnd() const
{
return row_num == offsets.size();
}
size_t rowNum() const
{
return row_num;
}
void reserve(size_t num_elements)
{
elements.reserve(num_elements);
}
};
struct StringSink
{
typename ColumnString::Chars_t & elements;
typename ColumnString::Offsets_t & offsets;
size_t row_num = 0;
ColumnString::Offset_t current_offset = 0;
StringSink(ColumnString & col, size_t column_size)
: elements(col.getChars()), offsets(col.getOffsets())
{
offsets.resize(column_size);
}
void ALWAYS_INLINE next()
{
elements.push_back(0);
++current_offset;
offsets[row_num] = current_offset;
++row_num;
}
bool isEnd() const
{
return row_num == offsets.size();
}
size_t rowNum() const
{
return row_num;
}
void reserve(size_t num_elements)
{
elements.reserve(num_elements);
}
};
struct FixedStringSink
{
typename ColumnString::Chars_t & elements;
@ -421,6 +441,11 @@ struct FixedStringSink
{
return row_num;
}
void reserve(size_t num_elements)
{
elements.reserve(num_elements);
}
};
@ -430,6 +455,7 @@ struct IStringSource
virtual void next() = 0;
virtual bool isEnd() const = 0;
virtual size_t getSizeForReserve() const = 0;
virtual Slice getWhole() const = 0;
virtual ~IStringSource() {}
};
@ -441,9 +467,10 @@ struct DynamicStringSource final : IStringSource
DynamicStringSource(const IColumn & col) : impl(static_cast<const typename Impl::Column &>(col)) {}
void next() override { impl.next(); };
bool isEnd() const override { return impl.isEnd(); };
Slice getWhole() const override { return impl.getWhole(); };
void next() override { impl.next(); }
bool isEnd() const override { return impl.isEnd(); }
size_t getSizeForReserve() const override { return impl.getSizeForReserve(); }
Slice getWhole() const override { return impl.getWhole(); }
};
inline std::unique_ptr<IStringSource> createDynamicStringSource(const IColumn & col)
@ -502,6 +529,11 @@ struct GenericArraySource
return row_num;
}
size_t getSizeForReserve() const
{
return elements.size();
}
Slice getWhole() const
{
return {&elements, prev_offset, offsets[row_num] - prev_offset};
@ -569,6 +601,11 @@ struct GenericArraySink
{
return row_num;
}
void reserve(size_t num_elements)
{
elements.reserve(num_elements);
}
};
@ -618,6 +655,8 @@ inline ALWAYS_INLINE void writeSlice(const GenericArraySlice & slice, GenericArr
template <typename SourceA, typename SourceB, typename Sink>
void NO_INLINE concat(SourceA && src_a, SourceB && src_b, Sink && sink)
{
sink.reserve(src_a.getSizeForReserve() + src_b.getSizeForReserve());
while (!src_a.isEnd())
{
writeSlice(src_a.getWhole(), sink);
@ -743,6 +782,8 @@ void NO_INLINE sliceDynamicOffsetBounded(Source && src, Sink && sink, IColumn &
template <typename SourceA, typename SourceB, typename Sink>
void NO_INLINE conditional(SourceA && src_a, SourceB && src_b, Sink && sink, const PaddedPODArray<UInt8> & condition)
{
sink.reserve(std::max(src_a.getSizeForReserve(), src_b.getSizeForReserve()));
const UInt8 * cond_pos = &condition[0];
const UInt8 * cond_end = cond_pos + condition.size();