mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-04 21:42:39 +00:00
194 lines
4.9 KiB
C++
194 lines
4.9 KiB
C++
|
#pragma once
|
||
|
|
||
|
#include <Analyzer/IQueryTreeNode.h>
|
||
|
#include <Analyzer/ListNode.h>
|
||
|
|
||
|
#include <Interpreters/WindowDescription.h>
|
||
|
|
||
|
namespace DB
|
||
|
{
|
||
|
|
||
|
/** Window node represents window function window description.
|
||
|
*
|
||
|
* Example: SELECT * FROM test_table WINDOW window AS (PARTITION BY id);
|
||
|
* window AS (PARTITION BY id) - window node.
|
||
|
*
|
||
|
* Example: SELECT count() OVER (PARTITION BY id) FROM test_table;
|
||
|
* PARTITION BY id - window node.
|
||
|
*
|
||
|
* Window node can also refer to its parent window node.
|
||
|
* Example: SELECT count() OVER (parent_window ORDER BY id) FROM test_table WINDOW parent_window AS (PARTITION BY id);
|
||
|
* parent_window ORDER BY id - window node.
|
||
|
*
|
||
|
* Window node initially initialized with window frame.
|
||
|
*
|
||
|
* If window frame has OFFSET begin type, additionally frame begin offset node must be initialized.
|
||
|
* If window frame has OFFSET end type, additionally frame end offset node must be initialized.
|
||
|
* During query analysis pass they must be resolved, validated and window node window frame offset constants must be updated.
|
||
|
*/
|
||
|
class WindowNode;
|
||
|
using WindowNodePtr = std::shared_ptr<WindowNode>;
|
||
|
|
||
|
class WindowNode final : public IQueryTreeNode
|
||
|
{
|
||
|
public:
|
||
|
/// Initialize window node with window frame
|
||
|
explicit WindowNode(WindowFrame window_frame_);
|
||
|
|
||
|
/// Get window node window frame
|
||
|
const WindowFrame & getWindowFrame() const
|
||
|
{
|
||
|
return window_frame;
|
||
|
}
|
||
|
|
||
|
/// Get window node window frame
|
||
|
WindowFrame & getWindowFrame()
|
||
|
{
|
||
|
return window_frame;
|
||
|
}
|
||
|
|
||
|
/// Has parent window name
|
||
|
bool hasParentWindowName() const
|
||
|
{
|
||
|
return parent_window_name.empty();
|
||
|
}
|
||
|
|
||
|
/// Get parent window name
|
||
|
const String & getParentWindowName() const
|
||
|
{
|
||
|
return parent_window_name;
|
||
|
}
|
||
|
|
||
|
/// Set parent window name
|
||
|
void setParentWindowName(String parent_window_name_value)
|
||
|
{
|
||
|
parent_window_name = std::move(parent_window_name_value);
|
||
|
}
|
||
|
|
||
|
/// Has order by
|
||
|
bool hasOrderBy() const
|
||
|
{
|
||
|
return !getOrderBy().getNodes().empty();
|
||
|
}
|
||
|
|
||
|
/// Get order by
|
||
|
const ListNode & getOrderBy() const
|
||
|
{
|
||
|
return children[order_by_child_index]->as<const ListNode &>();
|
||
|
}
|
||
|
|
||
|
/// Get order by
|
||
|
ListNode & getOrderBy()
|
||
|
{
|
||
|
return children[order_by_child_index]->as<ListNode &>();
|
||
|
}
|
||
|
|
||
|
/// Get order by
|
||
|
const QueryTreeNodePtr & getOrderByNode() const
|
||
|
{
|
||
|
return children[order_by_child_index];
|
||
|
}
|
||
|
|
||
|
/// Get order by
|
||
|
QueryTreeNodePtr & getOrderByNode()
|
||
|
{
|
||
|
return children[order_by_child_index];
|
||
|
}
|
||
|
|
||
|
/// Has partition by
|
||
|
bool hasPartitionBy() const
|
||
|
{
|
||
|
return !getPartitionBy().getNodes().empty();
|
||
|
}
|
||
|
|
||
|
/// Get partition by
|
||
|
const ListNode & getPartitionBy() const
|
||
|
{
|
||
|
return children[partition_by_child_index]->as<const ListNode &>();
|
||
|
}
|
||
|
|
||
|
/// Get partition by
|
||
|
ListNode & getPartitionBy()
|
||
|
{
|
||
|
return children[partition_by_child_index]->as<ListNode &>();
|
||
|
}
|
||
|
|
||
|
/// Get partition by
|
||
|
const QueryTreeNodePtr & getPartitionByNode() const
|
||
|
{
|
||
|
return children[partition_by_child_index];
|
||
|
}
|
||
|
|
||
|
/// Get partition by
|
||
|
QueryTreeNodePtr & getPartitionByNode()
|
||
|
{
|
||
|
return children[partition_by_child_index];
|
||
|
}
|
||
|
|
||
|
/// Has frame begin offset
|
||
|
bool hasFrameBeginOffset() const
|
||
|
{
|
||
|
return getFrameBeginOffsetNode() != nullptr;
|
||
|
}
|
||
|
|
||
|
/// Get FRAME begin offset node
|
||
|
const QueryTreeNodePtr & getFrameBeginOffsetNode() const
|
||
|
{
|
||
|
return children[frame_begin_offset_child_index];
|
||
|
}
|
||
|
|
||
|
/// Get FRAME begin offset node
|
||
|
QueryTreeNodePtr & getFrameBeginOffsetNode()
|
||
|
{
|
||
|
return children[frame_begin_offset_child_index];
|
||
|
}
|
||
|
|
||
|
/// Has frame end offset
|
||
|
bool hasFrameEndOffset() const
|
||
|
{
|
||
|
return getFrameEndOffsetNode() != nullptr;
|
||
|
}
|
||
|
|
||
|
/// Get FRAME begin offset node
|
||
|
const QueryTreeNodePtr & getFrameEndOffsetNode() const
|
||
|
{
|
||
|
return children[frame_end_offset_child_index];
|
||
|
}
|
||
|
|
||
|
/// Get FRAME begin offset node
|
||
|
QueryTreeNodePtr & getFrameEndOffsetNode()
|
||
|
{
|
||
|
return children[frame_end_offset_child_index];
|
||
|
}
|
||
|
|
||
|
QueryTreeNodeType getNodeType() const override
|
||
|
{
|
||
|
return QueryTreeNodeType::WINDOW;
|
||
|
}
|
||
|
|
||
|
String getName() const override;
|
||
|
|
||
|
void dumpTreeImpl(WriteBuffer & buffer, FormatState & format_state, size_t indent) const override;
|
||
|
|
||
|
bool isEqualImpl(const IQueryTreeNode & rhs) const override;
|
||
|
|
||
|
void updateTreeHashImpl(HashState & hash_state) const override;
|
||
|
|
||
|
protected:
|
||
|
ASTPtr toASTImpl() const override;
|
||
|
|
||
|
QueryTreeNodePtr cloneImpl() const override;
|
||
|
|
||
|
private:
|
||
|
static constexpr size_t order_by_child_index = 0;
|
||
|
static constexpr size_t partition_by_child_index = 1;
|
||
|
static constexpr size_t frame_begin_offset_child_index = 3;
|
||
|
static constexpr size_t frame_end_offset_child_index = 4;
|
||
|
static constexpr size_t children_size = frame_end_offset_child_index + 1;
|
||
|
|
||
|
WindowFrame window_frame;
|
||
|
String parent_window_name;
|
||
|
};
|
||
|
|
||
|
}
|