1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#pragma once
#include "FunctionArrayMapped.h"
#include <base/sort.h>
namespace DB
{
namespace ErrorCodes
{
extern const int ILLEGAL_TYPE_OF_ARGUMENT;
extern const int LOGICAL_ERROR;
}
/** Sort arrays, by values of its elements, or by values of corresponding elements of calculated expression (known as "schwartzsort").
*/
template <bool positive, bool is_partial>
struct ArraySortImpl
{
static constexpr auto num_fixed_params = is_partial;
static bool needBoolean() { return false; }
static bool needExpression() { return false; }
static bool needOneArray() { return false; }
static DataTypePtr getReturnType(const DataTypePtr & /*expression_return*/, const DataTypePtr & array_element)
{
return std::make_shared<DataTypeArray>(array_element);
}
static void checkArguments(
const String & name,
const ColumnWithTypeAndName * fixed_arguments)
requires(num_fixed_params)
{
if (!fixed_arguments)
throw Exception(
ErrorCodes::LOGICAL_ERROR,
"Expected fixed arguments to get the limit for partial array sort");
WhichDataType which(fixed_arguments[0].type.get());
if (!which.isUInt() && !which.isInt())
throw Exception(
ErrorCodes::ILLEGAL_TYPE_OF_ARGUMENT,
"Illegal type {} of limit argument of function {} (must be UInt or Int)",
fixed_arguments[0].type->getName(),
name);
}
static ColumnPtr execute(
const ColumnArray & array,
ColumnPtr mapped,
const ColumnWithTypeAndName * fixed_arguments [[maybe_unused]] = nullptr);
};
struct NameArraySort
{
static constexpr auto name = "arraySort";
};
struct NameArrayReverseSort
{
static constexpr auto name = "arrayReverseSort";
};
struct NameArrayPartialSort
{
static constexpr auto name = "arrayPartialSort";
};
struct NameArrayPartialReverseSort
{
static constexpr auto name = "arrayPartialReverseSort";
};
using FunctionArraySort = FunctionArrayMapped<ArraySortImpl<true, false>, NameArraySort>;
using FunctionArrayReverseSort = FunctionArrayMapped<ArraySortImpl<false, false>, NameArrayReverseSort>;
using FunctionArrayPartialSort = FunctionArrayMapped<ArraySortImpl<true, true>, NameArrayPartialSort>;
using FunctionArrayPartialReverseSort = FunctionArrayMapped<ArraySortImpl<false, true>, NameArrayPartialReverseSort>;
}
|
