ClickHouse/src/Interpreters/ArrayJoinAction.h

46 lines
1.2 KiB
C++
Raw Normal View History

#pragma once
#include <Core/Names.h>
#include <Core/Block.h>
#include <Interpreters/Context_fwd.h>
namespace DB
{
class IFunctionOverloadResolver;
using FunctionOverloadResolverPtr = std::shared_ptr<IFunctionOverloadResolver>;
2022-11-15 01:42:46 +00:00
class DataTypeArray;
class ColumnArray;
std::shared_ptr<const DataTypeArray> getArrayJoinDataType(DataTypePtr type);
const ColumnArray * getArrayJoinColumnRawPtr(const ColumnPtr & column);
2022-11-15 01:42:46 +00:00
2022-11-21 10:03:46 +00:00
/// If input array join column has map type, convert it to array type.
/// Otherwise do nothing.
ColumnWithTypeAndName convertArrayJoinColumn(const ColumnWithTypeAndName & src_col);
2022-11-15 01:42:46 +00:00
class ArrayJoinAction
{
public:
NameSet columns;
bool is_left = false;
bool is_unaligned = false;
/// For unaligned [LEFT] ARRAY JOIN
FunctionOverloadResolverPtr function_length;
FunctionOverloadResolverPtr function_greatest;
2022-11-15 01:42:46 +00:00
FunctionOverloadResolverPtr function_array_resize;
/// For LEFT ARRAY JOIN.
FunctionOverloadResolverPtr function_builder;
ArrayJoinAction(const NameSet & array_joined_columns_, bool array_join_is_left, ContextPtr context);
2020-09-10 18:36:51 +00:00
void prepare(ColumnsWithTypeAndName & sample) const;
2020-08-13 20:17:18 +00:00
void execute(Block & block);
};
using ArrayJoinActionPtr = std::shared_ptr<ArrayJoinAction>;
}