mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 08:32:02 +00:00
51 lines
1.7 KiB
C++
51 lines
1.7 KiB
C++
#pragma once
|
|
|
|
#include <Storages/SelectQueryInfo.h>
|
|
|
|
#include <Interpreters/PreparedSets.h>
|
|
#include <Core/Field.h>
|
|
|
|
#include <IO/ReadBufferFromString.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
using FieldVectorPtr = std::shared_ptr<FieldVector>;
|
|
|
|
class IDataType;
|
|
using DataTypePtr = std::shared_ptr<const IDataType>;
|
|
|
|
/** Retrieve from the query a condition of the form `key = 'key'`, `key in ('xxx_'), from conjunctions in the WHERE clause.
|
|
* TODO support key like search
|
|
*/
|
|
std::pair<FieldVectorPtr, bool> getFilterKeys(
|
|
const std::string & primary_key, const DataTypePtr & primary_key_type, const SelectQueryInfo & query_info, const ContextPtr & context);
|
|
|
|
std::pair<FieldVectorPtr, bool> getFilterKeys(
|
|
const String & primary_key, const DataTypePtr & primary_key_type, const ActionDAGNodes & filter_nodes, const ContextPtr & context);
|
|
|
|
template <typename K, typename V>
|
|
void fillColumns(const K & key, const V & value, size_t key_pos, const Block & header, MutableColumns & columns)
|
|
{
|
|
ReadBufferFromString key_buffer(key);
|
|
ReadBufferFromString value_buffer(value);
|
|
for (size_t i = 0; i < header.columns(); ++i)
|
|
{
|
|
const auto & serialization = header.getByPosition(i).type->getDefaultSerialization();
|
|
serialization->deserializeBinary(*columns[i], i == key_pos ? key_buffer : value_buffer, {});
|
|
}
|
|
}
|
|
|
|
std::vector<std::string> serializeKeysToRawString(
|
|
FieldVector::const_iterator & it,
|
|
FieldVector::const_iterator end,
|
|
DataTypePtr key_column_type,
|
|
size_t max_block_size);
|
|
|
|
std::vector<std::string> serializeKeysToRawString(const ColumnWithTypeAndName & keys);
|
|
|
|
/// In current implementation key with only column is supported.
|
|
size_t getPrimaryKeyPos(const Block & header, const Names & primary_key);
|
|
|
|
}
|