duckdb 0.7.2-dev1803.0 → 0.7.2-dev1867.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/package.json +1 -1
- package/src/duckdb/src/catalog/catalog.cpp +27 -27
- package/src/duckdb/src/catalog/catalog_entry/duck_schema_entry.cpp +6 -6
- package/src/duckdb/src/catalog/catalog_set.cpp +27 -25
- package/src/duckdb/src/catalog/default/default_functions.cpp +4 -4
- package/src/duckdb/src/catalog/default/default_types.cpp +4 -4
- package/src/duckdb/src/catalog/default/default_views.cpp +4 -4
- package/src/duckdb/src/catalog/dependency_list.cpp +7 -6
- package/src/duckdb/src/catalog/dependency_manager.cpp +44 -38
- package/src/duckdb/src/common/serializer/buffered_file_reader.cpp +11 -6
- package/src/duckdb/src/common/types/batched_data_collection.cpp +2 -1
- package/src/duckdb/src/common/types/column_data_allocator.cpp +1 -0
- package/src/duckdb/src/common/types/vector.cpp +2 -2
- package/src/duckdb/src/common/vector_operations/vector_copy.cpp +14 -11
- package/src/duckdb/src/execution/operator/aggregate/distinct_aggregate_data.cpp +1 -1
- package/src/duckdb/src/execution/operator/aggregate/physical_window.cpp +51 -50
- package/src/duckdb/src/execution/operator/persistent/physical_batch_insert.cpp +14 -13
- package/src/duckdb/src/execution/operator/persistent/physical_insert.cpp +20 -20
- package/src/duckdb/src/execution/operator/schema/physical_create_table.cpp +2 -2
- package/src/duckdb/src/execution/physical_plan/plan_create_index.cpp +1 -1
- package/src/duckdb/src/execution/physical_plan/plan_create_table.cpp +3 -3
- package/src/duckdb/src/execution/physical_plan/plan_delete.cpp +1 -1
- package/src/duckdb/src/execution/physical_plan/plan_insert.cpp +1 -1
- package/src/duckdb/src/execution/physical_plan/plan_update.cpp +1 -1
- package/src/duckdb/src/function/aggregate/sorted_aggregate_function.cpp +3 -3
- package/src/duckdb/src/function/cast/cast_function_set.cpp +2 -1
- package/src/duckdb/src/function/scalar/sequence/nextval.cpp +29 -29
- package/src/duckdb/src/function/scalar/string/damerau_levenshtein.cpp +106 -0
- package/src/duckdb/src/function/scalar/string/regexp.cpp +145 -28
- package/src/duckdb/src/function/scalar/string_functions.cpp +1 -0
- package/src/duckdb/src/function/table/checkpoint.cpp +4 -4
- package/src/duckdb/src/function/table/system/duckdb_columns.cpp +24 -24
- package/src/duckdb/src/function/table/system/duckdb_constraints.cpp +7 -6
- package/src/duckdb/src/function/table/system/duckdb_databases.cpp +1 -1
- package/src/duckdb/src/function/table/system/duckdb_dependencies.cpp +11 -11
- package/src/duckdb/src/function/table/system/pragma_database_size.cpp +1 -1
- package/src/duckdb/src/function/table/system/pragma_table_info.cpp +17 -18
- package/src/duckdb/src/function/table/table_scan.cpp +8 -11
- package/src/duckdb/src/function/table/version/pragma_version.cpp +2 -2
- package/src/duckdb/src/include/duckdb/catalog/catalog.hpp +9 -9
- package/src/duckdb/src/include/duckdb/catalog/catalog_entry_map.hpp +38 -0
- package/src/duckdb/src/include/duckdb/catalog/catalog_transaction.hpp +4 -3
- package/src/duckdb/src/include/duckdb/catalog/default/default_functions.hpp +2 -2
- package/src/duckdb/src/include/duckdb/catalog/default/default_types.hpp +2 -2
- package/src/duckdb/src/include/duckdb/catalog/default/default_views.hpp +2 -2
- package/src/duckdb/src/include/duckdb/catalog/dependency.hpp +4 -5
- package/src/duckdb/src/include/duckdb/catalog/dependency_list.hpp +4 -5
- package/src/duckdb/src/include/duckdb/catalog/dependency_manager.hpp +10 -9
- package/src/duckdb/src/include/duckdb/common/allocator.hpp +2 -1
- package/src/duckdb/src/include/duckdb/common/field_writer.hpp +1 -1
- package/src/duckdb/src/include/duckdb/common/helper.hpp +9 -0
- package/src/duckdb/src/include/duckdb/common/optional_ptr.hpp +29 -6
- package/src/duckdb/src/include/duckdb/common/serializer/buffered_file_reader.hpp +6 -5
- package/src/duckdb/src/include/duckdb/common/serializer.hpp +1 -1
- package/src/duckdb/src/include/duckdb/common/types/row_data_collection.hpp +1 -0
- package/src/duckdb/src/include/duckdb/execution/operator/persistent/physical_batch_insert.hpp +2 -2
- package/src/duckdb/src/include/duckdb/execution/operator/persistent/physical_insert.hpp +5 -5
- package/src/duckdb/src/include/duckdb/execution/operator/schema/physical_create_table.hpp +2 -2
- package/src/duckdb/src/include/duckdb/function/cast/default_casts.hpp +3 -2
- package/src/duckdb/src/include/duckdb/function/scalar/string_functions.hpp +4 -0
- package/src/duckdb/src/include/duckdb/main/database_manager.hpp +4 -3
- package/src/duckdb/src/include/duckdb/main/query_result.hpp +3 -2
- package/src/duckdb/src/include/duckdb/optimizer/filter_combiner.hpp +7 -7
- package/src/duckdb/src/include/duckdb/optimizer/matcher/expression_matcher.hpp +11 -11
- package/src/duckdb/src/include/duckdb/optimizer/matcher/set_matcher.hpp +8 -8
- package/src/duckdb/src/include/duckdb/optimizer/rule/arithmetic_simplification.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/case_simplification.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/comparison_simplification.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/conjunction_simplification.hpp +2 -2
- package/src/duckdb/src/include/duckdb/optimizer/rule/constant_folding.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/date_part_simplification.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/distributivity.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/empty_needle_removal.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/enum_comparison.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/equal_or_null_simplification.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/in_clause_simplification.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/like_optimizations.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/move_constants.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/ordered_aggregate_optimizer.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule/regex_optimizations.hpp +1 -1
- package/src/duckdb/src/include/duckdb/optimizer/rule.hpp +2 -2
- package/src/duckdb/src/include/duckdb/parser/base_expression.hpp +1 -1
- package/src/duckdb/src/include/duckdb/parser/expression_map.hpp +19 -6
- package/src/duckdb/src/include/duckdb/parser/expression_util.hpp +1 -1
- package/src/duckdb/src/include/duckdb/planner/expression.hpp +5 -2
- package/src/duckdb/src/include/duckdb/planner/expression_binder/base_select_binder.hpp +1 -1
- package/src/duckdb/src/include/duckdb/planner/expression_binder/order_binder.hpp +3 -3
- package/src/duckdb/src/include/duckdb/storage/buffer/block_handle.hpp +10 -2
- package/src/duckdb/src/include/duckdb/storage/buffer/buffer_pool.hpp +1 -0
- package/src/duckdb/src/include/duckdb/storage/buffer_manager.hpp +49 -126
- package/src/duckdb/src/include/duckdb/storage/meta_block_reader.hpp +5 -5
- package/src/duckdb/src/include/duckdb/storage/standard_buffer_manager.hpp +159 -0
- package/src/duckdb/src/include/duckdb/storage/table/column_segment.hpp +1 -0
- package/src/duckdb/src/include/duckdb/transaction/meta_transaction.hpp +6 -5
- package/src/duckdb/src/main/client_context.cpp +1 -1
- package/src/duckdb/src/main/database.cpp +2 -1
- package/src/duckdb/src/main/database_manager.cpp +4 -4
- package/src/duckdb/src/optimizer/common_aggregate_optimizer.cpp +2 -2
- package/src/duckdb/src/optimizer/cse_optimizer.cpp +4 -4
- package/src/duckdb/src/optimizer/deliminator.cpp +13 -11
- package/src/duckdb/src/optimizer/expression_rewriter.cpp +2 -2
- package/src/duckdb/src/optimizer/filter_combiner.cpp +67 -65
- package/src/duckdb/src/optimizer/join_order/cardinality_estimator.cpp +1 -0
- package/src/duckdb/src/optimizer/join_order/join_order_optimizer.cpp +26 -25
- package/src/duckdb/src/optimizer/matcher/expression_matcher.cpp +23 -21
- package/src/duckdb/src/optimizer/rule/arithmetic_simplification.cpp +3 -3
- package/src/duckdb/src/optimizer/rule/case_simplification.cpp +2 -2
- package/src/duckdb/src/optimizer/rule/comparison_simplification.cpp +6 -7
- package/src/duckdb/src/optimizer/rule/conjunction_simplification.cpp +9 -8
- package/src/duckdb/src/optimizer/rule/constant_folding.cpp +7 -7
- package/src/duckdb/src/optimizer/rule/date_part_simplification.cpp +3 -3
- package/src/duckdb/src/optimizer/rule/distributivity.cpp +5 -5
- package/src/duckdb/src/optimizer/rule/empty_needle_removal.cpp +6 -6
- package/src/duckdb/src/optimizer/rule/enum_comparison.cpp +4 -4
- package/src/duckdb/src/optimizer/rule/equal_or_null_simplification.cpp +23 -26
- package/src/duckdb/src/optimizer/rule/in_clause_simplification_rule.cpp +2 -3
- package/src/duckdb/src/optimizer/rule/like_optimizations.cpp +3 -3
- package/src/duckdb/src/optimizer/rule/move_constants.cpp +6 -6
- package/src/duckdb/src/optimizer/rule/ordered_aggregate_optimizer.cpp +2 -2
- package/src/duckdb/src/optimizer/rule/regex_optimizations.cpp +3 -3
- package/src/duckdb/src/parser/expression_util.cpp +6 -6
- package/src/duckdb/src/parser/transform/helpers/transform_groupby.cpp +3 -3
- package/src/duckdb/src/planner/binder/expression/bind_aggregate_expression.cpp +2 -2
- package/src/duckdb/src/planner/binder/query_node/bind_select_node.cpp +3 -3
- package/src/duckdb/src/planner/binder/query_node/bind_setop_node.cpp +5 -5
- package/src/duckdb/src/planner/binder/statement/bind_create_table.cpp +2 -2
- package/src/duckdb/src/planner/expression_binder/base_select_binder.cpp +4 -4
- package/src/duckdb/src/planner/expression_binder/order_binder.cpp +3 -3
- package/src/duckdb/src/storage/buffer/block_handle.cpp +3 -2
- package/src/duckdb/src/storage/buffer/block_manager.cpp +3 -1
- package/src/duckdb/src/storage/buffer/buffer_handle.cpp +1 -0
- package/src/duckdb/src/storage/buffer/buffer_pool_reservation.cpp +3 -0
- package/src/duckdb/src/storage/buffer_manager.cpp +35 -726
- package/src/duckdb/src/storage/checkpoint_manager.cpp +2 -2
- package/src/duckdb/src/storage/meta_block_reader.cpp +6 -5
- package/src/duckdb/src/storage/standard_buffer_manager.cpp +801 -0
- package/src/duckdb/src/storage/wal_replay.cpp +2 -2
- package/src/duckdb/src/transaction/meta_transaction.cpp +13 -13
- package/src/duckdb/src/transaction/transaction.cpp +1 -1
- package/src/duckdb/src/transaction/transaction_context.cpp +1 -1
- package/src/duckdb/ub_src_function_scalar_string.cpp +2 -0
- package/src/duckdb/ub_src_storage.cpp +2 -0
@@ -218,6 +218,103 @@ bool RegexpExtractBindData::Equals(const FunctionData &other_p) const {
|
|
218
218
|
return RegexpBaseBindData::Equals(other) && group_string == other.group_string;
|
219
219
|
}
|
220
220
|
|
221
|
+
static void RegexExtractFunction(DataChunk &args, ExpressionState &state, Vector &result) {
|
222
|
+
auto &func_expr = state.expr.Cast<BoundFunctionExpression>();
|
223
|
+
const auto &info = func_expr.bind_info->Cast<RegexpExtractBindData>();
|
224
|
+
|
225
|
+
auto &strings = args.data[0];
|
226
|
+
auto &patterns = args.data[1];
|
227
|
+
if (info.constant_pattern) {
|
228
|
+
auto &lstate = ExecuteFunctionState::GetFunctionState(state)->Cast<RegexLocalState>();
|
229
|
+
UnaryExecutor::Execute<string_t, string_t>(strings, result, args.size(), [&](string_t input) {
|
230
|
+
return Extract(input, result, lstate.constant_pattern, info.rewrite);
|
231
|
+
});
|
232
|
+
} else {
|
233
|
+
BinaryExecutor::Execute<string_t, string_t, string_t>(strings, patterns, result, args.size(),
|
234
|
+
[&](string_t input, string_t pattern) {
|
235
|
+
RE2 re(CreateStringPiece(pattern), info.options);
|
236
|
+
return Extract(input, result, re, info.rewrite);
|
237
|
+
});
|
238
|
+
}
|
239
|
+
}
|
240
|
+
|
241
|
+
//===--------------------------------------------------------------------===//
|
242
|
+
// Regexp Extract Struct
|
243
|
+
//===--------------------------------------------------------------------===//
|
244
|
+
static void RegexExtractStructFunction(DataChunk &args, ExpressionState &state, Vector &result) {
|
245
|
+
auto &lstate = ExecuteFunctionState::GetFunctionState(state)->Cast<RegexLocalState>();
|
246
|
+
|
247
|
+
const auto count = args.size();
|
248
|
+
auto &input = args.data[0];
|
249
|
+
|
250
|
+
auto &child_entries = StructVector::GetEntries(result);
|
251
|
+
const auto groupSize = child_entries.size();
|
252
|
+
// Reference the 'input' StringBuffer, because we won't need to allocate new data
|
253
|
+
// for the result, all returned strings are substrings of the originals
|
254
|
+
for (auto &child_entry : child_entries) {
|
255
|
+
child_entry->SetAuxiliary(input.GetAuxiliary());
|
256
|
+
}
|
257
|
+
|
258
|
+
vector<RE2::Arg> argv(groupSize);
|
259
|
+
vector<RE2::Arg *> groups(groupSize);
|
260
|
+
vector<duckdb_re2::StringPiece> ws(groupSize);
|
261
|
+
for (size_t i = 0; i < groupSize; ++i) {
|
262
|
+
groups[i] = &argv[i];
|
263
|
+
argv[i] = &ws[i];
|
264
|
+
}
|
265
|
+
|
266
|
+
if (input.GetVectorType() == VectorType::CONSTANT_VECTOR) {
|
267
|
+
result.SetVectorType(VectorType::CONSTANT_VECTOR);
|
268
|
+
|
269
|
+
if (ConstantVector::IsNull(input)) {
|
270
|
+
ConstantVector::SetNull(result, true);
|
271
|
+
} else {
|
272
|
+
ConstantVector::SetNull(result, false);
|
273
|
+
auto idata = ConstantVector::GetData<string_t>(input);
|
274
|
+
auto str = CreateStringPiece(idata[0]);
|
275
|
+
auto match = duckdb_re2::RE2::PartialMatchN(str, lstate.constant_pattern, groups.data(), groups.size());
|
276
|
+
for (size_t col = 0; col < child_entries.size(); ++col) {
|
277
|
+
auto &child_entry = child_entries[col];
|
278
|
+
ConstantVector::SetNull(*child_entry, false);
|
279
|
+
auto &extracted = ws[col];
|
280
|
+
auto cdata = ConstantVector::GetData<string_t>(*child_entry);
|
281
|
+
cdata[0] = string_t(extracted.data(), match ? extracted.size() : 0);
|
282
|
+
}
|
283
|
+
}
|
284
|
+
} else {
|
285
|
+
UnifiedVectorFormat iunified;
|
286
|
+
input.ToUnifiedFormat(count, iunified);
|
287
|
+
|
288
|
+
const auto &ivalidity = iunified.validity;
|
289
|
+
auto idata = (const string_t *)iunified.data;
|
290
|
+
|
291
|
+
// Start with a valid flat vector
|
292
|
+
result.SetVectorType(VectorType::FLAT_VECTOR);
|
293
|
+
|
294
|
+
// Start with valid children
|
295
|
+
for (size_t col = 0; col < child_entries.size(); ++col) {
|
296
|
+
auto &child_entry = child_entries[col];
|
297
|
+
child_entry->SetVectorType(VectorType::FLAT_VECTOR);
|
298
|
+
}
|
299
|
+
|
300
|
+
for (idx_t i = 0; i < count; ++i) {
|
301
|
+
const auto idx = iunified.sel->get_index(i);
|
302
|
+
if (ivalidity.RowIsValid(idx)) {
|
303
|
+
auto str = CreateStringPiece(idata[idx]);
|
304
|
+
auto match = duckdb_re2::RE2::PartialMatchN(str, lstate.constant_pattern, groups.data(), groups.size());
|
305
|
+
for (size_t col = 0; col < child_entries.size(); ++col) {
|
306
|
+
auto &child_entry = child_entries[col];
|
307
|
+
auto cdata = FlatVector::GetData<string_t>(*child_entry);
|
308
|
+
auto &extracted = ws[col];
|
309
|
+
cdata[i] = string_t(extracted.data(), match ? extracted.size() : 0);
|
310
|
+
}
|
311
|
+
} else {
|
312
|
+
FlatVector::SetNull(result, i, true);
|
313
|
+
}
|
314
|
+
}
|
315
|
+
}
|
316
|
+
}
|
317
|
+
|
221
318
|
static unique_ptr<FunctionData> RegexExtractBind(ClientContext &context, ScalarFunction &bound_function,
|
222
319
|
vector<unique_ptr<Expression>> &arguments) {
|
223
320
|
D_ASSERT(arguments.size() >= 2);
|
@@ -227,52 +324,62 @@ static unique_ptr<FunctionData> RegexExtractBind(ClientContext &context, ScalarF
|
|
227
324
|
string constant_string;
|
228
325
|
bool constant_pattern = TryParseConstantPattern(context, *arguments[1], constant_string);
|
229
326
|
|
230
|
-
|
327
|
+
if (arguments.size() >= 4) {
|
328
|
+
ParseRegexOptions(context, *arguments[3], options);
|
329
|
+
}
|
330
|
+
|
331
|
+
string group_string = "\\0";
|
231
332
|
if (arguments.size() >= 3) {
|
232
333
|
if (arguments[2]->HasParameter()) {
|
233
334
|
throw ParameterNotResolvedException();
|
234
335
|
}
|
235
336
|
if (!arguments[2]->IsFoldable()) {
|
236
|
-
throw InvalidInputException("Group
|
337
|
+
throw InvalidInputException("Group specification field must be a constant!");
|
237
338
|
}
|
238
339
|
Value group = ExpressionExecutor::EvaluateScalar(context, *arguments[2]);
|
239
|
-
if (
|
340
|
+
if (group.IsNull()) {
|
341
|
+
group_string = "";
|
342
|
+
} else if (group.type().id() == LogicalTypeId::LIST) {
|
343
|
+
if (!constant_pattern) {
|
344
|
+
throw BinderException("%s with LIST requires a constant pattern", bound_function.name);
|
345
|
+
}
|
346
|
+
auto &list_children = ListValue::GetChildren(group);
|
347
|
+
if (list_children.empty()) {
|
348
|
+
throw BinderException("%s requires non-empty lists of capture names", bound_function.name);
|
349
|
+
}
|
350
|
+
case_insensitive_set_t name_collision_set;
|
351
|
+
child_list_t<LogicalType> struct_children;
|
352
|
+
for (const auto &child : list_children) {
|
353
|
+
if (child.IsNull()) {
|
354
|
+
throw BinderException("NULL group name in %s", bound_function.name);
|
355
|
+
}
|
356
|
+
const auto group_name = child.ToString();
|
357
|
+
if (name_collision_set.find(group_name) != name_collision_set.end()) {
|
358
|
+
throw BinderException("Duplicate group name \"%s\" in %s", group_name, bound_function.name);
|
359
|
+
}
|
360
|
+
name_collision_set.insert(group_name);
|
361
|
+
struct_children.emplace_back(make_pair(group_name, LogicalType::VARCHAR));
|
362
|
+
}
|
363
|
+
bound_function.return_type = LogicalType::STRUCT(struct_children);
|
364
|
+
|
365
|
+
duckdb_re2::StringPiece constant_piece(constant_string.c_str(), constant_string.size());
|
366
|
+
RE2 constant_pattern(constant_piece, options);
|
367
|
+
if (size_t(constant_pattern.NumberOfCapturingGroups()) < list_children.size()) {
|
368
|
+
throw BinderException("Not enough group names in %s", bound_function.name);
|
369
|
+
}
|
370
|
+
} else {
|
240
371
|
auto group_idx = group.GetValue<int32_t>();
|
241
372
|
if (group_idx < 0 || group_idx > 9) {
|
242
373
|
throw InvalidInputException("Group index must be between 0 and 9!");
|
243
374
|
}
|
244
375
|
group_string = "\\" + to_string(group_idx);
|
245
376
|
}
|
246
|
-
} else {
|
247
|
-
group_string = "\\0";
|
248
|
-
}
|
249
|
-
if (arguments.size() >= 4) {
|
250
|
-
ParseRegexOptions(context, *arguments[3], options);
|
251
377
|
}
|
378
|
+
|
252
379
|
return make_uniq<RegexpExtractBindData>(options, std::move(constant_string), constant_pattern,
|
253
380
|
std::move(group_string));
|
254
381
|
}
|
255
382
|
|
256
|
-
static void RegexExtractFunction(DataChunk &args, ExpressionState &state, Vector &result) {
|
257
|
-
auto &func_expr = state.expr.Cast<BoundFunctionExpression>();
|
258
|
-
const auto &info = func_expr.bind_info->Cast<RegexpExtractBindData>();
|
259
|
-
|
260
|
-
auto &strings = args.data[0];
|
261
|
-
auto &patterns = args.data[1];
|
262
|
-
if (info.constant_pattern) {
|
263
|
-
auto &lstate = ExecuteFunctionState::GetFunctionState(state)->Cast<RegexLocalState>();
|
264
|
-
UnaryExecutor::Execute<string_t, string_t>(strings, result, args.size(), [&](string_t input) {
|
265
|
-
return Extract(input, result, lstate.constant_pattern, info.rewrite);
|
266
|
-
});
|
267
|
-
} else {
|
268
|
-
BinaryExecutor::Execute<string_t, string_t, string_t>(strings, patterns, result, args.size(),
|
269
|
-
[&](string_t input, string_t pattern) {
|
270
|
-
RE2 re(CreateStringPiece(pattern), info.options);
|
271
|
-
return Extract(input, result, re, info.rewrite);
|
272
|
-
});
|
273
|
-
}
|
274
|
-
}
|
275
|
-
|
276
383
|
void RegexpFun::RegisterFunction(BuiltinFunctions &set) {
|
277
384
|
ScalarFunctionSet regexp_full_match("regexp_full_match");
|
278
385
|
regexp_full_match.AddFunction(ScalarFunction(
|
@@ -315,6 +422,16 @@ void RegexpFun::RegisterFunction(BuiltinFunctions &set) {
|
|
315
422
|
{LogicalType::VARCHAR, LogicalType::VARCHAR, LogicalType::INTEGER, LogicalType::VARCHAR}, LogicalType::VARCHAR,
|
316
423
|
RegexExtractFunction, RegexExtractBind, nullptr, nullptr, RegexInitLocalState, LogicalType::INVALID,
|
317
424
|
FunctionSideEffects::NO_SIDE_EFFECTS, FunctionNullHandling::SPECIAL_HANDLING));
|
425
|
+
// REGEXP_EXTRACT(<string>, <pattern>, [<group 1 name>[, <group n name>]...])
|
426
|
+
regexp_extract.AddFunction(ScalarFunction(
|
427
|
+
{LogicalType::VARCHAR, LogicalType::VARCHAR, LogicalType::LIST(LogicalType::VARCHAR)}, LogicalType::VARCHAR,
|
428
|
+
RegexExtractStructFunction, RegexExtractBind, nullptr, nullptr, RegexInitLocalState, LogicalType::INVALID,
|
429
|
+
FunctionSideEffects::NO_SIDE_EFFECTS, FunctionNullHandling::SPECIAL_HANDLING));
|
430
|
+
// REGEXP_EXTRACT(<string>, <pattern>, [<group 1 name>[, <group n name>]...], <options>)
|
431
|
+
regexp_extract.AddFunction(ScalarFunction(
|
432
|
+
{LogicalType::VARCHAR, LogicalType::VARCHAR, LogicalType::LIST(LogicalType::VARCHAR), LogicalType::VARCHAR},
|
433
|
+
LogicalType::VARCHAR, RegexExtractStructFunction, RegexExtractBind, nullptr, nullptr, RegexInitLocalState,
|
434
|
+
LogicalType::INVALID, FunctionSideEffects::NO_SIDE_EFFECTS, FunctionNullHandling::SPECIAL_HANDLING));
|
318
435
|
|
319
436
|
ScalarFunctionSet regexp_extract_all("regexp_extract_all");
|
320
437
|
regexp_extract_all.AddFunction(ScalarFunction(
|
@@ -8,10 +8,10 @@
|
|
8
8
|
namespace duckdb {
|
9
9
|
|
10
10
|
struct CheckpointBindData : public FunctionData {
|
11
|
-
explicit CheckpointBindData(AttachedDatabase
|
11
|
+
explicit CheckpointBindData(optional_ptr<AttachedDatabase> db) : db(db) {
|
12
12
|
}
|
13
13
|
|
14
|
-
AttachedDatabase
|
14
|
+
optional_ptr<AttachedDatabase> db;
|
15
15
|
|
16
16
|
public:
|
17
17
|
unique_ptr<FunctionData> Copy() const override {
|
@@ -29,7 +29,7 @@ static unique_ptr<FunctionData> CheckpointBind(ClientContext &context, TableFunc
|
|
29
29
|
return_types.emplace_back(LogicalType::BOOLEAN);
|
30
30
|
names.emplace_back("Success");
|
31
31
|
|
32
|
-
AttachedDatabase
|
32
|
+
optional_ptr<AttachedDatabase> db;
|
33
33
|
auto &db_manager = DatabaseManager::Get(context);
|
34
34
|
if (!input.inputs.empty()) {
|
35
35
|
auto &db_name = StringValue::Get(input.inputs[0]);
|
@@ -46,7 +46,7 @@ static unique_ptr<FunctionData> CheckpointBind(ClientContext &context, TableFunc
|
|
46
46
|
template <bool FORCE>
|
47
47
|
static void TemplatedCheckpointFunction(ClientContext &context, TableFunctionInput &data_p, DataChunk &output) {
|
48
48
|
auto &bind_data = data_p.bind_data->Cast<CheckpointBindData>();
|
49
|
-
auto &transaction_manager = TransactionManager::Get(*bind_data.db);
|
49
|
+
auto &transaction_manager = TransactionManager::Get(*bind_data.db.get_mutable());
|
50
50
|
transaction_manager.Checkpoint(context, FORCE);
|
51
51
|
}
|
52
52
|
|
@@ -16,7 +16,7 @@ struct DuckDBColumnsData : public GlobalTableFunctionState {
|
|
16
16
|
DuckDBColumnsData() : offset(0), column_offset(0) {
|
17
17
|
}
|
18
18
|
|
19
|
-
vector<CatalogEntry
|
19
|
+
vector<optional_ptr<CatalogEntry>> entries;
|
20
20
|
idx_t offset;
|
21
21
|
idx_t column_offset;
|
22
22
|
};
|
@@ -90,12 +90,12 @@ unique_ptr<GlobalTableFunctionState> DuckDBColumnsInit(ClientContext &context, T
|
|
90
90
|
|
91
91
|
class ColumnHelper {
|
92
92
|
public:
|
93
|
-
static unique_ptr<ColumnHelper> Create(CatalogEntry
|
93
|
+
static unique_ptr<ColumnHelper> Create(CatalogEntry &entry);
|
94
94
|
|
95
95
|
virtual ~ColumnHelper() {
|
96
96
|
}
|
97
97
|
|
98
|
-
virtual StandardEntry
|
98
|
+
virtual StandardEntry &Entry() = 0;
|
99
99
|
virtual idx_t NumColumns() = 0;
|
100
100
|
virtual const string &ColumnName(idx_t col) = 0;
|
101
101
|
virtual const LogicalType &ColumnType(idx_t col) = 0;
|
@@ -107,8 +107,8 @@ public:
|
|
107
107
|
|
108
108
|
class TableColumnHelper : public ColumnHelper {
|
109
109
|
public:
|
110
|
-
explicit TableColumnHelper(TableCatalogEntry
|
111
|
-
for (auto &constraint : entry
|
110
|
+
explicit TableColumnHelper(TableCatalogEntry &entry) : entry(entry) {
|
111
|
+
for (auto &constraint : entry.GetConstraints()) {
|
112
112
|
if (constraint->type == ConstraintType::NOT_NULL) {
|
113
113
|
auto ¬_null = *reinterpret_cast<NotNullConstraint *>(constraint.get());
|
114
114
|
not_null_cols.insert(not_null.index.index);
|
@@ -116,20 +116,20 @@ public:
|
|
116
116
|
}
|
117
117
|
}
|
118
118
|
|
119
|
-
StandardEntry
|
119
|
+
StandardEntry &Entry() override {
|
120
120
|
return entry;
|
121
121
|
}
|
122
122
|
idx_t NumColumns() override {
|
123
|
-
return entry
|
123
|
+
return entry.GetColumns().LogicalColumnCount();
|
124
124
|
}
|
125
125
|
const string &ColumnName(idx_t col) override {
|
126
|
-
return entry
|
126
|
+
return entry.GetColumn(LogicalIndex(col)).Name();
|
127
127
|
}
|
128
128
|
const LogicalType &ColumnType(idx_t col) override {
|
129
|
-
return entry
|
129
|
+
return entry.GetColumn(LogicalIndex(col)).Type();
|
130
130
|
}
|
131
131
|
const Value ColumnDefault(idx_t col) override {
|
132
|
-
auto &column = entry
|
132
|
+
auto &column = entry.GetColumn(LogicalIndex(col));
|
133
133
|
if (column.DefaultValue()) {
|
134
134
|
return Value(column.DefaultValue()->ToString());
|
135
135
|
}
|
@@ -140,26 +140,26 @@ public:
|
|
140
140
|
}
|
141
141
|
|
142
142
|
private:
|
143
|
-
TableCatalogEntry
|
143
|
+
TableCatalogEntry &entry;
|
144
144
|
std::set<idx_t> not_null_cols;
|
145
145
|
};
|
146
146
|
|
147
147
|
class ViewColumnHelper : public ColumnHelper {
|
148
148
|
public:
|
149
|
-
explicit ViewColumnHelper(ViewCatalogEntry
|
149
|
+
explicit ViewColumnHelper(ViewCatalogEntry &entry) : entry(entry) {
|
150
150
|
}
|
151
151
|
|
152
|
-
StandardEntry
|
152
|
+
StandardEntry &Entry() override {
|
153
153
|
return entry;
|
154
154
|
}
|
155
155
|
idx_t NumColumns() override {
|
156
|
-
return entry
|
156
|
+
return entry.types.size();
|
157
157
|
}
|
158
158
|
const string &ColumnName(idx_t col) override {
|
159
|
-
return entry
|
159
|
+
return entry.aliases[col];
|
160
160
|
}
|
161
161
|
const LogicalType &ColumnType(idx_t col) override {
|
162
|
-
return entry
|
162
|
+
return entry.types[col];
|
163
163
|
}
|
164
164
|
const Value ColumnDefault(idx_t col) override {
|
165
165
|
return Value();
|
@@ -169,15 +169,15 @@ public:
|
|
169
169
|
}
|
170
170
|
|
171
171
|
private:
|
172
|
-
ViewCatalogEntry
|
172
|
+
ViewCatalogEntry &entry;
|
173
173
|
};
|
174
174
|
|
175
|
-
unique_ptr<ColumnHelper> ColumnHelper::Create(CatalogEntry
|
176
|
-
switch (entry
|
175
|
+
unique_ptr<ColumnHelper> ColumnHelper::Create(CatalogEntry &entry) {
|
176
|
+
switch (entry.type) {
|
177
177
|
case CatalogType::TABLE_ENTRY:
|
178
|
-
return make_uniq<TableColumnHelper>((
|
178
|
+
return make_uniq<TableColumnHelper>(entry.Cast<TableCatalogEntry>());
|
179
179
|
case CatalogType::VIEW_ENTRY:
|
180
|
-
return make_uniq<ViewColumnHelper>((
|
180
|
+
return make_uniq<ViewColumnHelper>(entry.Cast<ViewCatalogEntry>());
|
181
181
|
default:
|
182
182
|
throw NotImplementedException("Unsupported catalog type for duckdb_columns");
|
183
183
|
}
|
@@ -186,7 +186,7 @@ unique_ptr<ColumnHelper> ColumnHelper::Create(CatalogEntry *entry) {
|
|
186
186
|
void ColumnHelper::WriteColumns(idx_t start_index, idx_t start_col, idx_t end_col, DataChunk &output) {
|
187
187
|
for (idx_t i = start_col; i < end_col; i++) {
|
188
188
|
auto index = start_index + (i - start_col);
|
189
|
-
auto &entry =
|
189
|
+
auto &entry = Entry();
|
190
190
|
|
191
191
|
idx_t col = 0;
|
192
192
|
// database_name, VARCHAR
|
@@ -284,7 +284,7 @@ void ColumnHelper::WriteColumns(idx_t start_index, idx_t start_col, idx_t end_co
|
|
284
284
|
}
|
285
285
|
|
286
286
|
void DuckDBColumnsFunction(ClientContext &context, TableFunctionInput &data_p, DataChunk &output) {
|
287
|
-
auto &data =
|
287
|
+
auto &data = data_p.global_state->Cast<DuckDBColumnsData>();
|
288
288
|
if (data.offset >= data.entries.size()) {
|
289
289
|
// finished returning values
|
290
290
|
return;
|
@@ -298,7 +298,7 @@ void DuckDBColumnsFunction(ClientContext &context, TableFunctionInput &data_p, D
|
|
298
298
|
idx_t column_offset = data.column_offset;
|
299
299
|
idx_t index = 0;
|
300
300
|
while (next < data.entries.size() && index < STANDARD_VECTOR_SIZE) {
|
301
|
-
auto column_helper = ColumnHelper::Create(data.entries[next]);
|
301
|
+
auto column_helper = ColumnHelper::Create(*data.entries[next]);
|
302
302
|
idx_t columns = column_helper->NumColumns();
|
303
303
|
|
304
304
|
// Check to see if we are going to exceed the maximum index for a DataChunk
|
@@ -19,7 +19,8 @@
|
|
19
19
|
namespace duckdb {
|
20
20
|
|
21
21
|
struct UniqueKeyInfo {
|
22
|
-
string schema
|
22
|
+
string schema;
|
23
|
+
string table;
|
23
24
|
vector<LogicalIndex> columns;
|
24
25
|
|
25
26
|
bool operator==(const UniqueKeyInfo &other) const {
|
@@ -52,7 +53,7 @@ struct DuckDBConstraintsData : public GlobalTableFunctionState {
|
|
52
53
|
DuckDBConstraintsData() : offset(0), constraint_offset(0), unique_constraint_offset(0) {
|
53
54
|
}
|
54
55
|
|
55
|
-
vector<CatalogEntry
|
56
|
+
vector<optional_ptr<CatalogEntry>> entries;
|
56
57
|
idx_t offset;
|
57
58
|
idx_t constraint_offset;
|
58
59
|
idx_t unique_constraint_offset;
|
@@ -125,7 +126,7 @@ unique_ptr<GlobalTableFunctionState> DuckDBConstraintsInit(ClientContext &contex
|
|
125
126
|
}
|
126
127
|
|
127
128
|
void DuckDBConstraintsFunction(ClientContext &context, TableFunctionInput &data_p, DataChunk &output) {
|
128
|
-
auto &data =
|
129
|
+
auto &data = data_p.global_state->Cast<DuckDBConstraintsData>();
|
129
130
|
if (data.offset >= data.entries.size()) {
|
130
131
|
// finished returning values
|
131
132
|
return;
|
@@ -134,10 +135,10 @@ void DuckDBConstraintsFunction(ClientContext &context, TableFunctionInput &data_
|
|
134
135
|
// either fill up the chunk or return all the remaining columns
|
135
136
|
idx_t count = 0;
|
136
137
|
while (data.offset < data.entries.size() && count < STANDARD_VECTOR_SIZE) {
|
137
|
-
auto &entry = data.entries[data.offset];
|
138
|
-
D_ASSERT(entry
|
138
|
+
auto &entry = *data.entries[data.offset];
|
139
|
+
D_ASSERT(entry.type == CatalogType::TABLE_ENTRY);
|
139
140
|
|
140
|
-
auto &table = entry
|
141
|
+
auto &table = entry.Cast<TableCatalogEntry>();
|
141
142
|
auto &constraints = table.GetConstraints();
|
142
143
|
bool is_duck_table = table.IsDuckTable();
|
143
144
|
for (; data.constraint_offset < constraints.size() && count < STANDARD_VECTOR_SIZE; data.constraint_offset++) {
|
@@ -9,8 +9,12 @@
|
|
9
9
|
namespace duckdb {
|
10
10
|
|
11
11
|
struct DependencyInformation {
|
12
|
-
CatalogEntry
|
13
|
-
|
12
|
+
DependencyInformation(CatalogEntry &object, CatalogEntry &dependent, DependencyType type)
|
13
|
+
: object(object), dependent(dependent), type(type) {
|
14
|
+
}
|
15
|
+
|
16
|
+
CatalogEntry &object;
|
17
|
+
CatalogEntry &dependent;
|
14
18
|
DependencyType type;
|
15
19
|
};
|
16
20
|
|
@@ -54,14 +58,10 @@ unique_ptr<GlobalTableFunctionState> DuckDBDependenciesInit(ClientContext &conte
|
|
54
58
|
// scan all the schemas and collect them
|
55
59
|
auto &catalog = Catalog::GetCatalog(context, INVALID_CATALOG);
|
56
60
|
if (catalog.IsDuckCatalog()) {
|
57
|
-
auto &duck_catalog = (
|
61
|
+
auto &duck_catalog = catalog.Cast<DuckCatalog>();
|
58
62
|
auto &dependency_manager = duck_catalog.GetDependencyManager();
|
59
|
-
dependency_manager.Scan([&](CatalogEntry
|
60
|
-
|
61
|
-
info.object = obj;
|
62
|
-
info.dependent = dependent;
|
63
|
-
info.type = type;
|
64
|
-
result->entries.push_back(info);
|
63
|
+
dependency_manager.Scan([&](CatalogEntry &obj, CatalogEntry &dependent, DependencyType type) {
|
64
|
+
result->entries.emplace_back(obj, dependent, type);
|
65
65
|
});
|
66
66
|
}
|
67
67
|
|
@@ -84,13 +84,13 @@ void DuckDBDependenciesFunction(ClientContext &context, TableFunctionInput &data
|
|
84
84
|
// classid, LogicalType::BIGINT
|
85
85
|
output.SetValue(0, count, Value::BIGINT(0));
|
86
86
|
// objid, LogicalType::BIGINT
|
87
|
-
output.SetValue(1, count, Value::BIGINT(entry.object
|
87
|
+
output.SetValue(1, count, Value::BIGINT(entry.object.oid));
|
88
88
|
// objsubid, LogicalType::INTEGER
|
89
89
|
output.SetValue(2, count, Value::INTEGER(0));
|
90
90
|
// refclassid, LogicalType::BIGINT
|
91
91
|
output.SetValue(3, count, Value::BIGINT(0));
|
92
92
|
// refobjid, LogicalType::BIGINT
|
93
|
-
output.SetValue(4, count, Value::BIGINT(entry.dependent
|
93
|
+
output.SetValue(4, count, Value::BIGINT(entry.dependent.oid));
|
94
94
|
// refobjsubid, LogicalType::INTEGER
|
95
95
|
output.SetValue(5, count, Value::INTEGER(0));
|
96
96
|
// deptype, LogicalType::VARCHAR
|
@@ -17,10 +17,10 @@
|
|
17
17
|
namespace duckdb {
|
18
18
|
|
19
19
|
struct PragmaTableFunctionData : public TableFunctionData {
|
20
|
-
explicit PragmaTableFunctionData(CatalogEntry
|
20
|
+
explicit PragmaTableFunctionData(CatalogEntry &entry_p) : entry(entry_p) {
|
21
21
|
}
|
22
22
|
|
23
|
-
CatalogEntry
|
23
|
+
CatalogEntry &entry;
|
24
24
|
};
|
25
25
|
|
26
26
|
struct PragmaTableOperatorData : public GlobalTableFunctionState {
|
@@ -55,20 +55,20 @@ static unique_ptr<FunctionData> PragmaTableInfoBind(ClientContext &context, Tabl
|
|
55
55
|
// look up the table name in the catalog
|
56
56
|
Binder::BindSchemaOrCatalog(context, qname.catalog, qname.schema);
|
57
57
|
auto entry = Catalog::GetEntry(context, CatalogType::TABLE_ENTRY, qname.catalog, qname.schema, qname.name);
|
58
|
-
return make_uniq<PragmaTableFunctionData>(entry);
|
58
|
+
return make_uniq<PragmaTableFunctionData>(*entry);
|
59
59
|
}
|
60
60
|
|
61
61
|
unique_ptr<GlobalTableFunctionState> PragmaTableInfoInit(ClientContext &context, TableFunctionInitInput &input) {
|
62
62
|
return make_uniq<PragmaTableOperatorData>();
|
63
63
|
}
|
64
64
|
|
65
|
-
static void CheckConstraints(TableCatalogEntry
|
65
|
+
static void CheckConstraints(TableCatalogEntry &table, const ColumnDefinition &column, bool &out_not_null,
|
66
66
|
bool &out_pk) {
|
67
67
|
out_not_null = false;
|
68
68
|
out_pk = false;
|
69
69
|
// check all constraints
|
70
70
|
// FIXME: this is pretty inefficient, it probably doesn't matter
|
71
|
-
for (auto &constraint : table
|
71
|
+
for (auto &constraint : table.GetConstraints()) {
|
72
72
|
switch (constraint->type) {
|
73
73
|
case ConstraintType::NOT_NULL: {
|
74
74
|
auto ¬_null = constraint->Cast<NotNullConstraint>();
|
@@ -79,7 +79,6 @@ static void CheckConstraints(TableCatalogEntry *table, const ColumnDefinition &c
|
|
79
79
|
}
|
80
80
|
case ConstraintType::UNIQUE: {
|
81
81
|
auto &unique = constraint->Cast<UniqueConstraint>();
|
82
|
-
|
83
82
|
if (unique.is_primary_key) {
|
84
83
|
if (unique.index == column.Logical()) {
|
85
84
|
out_pk = true;
|
@@ -96,20 +95,20 @@ static void CheckConstraints(TableCatalogEntry *table, const ColumnDefinition &c
|
|
96
95
|
}
|
97
96
|
}
|
98
97
|
|
99
|
-
static void PragmaTableInfoTable(PragmaTableOperatorData &data, TableCatalogEntry
|
100
|
-
if (data.offset >= table
|
98
|
+
static void PragmaTableInfoTable(PragmaTableOperatorData &data, TableCatalogEntry &table, DataChunk &output) {
|
99
|
+
if (data.offset >= table.GetColumns().LogicalColumnCount()) {
|
101
100
|
// finished returning values
|
102
101
|
return;
|
103
102
|
}
|
104
103
|
// start returning values
|
105
104
|
// either fill up the chunk or return all the remaining columns
|
106
|
-
idx_t next = MinValue<idx_t>(data.offset + STANDARD_VECTOR_SIZE, table
|
105
|
+
idx_t next = MinValue<idx_t>(data.offset + STANDARD_VECTOR_SIZE, table.GetColumns().LogicalColumnCount());
|
107
106
|
output.SetCardinality(next - data.offset);
|
108
107
|
|
109
108
|
for (idx_t i = data.offset; i < next; i++) {
|
110
109
|
bool not_null, pk;
|
111
110
|
auto index = i - data.offset;
|
112
|
-
auto &column = table
|
111
|
+
auto &column = table.GetColumn(LogicalIndex(i));
|
113
112
|
D_ASSERT(column.Oid() < (idx_t)NumericLimits<int32_t>::Maximum());
|
114
113
|
CheckConstraints(table, column, not_null, pk);
|
115
114
|
|
@@ -131,20 +130,20 @@ static void PragmaTableInfoTable(PragmaTableOperatorData &data, TableCatalogEntr
|
|
131
130
|
data.offset = next;
|
132
131
|
}
|
133
132
|
|
134
|
-
static void PragmaTableInfoView(PragmaTableOperatorData &data, ViewCatalogEntry
|
135
|
-
if (data.offset >= view
|
133
|
+
static void PragmaTableInfoView(PragmaTableOperatorData &data, ViewCatalogEntry &view, DataChunk &output) {
|
134
|
+
if (data.offset >= view.types.size()) {
|
136
135
|
// finished returning values
|
137
136
|
return;
|
138
137
|
}
|
139
138
|
// start returning values
|
140
139
|
// either fill up the chunk or return all the remaining columns
|
141
|
-
idx_t next = MinValue<idx_t>(data.offset + STANDARD_VECTOR_SIZE, view
|
140
|
+
idx_t next = MinValue<idx_t>(data.offset + STANDARD_VECTOR_SIZE, view.types.size());
|
142
141
|
output.SetCardinality(next - data.offset);
|
143
142
|
|
144
143
|
for (idx_t i = data.offset; i < next; i++) {
|
145
144
|
auto index = i - data.offset;
|
146
|
-
auto type = view
|
147
|
-
auto &name = view
|
145
|
+
auto type = view.types[i];
|
146
|
+
auto &name = view.aliases[i];
|
148
147
|
// return values:
|
149
148
|
// "cid", PhysicalType::INT32
|
150
149
|
|
@@ -166,12 +165,12 @@ static void PragmaTableInfoView(PragmaTableOperatorData &data, ViewCatalogEntry
|
|
166
165
|
static void PragmaTableInfoFunction(ClientContext &context, TableFunctionInput &data_p, DataChunk &output) {
|
167
166
|
auto &bind_data = (PragmaTableFunctionData &)*data_p.bind_data;
|
168
167
|
auto &state = data_p.global_state->Cast<PragmaTableOperatorData>();
|
169
|
-
switch (bind_data.entry
|
168
|
+
switch (bind_data.entry.type) {
|
170
169
|
case CatalogType::TABLE_ENTRY:
|
171
|
-
PragmaTableInfoTable(state,
|
170
|
+
PragmaTableInfoTable(state, bind_data.entry.Cast<TableCatalogEntry>(), output);
|
172
171
|
break;
|
173
172
|
case CatalogType::VIEW_ENTRY:
|
174
|
-
PragmaTableInfoView(state,
|
173
|
+
PragmaTableInfoView(state, bind_data.entry.Cast<ViewCatalogEntry>(), output);
|
175
174
|
break;
|
176
175
|
default:
|
177
176
|
throw NotImplementedException("Unimplemented catalog type for pragma_table_info");
|
@@ -185,7 +185,7 @@ BindInfo TableScanGetBindInfo(const FunctionData *bind_data) {
|
|
185
185
|
|
186
186
|
void TableScanDependency(DependencyList &entries, const FunctionData *bind_data_p) {
|
187
187
|
auto &bind_data = bind_data_p->Cast<TableScanBindData>();
|
188
|
-
entries.AddDependency(bind_data.table);
|
188
|
+
entries.AddDependency(*bind_data.table);
|
189
189
|
}
|
190
190
|
|
191
191
|
unique_ptr<NodeStatistics> TableScanCardinality(ClientContext &context, const FunctionData *bind_data_p) {
|
@@ -302,30 +302,27 @@ void TableScanPushdownComplexFilter(ClientContext &context, LogicalGet &get, Fun
|
|
302
302
|
ExpressionType low_comparison_type = ExpressionType::INVALID, high_comparison_type = ExpressionType::INVALID;
|
303
303
|
// try to find a matching index for any of the filter expressions
|
304
304
|
for (auto &filter : filters) {
|
305
|
-
auto expr = filter
|
305
|
+
auto &expr = *filter;
|
306
306
|
|
307
307
|
// create a matcher for a comparison with a constant
|
308
308
|
ComparisonExpressionMatcher matcher;
|
309
309
|
// match on a comparison type
|
310
310
|
matcher.expr_type = make_uniq<ComparisonExpressionTypeMatcher>();
|
311
311
|
// match on a constant comparison with the indexed expression
|
312
|
-
matcher.matchers.push_back(make_uniq<ExpressionEqualityMatcher>(index_expression
|
312
|
+
matcher.matchers.push_back(make_uniq<ExpressionEqualityMatcher>(*index_expression));
|
313
313
|
matcher.matchers.push_back(make_uniq<ConstantExpressionMatcher>());
|
314
314
|
|
315
315
|
matcher.policy = SetMatcher::Policy::UNORDERED;
|
316
316
|
|
317
|
-
vector<Expression
|
317
|
+
vector<reference<Expression>> bindings;
|
318
318
|
if (matcher.Match(expr, bindings)) {
|
319
319
|
// range or equality comparison with constant value
|
320
320
|
// we can use our index here
|
321
321
|
// bindings[0] = the expression
|
322
322
|
// bindings[1] = the index expression
|
323
323
|
// bindings[2] = the constant
|
324
|
-
auto &comparison = bindings[0]
|
325
|
-
|
326
|
-
D_ASSERT(bindings[2]->type == ExpressionType::VALUE_CONSTANT);
|
327
|
-
|
328
|
-
auto constant_value = bindings[2]->Cast<BoundConstantExpression>().value;
|
324
|
+
auto &comparison = bindings[0].get().Cast<BoundComparisonExpression>();
|
325
|
+
auto constant_value = bindings[2].get().Cast<BoundConstantExpression>().value;
|
329
326
|
auto comparison_type = comparison.type;
|
330
327
|
if (comparison.left->type == ExpressionType::VALUE_CONSTANT) {
|
331
328
|
// the expression is on the right side, we flip them around
|
@@ -346,9 +343,9 @@ void TableScanPushdownComplexFilter(ClientContext &context, LogicalGet &get, Fun
|
|
346
343
|
high_value = constant_value;
|
347
344
|
high_comparison_type = comparison_type;
|
348
345
|
}
|
349
|
-
} else if (expr
|
346
|
+
} else if (expr.type == ExpressionType::COMPARE_BETWEEN) {
|
350
347
|
// BETWEEN expression
|
351
|
-
auto &between = expr
|
348
|
+
auto &between = expr.Cast<BoundBetweenExpression>();
|
352
349
|
if (!between.input->Equals(index_expression.get())) {
|
353
350
|
// expression doesn't match the current index expression
|
354
351
|
continue;
|