mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-17 03:42:48 +00:00
32ed192434
Before it always uses the first column from sample block, while it is not guaranteed. Signed-off-by: Azat Khuzhin <a.khuzhin@semrush.com>
49 lines
1.5 KiB
C++
49 lines
1.5 KiB
C++
#pragma once
|
|
|
|
#include <Analyzer/IQueryTreeNode.h>
|
|
#include <Interpreters/InDepthNodeVisitor.h>
|
|
#include <Interpreters/Cluster.h>
|
|
|
|
namespace DB
|
|
{
|
|
|
|
class ExpressionActions;
|
|
using ExpressionActionsPtr = std::shared_ptr<ExpressionActions>;
|
|
|
|
class ASTFunction;
|
|
|
|
/// Rewrite `sharding_key IN (...)` for specific shard,
|
|
/// so that it will contain only values that belong to this specific shard.
|
|
///
|
|
/// See also:
|
|
/// - evaluateExpressionOverConstantCondition()
|
|
/// - StorageDistributed::createSelector()
|
|
/// - createBlockSelector()
|
|
struct OptimizeShardingKeyRewriteInMatcher
|
|
{
|
|
/// Cluster::SlotToShard
|
|
using SlotToShard = std::vector<UInt64>;
|
|
|
|
struct Data
|
|
{
|
|
/// Expression of sharding_key for the Distributed() table
|
|
const ExpressionActionsPtr & sharding_key_expr;
|
|
/// Name of the column for sharding_expr
|
|
const std::string & sharding_key_column_name;
|
|
/// Info for the current shard (to compare shard_num with calculated)
|
|
const Cluster::ShardInfo & shard_info;
|
|
/// weight -> shard mapping
|
|
const Cluster::SlotToShard & slots;
|
|
};
|
|
|
|
static bool needChildVisit(ASTPtr & /*node*/, const ASTPtr & /*child*/);
|
|
static void visit(ASTPtr & node, Data & data);
|
|
static void visit(ASTFunction & function, Data & data);
|
|
};
|
|
|
|
using OptimizeShardingKeyRewriteInVisitor = InDepthNodeVisitor<OptimizeShardingKeyRewriteInMatcher, true>;
|
|
|
|
void optimizeShardingKeyRewriteIn(QueryTreeNodePtr & node, OptimizeShardingKeyRewriteInVisitor::Data data, ContextPtr context);
|
|
|
|
}
|