Coverage Report

Created: 2024-11-20 12:30

/root/doris/be/src/vec/functions/if.cpp
Line
Count
Source (jump to first uncovered line)
1
// Licensed to the Apache Software Foundation (ASF) under one
2
// or more contributor license agreements.  See the NOTICE file
3
// distributed with this work for additional information
4
// regarding copyright ownership.  The ASF licenses this file
5
// to you under the Apache License, Version 2.0 (the
6
// "License"); you may not use this file except in compliance
7
// with the License.  You may obtain a copy of the License at
8
//
9
//   http://www.apache.org/licenses/LICENSE-2.0
10
//
11
// Unless required by applicable law or agreed to in writing,
12
// software distributed under the License is distributed on an
13
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14
// KIND, either express or implied.  See the License for the
15
// specific language governing permissions and limitations
16
// under the License.
17
// This file is copied from
18
// https://github.com/ClickHouse/ClickHouse/blob/master/src/Functions/If.cpp
19
// and modified by Doris
20
21
#include <glog/logging.h>
22
#include <stddef.h>
23
24
#include <algorithm>
25
#include <boost/iterator/iterator_facade.hpp>
26
#include <memory>
27
#include <ostream>
28
#include <type_traits>
29
#include <utility>
30
31
#include "common/status.h"
32
#include "util/simd/bits.h"
33
#include "vec/aggregate_functions/aggregate_function.h"
34
#include "vec/columns/column.h"
35
#include "vec/columns/column_const.h"
36
#include "vec/columns/column_nullable.h"
37
#include "vec/columns/column_vector.h"
38
#include "vec/columns/columns_number.h"
39
#include "vec/common/assert_cast.h"
40
#include "vec/common/pod_array_fwd.h"
41
#include "vec/common/typeid_cast.h"
42
#include "vec/core/block.h"
43
#include "vec/core/call_on_type_index.h"
44
#include "vec/core/column_numbers.h"
45
#include "vec/core/column_with_type_and_name.h"
46
#include "vec/core/types.h"
47
#include "vec/data_types/data_type.h"
48
#include "vec/data_types/data_type_nullable.h"
49
#include "vec/data_types/data_type_number.h"
50
#include "vec/functions/function.h"
51
#include "vec/functions/function_helpers.h"
52
#include "vec/functions/simple_function_factory.h"
53
namespace doris {
54
class FunctionContext;
55
56
namespace vectorized {
57
namespace NumberTraits {
58
struct Error;
59
} // namespace NumberTraits
60
} // namespace vectorized
61
} // namespace doris
62
63
namespace doris::vectorized {
64
65
template <typename A, typename B, typename ResultType>
66
struct NumIfImpl {
67
    using ArrayCond = PaddedPODArray<UInt8>;
68
    using ArrayA = PaddedPODArray<A>;
69
    using ArrayB = PaddedPODArray<B>;
70
    using ColVecResult = ColumnVector<ResultType>;
71
72
    static void vector_vector(const ArrayCond& cond, const ArrayA& a, const ArrayB& b, Block& block,
73
3
                              size_t result, UInt32) {
74
3
        size_t size = cond.size();
75
3
        auto col_res = ColVecResult::create(size);
76
3
        typename ColVecResult::Container& res = col_res->get_data();
77
78
12
        for (size_t i = 0; i < size; ++i)
79
9
            res[i] = cond[i] ? static_cast<ResultType>(a[i]) : static_cast<ResultType>(b[i]);
80
3
        block.replace_by_position(result, std::move(col_res));
81
3
    }
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIhhhE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEES9_S9_RNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplItttE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_ItLm4096ES6_Lm16ELm16EEESC_RNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIjjjE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IjLm4096ES6_Lm16ELm16EEESC_RNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplImmmE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_ImLm4096ES6_Lm16ELm16EEESC_RNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIaaaE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IaLm4096ES6_Lm16ELm16EEESC_RNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIsssE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IsLm4096ES6_Lm16ELm16EEESC_RNS0_5BlockEmj
_ZN5doris10vectorized9NumIfImplIiiiE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IiLm4096ES6_Lm16ELm16EEESC_RNS0_5BlockEmj
Line
Count
Source
73
2
                              size_t result, UInt32) {
74
2
        size_t size = cond.size();
75
2
        auto col_res = ColVecResult::create(size);
76
2
        typename ColVecResult::Container& res = col_res->get_data();
77
78
8
        for (size_t i = 0; i < size; ++i)
79
6
            res[i] = cond[i] ? static_cast<ResultType>(a[i]) : static_cast<ResultType>(b[i]);
80
2
        block.replace_by_position(result, std::move(col_res));
81
2
    }
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIlllE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IlLm4096ES6_Lm16ELm16EEESC_RNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplInnnE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_InLm4096ES6_Lm16ELm16EEESC_RNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIfffE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IfLm4096ES6_Lm16ELm16EEESC_RNS0_5BlockEmj
_ZN5doris10vectorized9NumIfImplIdddE13vector_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IdLm4096ES6_Lm16ELm16EEESC_RNS0_5BlockEmj
Line
Count
Source
73
1
                              size_t result, UInt32) {
74
1
        size_t size = cond.size();
75
1
        auto col_res = ColVecResult::create(size);
76
1
        typename ColVecResult::Container& res = col_res->get_data();
77
78
4
        for (size_t i = 0; i < size; ++i)
79
3
            res[i] = cond[i] ? static_cast<ResultType>(a[i]) : static_cast<ResultType>(b[i]);
80
1
        block.replace_by_position(result, std::move(col_res));
81
1
    }
