thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-12-16 19:32:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix escape function

Browse Source
This commit is contained in:
Arthur Passos 2022-12-20 07:38:49 -03:00
parent 51e71bd10e
commit 1104f218aa
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/Functions/extractKeyValuePairs.cpp
View File

@ -210,17 +210,19 @@ ExtractKeyValuePairs::RawColumns ExtractKeyValuePairs::extract(std::shared_ptr<K
return {std::move(keys), std::move(values), std::move(offsets)};
}
ColumnPtr ExtractKeyValuePairs::escape(RawColumns & raw_columns)
ColumnPtr ExtractKeyValuePairs::escape(RawColumns & raw_columns, char escape_character)
{
auto & [raw_keys, raw_values, offsets] = raw_columns;
auto escaped_keys = ColumnString::create();
auto escaped_values = ColumnString::create();
auto escape_character_string = std::string('1', escape_character);
using ReplaceString = ReplaceStringImpl<ReplaceStringTraits::Replace::All>;
ReplaceString::vector(raw_keys->getChars(), raw_keys->getOffsets(), "\\", "", escaped_keys->getChars(), escaped_keys->getOffsets());
ReplaceString::vector(raw_values->getChars(), raw_values->getOffsets(), "\\", "", escaped_values->getChars(), escaped_values->getOffsets());
ReplaceString::vector(raw_keys->getChars(), raw_keys->getOffsets(), escape_character_string, "", escaped_keys->getChars(), escaped_keys->getOffsets());
ReplaceString::vector(raw_values->getChars(), raw_values->getOffsets(), escape_character_string, "", escaped_values->getChars(), escaped_values->getOffsets());
ColumnPtr keys_ptr = std::move(escaped_keys);

2
src/Functions/extractKeyValuePairs.h
View File

@ -72,7 +72,7 @@ private:
static RawColumns extract(std::shared_ptr<KeyValuePairExtractor<EscapingProcessorOutput>> extractor, ColumnPtr data_column);
static ColumnPtr escape(RawColumns & raw_columns);
static ColumnPtr escape(RawColumns & raw_columns, char escape_character);
ColumnNumbers getArgumentsThatAreAlwaysConstant() const override;
};