82
83
    static void vector_constant(const ArrayCond& cond, const ArrayA& a, B b, Block& block,
84
0
                                size_t result, UInt32) {
85
0
        size_t size = cond.size();
86
0
        auto col_res = ColVecResult::create(size);
87
0
        typename ColVecResult::Container& res = col_res->get_data();
88
89
0
        for (size_t i = 0; i < size; ++i)
90
0
            res[i] = cond[i] ? static_cast<ResultType>(a[i]) : static_cast<ResultType>(b);
91
0
        block.replace_by_position(result, std::move(col_res));
92
0
    }
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIhhhE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEES9_hRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplItttE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_ItLm4096ES6_Lm16ELm16EEEtRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIjjjE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IjLm4096ES6_Lm16ELm16EEEjRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplImmmE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_ImLm4096ES6_Lm16ELm16EEEmRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIaaaE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IaLm4096ES6_Lm16ELm16EEEaRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIsssE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IsLm4096ES6_Lm16ELm16EEEsRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIiiiE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IiLm4096ES6_Lm16ELm16EEEiRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIlllE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IlLm4096ES6_Lm16ELm16EEElRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplInnnE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_InLm4096ES6_Lm16ELm16EEEnRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIfffE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IfLm4096ES6_Lm16ELm16EEEfRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIdddE15vector_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEERKNS3_IdLm4096ES6_Lm16ELm16EEEdRNS0_5BlockEmj
93
94
    static void constant_vector(const ArrayCond& cond, A a, const ArrayB& b, Block& block,
95
0
                                size_t result, UInt32) {
96
0
        size_t size = cond.size();
97
0
        auto col_res = ColVecResult::create(size);
98
0
        typename ColVecResult::Container& res = col_res->get_data();
99
100
0
        for (size_t i = 0; i < size; ++i)
101
0
            res[i] = cond[i] ? static_cast<ResultType>(a) : static_cast<ResultType>(b[i]);
102
0
        block.replace_by_position(result, std::move(col_res));
103
0
    }
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIhhhE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEhS9_RNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplItttE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEtRKNS3_ItLm4096ES6_Lm16ELm16EEERNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIjjjE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEjRKNS3_IjLm4096ES6_Lm16ELm16EEERNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplImmmE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEmRKNS3_ImLm4096ES6_Lm16ELm16EEERNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIaaaE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEaRKNS3_IaLm4096ES6_Lm16ELm16EEERNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIsssE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEsRKNS3_IsLm4096ES6_Lm16ELm16EEERNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIiiiE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEiRKNS3_IiLm4096ES6_Lm16ELm16EEERNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIlllE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEElRKNS3_IlLm4096ES6_Lm16ELm16EEERNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplInnnE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEnRKNS3_InLm4096ES6_Lm16ELm16EEERNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIfffE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEfRKNS3_IfLm4096ES6_Lm16ELm16EEERNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIdddE15constant_vectorERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEdRKNS3_IdLm4096ES6_Lm16ELm16EEERNS0_5BlockEmj
104
105
    static void constant_constant(const ArrayCond& cond, A a, B b, Block& block, size_t result,
106
0
                                  UInt32) {
107
0
        size_t size = cond.size();
108
0
        auto col_res = ColVecResult::create(size);
109
0
        typename ColVecResult::Container& res = col_res->get_data();
110
111
0
        for (size_t i = 0; i < size; ++i)
112
0
            res[i] = cond[i] ? static_cast<ResultType>(a) : static_cast<ResultType>(b);
113
0
        block.replace_by_position(result, std::move(col_res));
114
0
    }
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIhhhE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEhhRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplItttE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEttRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIjjjE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEjjRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplImmmE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEmmRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIaaaE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEaaRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIsssE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEssRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIiiiE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEiiRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIlllE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEllRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplInnnE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEnnRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIfffE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEffRNS0_5BlockEmj
Unexecuted instantiation: _ZN5doris10vectorized9NumIfImplIdddE17constant_constantERKNS0_8PODArrayIhLm4096E9AllocatorILb0ELb0ELb0E22DefaultMemoryAllocatorELm16ELm16EEEddRNS0_5BlockEmj
115
};
116
117
template <typename A, typename B>
118
struct NumIfImpl<A, B, NumberTraits::Error> {
119
private:
120
    [[noreturn]] static void throw_error() {
121
        throw doris::Exception(ErrorCode::INTERNAL_ERROR,
122
                               "Internal logic error: invalid types of arguments 2 and 3 of if");
123
        __builtin_unreachable();
124
    }
125
126
public:
127
    template <typename... Args>
128
    static void vector_vector(Args&&...) {
129
        throw_error();
130
    }
131
    template <typename... Args>
132
    static void vector_constant(Args&&...) {
133
        throw_error();
134
    }
135
    template <typename... Args>
136
    static void constant_vector(Args&&...) {
137
        throw_error();
138
    }
139
    template <typename... Args>
140
    static void constant_constant(Args&&...) {
141
        throw_error();
142
    }
143
};
144
145
8
size_t count_true_with_notnull(const ColumnPtr& col) {
146
8
    if (col->only_null()) {
147
0
        return 0;
148
0
    }
149
150
8
    if (const auto* const_col = check_and_get_column_const<ColumnVector<UInt8>>(col.get())) {
151
0
        bool is_true = const_col->get_bool(0);
152
0
        return is_true ? col->size() : 0;
153
0
    }
154
155
8
    auto count = col->size();
156
8
    if (col->is_nullable()) {
157
3
        const auto* nullable = assert_cast<const ColumnNullable*>(col.get());
158
3
        const auto* __restrict null_data = nullable->get_null_map_data().data();
159
3
        const auto* __restrict bool_data =
160
3
                ((const ColumnVector<UInt8>&)(nullable->get_nested_column())).get_data().data();
161
162
3
        size_t null_count = count - simd::count_zero_num((const int8_t*)null_data, count);
163
164
3
        if (null_count == count) {
165
0
            return 0;
166
3
        } else if (null_count == 0) {
167
0
            size_t true_count = count - simd::count_zero_num((const int8_t*)bool_data, count);
168
0
            return true_count;
169
3
        } else {
170
            // In fact, the null_count maybe is different with true_count, but it's no impact
171
3
            return null_count;
172
3
        }
173
5
    } else {
174
5
        const auto* bool_col = typeid_cast<const ColumnUInt8*>(col.get());
175
5
        const auto* __restrict bool_data = bool_col->get_data().data();
176
5
        return count - simd::count_zero_num((const int8_t*)bool_data, count);
177
5
    }
178
8
}
179
// todo(wb) support llvm codegen
180
class FunctionIf : public IFunction {
181
public:
182
    static constexpr auto name = "if";
183
184
9
    static FunctionPtr create() { return std::make_shared<FunctionIf>(); }
185
0
    String get_name() const override { return name; }
186
187
8
    size_t get_number_of_arguments() const override { return 3; }
188
16
    bool use_default_implementation_for_nulls() const override { return false; }
189
8
    DataTypePtr get_return_type_impl(const DataTypes& arguments) const override {
190
        // if return type is custom, one of nullable return type will be nullable
191
8
        bool nullable = arguments[1]->is_nullable() || arguments[2]->is_nullable();
192
8
        if (nullable) {
193
8
            return make_nullable(arguments[1]);
194
8
        } else {
195
0
            return arguments[1];
196
0
        }
197
8
    }
198
199
39
    static ColumnPtr materialize_column_if_const(const ColumnPtr& column) {
200
39
        return column->convert_to_full_column_if_const();
201
39
    }
202
203
0
    static ColumnPtr make_nullable_column_if_not(const ColumnPtr& column) {
204
0
        if (is_column_nullable(*column)) return column;
205
206
0
        return ColumnNullable::create(materialize_column_if_const(column),
207
0
                                      ColumnUInt8::create(column->size(), 0));
208
0
    }
209
210
10
    static ColumnPtr get_nested_column(const ColumnPtr& column) {
211
10
        if (auto* nullable = check_and_get_column<ColumnNullable>(*column))
212
10
            return nullable->get_nested_column_ptr();
213
0
        else if (const auto* column_const = check_and_get_column<ColumnConst>(*column))
214
0
            return ColumnConst::create(get_nested_column(column_const->get_data_column_ptr()),
215
0
                                       column->size());
216
217
0
        return column;
218
10
    }
219
220
    Status execute_generic(Block& block, const ColumnUInt8* cond_col,
221
                           const ColumnWithTypeAndName& then_col_type_name,
222
                           const ColumnWithTypeAndName& else_col_type_name, size_t result,
223
7
                           size_t input_row_count) const {
224
7
        MutableColumnPtr result_column = block.get_by_position(result).type->create_column();
225
7
        result_column->reserve(input_row_count);
226
227
7
        const IColumn& then_col = *then_col_type_name.column;
228
7
        const IColumn& else_col = *else_col_type_name.column;
229
7
        bool then_is_const = is_column_const(then_col);
230
7
        bool else_is_const = is_column_const(else_col);
231
232
7
        const auto& cond_array = cond_col->get_data();
233
234
7
        if (then_is_const && else_is_const) {
235
0
            const IColumn& then_nested_column =
236
0
                    assert_cast<const ColumnConst&>(then_col).get_data_column();
237
0
            const IColumn& else_nested_column =
238
0
                    assert_cast<const ColumnConst&>(else_col).get_data_column();
239
0
            for (size_t i = 0; i < input_row_count; i++) {
240
0
                if (cond_array[i])
241
0
                    result_column->insert_from(then_nested_column, 0);
242
0
                else
243
0
                    result_column->insert_from(else_nested_column, 0);
244
0
            }
245
7
        } else if (then_is_const) {
246
0
            const IColumn& then_nested_column =
247
0
                    assert_cast<const ColumnConst&>(then_col).get_data_column();
248
249
0
            for (size_t i = 0; i < input_row_count; i++) {
250
0
                if (cond_array[i])
251
0
                    result_column->insert_from(then_nested_column, 0);
252
0
                else
253
0
                    result_column->insert_from(else_col, i);
254
0
            }
255
7
        } else if (else_is_const) {
256
0
            const IColumn& else_nested_column =
257
0
                    assert_cast<const ColumnConst&>(else_col).get_data_column();
258
259
0
            for (size_t i = 0; i < input_row_count; i++) {
260
0
                if (cond_array[i])
261
0
                    result_column->insert_from(then_col, i);
262
0
                else
263
0
                    result_column->insert_from(else_nested_column, 0);
264
0
            }
265
7
        } else {
266
26
            for (size_t i = 0; i < input_row_count; i++) {
267
19
                result_column->insert_from(cond_array[i] ? then_col : else_col, i);
268
19
            }
269
7
        }
270
7
        block.replace_by_position(result, std::move(result_column));
271
7
        return Status::OK();
272
7
    }
273
274
    void execute_basic_type(Block& block, const ColumnUInt8* cond_col,
275
                            const ColumnWithTypeAndName& then_col,
276
                            const ColumnWithTypeAndName& else_col, size_t result,
277
3
                            Status& status) const {
278
3
        auto call = [&](const auto& types) -> bool {
279
3
            using Types = std::decay_t<decltype(types)>;
280
3
            using T0 = typename Types::LeftType;
281
3
            using result_type = typename Types::LeftType;
282
283
            // for doris, args type and return type must be sanme beacause of type cast has already done before, so here just need one type;
284
            // but code still need a better impelement
285
3
            using ColVecT0 = ColumnVector<T0>;
286
287
3
            if (auto col_then = check_and_get_column<ColVecT0>(then_col.column.get())) {
288
3
                if (auto col_else = check_and_get_column<ColVecT0>(else_col.column.get())) {
289
3
                    NumIfImpl<T0, T0, result_type>::vector_vector(
290
3
                            cond_col->get_data(), col_then->get_data(), col_else->get_data(), block,
291
3
                            result, 0);
292
3
                } else if (auto col_const_else =
293
0
                                   check_and_get_column_const<ColVecT0>(else_col.column.get())) {
294
0
                    NumIfImpl<T0, T0, result_type>::vector_constant(
295
0
                            cond_col->get_data(), col_then->get_data(),
296
0
                            col_const_else->template get_value<T0>(), block, result, 0);
297
0
                }
298
3
            } else if (auto col_const_then =
299
0
                               check_and_get_column_const<ColVecT0>(then_col.column.get())) {
300
0
                if (auto col_else = check_and_get_column<ColVecT0>(else_col.column.get())) {
301
0
                    NumIfImpl<T0, T0, result_type>::constant_vector(
302
0
                            cond_col->get_data(), col_const_then->template get_value<T0>(),
303
0
                            col_else->get_data(), block, result, 0);
304
0
                } else if (auto col_const_else =
305
0
                                   check_and_get_column_const<ColVecT0>(else_col.column.get())) {
306
0
                    NumIfImpl<T0, T0, result_type>::constant_constant(
307
0
                            cond_col->get_data(), col_const_then->template get_value<T0>(),
308
0
                            col_const_else->template get_value<T0>(), block, result, 0);
309
0
                }
310
0
            } else {
311
0
                status = Status::InternalError("unexpected args column type");
312
0
            }
313
3
            return true;
314
3
        };
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhhEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhtEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhjEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhmEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhsEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhiEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhlEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhnEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhfEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIhdEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIthEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIttEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairItjEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairItmEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairItaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairItsEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairItiEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairItlEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairItnEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairItfEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairItdEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjhEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjtEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjjEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjmEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjsEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjiEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjlEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjnEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjfEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIjdEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImhEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImtEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImjEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImmEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImsEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImiEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImlEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImnEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImfEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairImdEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIahEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIatEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIajEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIamEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIaaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIasEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIaiEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIalEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIanEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIafEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIadEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIshEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIstEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIsjEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIsmEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIsaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIssEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIsiEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIslEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIsnEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIsfEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIsdEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIihEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIitEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIijEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIimEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIiaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIisEEEEbSF_
_ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIiiEEEEbSF_
Line
Count
Source
278
2
        auto call = [&](const auto& types) -> bool {
279
2
            using Types = std::decay_t<decltype(types)>;
280
2
            using T0 = typename Types::LeftType;
281
2
            using result_type = typename Types::LeftType;
282
283
            // for doris, args type and return type must be sanme beacause of type cast has already done before, so here just need one type;
284
            // but code still need a better impelement
285
2
            using ColVecT0 = ColumnVector<T0>;
286
287
2
            if (auto col_then = check_and_get_column<ColVecT0>(then_col.column.get())) {
288
2
                if (auto col_else = check_and_get_column<ColVecT0>(else_col.column.get())) {
289
2
                    NumIfImpl<T0, T0, result_type>::vector_vector(
290
2
                            cond_col->get_data(), col_then->get_data(), col_else->get_data(), block,
291
2
                            result, 0);
292
2
                } else if (auto col_const_else =
293
0
                                   check_and_get_column_const<ColVecT0>(else_col.column.get())) {
294
0
                    NumIfImpl<T0, T0, result_type>::vector_constant(
295
0
                            cond_col->get_data(), col_then->get_data(),
296
0
                            col_const_else->template get_value<T0>(), block, result, 0);
297
0
                }
298
2
            } else if (auto col_const_then =
299
0
                               check_and_get_column_const<ColVecT0>(then_col.column.get())) {
300
0
                if (auto col_else = check_and_get_column<ColVecT0>(else_col.column.get())) {
301
0
                    NumIfImpl<T0, T0, result_type>::constant_vector(
302
0
                            cond_col->get_data(), col_const_then->template get_value<T0>(),
303
0
                            col_else->get_data(), block, result, 0);
304
0
                } else if (auto col_const_else =
305
0
                                   check_and_get_column_const<ColVecT0>(else_col.column.get())) {
306
0
                    NumIfImpl<T0, T0, result_type>::constant_constant(
307
0
                            cond_col->get_data(), col_const_then->template get_value<T0>(),
308
0
                            col_const_else->template get_value<T0>(), block, result, 0);
309
0
                }
310
0
            } else {
311
0
                status = Status::InternalError("unexpected args column type");
312
0
            }
313
2
            return true;
314
2
        };
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIilEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIinEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIifEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIidEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIlhEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIltEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIljEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIlmEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIlaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIlsEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIliEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIllEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIlnEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIlfEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIldEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairInhEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIntEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairInjEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairInmEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairInaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairInsEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIniEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairInlEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairInnEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairInfEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIndEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIfhEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIftEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIfjEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIfmEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIfaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIfsEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIfiEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIflEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIfnEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIffEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIfdEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIdhEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIdtEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIdjEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIdmEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIdaEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIdsEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIdiEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIdlEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIdnEEEEbSF_
Unexecuted instantiation: _ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIdfEEEEbSF_
_ZZNK5doris10vectorized10FunctionIf18execute_basic_typeERNS0_5BlockEPKNS0_12ColumnVectorIhEERKNS0_21ColumnWithTypeAndNameESA_mRNS_6StatusEENKUlRKT_E_clINS0_8TypePairIddEEEEbSF_
Line
Count
Source
278
1
        auto call = [&](const auto& types) -> bool {
279
1
            using Types = std::decay_t<decltype(types)>;
280
1
            using T0 = typename Types::LeftType;
281
1
            using result_type = typename Types::LeftType;
282
283
            // for doris, args type and return type must be sanme beacause of type cast has already done before, so here just need one type;
284
            // but code still need a better impelement
285
1
            using ColVecT0 = ColumnVector<T0>;
286
287
1
            if (auto col_then = check_and_get_column<ColVecT0>(then_col.column.get())) {
288
1
                if (auto col_else = check_and_get_column<ColVecT0>(else_col.column.get())) {
289
1
                    NumIfImpl<T0, T0, result_type>::vector_vector(
290
1
                            cond_col->get_data(), col_then->get_data(), col_else->get_data(), block,
291
1
                            result, 0);
292
1
                } else if (auto col_const_else =
293
0
                                   check_and_get_column_const<ColVecT0>(else_col.column.get())) {
294
0
                    NumIfImpl<T0, T0, result_type>::vector_constant(
295
0
                            cond_col->get_data(), col_then->get_data(),
296
0
                            col_const_else->template get_value<T0>(), block, result, 0);
297
0
                }
298
1
            } else if (auto col_const_then =
299
0
                               check_and_get_column_const<ColVecT0>(then_col.column.get())) {
300
0
                if (auto col_else = check_and_get_column<ColVecT0>(else_col.column.get())) {
301
0
                    NumIfImpl<T0, T0, result_type>::constant_vector(
302
0
                            cond_col->get_data(), col_const_then->template get_value<T0>(),
303
0
                            col_else->get_data(), block, result, 0);
304
0
                } else if (auto col_const_else =
305
0
                                   check_and_get_column_const<ColVecT0>(else_col.column.get())) {
306
0
                    NumIfImpl<T0, T0, result_type>::constant_constant(
307
0
                            cond_col->get_data(), col_const_then->template get_value<T0>(),
308
0
                            col_const_else->template get_value<T0>(), block, result, 0);
309
0
                }
310
0
            } else {
311
0
                status = Status::InternalError("unexpected args column type");
312
0
            }
313
1
            return true;
314
1
        };
315
316
        // todo(wb): a better way to determine type
317
3
        call_on_basic_types<true, true, false, false>(then_col.type->get_type_id(),
318
3
                                                      else_col.type->get_type_id(), call);
319
3
    }
320
321
    Status execute_for_null_then_else(FunctionContext* context, Block& block,
322
                                      const ColumnWithTypeAndName& arg_cond,
323
                                      const ColumnWithTypeAndName& arg_then,
324
                                      const ColumnWithTypeAndName& arg_else, size_t result,
325
18
                                      size_t input_rows_count, bool& handled) const {
326
18
        bool then_is_null = arg_then.column->only_null();
327
18
        bool else_is_null = arg_else.column->only_null();
328
329
18
        handled = false;
330
18
        if (!then_is_null && !else_is_null) {
331
15
            return Status::OK();
332
15
        }
333
334
3
        if (then_is_null && else_is_null) {
335
0
            block.get_by_position(result).column =
336
0
                    block.get_by_position(result).type->create_column_const_with_default_value(
337
0
                            input_rows_count);
338
0
            handled = true;
339
0
            return Status::OK();
340
0
        }
341
342
3
        const auto* cond_col = typeid_cast<const ColumnUInt8*>(arg_cond.column.get());
343
3
        const ColumnConst* cond_const_col =
344
3
                check_and_get_column_const<ColumnVector<UInt8>>(arg_cond.column.get());
345
346
        /// If then is NULL, we create Nullable column with null mask OR-ed with condition.
347
3
        if (then_is_null) {
348
3
            if (cond_col) {
349
3
                if (is_column_nullable(*arg_else.column)) { // if(cond, null, nullable)
350
3
                    auto arg_else_column = arg_else.column;
351
3
                    auto result_column = (*std::move(arg_else_column)).mutate();
352
3
                    assert_cast<ColumnNullable&>(*result_column)
353
3
                            .apply_null_map(assert_cast<const ColumnUInt8&>(*arg_cond.column));
354
3
                    block.replace_by_position(result, std::move(result_column));
355
3
                } else { // if(cond, null, not_nullable)
356
0
                    block.replace_by_position(
357
0
                            result,
358
0
                            ColumnNullable::create(materialize_column_if_const(arg_else.column),
359
0
                                                   arg_cond.column));
360
0
                }
361
3
            } else if (cond_const_col) {
362
0
                if (cond_const_col->get_value<UInt8>()) { // if(true, null, else)
363
0
                    block.get_by_position(result).column =
364
0
                            block.get_by_position(result).type->create_column()->clone_resized(
365
0
                                    input_rows_count);
366
0
                } else { // if(false, null, else)
367
0
                    block.get_by_position(result).column =
368
0
                            make_nullable_column_if_not(arg_else.column);
369
0
                }
370
0
            } else {
371
0
                return Status::InternalError(
372
0
                        "Illegal column {} of first argument of function {}. Must be ColumnUInt8 "
373
0
                        "or ColumnConstUInt8.",
374
0
                        arg_cond.column->get_name(), get_name());
375
0
            }
376
3
        } else { /// If else is NULL, we create Nullable column with null mask OR-ed with negated condition.
377
0
            if (cond_col) {
378
0
                size_t size = input_rows_count;
379
380
0
                if (is_column_nullable(*arg_then.column)) { // if(cond, nullable, NULL)
381
0
                    auto arg_then_column = arg_then.column;
382
0
                    auto result_column = (*std::move(arg_then_column)).mutate();
383
0
                    assert_cast<ColumnNullable&>(*result_column)
384
0
                            .apply_negated_null_map(
385
0
                                    assert_cast<const ColumnUInt8&>(*arg_cond.column));
386
0
                    block.replace_by_position(result, std::move(result_column));
387
0
                } else { // if(cond, not_nullable, NULL)
388
0
                    const auto& null_map_data = cond_col->get_data();
389
0
                    auto negated_null_map = ColumnUInt8::create();
390
0
                    auto& negated_null_map_data = negated_null_map->get_data();
391
0
                    negated_null_map_data.resize(size);
392
393
0
                    for (size_t i = 0; i < size; ++i) {
394
0
                        negated_null_map_data[i] = !null_map_data[i];
395
0
                    }
396
397
0
                    block.replace_by_position(
398
0
                            result,
399
0
                            ColumnNullable::create(materialize_column_if_const(arg_then.column),
400
0
                                                   std::move(negated_null_map)));
401
0
                }
402
0
            } else if (cond_const_col) {
403
0
                if (cond_const_col->get_value<UInt8>()) { // if(true, then, NULL)
404
0
                    block.get_by_position(result).column =
405
0
                            make_nullable_column_if_not(arg_then.column);
406
0
                } else { // if(false, then, NULL)
407
0
                    block.get_by_position(result).column =
408
0
                            block.get_by_position(result).type->create_column()->clone_resized(
409
0
                                    input_rows_count);
410
0
                }
411
0
            } else {
412
0
                return Status::InternalError(
413
0
                        "Illegal column {} of first argument of function {}. Must be ColumnUInt8 "
414
0
                        "or ColumnConstUInt8.",
415
0
                        arg_cond.column->get_name(), get_name());
416
0
            }
417
0
        }
418
3
        handled = true;
419
3
        return Status::OK();
420
3
    }
421
422
    Status execute_for_nullable_then_else(FunctionContext* context, Block& block,
423
                                          const ColumnWithTypeAndName& arg_cond,
424
                                          const ColumnWithTypeAndName& arg_then,
425
                                          const ColumnWithTypeAndName& arg_else, size_t result,
426
15
                                          size_t input_rows_count, bool& handled) const {
427
15
        auto then_type_is_nullable = arg_then.type->is_nullable();
428
15
        auto else_type_is_nullable = arg_else.type->is_nullable();
429
15
        handled = false;
430
15
        if (!then_type_is_nullable && !else_type_is_nullable) {
431
10
            return Status::OK();
432
10
        }
433
434
5
        auto* then_is_nullable = check_and_get_column<ColumnNullable>(*arg_then.column);
435
5
        auto* else_is_nullable = check_and_get_column<ColumnNullable>(*arg_else.column);
436
5
        bool then_column_is_const_nullable = false;
437
5
        bool else_column_is_const_nullable = false;
438
5
        if (then_type_is_nullable && then_is_nullable == nullptr) {
439
            //this case is a const(nullable column)
440
0
            auto& const_column = assert_cast<const ColumnConst&>(*arg_then.column);
441
0
            then_is_nullable =
442
0
                    assert_cast<const ColumnNullable*>(const_column.get_data_column_ptr().get());
443
0
            then_column_is_const_nullable = true;
444
0
        }
445
446
5
        if (else_type_is_nullable && else_is_nullable == nullptr) {
447
            //this case is a const(nullable column)
448
0
            auto& const_column = assert_cast<const ColumnConst&>(*arg_else.column);
449
0
            else_is_nullable =
450
0
                    assert_cast<const ColumnNullable*>(const_column.get_data_column_ptr().get());
451
0
            else_column_is_const_nullable = true;
452
0
        }
453
454
        /** Calculate null mask of result and nested column separately.
455
          */
456
5
        ColumnPtr result_null_mask;
457
5
        {
458
            // get null map from column:
459
            // a. get_null_map_column_ptr() : it's a real nullable column, so could get it from nullable column
460
            // b. create a const_nullmap_column: it's a not nullable column or a const nullable column, contain a const value
461
5
            Block temporary_block;
462
5
            temporary_block.insert(arg_cond);
463
5
            auto then_nested_null_map =
464
5
                    (then_type_is_nullable && !then_column_is_const_nullable)
465
5
                            ? then_is_nullable->get_null_map_column_ptr()
466
5
                            : DataTypeUInt8().create_column_const_with_default_value(
467
0
                                      input_rows_count);
468
5
            temporary_block.insert({then_nested_null_map, std::make_shared<DataTypeUInt8>(),
469
5
                                    "then_column_null_map"});
470
471
5
            auto else_nested_null_map =
472
5
                    (else_type_is_nullable && !else_column_is_const_nullable)
473
5
                            ? else_is_nullable->get_null_map_column_ptr()
474
5
                            : DataTypeUInt8().create_column_const_with_default_value(
475
0
                                      input_rows_count);
476
5
            temporary_block.insert({else_nested_null_map, std::make_shared<DataTypeUInt8>(),
477
5
                                    "else_column_null_map"});
478
5
            temporary_block.insert(
479
5
                    {nullptr, std::make_shared<DataTypeUInt8>(), "result_column_null_map"});
480
481
5
            RETURN_IF_ERROR(_execute_impl_internal(context, temporary_block, {0, 1, 2}, 3,
482
5
                                                   temporary_block.rows()));
483
484
5
            result_null_mask = temporary_block.get_by_position(3).column;
485
5
        }
486
487
0
        ColumnPtr result_nested_column;
488
489
5
        {
490
5
            Block temporary_block(
491
5
                    {arg_cond,
492
5
                     {get_nested_column(arg_then.column), remove_nullable(arg_then.type), ""},
493
5
                     {get_nested_column(arg_else.column), remove_nullable(arg_else.type), ""},
494
5
                     {nullptr, remove_nullable(block.get_by_position(result).type), ""}});
495
496
5
            RETURN_IF_ERROR(_execute_impl_internal(context, temporary_block, {0, 1, 2}, 3,
497
5
                                                   temporary_block.rows()));
498
499
5
            result_nested_column = temporary_block.get_by_position(3).column;
500
5
        }
501
502
0
        auto column = ColumnNullable::create(materialize_column_if_const(result_nested_column),
503
5
                                             materialize_column_if_const(result_null_mask));
504
5
        block.replace_by_position(result, std::move(column));
505
5
        handled = true;
506
5
        return Status::OK();
507
5
    }
508
509
    Status execute_for_null_condition(FunctionContext* context, Block& block,
510
                                      const ColumnNumbers& arguments,
511
                                      const ColumnWithTypeAndName& arg_cond,
512
                                      const ColumnWithTypeAndName& arg_then,
513
                                      const ColumnWithTypeAndName& arg_else, size_t result,
514
21
                                      bool& handled) const {
515
21
        bool cond_is_null = arg_cond.column->only_null();
516
21
        handled = false;
517
518
21
        if (cond_is_null) {
519
0
            block.replace_by_position(result,
520
0
                                      arg_else.column->clone_resized(arg_cond.column->size()));
521
0
            handled = true;
522
0
            return Status::OK();
523
0
        }
524
525
21
        if (const auto* nullable = check_and_get_column<ColumnNullable>(*arg_cond.column)) {
526
3
            DCHECK(remove_nullable(arg_cond.type)->get_type_id() == TypeIndex::UInt8);
527
528
            // update nested column by null map
529
3
            const auto* __restrict null_map = nullable->get_null_map_data().data();
530
3
            auto* __restrict nested_bool_data =
531
3
                    ((ColumnVector<UInt8>&)(nullable->get_nested_column())).get_data().data();
532
3
            auto rows = nullable->size();
533
12
            for (size_t i = 0; i < rows; i++) {
534
9
                nested_bool_data[i] &= !null_map[i];
535
9
            }
536
3
            auto column_size = block.columns();
537
3
            block.insert({nullable->get_nested_column_ptr(), remove_nullable(arg_cond.type),
538
3
                          arg_cond.name});
539
540
3
            handled = true;
541
3
            return _execute_impl_internal(context, block, {column_size, arguments[1], arguments[2]},
542
3
                                          result, rows);
543
3
        }
544
18
        return Status::OK();
545
21
    }
546
547
    Status execute_impl(FunctionContext* context, Block& block, const ColumnNumbers& arguments,
548
8
                        size_t result, size_t input_rows_count) const override {
549
8
        const ColumnWithTypeAndName& arg_then = block.get_by_position(arguments[1]);
550
8
        const ColumnWithTypeAndName& arg_else = block.get_by_position(arguments[2]);
551
552
        /// A case for identical then and else (pointers are the same).
553
8
        if (arg_then.column.get() == arg_else.column.get()) {
554
            /// Just point result to them.
555
0
            block.replace_by_position(result, arg_then.column);
556
0
            return Status::OK();
557
0
        }
558
559
8
        ColumnWithTypeAndName& cond_column = block.get_by_position(arguments[0]);
560
8
        cond_column.column = materialize_column_if_const(cond_column.column);
561
8
        const ColumnWithTypeAndName& arg_cond = block.get_by_position(arguments[0]);
562
563
8
        auto true_count = count_true_with_notnull(arg_cond.column);
564
8
        auto item_count = arg_cond.column->size();
565
8
        if (true_count == item_count || true_count == 0) {
566
0
            bool result_nullable = block.get_by_position(result).type->is_nullable();
567
0
            if (true_count == item_count) {
568
0
                block.replace_by_position(
569
0
                        result,
570
0
                        result_nullable
571
0
                                ? make_nullable(arg_then.column->clone_resized(input_rows_count))
572
0
                                : arg_then.column->clone_resized(input_rows_count));
573
0
            } else {
574
0
                block.replace_by_position(
575
0
                        result,
576
0
                        result_nullable
577
0
                                ? make_nullable(arg_else.column->clone_resized(input_rows_count))
578
0
                                : arg_else.column->clone_resized(input_rows_count));
579
0
            }
580
0
            return Status::OK();
581
0
        }
582
583
8
        return _execute_impl_internal(context, block, arguments, result, input_rows_count);
584
8
    }
585
586
    Status _execute_impl_internal(FunctionContext* context, Block& block,
587
                                  const ColumnNumbers& arguments, size_t result,
588
21
                                  size_t input_rows_count) const {
589
21
        const ColumnWithTypeAndName& arg_then = block.get_by_position(arguments[1]);
590
21
        const ColumnWithTypeAndName& arg_else = block.get_by_position(arguments[2]);
591
21
        ColumnWithTypeAndName& cond_column = block.get_by_position(arguments[0]);
592
21
        cond_column.column = materialize_column_if_const(cond_column.column);
593
21
        const ColumnWithTypeAndName& arg_cond = block.get_by_position(arguments[0]);
594
595
21
        Status ret = Status::OK();
596
21
        bool handled = false;
597
21
        RETURN_IF_ERROR(execute_for_null_condition(context, block, arguments, arg_cond, arg_then,
598
21
                                                   arg_else, result, handled));
599
600
21
        if (!handled) {
601
18
            RETURN_IF_ERROR(execute_for_null_then_else(context, block, arg_cond, arg_then, arg_else,
602
18
                                                       result, input_rows_count, handled));
603
18
        }
604
605
21
        if (!handled) {
606
15
            RETURN_IF_ERROR(execute_for_nullable_then_else(context, block, arg_cond, arg_then,
607
15
                                                           arg_else, result, input_rows_count,
608
15
                                                           handled));
609
15
        }
610
611
21
        if (handled) {
612
11
            return Status::OK();
613
11
        }
614
615
10
        const auto* cond_col = typeid_cast<const ColumnUInt8*>(arg_cond.column.get());
616
10
        const ColumnConst* cond_const_col =
617
10
                check_and_get_column_const<ColumnVector<UInt8>>(arg_cond.column.get());
618
619
10
        if (cond_const_col) {
620
0
            block.get_by_position(result).column =
621
0
                    cond_const_col->get_value<UInt8>() ? arg_then.column : arg_else.column;
622
0
            return Status::OK();
623
0
        }
624
625
10
        if (!cond_col) {
626
0
            return Status::InvalidArgument(
627
0
                    "Illegal column {} of first argument of function {},Must be ColumnUInt8 or "
628
0
                    "ColumnConstUInt8.",
629
0
                    arg_cond.column->get_name(), get_name());
630
0
        }
631
632
10
        WhichDataType which_type(arg_then.type);
633
10
        if (which_type.is_int() || which_type.is_float()) {
634
3
            Status status;
635
3
            execute_basic_type(block, cond_col, arg_then, arg_else, result, status);
636
3
            return status;
637
7
        } else {
638
7
            return execute_generic(block, cond_col, arg_then, arg_else, result, input_rows_count);
639
7
        }
640
10
    }
641
};
642
643
1
void register_function_if(SimpleFunctionFactory& factory) {
644
1
    factory.register_function<FunctionIf>();
645
1
}
646
647
} // namespace doris::vectorized