duckdb 0.8.2-dev87.0 → 0.9.1-dev0.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.
Files changed (1321) hide show
  1. package/README.md +7 -0
  2. package/binding.gyp +30 -15
  3. package/binding.gyp.in +1 -2
  4. package/configure.py +11 -3
  5. package/duckdb_extension_config.cmake +10 -0
  6. package/lib/duckdb.js +14 -4
  7. package/package.json +3 -1
  8. package/src/connection.cpp +67 -38
  9. package/src/data_chunk.cpp +1 -3
  10. package/src/database.cpp +9 -17
  11. package/src/duckdb/extension/icu/icu-dateadd.cpp +13 -21
  12. package/src/duckdb/extension/icu/icu-datefunc.cpp +10 -1
  13. package/src/duckdb/extension/icu/icu-datepart.cpp +216 -100
  14. package/src/duckdb/extension/icu/icu-datesub.cpp +13 -17
  15. package/src/duckdb/extension/icu/icu-datetrunc.cpp +8 -9
  16. package/src/duckdb/extension/icu/icu-list-range.cpp +7 -9
  17. package/src/duckdb/extension/icu/icu-makedate.cpp +27 -16
  18. package/src/duckdb/extension/icu/icu-strptime.cpp +36 -65
  19. package/src/duckdb/extension/icu/icu-table-range.cpp +11 -14
  20. package/src/duckdb/extension/icu/icu-timebucket.cpp +21 -23
  21. package/src/duckdb/extension/icu/icu-timezone.cpp +26 -37
  22. package/src/duckdb/extension/icu/icu_extension.cpp +29 -36
  23. package/src/duckdb/extension/icu/include/icu-dateadd.hpp +1 -1
  24. package/src/duckdb/extension/icu/include/icu-datepart.hpp +1 -1
  25. package/src/duckdb/extension/icu/include/icu-datesub.hpp +1 -1
  26. package/src/duckdb/extension/icu/include/icu-datetrunc.hpp +1 -1
  27. package/src/duckdb/extension/icu/include/icu-list-range.hpp +1 -1
  28. package/src/duckdb/extension/icu/include/icu-makedate.hpp +1 -1
  29. package/src/duckdb/extension/icu/include/icu-strptime.hpp +1 -1
  30. package/src/duckdb/extension/icu/include/icu-table-range.hpp +1 -1
  31. package/src/duckdb/extension/icu/include/icu-timebucket.hpp +1 -1
  32. package/src/duckdb/extension/icu/include/icu-timezone.hpp +1 -1
  33. package/src/duckdb/extension/icu/third_party/icu/i18n/unicode/gregocal.h +1 -1
  34. package/src/duckdb/extension/json/buffered_json_reader.cpp +82 -92
  35. package/src/duckdb/extension/json/include/buffered_json_reader.hpp +38 -52
  36. package/src/duckdb/extension/json/include/json_common.hpp +47 -231
  37. package/src/duckdb/extension/json/include/json_deserializer.hpp +9 -18
  38. package/src/duckdb/extension/json/include/json_enums.hpp +60 -0
  39. package/src/duckdb/extension/json/include/json_executors.hpp +49 -13
  40. package/src/duckdb/extension/json/include/json_functions.hpp +2 -1
  41. package/src/duckdb/extension/json/include/json_scan.hpp +21 -21
  42. package/src/duckdb/extension/json/include/json_serializer.hpp +12 -18
  43. package/src/duckdb/extension/json/include/json_transform.hpp +2 -2
  44. package/src/duckdb/extension/json/json_common.cpp +272 -40
  45. package/src/duckdb/extension/json/json_deserializer.cpp +37 -73
  46. package/src/duckdb/extension/json/json_enums.cpp +105 -0
  47. package/src/duckdb/extension/json/json_functions/copy_json.cpp +1 -1
  48. package/src/duckdb/extension/json/json_functions/json_create.cpp +21 -2
  49. package/src/duckdb/extension/json/json_functions/json_serialize_sql.cpp +4 -1
  50. package/src/duckdb/extension/json/json_functions/json_structure.cpp +1 -1
  51. package/src/duckdb/extension/json/json_functions/json_transform.cpp +98 -57
  52. package/src/duckdb/extension/json/json_functions/json_type.cpp +1 -1
  53. package/src/duckdb/extension/json/json_functions/read_json.cpp +21 -2
  54. package/src/duckdb/extension/json/json_functions.cpp +30 -31
  55. package/src/duckdb/extension/json/json_scan.cpp +148 -175
  56. package/src/duckdb/extension/json/json_serializer.cpp +29 -72
  57. package/src/duckdb/extension/json/serialize_json.cpp +92 -0
  58. package/src/duckdb/extension/parquet/column_reader.cpp +37 -25
  59. package/src/duckdb/extension/parquet/column_writer.cpp +121 -104
  60. package/src/duckdb/extension/parquet/include/cast_column_reader.hpp +2 -2
  61. package/src/duckdb/extension/parquet/include/column_reader.hpp +14 -16
  62. package/src/duckdb/extension/parquet/include/column_writer.hpp +11 -9
  63. package/src/duckdb/extension/parquet/include/list_column_reader.hpp +2 -2
  64. package/src/duckdb/extension/parquet/include/parquet_dbp_decoder.hpp +3 -3
  65. package/src/duckdb/extension/parquet/include/parquet_decimal_utils.hpp +3 -3
  66. package/src/duckdb/extension/parquet/include/parquet_file_metadata_cache.hpp +2 -2
  67. package/src/duckdb/extension/parquet/include/parquet_reader.hpp +3 -2
  68. package/src/duckdb/extension/parquet/include/parquet_rle_bp_encoder.hpp +4 -4
  69. package/src/duckdb/extension/parquet/include/parquet_statistics.hpp +2 -2
  70. package/src/duckdb/extension/parquet/include/parquet_support.hpp +9 -11
  71. package/src/duckdb/extension/parquet/include/parquet_timestamp.hpp +1 -0
  72. package/src/duckdb/extension/parquet/include/parquet_writer.hpp +29 -5
  73. package/src/duckdb/extension/parquet/include/string_column_reader.hpp +1 -1
  74. package/src/duckdb/extension/parquet/include/struct_column_reader.hpp +2 -3
  75. package/src/duckdb/extension/parquet/include/zstd_file_system.hpp +2 -2
  76. package/src/duckdb/extension/parquet/parquet_extension.cpp +340 -93
  77. package/src/duckdb/extension/parquet/parquet_reader.cpp +10 -10
  78. package/src/duckdb/extension/parquet/parquet_statistics.cpp +25 -8
  79. package/src/duckdb/extension/parquet/parquet_timestamp.cpp +6 -0
  80. package/src/duckdb/extension/parquet/parquet_writer.cpp +170 -33
  81. package/src/duckdb/extension/parquet/serialize_parquet.cpp +26 -0
  82. package/src/duckdb/extension/parquet/zstd_file_system.cpp +2 -2
  83. package/src/duckdb/src/catalog/catalog.cpp +177 -70
  84. package/src/duckdb/src/catalog/catalog_entry/duck_index_entry.cpp +5 -0
  85. package/src/duckdb/src/catalog/catalog_entry/duck_table_entry.cpp +8 -11
  86. package/src/duckdb/src/catalog/catalog_entry/index_catalog_entry.cpp +17 -42
  87. package/src/duckdb/src/catalog/catalog_entry/macro_catalog_entry.cpp +2 -11
  88. package/src/duckdb/src/catalog/catalog_entry/schema_catalog_entry.cpp +4 -15
  89. package/src/duckdb/src/catalog/catalog_entry/sequence_catalog_entry.cpp +11 -29
  90. package/src/duckdb/src/catalog/catalog_entry/table_catalog_entry.cpp +11 -44
  91. package/src/duckdb/src/catalog/catalog_entry/type_catalog_entry.cpp +8 -29
  92. package/src/duckdb/src/catalog/catalog_entry/view_catalog_entry.cpp +11 -28
  93. package/src/duckdb/src/catalog/catalog_entry.cpp +16 -1
  94. package/src/duckdb/src/catalog/catalog_search_path.cpp +5 -4
  95. package/src/duckdb/src/catalog/catalog_set.cpp +16 -77
  96. package/src/duckdb/src/catalog/default/default_functions.cpp +23 -2
  97. package/src/duckdb/src/catalog/dependency_manager.cpp +0 -36
  98. package/src/duckdb/src/catalog/duck_catalog.cpp +4 -0
  99. package/src/duckdb/src/common/adbc/adbc.cpp +541 -171
  100. package/src/duckdb/src/common/adbc/driver_manager.cpp +92 -39
  101. package/src/duckdb/src/common/adbc/nanoarrow/allocator.cpp +57 -0
  102. package/src/duckdb/src/common/adbc/nanoarrow/metadata.cpp +121 -0
  103. package/src/duckdb/src/common/adbc/nanoarrow/schema.cpp +474 -0
  104. package/src/duckdb/src/common/adbc/nanoarrow/single_batch_array_stream.cpp +84 -0
  105. package/src/duckdb/src/common/allocator.cpp +14 -2
  106. package/src/duckdb/src/common/arrow/appender/bool_data.cpp +44 -0
  107. package/src/duckdb/src/common/arrow/appender/list_data.cpp +78 -0
  108. package/src/duckdb/src/common/arrow/appender/map_data.cpp +86 -0
  109. package/src/duckdb/src/common/arrow/appender/struct_data.cpp +45 -0
  110. package/src/duckdb/src/common/arrow/appender/union_data.cpp +70 -0
  111. package/src/duckdb/src/common/arrow/arrow_appender.cpp +95 -666
  112. package/src/duckdb/src/common/arrow/arrow_converter.cpp +68 -38
  113. package/src/duckdb/src/common/arrow/arrow_wrapper.cpp +37 -42
  114. package/src/duckdb/src/common/assert.cpp +3 -0
  115. package/src/duckdb/src/common/constants.cpp +2 -2
  116. package/src/duckdb/src/common/crypto/md5.cpp +2 -12
  117. package/src/duckdb/src/common/enum_util.cpp +4967 -4412
  118. package/src/duckdb/src/common/enums/date_part_specifier.cpp +2 -0
  119. package/src/duckdb/src/common/enums/logical_operator_type.cpp +4 -0
  120. package/src/duckdb/src/common/enums/optimizer_type.cpp +2 -0
  121. package/src/duckdb/src/common/enums/physical_operator_type.cpp +4 -0
  122. package/src/duckdb/src/common/exception.cpp +15 -2
  123. package/src/duckdb/src/common/extra_type_info.cpp +315 -0
  124. package/src/duckdb/src/common/file_buffer.cpp +2 -2
  125. package/src/duckdb/src/common/file_system.cpp +46 -12
  126. package/src/duckdb/src/common/filename_pattern.cpp +1 -1
  127. package/src/duckdb/src/common/gzip_file_system.cpp +7 -12
  128. package/src/duckdb/src/common/hive_partitioning.cpp +10 -6
  129. package/src/duckdb/src/common/http_state.cpp +78 -0
  130. package/src/duckdb/src/common/local_file_system.cpp +36 -28
  131. package/src/duckdb/src/common/multi_file_reader.cpp +175 -70
  132. package/src/duckdb/src/common/operator/cast_operators.cpp +110 -1
  133. package/src/duckdb/src/common/operator/string_cast.cpp +45 -8
  134. package/src/duckdb/src/common/radix_partitioning.cpp +34 -39
  135. package/src/duckdb/src/common/re2_regex.cpp +1 -1
  136. package/src/duckdb/src/common/row_operations/row_aggregate.cpp +18 -3
  137. package/src/duckdb/src/common/row_operations/row_external.cpp +1 -1
  138. package/src/duckdb/src/common/row_operations/row_matcher.cpp +375 -0
  139. package/src/duckdb/src/common/serializer/binary_deserializer.cpp +63 -73
  140. package/src/duckdb/src/common/serializer/binary_serializer.cpp +88 -78
  141. package/src/duckdb/src/common/serializer/buffered_file_reader.cpp +3 -20
  142. package/src/duckdb/src/common/serializer/buffered_file_writer.cpp +1 -1
  143. package/src/duckdb/src/common/serializer/memory_stream.cpp +61 -0
  144. package/src/duckdb/src/common/serializer/serializer.cpp +15 -0
  145. package/src/duckdb/src/common/sort/merge_sorter.cpp +9 -16
  146. package/src/duckdb/src/common/sort/partition_state.cpp +210 -100
  147. package/src/duckdb/src/common/sort/sort_state.cpp +1 -1
  148. package/src/duckdb/src/common/sort/sorted_block.cpp +1 -1
  149. package/src/duckdb/src/common/types/batched_data_collection.cpp +7 -2
  150. package/src/duckdb/src/common/types/bit.cpp +51 -0
  151. package/src/duckdb/src/common/types/column/column_data_allocator.cpp +9 -6
  152. package/src/duckdb/src/common/types/column/column_data_collection.cpp +68 -2
  153. package/src/duckdb/src/common/types/column/column_data_collection_segment.cpp +20 -6
  154. package/src/duckdb/src/common/types/column/partitioned_column_data.cpp +2 -2
  155. package/src/duckdb/src/common/types/data_chunk.cpp +77 -36
  156. package/src/duckdb/src/common/types/date.cpp +15 -0
  157. package/src/duckdb/src/common/types/hugeint.cpp +40 -0
  158. package/src/duckdb/src/common/types/hyperloglog.cpp +9 -7
  159. package/src/duckdb/src/common/types/interval.cpp +6 -0
  160. package/src/duckdb/src/common/types/list_segment.cpp +56 -198
  161. package/src/duckdb/src/common/types/row/partitioned_tuple_data.cpp +251 -131
  162. package/src/duckdb/src/common/types/row/row_data_collection_scanner.cpp +35 -5
  163. package/src/duckdb/src/common/types/row/row_layout.cpp +3 -31
  164. package/src/duckdb/src/common/types/row/tuple_data_allocator.cpp +43 -35
  165. package/src/duckdb/src/common/types/row/tuple_data_collection.cpp +74 -41
  166. package/src/duckdb/src/common/types/row/tuple_data_layout.cpp +11 -1
  167. package/src/duckdb/src/common/types/row/tuple_data_scatter_gather.cpp +46 -45
  168. package/src/duckdb/src/common/types/row/tuple_data_segment.cpp +21 -16
  169. package/src/duckdb/src/common/types/string_heap.cpp +4 -0
  170. package/src/duckdb/src/common/types/time.cpp +105 -0
  171. package/src/duckdb/src/common/types/timestamp.cpp +7 -0
  172. package/src/duckdb/src/common/types/uuid.cpp +2 -2
  173. package/src/duckdb/src/common/types/validity_mask.cpp +89 -0
  174. package/src/duckdb/src/common/types/value.cpp +103 -189
  175. package/src/duckdb/src/common/types/vector.cpp +79 -216
  176. package/src/duckdb/src/common/types.cpp +50 -745
  177. package/src/duckdb/src/common/vector_operations/vector_hash.cpp +1 -0
  178. package/src/duckdb/src/common/virtual_file_system.cpp +142 -1
  179. package/src/duckdb/src/core_functions/aggregate/algebraic/avg.cpp +0 -1
  180. package/src/duckdb/src/core_functions/aggregate/distributive/string_agg.cpp +6 -7
  181. package/src/duckdb/src/core_functions/aggregate/holistic/approximate_quantile.cpp +11 -8
  182. package/src/duckdb/src/core_functions/aggregate/holistic/mode.cpp +6 -8
  183. package/src/duckdb/src/core_functions/aggregate/holistic/quantile.cpp +137 -92
  184. package/src/duckdb/src/core_functions/aggregate/holistic/reservoir_quantile.cpp +15 -10
  185. package/src/duckdb/src/core_functions/aggregate/nested/histogram.cpp +1 -0
  186. package/src/duckdb/src/core_functions/aggregate/nested/list.cpp +83 -59
  187. package/src/duckdb/src/core_functions/aggregate/regression/regr_avg.cpp +4 -4
  188. package/src/duckdb/src/core_functions/aggregate/regression/regr_intercept.cpp +4 -4
  189. package/src/duckdb/src/core_functions/aggregate/regression/regr_r2.cpp +5 -4
  190. package/src/duckdb/src/core_functions/aggregate/regression/regr_sxx_syy.cpp +8 -8
  191. package/src/duckdb/src/core_functions/aggregate/regression/regr_sxy.cpp +4 -3
  192. package/src/duckdb/src/core_functions/function_list.cpp +20 -6
  193. package/src/duckdb/src/core_functions/scalar/date/date_diff.cpp +2 -0
  194. package/src/duckdb/src/core_functions/scalar/date/date_part.cpp +450 -126
  195. package/src/duckdb/src/core_functions/scalar/date/date_sub.cpp +2 -0
  196. package/src/duckdb/src/core_functions/scalar/date/date_trunc.cpp +4 -0
  197. package/src/duckdb/src/core_functions/scalar/date/epoch.cpp +10 -24
  198. package/src/duckdb/src/core_functions/scalar/date/make_date.cpp +19 -4
  199. package/src/duckdb/src/core_functions/scalar/date/strftime.cpp +12 -2
  200. package/src/duckdb/src/core_functions/scalar/debug/vector_type.cpp +23 -0
  201. package/src/duckdb/src/core_functions/scalar/enum/enum_functions.cpp +16 -12
  202. package/src/duckdb/src/core_functions/scalar/generic/current_setting.cpp +3 -1
  203. package/src/duckdb/src/core_functions/scalar/generic/hash.cpp +3 -0
  204. package/src/duckdb/src/core_functions/scalar/list/array_slice.cpp +324 -82
  205. package/src/duckdb/src/core_functions/scalar/list/list_aggregates.cpp +25 -18
  206. package/src/duckdb/src/core_functions/scalar/list/list_cosine_similarity.cpp +78 -0
  207. package/src/duckdb/src/core_functions/scalar/list/list_distance.cpp +72 -0
  208. package/src/duckdb/src/core_functions/scalar/list/list_inner_product.cpp +70 -0
  209. package/src/duckdb/src/core_functions/scalar/list/list_lambdas.cpp +18 -6
  210. package/src/duckdb/src/core_functions/scalar/list/list_sort.cpp +10 -1
  211. package/src/duckdb/src/core_functions/scalar/map/map_concat.cpp +0 -2
  212. package/src/duckdb/src/core_functions/scalar/map/map_entries.cpp +2 -2
  213. package/src/duckdb/src/core_functions/scalar/string/repeat.cpp +8 -5
  214. package/src/duckdb/src/core_functions/scalar/string/sha256.cpp +32 -0
  215. package/src/duckdb/src/core_functions/scalar/string/to_base.cpp +66 -0
  216. package/src/duckdb/src/core_functions/scalar/struct/struct_pack.cpp +24 -14
  217. package/src/duckdb/src/core_functions/scalar/union/union_tag.cpp +1 -1
  218. package/src/duckdb/src/execution/aggregate_hashtable.cpp +229 -348
  219. package/src/duckdb/src/execution/column_binding_resolver.cpp +11 -7
  220. package/src/duckdb/src/execution/expression_executor/execute_parameter.cpp +2 -2
  221. package/src/duckdb/src/execution/expression_executor.cpp +1 -1
  222. package/src/duckdb/src/execution/index/art/art.cpp +279 -296
  223. package/src/duckdb/src/execution/index/art/art_key.cpp +0 -11
  224. package/src/duckdb/src/execution/index/art/iterator.cpp +127 -217
  225. package/src/duckdb/src/execution/index/art/leaf.cpp +228 -272
  226. package/src/duckdb/src/execution/index/art/node.cpp +225 -313
  227. package/src/duckdb/src/execution/index/art/node16.cpp +34 -75
  228. package/src/duckdb/src/execution/index/art/node256.cpp +40 -63
  229. package/src/duckdb/src/execution/index/art/node4.cpp +44 -77
  230. package/src/duckdb/src/execution/index/art/node48.cpp +45 -79
  231. package/src/duckdb/src/execution/index/art/prefix.cpp +246 -347
  232. package/src/duckdb/src/execution/index/fixed_size_allocator.cpp +323 -0
  233. package/src/duckdb/src/execution/index/fixed_size_buffer.cpp +285 -0
  234. package/src/duckdb/src/execution/join_hashtable.cpp +80 -69
  235. package/src/duckdb/src/execution/nested_loop_join/nested_loop_join_inner.cpp +20 -27
  236. package/src/duckdb/src/execution/nested_loop_join/nested_loop_join_mark.cpp +21 -9
  237. package/src/duckdb/src/execution/operator/aggregate/aggregate_object.cpp +1 -0
  238. package/src/duckdb/src/execution/operator/aggregate/physical_hash_aggregate.cpp +257 -324
  239. package/src/duckdb/src/execution/operator/aggregate/physical_perfecthash_aggregate.cpp +6 -4
  240. package/src/duckdb/src/execution/operator/aggregate/physical_streaming_window.cpp +8 -3
  241. package/src/duckdb/src/execution/operator/aggregate/physical_ungrouped_aggregate.cpp +231 -190
  242. package/src/duckdb/src/execution/operator/aggregate/physical_window.cpp +368 -1070
  243. package/src/duckdb/src/execution/operator/{persistent → csv_scanner}/base_csv_reader.cpp +162 -175
  244. package/src/duckdb/src/execution/operator/csv_scanner/buffered_csv_reader.cpp +434 -0
  245. package/src/duckdb/src/execution/operator/csv_scanner/csv_buffer.cpp +89 -0
  246. package/src/duckdb/src/execution/operator/csv_scanner/csv_buffer_manager.cpp +90 -0
  247. package/src/duckdb/src/execution/operator/csv_scanner/csv_file_handle.cpp +95 -0
  248. package/src/duckdb/src/execution/operator/csv_scanner/csv_reader_options.cpp +494 -0
  249. package/src/duckdb/src/execution/operator/csv_scanner/csv_state_machine.cpp +35 -0
  250. package/src/duckdb/src/execution/operator/csv_scanner/csv_state_machine_cache.cpp +106 -0
  251. package/src/duckdb/src/execution/operator/{persistent → csv_scanner}/parallel_csv_reader.cpp +69 -51
  252. package/src/duckdb/src/execution/operator/csv_scanner/sniffer/csv_sniffer.cpp +61 -0
  253. package/src/duckdb/src/execution/operator/csv_scanner/sniffer/dialect_detection.cpp +339 -0
  254. package/src/duckdb/src/execution/operator/csv_scanner/sniffer/header_detection.cpp +171 -0
  255. package/src/duckdb/src/execution/operator/csv_scanner/sniffer/type_detection.cpp +415 -0
  256. package/src/duckdb/src/execution/operator/csv_scanner/sniffer/type_refinement.cpp +196 -0
  257. package/src/duckdb/src/execution/operator/csv_scanner/sniffer/type_replacement.cpp +39 -0
  258. package/src/duckdb/src/execution/operator/filter/physical_filter.cpp +1 -1
  259. package/src/duckdb/src/execution/operator/helper/physical_batch_collector.cpp +12 -9
  260. package/src/duckdb/src/execution/operator/helper/physical_explain_analyze.cpp +2 -2
  261. package/src/duckdb/src/execution/operator/helper/physical_limit.cpp +10 -8
  262. package/src/duckdb/src/execution/operator/helper/physical_load.cpp +2 -1
  263. package/src/duckdb/src/execution/operator/helper/physical_materialized_collector.cpp +7 -5
  264. package/src/duckdb/src/execution/operator/helper/physical_reset.cpp +3 -1
  265. package/src/duckdb/src/execution/operator/helper/physical_set.cpp +3 -1
  266. package/src/duckdb/src/execution/operator/helper/physical_vacuum.cpp +7 -5
  267. package/src/duckdb/src/execution/operator/join/physical_asof_join.cpp +465 -289
  268. package/src/duckdb/src/execution/operator/join/physical_blockwise_nl_join.cpp +2 -2
  269. package/src/duckdb/src/execution/operator/join/physical_comparison_join.cpp +1 -2
  270. package/src/duckdb/src/execution/operator/join/physical_delim_join.cpp +13 -6
  271. package/src/duckdb/src/execution/operator/join/physical_hash_join.cpp +37 -19
  272. package/src/duckdb/src/execution/operator/join/physical_iejoin.cpp +35 -17
  273. package/src/duckdb/src/execution/operator/join/physical_join.cpp +1 -1
  274. package/src/duckdb/src/execution/operator/join/physical_nested_loop_join.cpp +7 -4
  275. package/src/duckdb/src/execution/operator/join/physical_piecewise_merge_join.cpp +31 -10
  276. package/src/duckdb/src/execution/operator/join/physical_range_join.cpp +41 -5
  277. package/src/duckdb/src/execution/operator/order/physical_order.cpp +7 -5
  278. package/src/duckdb/src/execution/operator/order/physical_top_n.cpp +7 -5
  279. package/src/duckdb/src/execution/operator/persistent/csv_rejects_table.cpp +48 -0
  280. package/src/duckdb/src/execution/operator/persistent/physical_batch_copy_to_file.cpp +14 -10
  281. package/src/duckdb/src/execution/operator/persistent/physical_batch_insert.cpp +69 -47
  282. package/src/duckdb/src/execution/operator/persistent/physical_copy_to_file.cpp +9 -7
  283. package/src/duckdb/src/execution/operator/persistent/physical_export.cpp +1 -1
  284. package/src/duckdb/src/execution/operator/persistent/physical_fixed_batch_copy.cpp +14 -12
  285. package/src/duckdb/src/execution/operator/persistent/physical_insert.cpp +12 -12
  286. package/src/duckdb/src/execution/operator/persistent/physical_update.cpp +4 -2
  287. package/src/duckdb/src/execution/operator/projection/physical_pivot.cpp +2 -1
  288. package/src/duckdb/src/execution/operator/projection/physical_unnest.cpp +24 -27
  289. package/src/duckdb/src/execution/operator/scan/physical_column_data_scan.cpp +19 -0
  290. package/src/duckdb/src/execution/operator/scan/physical_table_scan.cpp +7 -12
  291. package/src/duckdb/src/execution/operator/schema/physical_attach.cpp +6 -2
  292. package/src/duckdb/src/execution/operator/schema/physical_create_art_index.cpp +198 -0
  293. package/src/duckdb/src/execution/operator/schema/physical_create_type.cpp +2 -6
  294. package/src/duckdb/src/execution/operator/set/physical_cte.cpp +160 -0
  295. package/src/duckdb/src/execution/operator/set/physical_recursive_cte.cpp +16 -7
  296. package/src/duckdb/src/execution/perfect_aggregate_hashtable.cpp +37 -6
  297. package/src/duckdb/src/execution/physical_operator.cpp +3 -2
  298. package/src/duckdb/src/execution/physical_plan/plan_aggregate.cpp +43 -10
  299. package/src/duckdb/src/execution/physical_plan/plan_asof_join.cpp +57 -35
  300. package/src/duckdb/src/execution/physical_plan/plan_comparison_join.cpp +45 -25
  301. package/src/duckdb/src/execution/physical_plan/plan_create_index.cpp +45 -34
  302. package/src/duckdb/src/execution/physical_plan/plan_cte.cpp +33 -0
  303. package/src/duckdb/src/execution/physical_plan/plan_delim_join.cpp +2 -5
  304. package/src/duckdb/src/execution/physical_plan/plan_get.cpp +2 -2
  305. package/src/duckdb/src/execution/physical_plan/plan_recursive_cte.cpp +25 -4
  306. package/src/duckdb/src/execution/physical_plan_generator.cpp +6 -11
  307. package/src/duckdb/src/execution/radix_partitioned_hashtable.cpp +658 -363
  308. package/src/duckdb/src/execution/reservoir_sample.cpp +3 -9
  309. package/src/duckdb/src/execution/window_executor.cpp +1285 -0
  310. package/src/duckdb/src/execution/window_segment_tree.cpp +408 -144
  311. package/src/duckdb/src/function/aggregate/distributive/count.cpp +2 -13
  312. package/src/duckdb/src/function/aggregate/sorted_aggregate_function.cpp +6 -12
  313. package/src/duckdb/src/function/cast/bit_cast.cpp +34 -2
  314. package/src/duckdb/src/function/cast/blob_cast.cpp +3 -0
  315. package/src/duckdb/src/function/cast/cast_function_set.cpp +1 -0
  316. package/src/duckdb/src/function/cast/numeric_casts.cpp +2 -0
  317. package/src/duckdb/src/function/cast/string_cast.cpp +2 -2
  318. package/src/duckdb/src/function/cast/struct_cast.cpp +8 -0
  319. package/src/duckdb/src/function/cast/time_casts.cpp +19 -6
  320. package/src/duckdb/src/function/cast/union/from_struct.cpp +114 -0
  321. package/src/duckdb/src/function/cast/union_casts.cpp +25 -36
  322. package/src/duckdb/src/function/cast/vector_cast_helpers.cpp +15 -4
  323. package/src/duckdb/src/function/function.cpp +3 -1
  324. package/src/duckdb/src/function/function_binder.cpp +18 -8
  325. package/src/duckdb/src/function/macro_function.cpp +0 -42
  326. package/src/duckdb/src/function/pragma/pragma_functions.cpp +5 -0
  327. package/src/duckdb/src/function/pragma/pragma_queries.cpp +15 -1
  328. package/src/duckdb/src/function/scalar/compressed_materialization/compress_integral.cpp +214 -0
  329. package/src/duckdb/src/function/scalar/compressed_materialization/compress_string.cpp +250 -0
  330. package/src/duckdb/src/function/scalar/compressed_materialization_functions.cpp +29 -0
  331. package/src/duckdb/src/function/scalar/list/list_resize.cpp +162 -0
  332. package/src/duckdb/src/function/scalar/nested_functions.cpp +1 -0
  333. package/src/duckdb/src/function/scalar/operators/add.cpp +9 -0
  334. package/src/duckdb/src/function/scalar/operators/arithmetic.cpp +15 -14
  335. package/src/duckdb/src/function/scalar/strftime_format.cpp +33 -12
  336. package/src/duckdb/src/function/scalar/string/like.cpp +12 -7
  337. package/src/duckdb/src/function/scalar/string/suffix.cpp +1 -1
  338. package/src/duckdb/src/function/scalar/system/aggregate_export.cpp +19 -11
  339. package/src/duckdb/src/function/scalar_function.cpp +3 -19
  340. package/src/duckdb/src/function/scalar_macro_function.cpp +0 -10
  341. package/src/duckdb/src/function/table/arrow/arrow_duck_schema.cpp +57 -0
  342. package/src/duckdb/src/function/table/arrow.cpp +131 -100
  343. package/src/duckdb/src/function/table/arrow_conversion.cpp +95 -74
  344. package/src/duckdb/src/function/table/copy_csv.cpp +133 -127
  345. package/src/duckdb/src/function/table/read_csv.cpp +198 -381
  346. package/src/duckdb/src/function/table/system/duckdb_columns.cpp +3 -1
  347. package/src/duckdb/src/function/table/system/pragma_metadata_info.cpp +83 -0
  348. package/src/duckdb/src/function/table/system/pragma_storage_info.cpp +5 -0
  349. package/src/duckdb/src/function/table/system/test_all_types.cpp +48 -21
  350. package/src/duckdb/src/function/table/system_functions.cpp +2 -0
  351. package/src/duckdb/src/function/table/table_scan.cpp +29 -24
  352. package/src/duckdb/src/function/table/version/pragma_version.cpp +49 -2
  353. package/src/duckdb/src/function/table_macro_function.cpp +0 -10
  354. package/src/duckdb/src/include/duckdb/catalog/catalog.hpp +23 -5
  355. package/src/duckdb/src/include/duckdb/catalog/catalog_entry/duck_index_entry.hpp +2 -0
  356. package/src/duckdb/src/include/duckdb/catalog/catalog_entry/index_catalog_entry.hpp +3 -3
  357. package/src/duckdb/src/include/duckdb/catalog/catalog_entry/macro_catalog_entry.hpp +1 -4
  358. package/src/duckdb/src/include/duckdb/catalog/catalog_entry/schema_catalog_entry.hpp +2 -7
  359. package/src/duckdb/src/include/duckdb/catalog/catalog_entry/sequence_catalog_entry.hpp +1 -8
  360. package/src/duckdb/src/include/duckdb/catalog/catalog_entry/table_catalog_entry.hpp +2 -13
  361. package/src/duckdb/src/include/duckdb/catalog/catalog_entry/type_catalog_entry.hpp +1 -6
  362. package/src/duckdb/src/include/duckdb/catalog/catalog_entry/view_catalog_entry.hpp +2 -5
  363. package/src/duckdb/src/include/duckdb/catalog/catalog_entry.hpp +9 -0
  364. package/src/duckdb/src/include/duckdb/catalog/catalog_set.hpp +0 -6
  365. package/src/duckdb/src/include/duckdb/catalog/duck_catalog.hpp +1 -0
  366. package/src/duckdb/src/include/duckdb/common/adbc/adbc.h +1 -0
  367. package/src/duckdb/src/include/duckdb/common/adbc/adbc.hpp +4 -1
  368. package/src/duckdb/src/include/duckdb/common/adbc/single_batch_array_stream.hpp +16 -0
  369. package/src/duckdb/src/include/duckdb/common/allocator.hpp +2 -0
  370. package/src/duckdb/src/include/duckdb/common/arrow/appender/append_data.hpp +109 -0
  371. package/src/duckdb/src/include/duckdb/common/arrow/appender/bool_data.hpp +15 -0
  372. package/src/duckdb/src/include/duckdb/common/arrow/appender/enum_data.hpp +69 -0
  373. package/src/duckdb/src/include/duckdb/common/arrow/appender/list.hpp +8 -0
  374. package/src/duckdb/src/include/duckdb/common/arrow/appender/list_data.hpp +18 -0
  375. package/src/duckdb/src/include/duckdb/common/arrow/appender/map_data.hpp +18 -0
  376. package/src/duckdb/src/include/duckdb/common/arrow/appender/scalar_data.hpp +88 -0
  377. package/src/duckdb/src/include/duckdb/common/arrow/appender/struct_data.hpp +18 -0
  378. package/src/duckdb/src/include/duckdb/common/arrow/appender/union_data.hpp +21 -0
  379. package/src/duckdb/src/include/duckdb/common/arrow/appender/varchar_data.hpp +105 -0
  380. package/src/duckdb/src/include/duckdb/common/arrow/arrow_appender.hpp +9 -4
  381. package/src/duckdb/src/include/duckdb/common/arrow/arrow_converter.hpp +3 -5
  382. package/src/duckdb/src/include/duckdb/common/arrow/arrow_wrapper.hpp +5 -3
  383. package/src/duckdb/src/include/duckdb/common/arrow/nanoarrow/nanoarrow.h +462 -0
  384. package/src/duckdb/src/include/duckdb/common/arrow/nanoarrow/nanoarrow.hpp +14 -0
  385. package/src/duckdb/src/include/duckdb/common/arrow/result_arrow_wrapper.hpp +4 -0
  386. package/src/duckdb/src/include/duckdb/common/assert.hpp +1 -1
  387. package/src/duckdb/src/include/duckdb/common/bitpacking.hpp +70 -55
  388. package/src/duckdb/src/include/duckdb/common/box_renderer.hpp +1 -1
  389. package/src/duckdb/src/include/duckdb/common/bswap.hpp +42 -0
  390. package/src/duckdb/src/include/duckdb/common/case_insensitive_map.hpp +1 -0
  391. package/src/duckdb/src/include/duckdb/common/constants.hpp +6 -15
  392. package/src/duckdb/src/include/duckdb/common/dl.hpp +3 -1
  393. package/src/duckdb/src/include/duckdb/common/enum_util.hpp +689 -577
  394. package/src/duckdb/src/include/duckdb/common/enums/cte_materialize.hpp +21 -0
  395. package/src/duckdb/src/include/duckdb/common/enums/date_part_specifier.hpp +18 -2
  396. package/src/duckdb/src/include/duckdb/common/enums/index_type.hpp +4 -3
  397. package/src/duckdb/src/include/duckdb/common/enums/joinref_type.hpp +2 -1
  398. package/src/duckdb/src/include/duckdb/common/enums/logical_operator_type.hpp +2 -0
  399. package/src/duckdb/src/include/duckdb/common/enums/operator_result_type.hpp +5 -1
  400. package/src/duckdb/src/include/duckdb/common/enums/optimizer_type.hpp +2 -0
  401. package/src/duckdb/src/include/duckdb/common/enums/pending_execution_result.hpp +1 -1
  402. package/src/duckdb/src/include/duckdb/common/enums/physical_operator_type.hpp +2 -0
  403. package/src/duckdb/src/include/duckdb/common/exception.hpp +15 -1
  404. package/src/duckdb/src/include/duckdb/common/extra_operator_info.hpp +27 -0
  405. package/src/duckdb/src/include/duckdb/common/extra_type_info.hpp +185 -0
  406. package/src/duckdb/src/include/duckdb/common/file_opener.hpp +9 -0
  407. package/src/duckdb/src/include/duckdb/common/file_system.hpp +10 -8
  408. package/src/duckdb/src/include/duckdb/common/filename_pattern.hpp +1 -1
  409. package/src/duckdb/src/include/duckdb/common/fixed_size_map.hpp +208 -0
  410. package/src/duckdb/src/include/duckdb/common/helper.hpp +8 -3
  411. package/src/duckdb/src/include/duckdb/common/hive_partitioning.hpp +1 -1
  412. package/src/duckdb/src/include/duckdb/common/http_state.hpp +61 -28
  413. package/src/duckdb/src/include/duckdb/common/hugeint.hpp +15 -0
  414. package/src/duckdb/src/include/duckdb/common/index_vector.hpp +12 -0
  415. package/src/duckdb/src/include/duckdb/common/limits.hpp +52 -149
  416. package/src/duckdb/src/include/duckdb/common/multi_file_reader.hpp +14 -7
  417. package/src/duckdb/src/include/duckdb/common/multi_file_reader_options.hpp +10 -44
  418. package/src/duckdb/src/include/duckdb/common/mutex.hpp +3 -0
  419. package/src/duckdb/src/include/duckdb/common/numeric_utils.hpp +48 -0
  420. package/src/duckdb/src/include/duckdb/common/opener_file_system.hpp +6 -2
  421. package/src/duckdb/src/include/duckdb/common/operator/add.hpp +5 -2
  422. package/src/duckdb/src/include/duckdb/common/operator/cast_operators.hpp +92 -4
  423. package/src/duckdb/src/include/duckdb/common/operator/comparison_operators.hpp +38 -2
  424. package/src/duckdb/src/include/duckdb/common/operator/multiply.hpp +3 -2
  425. package/src/duckdb/src/include/duckdb/common/operator/numeric_cast.hpp +10 -0
  426. package/src/duckdb/src/include/duckdb/common/operator/string_cast.hpp +1 -1
  427. package/src/duckdb/src/include/duckdb/common/operator/subtract.hpp +3 -2
  428. package/src/duckdb/src/include/duckdb/common/optional_idx.hpp +4 -1
  429. package/src/duckdb/src/include/duckdb/common/perfect_map_set.hpp +2 -1
  430. package/src/duckdb/src/include/duckdb/common/printer.hpp +11 -0
  431. package/src/duckdb/src/include/duckdb/common/radix.hpp +9 -20
  432. package/src/duckdb/src/include/duckdb/common/radix_partitioning.hpp +6 -21
  433. package/src/duckdb/src/include/duckdb/common/row_operations/row_matcher.hpp +63 -0
  434. package/src/duckdb/src/include/duckdb/common/row_operations/row_operations.hpp +3 -3
  435. package/src/duckdb/src/include/duckdb/common/serializer/binary_deserializer.hpp +103 -41
  436. package/src/duckdb/src/include/duckdb/common/serializer/binary_serializer.hpp +59 -42
  437. package/src/duckdb/src/include/duckdb/common/serializer/buffered_file_reader.hpp +4 -10
  438. package/src/duckdb/src/include/duckdb/common/serializer/buffered_file_writer.hpp +3 -3
  439. package/src/duckdb/src/include/duckdb/common/serializer/deserialization_data.hpp +181 -0
  440. package/src/duckdb/src/include/duckdb/common/serializer/{format_deserializer.hpp → deserializer.hpp} +222 -134
  441. package/src/duckdb/src/include/duckdb/common/serializer/encoding_util.hpp +132 -0
  442. package/src/duckdb/src/include/duckdb/common/serializer/memory_stream.hpp +62 -0
  443. package/src/duckdb/src/include/duckdb/common/serializer/read_stream.hpp +38 -0
  444. package/src/duckdb/src/include/duckdb/common/serializer/serialization_traits.hpp +172 -19
  445. package/src/duckdb/src/include/duckdb/common/serializer/serializer.hpp +307 -0
  446. package/src/duckdb/src/include/duckdb/common/serializer/write_stream.hpp +36 -0
  447. package/src/duckdb/src/include/duckdb/common/shared_ptr.hpp +8 -0
  448. package/src/duckdb/src/include/duckdb/common/sort/partition_state.hpp +48 -17
  449. package/src/duckdb/src/include/duckdb/common/stack_checker.hpp +34 -0
  450. package/src/duckdb/src/include/duckdb/common/string_util.hpp +11 -0
  451. package/src/duckdb/src/include/duckdb/common/type_util.hpp +8 -0
  452. package/src/duckdb/src/include/duckdb/common/typedefs.hpp +8 -0
  453. package/src/duckdb/src/include/duckdb/common/types/batched_data_collection.hpp +3 -1
  454. package/src/duckdb/src/include/duckdb/common/types/bit.hpp +81 -0
  455. package/src/duckdb/src/include/duckdb/common/types/column/column_data_allocator.hpp +11 -1
  456. package/src/duckdb/src/include/duckdb/common/types/column/column_data_collection.hpp +12 -1
  457. package/src/duckdb/src/include/duckdb/common/types/column/column_data_collection_segment.hpp +3 -1
  458. package/src/duckdb/src/include/duckdb/common/types/column/column_data_scan_states.hpp +3 -1
  459. package/src/duckdb/src/include/duckdb/common/types/data_chunk.hpp +4 -6
  460. package/src/duckdb/src/include/duckdb/common/types/date.hpp +9 -5
  461. package/src/duckdb/src/include/duckdb/common/types/datetime.hpp +46 -3
  462. package/src/duckdb/src/include/duckdb/common/types/hyperloglog.hpp +6 -6
  463. package/src/duckdb/src/include/duckdb/common/types/interval.hpp +7 -0
  464. package/src/duckdb/src/include/duckdb/common/types/list_segment.hpp +11 -15
  465. package/src/duckdb/src/include/duckdb/common/types/row/partitioned_tuple_data.hpp +46 -11
  466. package/src/duckdb/src/include/duckdb/common/types/row/row_data_collection_scanner.hpp +10 -1
  467. package/src/duckdb/src/include/duckdb/common/types/row/row_layout.hpp +1 -23
  468. package/src/duckdb/src/include/duckdb/common/types/row/tuple_data_allocator.hpp +14 -8
  469. package/src/duckdb/src/include/duckdb/common/types/row/tuple_data_collection.hpp +15 -5
  470. package/src/duckdb/src/include/duckdb/common/types/row/tuple_data_layout.hpp +7 -0
  471. package/src/duckdb/src/include/duckdb/common/types/row/tuple_data_segment.hpp +13 -8
  472. package/src/duckdb/src/include/duckdb/common/types/row/tuple_data_states.hpp +8 -4
  473. package/src/duckdb/src/include/duckdb/common/types/string_heap.hpp +3 -0
  474. package/src/duckdb/src/include/duckdb/common/types/string_type.hpp +9 -0
  475. package/src/duckdb/src/include/duckdb/common/types/time.hpp +5 -0
  476. package/src/duckdb/src/include/duckdb/common/types/timestamp.hpp +16 -10
  477. package/src/duckdb/src/include/duckdb/common/types/validity_mask.hpp +7 -1
  478. package/src/duckdb/src/include/duckdb/common/types/value.hpp +8 -7
  479. package/src/duckdb/src/include/duckdb/common/types/vector.hpp +13 -11
  480. package/src/duckdb/src/include/duckdb/common/types.hpp +8 -35
  481. package/src/duckdb/src/include/duckdb/common/vector.hpp +2 -2
  482. package/src/duckdb/src/include/duckdb/common/vector_operations/aggregate_executor.hpp +7 -2
  483. package/src/duckdb/src/include/duckdb/common/virtual_file_system.hpp +40 -97
  484. package/src/duckdb/src/include/duckdb/core_functions/aggregate/algebraic/corr.hpp +4 -4
  485. package/src/duckdb/src/include/duckdb/core_functions/aggregate/algebraic/covar.hpp +3 -1
  486. package/src/duckdb/src/include/duckdb/core_functions/aggregate/algebraic_functions.hpp +3 -1
  487. package/src/duckdb/src/include/duckdb/core_functions/aggregate/distributive_functions.hpp +4 -2
  488. package/src/duckdb/src/include/duckdb/core_functions/aggregate/holistic_functions.hpp +3 -1
  489. package/src/duckdb/src/include/duckdb/core_functions/aggregate/nested_functions.hpp +3 -1
  490. package/src/duckdb/src/include/duckdb/core_functions/aggregate/regression/regr_count.hpp +1 -0
  491. package/src/duckdb/src/include/duckdb/core_functions/aggregate/regression/regr_slope.hpp +3 -3
  492. package/src/duckdb/src/include/duckdb/core_functions/aggregate/regression_functions.hpp +3 -1
  493. package/src/duckdb/src/include/duckdb/core_functions/scalar/bit_functions.hpp +7 -5
  494. package/src/duckdb/src/include/duckdb/core_functions/scalar/blob_functions.hpp +7 -5
  495. package/src/duckdb/src/include/duckdb/core_functions/scalar/date_functions.hpp +44 -15
  496. package/src/duckdb/src/include/duckdb/core_functions/scalar/debug_functions.hpp +27 -0
  497. package/src/duckdb/src/include/duckdb/core_functions/scalar/enum_functions.hpp +10 -8
  498. package/src/duckdb/src/include/duckdb/core_functions/scalar/generic_functions.hpp +15 -13
  499. package/src/duckdb/src/include/duckdb/core_functions/scalar/list_functions.hpp +66 -13
  500. package/src/duckdb/src/include/duckdb/core_functions/scalar/map_functions.hpp +6 -4
  501. package/src/duckdb/src/include/duckdb/core_functions/scalar/math_functions.hpp +38 -36
  502. package/src/duckdb/src/include/duckdb/core_functions/scalar/operators_functions.hpp +6 -4
  503. package/src/duckdb/src/include/duckdb/core_functions/scalar/random_functions.hpp +6 -4
  504. package/src/duckdb/src/include/duckdb/core_functions/scalar/string_functions.hpp +34 -14
  505. package/src/duckdb/src/include/duckdb/core_functions/scalar/struct_functions.hpp +10 -5
  506. package/src/duckdb/src/include/duckdb/core_functions/scalar/union_functions.hpp +5 -3
  507. package/src/duckdb/src/include/duckdb/execution/aggregate_hashtable.hpp +132 -131
  508. package/src/duckdb/src/include/duckdb/execution/executor.hpp +3 -0
  509. package/src/duckdb/src/include/duckdb/execution/index/art/art.hpp +27 -16
  510. package/src/duckdb/src/include/duckdb/execution/index/art/art_key.hpp +0 -1
  511. package/src/duckdb/src/include/duckdb/execution/index/art/iterator.hpp +36 -32
  512. package/src/duckdb/src/include/duckdb/execution/index/art/leaf.hpp +47 -56
  513. package/src/duckdb/src/include/duckdb/execution/index/art/node.hpp +67 -64
  514. package/src/duckdb/src/include/duckdb/execution/index/art/node16.hpp +17 -20
  515. package/src/duckdb/src/include/duckdb/execution/index/art/node256.hpp +17 -25
  516. package/src/duckdb/src/include/duckdb/execution/index/art/node4.hpp +19 -22
  517. package/src/duckdb/src/include/duckdb/execution/index/art/node48.hpp +17 -26
  518. package/src/duckdb/src/include/duckdb/execution/index/art/prefix.hpp +60 -55
  519. package/src/duckdb/src/include/duckdb/execution/index/fixed_size_allocator.hpp +120 -0
  520. package/src/duckdb/src/include/duckdb/execution/index/fixed_size_buffer.hpp +109 -0
  521. package/src/duckdb/src/include/duckdb/execution/index/index_pointer.hpp +96 -0
  522. package/src/duckdb/src/include/duckdb/execution/join_hashtable.hpp +14 -8
  523. package/src/duckdb/src/include/duckdb/execution/operator/aggregate/physical_hash_aggregate.hpp +8 -7
  524. package/src/duckdb/src/include/duckdb/execution/operator/aggregate/physical_perfecthash_aggregate.hpp +1 -1
  525. package/src/duckdb/src/include/duckdb/execution/operator/aggregate/physical_ungrouped_aggregate.hpp +3 -3
  526. package/src/duckdb/src/include/duckdb/execution/operator/aggregate/physical_window.hpp +6 -5
  527. package/src/duckdb/src/include/duckdb/execution/operator/helper/physical_batch_collector.hpp +2 -2
  528. package/src/duckdb/src/include/duckdb/execution/operator/helper/physical_explain_analyze.hpp +1 -1
  529. package/src/duckdb/src/include/duckdb/execution/operator/helper/physical_limit.hpp +1 -1
  530. package/src/duckdb/src/include/duckdb/execution/operator/helper/physical_materialized_collector.hpp +1 -1
  531. package/src/duckdb/src/include/duckdb/execution/operator/helper/physical_vacuum.hpp +2 -2
  532. package/src/duckdb/src/include/duckdb/execution/operator/join/physical_asof_join.hpp +5 -12
  533. package/src/duckdb/src/include/duckdb/execution/operator/join/physical_blockwise_nl_join.hpp +1 -1
  534. package/src/duckdb/src/include/duckdb/execution/operator/join/physical_delim_join.hpp +2 -2
  535. package/src/duckdb/src/include/duckdb/execution/operator/join/physical_hash_join.hpp +2 -2
  536. package/src/duckdb/src/include/duckdb/execution/operator/join/physical_iejoin.hpp +3 -3
  537. package/src/duckdb/src/include/duckdb/execution/operator/join/physical_nested_loop_join.hpp +2 -2
  538. package/src/duckdb/src/include/duckdb/execution/operator/join/physical_piecewise_merge_join.hpp +3 -3
  539. package/src/duckdb/src/include/duckdb/execution/operator/join/physical_range_join.hpp +12 -1
  540. package/src/duckdb/src/include/duckdb/execution/operator/order/physical_order.hpp +2 -2
  541. package/src/duckdb/src/include/duckdb/execution/operator/order/physical_top_n.hpp +2 -2
  542. package/src/duckdb/src/include/duckdb/execution/operator/persistent/csv_rejects_table.hpp +36 -0
  543. package/src/duckdb/src/include/duckdb/execution/operator/persistent/physical_batch_copy_to_file.hpp +2 -2
  544. package/src/duckdb/src/include/duckdb/execution/operator/persistent/physical_batch_insert.hpp +2 -2
  545. package/src/duckdb/src/include/duckdb/execution/operator/persistent/physical_copy_to_file.hpp +2 -2
  546. package/src/duckdb/src/include/duckdb/execution/operator/persistent/physical_fixed_batch_copy.hpp +2 -2
  547. package/src/duckdb/src/include/duckdb/execution/operator/persistent/physical_insert.hpp +2 -2
  548. package/src/duckdb/src/include/duckdb/execution/operator/persistent/physical_update.hpp +1 -1
  549. package/src/duckdb/src/include/duckdb/execution/operator/{persistent → scan/csv}/base_csv_reader.hpp +23 -19
  550. package/src/duckdb/src/include/duckdb/execution/operator/scan/csv/buffered_csv_reader.hpp +72 -0
  551. package/src/duckdb/src/include/duckdb/execution/operator/scan/csv/csv_buffer.hpp +110 -0
  552. package/src/duckdb/src/include/duckdb/execution/operator/scan/csv/csv_buffer_manager.hpp +103 -0
  553. package/src/duckdb/src/include/duckdb/execution/operator/{persistent → scan/csv}/csv_file_handle.hpp +8 -15
  554. package/src/duckdb/src/include/duckdb/execution/operator/{persistent → scan/csv}/csv_line_info.hpp +9 -4
  555. package/src/duckdb/src/include/duckdb/execution/operator/{persistent → scan/csv}/csv_reader_options.hpp +79 -33
  556. package/src/duckdb/src/include/duckdb/execution/operator/scan/csv/csv_sniffer.hpp +129 -0
  557. package/src/duckdb/src/include/duckdb/execution/operator/scan/csv/csv_state_machine.hpp +75 -0
  558. package/src/duckdb/src/include/duckdb/execution/operator/scan/csv/csv_state_machine_cache.hpp +51 -0
  559. package/src/duckdb/src/include/duckdb/execution/operator/{persistent → scan/csv}/parallel_csv_reader.hpp +23 -28
  560. package/src/duckdb/src/include/duckdb/execution/operator/scan/csv/quote_rules.hpp +21 -0
  561. package/src/duckdb/src/include/duckdb/execution/operator/scan/physical_column_data_scan.hpp +10 -0
  562. package/src/duckdb/src/include/duckdb/execution/operator/scan/physical_table_scan.hpp +5 -5
  563. package/src/duckdb/src/include/duckdb/execution/operator/schema/{physical_create_index.hpp → physical_create_art_index.hpp} +14 -7
  564. package/src/duckdb/src/include/duckdb/execution/operator/set/physical_cte.hpp +62 -0
  565. package/src/duckdb/src/include/duckdb/execution/operator/set/physical_recursive_cte.hpp +8 -2
  566. package/src/duckdb/src/include/duckdb/execution/perfect_aggregate_hashtable.hpp +4 -2
  567. package/src/duckdb/src/include/duckdb/execution/physical_operator.hpp +3 -5
  568. package/src/duckdb/src/include/duckdb/execution/physical_operator_states.hpp +11 -0
  569. package/src/duckdb/src/include/duckdb/execution/physical_plan_generator.hpp +6 -2
  570. package/src/duckdb/src/include/duckdb/execution/radix_partitioned_hashtable.hpp +18 -21
  571. package/src/duckdb/src/include/duckdb/execution/window_executor.hpp +313 -0
  572. package/src/duckdb/src/include/duckdb/execution/window_segment_tree.hpp +79 -63
  573. package/src/duckdb/src/include/duckdb/function/aggregate_function.hpp +3 -5
  574. package/src/duckdb/src/include/duckdb/function/aggregate_state.hpp +2 -2
  575. package/src/duckdb/src/include/duckdb/function/built_in_functions.hpp +1 -0
  576. package/src/duckdb/src/include/duckdb/function/cast/bound_cast_data.hpp +32 -0
  577. package/src/duckdb/src/include/duckdb/function/compression_function.hpp +36 -4
  578. package/src/duckdb/src/include/duckdb/function/copy_function.hpp +11 -4
  579. package/src/duckdb/src/include/duckdb/function/function_serialization.hpp +58 -61
  580. package/src/duckdb/src/include/duckdb/function/macro_function.hpp +1 -4
  581. package/src/duckdb/src/include/duckdb/function/scalar/compressed_materialization_functions.hpp +49 -0
  582. package/src/duckdb/src/include/duckdb/function/scalar/list/contains_or_position.hpp +1 -1
  583. package/src/duckdb/src/include/duckdb/function/scalar/nested_functions.hpp +14 -10
  584. package/src/duckdb/src/include/duckdb/function/scalar/strftime_format.hpp +14 -5
  585. package/src/duckdb/src/include/duckdb/function/scalar/string_functions.hpp +2 -0
  586. package/src/duckdb/src/include/duckdb/function/scalar_function.hpp +2 -7
  587. package/src/duckdb/src/include/duckdb/function/scalar_macro_function.hpp +2 -4
  588. package/src/duckdb/src/include/duckdb/function/table/arrow/arrow_duck_schema.hpp +99 -0
  589. package/src/duckdb/src/include/duckdb/function/table/arrow.hpp +11 -36
  590. package/src/duckdb/src/include/duckdb/function/table/read_csv.hpp +22 -22
  591. package/src/duckdb/src/include/duckdb/function/table/system_functions.hpp +9 -1
  592. package/src/duckdb/src/include/duckdb/function/table_function.hpp +2 -3
  593. package/src/duckdb/src/include/duckdb/function/table_macro_function.hpp +2 -4
  594. package/src/duckdb/src/include/duckdb/function/udf_function.hpp +2 -1
  595. package/src/duckdb/src/include/duckdb/main/attached_database.hpp +1 -1
  596. package/src/duckdb/src/include/duckdb/main/capi/capi_internal.hpp +4 -3
  597. package/src/duckdb/src/include/duckdb/main/chunk_scan_state/query_result.hpp +29 -0
  598. package/src/duckdb/src/include/duckdb/main/chunk_scan_state.hpp +43 -0
  599. package/src/duckdb/src/include/duckdb/main/client_config.hpp +9 -2
  600. package/src/duckdb/src/include/duckdb/main/client_context.hpp +16 -14
  601. package/src/duckdb/src/include/duckdb/main/client_context_file_opener.hpp +1 -0
  602. package/src/duckdb/src/include/duckdb/main/client_data.hpp +2 -1
  603. package/src/duckdb/src/include/duckdb/main/client_properties.hpp +25 -0
  604. package/src/duckdb/src/include/duckdb/main/config.hpp +19 -1
  605. package/src/duckdb/src/include/duckdb/main/connection.hpp +3 -4
  606. package/src/duckdb/src/include/duckdb/main/extension/generated_extension_loader.hpp +27 -0
  607. package/src/duckdb/src/include/duckdb/main/extension_entries.hpp +258 -144
  608. package/src/duckdb/src/include/duckdb/main/extension_helper.hpp +42 -6
  609. package/src/duckdb/src/include/duckdb/main/extension_util.hpp +18 -0
  610. package/src/duckdb/src/include/duckdb/main/pending_query_result.hpp +5 -0
  611. package/src/duckdb/src/include/duckdb/main/prepared_statement.hpp +73 -5
  612. package/src/duckdb/src/include/duckdb/main/prepared_statement_data.hpp +7 -6
  613. package/src/duckdb/src/include/duckdb/main/query_result.hpp +3 -28
  614. package/src/duckdb/src/include/duckdb/main/relation/aggregate_relation.hpp +4 -1
  615. package/src/duckdb/src/include/duckdb/main/relation/cross_product_relation.hpp +4 -1
  616. package/src/duckdb/src/include/duckdb/main/relation/join_relation.hpp +5 -2
  617. package/src/duckdb/src/include/duckdb/main/relation/read_csv_relation.hpp +7 -7
  618. package/src/duckdb/src/include/duckdb/main/relation/table_function_relation.hpp +1 -0
  619. package/src/duckdb/src/include/duckdb/main/relation.hpp +17 -4
  620. package/src/duckdb/src/include/duckdb/main/settings.hpp +72 -12
  621. package/src/duckdb/src/include/duckdb/optimizer/column_binding_replacer.hpp +47 -0
  622. package/src/duckdb/src/include/duckdb/optimizer/compressed_materialization.hpp +132 -0
  623. package/src/duckdb/src/include/duckdb/optimizer/deliminator.hpp +13 -16
  624. package/src/duckdb/src/include/duckdb/optimizer/filter_pushdown.hpp +7 -0
  625. package/src/duckdb/src/include/duckdb/optimizer/join_order/cardinality_estimator.hpp +38 -64
  626. package/src/duckdb/src/include/duckdb/optimizer/join_order/cost_model.hpp +37 -0
  627. package/src/duckdb/src/include/duckdb/optimizer/join_order/estimated_properties.hpp +10 -1
  628. package/src/duckdb/src/include/duckdb/optimizer/join_order/join_node.hpp +14 -29
  629. package/src/duckdb/src/include/duckdb/optimizer/join_order/join_order_optimizer.hpp +8 -22
  630. package/src/duckdb/src/include/duckdb/optimizer/join_order/join_relation.hpp +1 -12
  631. package/src/duckdb/src/include/duckdb/optimizer/join_order/plan_enumerator.hpp +89 -0
  632. package/src/duckdb/src/include/duckdb/optimizer/join_order/query_graph.hpp +19 -30
  633. package/src/duckdb/src/include/duckdb/optimizer/join_order/query_graph_manager.hpp +113 -0
  634. package/src/duckdb/src/include/duckdb/optimizer/join_order/relation_manager.hpp +73 -0
  635. package/src/duckdb/src/include/duckdb/optimizer/join_order/relation_statistics_helper.hpp +73 -0
  636. package/src/duckdb/src/include/duckdb/optimizer/matcher/set_matcher.hpp +13 -0
  637. package/src/duckdb/src/include/duckdb/optimizer/optimizer.hpp +3 -0
  638. package/src/duckdb/src/include/duckdb/optimizer/remove_duplicate_groups.hpp +40 -0
  639. package/src/duckdb/src/include/duckdb/optimizer/rule/empty_needle_removal.hpp +1 -1
  640. package/src/duckdb/src/include/duckdb/optimizer/statistics_propagator.hpp +11 -3
  641. package/src/duckdb/src/include/duckdb/optimizer/topn_optimizer.hpp +2 -0
  642. package/src/duckdb/src/include/duckdb/parallel/event.hpp +12 -1
  643. package/src/duckdb/src/include/duckdb/parallel/pipeline.hpp +2 -3
  644. package/src/duckdb/src/include/duckdb/parallel/pipeline_executor.hpp +3 -2
  645. package/src/duckdb/src/include/duckdb/parallel/task_scheduler.hpp +9 -1
  646. package/src/duckdb/src/include/duckdb/parser/column_definition.hpp +4 -6
  647. package/src/duckdb/src/include/duckdb/parser/column_list.hpp +3 -3
  648. package/src/duckdb/src/include/duckdb/parser/common_table_expression_info.hpp +4 -2
  649. package/src/duckdb/src/include/duckdb/parser/constraint.hpp +3 -8
  650. package/src/duckdb/src/include/duckdb/parser/constraints/check_constraint.hpp +2 -2
  651. package/src/duckdb/src/include/duckdb/parser/constraints/foreign_key_constraint.hpp +5 -4
  652. package/src/duckdb/src/include/duckdb/parser/constraints/not_null_constraint.hpp +2 -4
  653. package/src/duckdb/src/include/duckdb/parser/constraints/unique_constraint.hpp +5 -4
  654. package/src/duckdb/src/include/duckdb/parser/expression/between_expression.hpp +5 -5
  655. package/src/duckdb/src/include/duckdb/parser/expression/bound_expression.hpp +1 -4
  656. package/src/duckdb/src/include/duckdb/parser/expression/case_expression.hpp +4 -6
  657. package/src/duckdb/src/include/duckdb/parser/expression/cast_expression.hpp +5 -4
  658. package/src/duckdb/src/include/duckdb/parser/expression/collate_expression.hpp +5 -4
  659. package/src/duckdb/src/include/duckdb/parser/expression/columnref_expression.hpp +5 -4
  660. package/src/duckdb/src/include/duckdb/parser/expression/comparison_expression.hpp +5 -4
  661. package/src/duckdb/src/include/duckdb/parser/expression/conjunction_expression.hpp +2 -4
  662. package/src/duckdb/src/include/duckdb/parser/expression/constant_expression.hpp +4 -4
  663. package/src/duckdb/src/include/duckdb/parser/expression/default_expression.hpp +2 -3
  664. package/src/duckdb/src/include/duckdb/parser/expression/function_expression.hpp +5 -4
  665. package/src/duckdb/src/include/duckdb/parser/expression/lambda_expression.hpp +5 -4
  666. package/src/duckdb/src/include/duckdb/parser/expression/operator_expression.hpp +22 -7
  667. package/src/duckdb/src/include/duckdb/parser/expression/parameter_expression.hpp +19 -5
  668. package/src/duckdb/src/include/duckdb/parser/expression/positional_reference_expression.hpp +5 -4
  669. package/src/duckdb/src/include/duckdb/parser/expression/star_expression.hpp +2 -4
  670. package/src/duckdb/src/include/duckdb/parser/expression/subquery_expression.hpp +2 -4
  671. package/src/duckdb/src/include/duckdb/parser/expression/window_expression.hpp +5 -4
  672. package/src/duckdb/src/include/duckdb/parser/group_by_node.hpp +11 -0
  673. package/src/duckdb/src/include/duckdb/parser/parsed_data/alter_info.hpp +12 -4
  674. package/src/duckdb/src/include/duckdb/parser/parsed_data/alter_scalar_function_info.hpp +0 -2
  675. package/src/duckdb/src/include/duckdb/parser/parsed_data/alter_table_function_info.hpp +0 -2
  676. package/src/duckdb/src/include/duckdb/parser/parsed_data/alter_table_info.hpp +66 -29
  677. package/src/duckdb/src/include/duckdb/parser/parsed_data/attach_info.hpp +6 -2
  678. package/src/duckdb/src/include/duckdb/parser/parsed_data/copy_info.hpp +8 -1
  679. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_collation_info.hpp +0 -3
  680. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_copy_function_info.hpp +0 -3
  681. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_function_info.hpp +0 -5
  682. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_index_info.hpp +7 -4
  683. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_info.hpp +7 -12
  684. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_macro_info.hpp +2 -5
  685. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_schema_info.hpp +2 -9
  686. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_sequence_info.hpp +2 -4
  687. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_table_info.hpp +3 -5
  688. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_type_info.hpp +2 -4
  689. package/src/duckdb/src/include/duckdb/parser/parsed_data/create_view_info.hpp +2 -4
  690. package/src/duckdb/src/include/duckdb/parser/parsed_data/detach_info.hpp +6 -1
  691. package/src/duckdb/src/include/duckdb/parser/parsed_data/drop_info.hpp +6 -2
  692. package/src/duckdb/src/include/duckdb/parser/parsed_data/exported_table_data.hpp +7 -0
  693. package/src/duckdb/src/include/duckdb/parser/parsed_data/load_info.hpp +14 -19
  694. package/src/duckdb/src/include/duckdb/parser/parsed_data/parse_info.hpp +22 -0
  695. package/src/duckdb/src/include/duckdb/parser/parsed_data/pragma_info.hpp +10 -0
  696. package/src/duckdb/src/include/duckdb/parser/parsed_data/sample_options.hpp +2 -4
  697. package/src/duckdb/src/include/duckdb/parser/parsed_data/show_select_info.hpp +7 -0
  698. package/src/duckdb/src/include/duckdb/parser/parsed_data/transaction_info.hpp +8 -1
  699. package/src/duckdb/src/include/duckdb/parser/parsed_data/vacuum_info.hpp +9 -1
  700. package/src/duckdb/src/include/duckdb/parser/parsed_expression.hpp +3 -16
  701. package/src/duckdb/src/include/duckdb/parser/parser.hpp +4 -0
  702. package/src/duckdb/src/include/duckdb/parser/query_node/cte_node.hpp +52 -0
  703. package/src/duckdb/src/include/duckdb/parser/query_node/list.hpp +1 -0
  704. package/src/duckdb/src/include/duckdb/parser/query_node/recursive_cte_node.hpp +2 -4
  705. package/src/duckdb/src/include/duckdb/parser/query_node/select_node.hpp +2 -4
  706. package/src/duckdb/src/include/duckdb/parser/query_node/set_operation_node.hpp +2 -4
  707. package/src/duckdb/src/include/duckdb/parser/query_node.hpp +9 -15
  708. package/src/duckdb/src/include/duckdb/parser/result_modifier.hpp +14 -32
  709. package/src/duckdb/src/include/duckdb/parser/statement/execute_statement.hpp +1 -1
  710. package/src/duckdb/src/include/duckdb/parser/statement/select_statement.hpp +4 -9
  711. package/src/duckdb/src/include/duckdb/parser/tableref/basetableref.hpp +2 -6
  712. package/src/duckdb/src/include/duckdb/parser/tableref/emptytableref.hpp +2 -5
  713. package/src/duckdb/src/include/duckdb/parser/tableref/expressionlistref.hpp +2 -6
  714. package/src/duckdb/src/include/duckdb/parser/tableref/joinref.hpp +3 -7
  715. package/src/duckdb/src/include/duckdb/parser/tableref/pivotref.hpp +6 -14
  716. package/src/duckdb/src/include/duckdb/parser/tableref/subqueryref.hpp +5 -6
  717. package/src/duckdb/src/include/duckdb/parser/tableref/table_function_ref.hpp +2 -6
  718. package/src/duckdb/src/include/duckdb/parser/tableref.hpp +2 -10
  719. package/src/duckdb/src/include/duckdb/parser/tokens.hpp +1 -0
  720. package/src/duckdb/src/include/duckdb/parser/transformer.hpp +24 -26
  721. package/src/duckdb/src/include/duckdb/planner/binder.hpp +16 -5
  722. package/src/duckdb/src/include/duckdb/planner/bound_constraint.hpp +0 -9
  723. package/src/duckdb/src/include/duckdb/planner/bound_parameter_map.hpp +30 -2
  724. package/src/duckdb/src/include/duckdb/planner/bound_result_modifier.hpp +2 -2
  725. package/src/duckdb/src/include/duckdb/planner/bound_tokens.hpp +1 -0
  726. package/src/duckdb/src/include/duckdb/planner/column_binding.hpp +9 -0
  727. package/src/duckdb/src/include/duckdb/planner/constraints/bound_unique_constraint.hpp +3 -3
  728. package/src/duckdb/src/include/duckdb/planner/expression/bound_aggregate_expression.hpp +3 -2
  729. package/src/duckdb/src/include/duckdb/planner/expression/bound_between_expression.hpp +6 -2
  730. package/src/duckdb/src/include/duckdb/planner/expression/bound_case_expression.hpp +3 -3
  731. package/src/duckdb/src/include/duckdb/planner/expression/bound_cast_expression.hpp +5 -2
  732. package/src/duckdb/src/include/duckdb/planner/expression/bound_columnref_expression.hpp +2 -5
  733. package/src/duckdb/src/include/duckdb/planner/expression/bound_comparison_expression.hpp +3 -2
  734. package/src/duckdb/src/include/duckdb/planner/expression/bound_conjunction_expression.hpp +2 -2
  735. package/src/duckdb/src/include/duckdb/planner/expression/bound_constant_expression.hpp +2 -2
  736. package/src/duckdb/src/include/duckdb/planner/expression/bound_default_expression.hpp +2 -2
  737. package/src/duckdb/src/include/duckdb/planner/expression/bound_function_expression.hpp +3 -2
  738. package/src/duckdb/src/include/duckdb/planner/expression/bound_lambda_expression.hpp +2 -2
  739. package/src/duckdb/src/include/duckdb/planner/expression/bound_lambdaref_expression.hpp +2 -5
  740. package/src/duckdb/src/include/duckdb/planner/expression/bound_operator_expression.hpp +2 -2
  741. package/src/duckdb/src/include/duckdb/planner/expression/bound_parameter_data.hpp +12 -29
  742. package/src/duckdb/src/include/duckdb/planner/expression/bound_parameter_expression.hpp +9 -5
  743. package/src/duckdb/src/include/duckdb/planner/expression/bound_reference_expression.hpp +2 -2
  744. package/src/duckdb/src/include/duckdb/planner/expression/bound_subquery_expression.hpp +0 -3
  745. package/src/duckdb/src/include/duckdb/planner/expression/bound_unnest_expression.hpp +2 -2
  746. package/src/duckdb/src/include/duckdb/planner/expression/bound_window_expression.hpp +2 -2
  747. package/src/duckdb/src/include/duckdb/planner/expression/list.hpp +1 -0
  748. package/src/duckdb/src/include/duckdb/planner/expression.hpp +2 -11
  749. package/src/duckdb/src/include/duckdb/planner/expression_binder/base_select_binder.hpp +1 -0
  750. package/src/duckdb/src/include/duckdb/planner/expression_binder/lateral_binder.hpp +0 -2
  751. package/src/duckdb/src/include/duckdb/planner/expression_binder.hpp +13 -1
  752. package/src/duckdb/src/include/duckdb/planner/extension_callback.hpp +26 -0
  753. package/src/duckdb/src/include/duckdb/planner/filter/conjunction_filter.hpp +4 -4
  754. package/src/duckdb/src/include/duckdb/planner/filter/constant_filter.hpp +2 -2
  755. package/src/duckdb/src/include/duckdb/planner/filter/null_filter.hpp +4 -4
  756. package/src/duckdb/src/include/duckdb/planner/joinside.hpp +1 -3
  757. package/src/duckdb/src/include/duckdb/planner/logical_operator.hpp +2 -16
  758. package/src/duckdb/src/include/duckdb/planner/logical_tokens.hpp +1 -2
  759. package/src/duckdb/src/include/duckdb/planner/operator/list.hpp +3 -3
  760. package/src/duckdb/src/include/duckdb/planner/operator/logical_aggregate.hpp +3 -2
  761. package/src/duckdb/src/include/duckdb/planner/operator/logical_any_join.hpp +3 -2
  762. package/src/duckdb/src/include/duckdb/planner/operator/logical_column_data_get.hpp +3 -2
  763. package/src/duckdb/src/include/duckdb/planner/operator/logical_comparison_join.hpp +12 -11
  764. package/src/duckdb/src/include/duckdb/planner/operator/logical_copy_to_file.hpp +2 -2
  765. package/src/duckdb/src/include/duckdb/planner/operator/logical_create.hpp +8 -8
  766. package/src/duckdb/src/include/duckdb/planner/operator/logical_create_index.hpp +11 -25
  767. package/src/duckdb/src/include/duckdb/planner/operator/logical_create_table.hpp +8 -8
  768. package/src/duckdb/src/include/duckdb/planner/operator/logical_cross_product.hpp +2 -2
  769. package/src/duckdb/src/include/duckdb/planner/operator/logical_cteref.hpp +9 -4
  770. package/src/duckdb/src/include/duckdb/planner/operator/logical_delete.hpp +6 -2
  771. package/src/duckdb/src/include/duckdb/planner/operator/logical_delim_get.hpp +3 -2
  772. package/src/duckdb/src/include/duckdb/planner/operator/logical_dependent_join.hpp +42 -0
  773. package/src/duckdb/src/include/duckdb/planner/operator/logical_distinct.hpp +6 -12
  774. package/src/duckdb/src/include/duckdb/planner/operator/logical_dummy_scan.hpp +2 -2
  775. package/src/duckdb/src/include/duckdb/planner/operator/logical_empty_result.hpp +2 -2
  776. package/src/duckdb/src/include/duckdb/planner/operator/logical_execute.hpp +0 -2
  777. package/src/duckdb/src/include/duckdb/planner/operator/logical_explain.hpp +3 -2
  778. package/src/duckdb/src/include/duckdb/planner/operator/logical_export.hpp +0 -3
  779. package/src/duckdb/src/include/duckdb/planner/operator/logical_expression_get.hpp +3 -2
  780. package/src/duckdb/src/include/duckdb/planner/operator/logical_extension_operator.hpp +7 -1
  781. package/src/duckdb/src/include/duckdb/planner/operator/logical_filter.hpp +3 -2
  782. package/src/duckdb/src/include/duckdb/planner/operator/logical_get.hpp +11 -3
  783. package/src/duckdb/src/include/duckdb/planner/operator/logical_insert.hpp +5 -2
  784. package/src/duckdb/src/include/duckdb/planner/operator/logical_join.hpp +1 -3
  785. package/src/duckdb/src/include/duckdb/planner/operator/logical_limit.hpp +2 -2
  786. package/src/duckdb/src/include/duckdb/planner/operator/logical_limit_percent.hpp +2 -2
  787. package/src/duckdb/src/include/duckdb/planner/operator/logical_materialized_cte.hpp +50 -0
  788. package/src/duckdb/src/include/duckdb/planner/operator/logical_order.hpp +6 -37
  789. package/src/duckdb/src/include/duckdb/planner/operator/logical_pivot.hpp +6 -2
  790. package/src/duckdb/src/include/duckdb/planner/operator/logical_positional_join.hpp +2 -2
  791. package/src/duckdb/src/include/duckdb/planner/operator/logical_pragma.hpp +0 -2
  792. package/src/duckdb/src/include/duckdb/planner/operator/logical_prepare.hpp +0 -2
  793. package/src/duckdb/src/include/duckdb/planner/operator/logical_projection.hpp +3 -2
  794. package/src/duckdb/src/include/duckdb/planner/operator/logical_recursive_cte.hpp +10 -9
  795. package/src/duckdb/src/include/duckdb/planner/operator/logical_reset.hpp +3 -2
  796. package/src/duckdb/src/include/duckdb/planner/operator/logical_sample.hpp +5 -2
  797. package/src/duckdb/src/include/duckdb/planner/operator/logical_set.hpp +3 -2
  798. package/src/duckdb/src/include/duckdb/planner/operator/logical_set_operation.hpp +3 -2
  799. package/src/duckdb/src/include/duckdb/planner/operator/logical_show.hpp +2 -2
  800. package/src/duckdb/src/include/duckdb/planner/operator/logical_simple.hpp +2 -2
  801. package/src/duckdb/src/include/duckdb/planner/operator/logical_top_n.hpp +4 -2
  802. package/src/duckdb/src/include/duckdb/planner/operator/logical_unnest.hpp +2 -2
  803. package/src/duckdb/src/include/duckdb/planner/operator/logical_update.hpp +6 -2
  804. package/src/duckdb/src/include/duckdb/planner/operator/logical_window.hpp +3 -2
  805. package/src/duckdb/src/include/duckdb/planner/operator_extension.hpp +1 -2
  806. package/src/duckdb/src/include/duckdb/planner/parsed_data/bound_create_table_info.hpp +0 -5
  807. package/src/duckdb/src/include/duckdb/planner/planner.hpp +5 -4
  808. package/src/duckdb/src/include/duckdb/planner/query_node/bound_cte_node.hpp +44 -0
  809. package/src/duckdb/src/include/duckdb/planner/query_node/list.hpp +1 -0
  810. package/src/duckdb/src/include/duckdb/planner/subquery/flatten_dependent_join.hpp +2 -2
  811. package/src/duckdb/src/include/duckdb/planner/subquery/has_correlated_expressions.hpp +4 -1
  812. package/src/duckdb/src/include/duckdb/planner/subquery/recursive_dependent_join_planner.hpp +31 -0
  813. package/src/duckdb/src/include/duckdb/planner/subquery/rewrite_correlated_expressions.hpp +8 -2
  814. package/src/duckdb/src/include/duckdb/planner/table_filter.hpp +4 -7
  815. package/src/duckdb/src/include/duckdb/planner/tableref/bound_cteref.hpp +5 -2
  816. package/src/duckdb/src/include/duckdb/planner/tableref/bound_pivotref.hpp +3 -0
  817. package/src/duckdb/src/include/duckdb/storage/arena_allocator.hpp +2 -1
  818. package/src/duckdb/src/include/duckdb/storage/block.hpp +36 -4
  819. package/src/duckdb/src/include/duckdb/storage/block_manager.hpp +11 -11
  820. package/src/duckdb/src/include/duckdb/storage/buffer/block_handle.hpp +3 -0
  821. package/src/duckdb/src/include/duckdb/storage/checkpoint/row_group_writer.hpp +8 -7
  822. package/src/duckdb/src/include/duckdb/storage/checkpoint/string_checkpoint_state.hpp +27 -4
  823. package/src/duckdb/src/include/duckdb/storage/checkpoint/table_data_reader.hpp +2 -2
  824. package/src/duckdb/src/include/duckdb/storage/checkpoint/table_data_writer.hpp +7 -7
  825. package/src/duckdb/src/include/duckdb/storage/checkpoint/write_overflow_strings_to_disk.hpp +4 -2
  826. package/src/duckdb/src/include/duckdb/storage/checkpoint_manager.hpp +29 -24
  827. package/src/duckdb/src/include/duckdb/storage/compression/bitpacking.hpp +1 -8
  828. package/src/duckdb/src/include/duckdb/storage/data_pointer.hpp +31 -4
  829. package/src/duckdb/src/include/duckdb/storage/data_table.hpp +4 -4
  830. package/src/duckdb/src/include/duckdb/storage/database_size.hpp +6 -0
  831. package/src/duckdb/src/include/duckdb/storage/in_memory_block_manager.hpp +2 -2
  832. package/src/duckdb/src/include/duckdb/storage/index.hpp +12 -10
  833. package/src/duckdb/src/include/duckdb/storage/metadata/metadata_manager.hpp +91 -0
  834. package/src/duckdb/src/include/duckdb/storage/metadata/metadata_reader.hpp +57 -0
  835. package/src/duckdb/src/include/duckdb/storage/metadata/metadata_writer.hpp +52 -0
  836. package/src/duckdb/src/include/duckdb/storage/object_cache.hpp +22 -0
  837. package/src/duckdb/src/include/duckdb/storage/partial_block_manager.hpp +37 -21
  838. package/src/duckdb/src/include/duckdb/storage/single_file_block_manager.hpp +8 -5
  839. package/src/duckdb/src/include/duckdb/storage/statistics/base_statistics.hpp +3 -8
  840. package/src/duckdb/src/include/duckdb/storage/statistics/column_statistics.hpp +3 -1
  841. package/src/duckdb/src/include/duckdb/storage/statistics/distinct_statistics.hpp +4 -7
  842. package/src/duckdb/src/include/duckdb/storage/statistics/list_stats.hpp +2 -4
  843. package/src/duckdb/src/include/duckdb/storage/statistics/node_statistics.hpp +0 -26
  844. package/src/duckdb/src/include/duckdb/storage/statistics/numeric_stats.hpp +2 -4
  845. package/src/duckdb/src/include/duckdb/storage/statistics/string_stats.hpp +6 -4
  846. package/src/duckdb/src/include/duckdb/storage/statistics/struct_stats.hpp +2 -4
  847. package/src/duckdb/src/include/duckdb/storage/storage_info.hpp +25 -8
  848. package/src/duckdb/src/include/duckdb/storage/storage_manager.hpp +4 -2
  849. package/src/duckdb/src/include/duckdb/storage/string_uncompressed.hpp +6 -1
  850. package/src/duckdb/src/include/duckdb/storage/table/chunk_info.hpp +28 -16
  851. package/src/duckdb/src/include/duckdb/storage/table/column_checkpoint_state.hpp +5 -20
  852. package/src/duckdb/src/include/duckdb/storage/table/column_data.hpp +3 -3
  853. package/src/duckdb/src/include/duckdb/storage/table/column_segment.hpp +7 -3
  854. package/src/duckdb/src/include/duckdb/storage/table/list_column_data.hpp +1 -1
  855. package/src/duckdb/src/include/duckdb/storage/table/persistent_table_data.hpp +2 -2
  856. package/src/duckdb/src/include/duckdb/storage/table/row_group.hpp +21 -21
  857. package/src/duckdb/src/include/duckdb/storage/table/row_group_collection.hpp +4 -4
  858. package/src/duckdb/src/include/duckdb/storage/table/row_group_segment_tree.hpp +2 -2
  859. package/src/duckdb/src/include/duckdb/storage/table/row_version_manager.hpp +59 -0
  860. package/src/duckdb/src/include/duckdb/storage/table/standard_column_data.hpp +2 -2
  861. package/src/duckdb/src/include/duckdb/storage/table/table_index_list.hpp +1 -1
  862. package/src/duckdb/src/include/duckdb/storage/table/table_statistics.hpp +4 -2
  863. package/src/duckdb/src/include/duckdb/storage/table/update_segment.hpp +1 -1
  864. package/src/duckdb/src/include/duckdb/storage/table_io_manager.hpp +3 -0
  865. package/src/duckdb/src/include/duckdb/storage/table_storage_info.hpp +1 -0
  866. package/src/duckdb/src/include/duckdb/storage/write_ahead_log.hpp +29 -32
  867. package/src/duckdb/src/include/duckdb/transaction/commit_state.hpp +1 -6
  868. package/src/duckdb/src/include/duckdb/transaction/delete_info.hpp +3 -2
  869. package/src/duckdb/src/include/duckdb/transaction/duck_transaction.hpp +4 -2
  870. package/src/duckdb/src/include/duckdb/transaction/local_storage.hpp +3 -4
  871. package/src/duckdb/src/include/duckdb/transaction/undo_buffer.hpp +0 -1
  872. package/src/duckdb/src/include/duckdb/verification/prepared_statement_verifier.hpp +1 -1
  873. package/src/duckdb/src/include/duckdb/verification/statement_verifier.hpp +0 -1
  874. package/src/duckdb/src/include/duckdb.h +101 -4
  875. package/src/duckdb/src/main/appender.cpp +3 -1
  876. package/src/duckdb/src/main/attached_database.cpp +2 -2
  877. package/src/duckdb/src/main/capi/arrow-c.cpp +196 -8
  878. package/src/duckdb/src/main/capi/duckdb-c.cpp +16 -0
  879. package/src/duckdb/src/main/capi/duckdb_value-c.cpp +1 -1
  880. package/src/duckdb/src/main/capi/logical_types-c.cpp +22 -0
  881. package/src/duckdb/src/main/capi/pending-c.cpp +23 -0
  882. package/src/duckdb/src/main/capi/prepared-c.cpp +114 -30
  883. package/src/duckdb/src/main/capi/result-c.cpp +3 -1
  884. package/src/duckdb/src/main/chunk_scan_state/query_result.cpp +53 -0
  885. package/src/duckdb/src/main/chunk_scan_state.cpp +48 -0
  886. package/src/duckdb/src/main/client_context.cpp +42 -25
  887. package/src/duckdb/src/main/client_context_file_opener.cpp +17 -0
  888. package/src/duckdb/src/main/client_verify.cpp +17 -1
  889. package/src/duckdb/src/main/config.cpp +10 -4
  890. package/src/duckdb/src/main/connection.cpp +5 -7
  891. package/src/duckdb/src/main/database.cpp +9 -12
  892. package/src/duckdb/src/main/db_instance_cache.cpp +14 -6
  893. package/src/duckdb/src/main/extension/extension_helper.cpp +195 -87
  894. package/src/duckdb/src/main/extension/extension_install.cpp +79 -15
  895. package/src/duckdb/src/main/extension/extension_load.cpp +63 -14
  896. package/src/duckdb/src/main/extension/extension_util.cpp +72 -0
  897. package/src/duckdb/src/main/pending_query_result.cpp +9 -1
  898. package/src/duckdb/src/main/prepared_statement.cpp +38 -11
  899. package/src/duckdb/src/main/prepared_statement_data.cpp +23 -18
  900. package/src/duckdb/src/main/query_result.cpp +16 -31
  901. package/src/duckdb/src/main/relation/aggregate_relation.cpp +20 -10
  902. package/src/duckdb/src/main/relation/cross_product_relation.cpp +4 -3
  903. package/src/duckdb/src/main/relation/join_relation.cpp +6 -6
  904. package/src/duckdb/src/main/relation/read_csv_relation.cpp +38 -13
  905. package/src/duckdb/src/main/relation/table_function_relation.cpp +8 -2
  906. package/src/duckdb/src/main/relation.cpp +35 -11
  907. package/src/duckdb/src/main/settings/settings.cpp +134 -43
  908. package/src/duckdb/src/optimizer/column_binding_replacer.cpp +43 -0
  909. package/src/duckdb/src/optimizer/column_lifetime_analyzer.cpp +11 -7
  910. package/src/duckdb/src/optimizer/compressed_materialization/compress_aggregate.cpp +140 -0
  911. package/src/duckdb/src/optimizer/compressed_materialization/compress_distinct.cpp +42 -0
  912. package/src/duckdb/src/optimizer/compressed_materialization/compress_order.cpp +65 -0
  913. package/src/duckdb/src/optimizer/compressed_materialization.cpp +477 -0
  914. package/src/duckdb/src/optimizer/deliminator.cpp +180 -323
  915. package/src/duckdb/src/optimizer/filter_pushdown.cpp +23 -6
  916. package/src/duckdb/src/optimizer/join_order/cardinality_estimator.cpp +79 -325
  917. package/src/duckdb/src/optimizer/join_order/cost_model.cpp +19 -0
  918. package/src/duckdb/src/optimizer/join_order/estimated_properties.cpp +7 -0
  919. package/src/duckdb/src/optimizer/join_order/join_node.cpp +5 -37
  920. package/src/duckdb/src/optimizer/join_order/join_order_optimizer.cpp +48 -1047
  921. package/src/duckdb/src/optimizer/join_order/join_relation_set.cpp +2 -6
  922. package/src/duckdb/src/optimizer/join_order/plan_enumerator.cpp +552 -0
  923. package/src/duckdb/src/optimizer/join_order/query_graph.cpp +52 -41
  924. package/src/duckdb/src/optimizer/join_order/query_graph_manager.cpp +409 -0
  925. package/src/duckdb/src/optimizer/join_order/relation_manager.cpp +356 -0
  926. package/src/duckdb/src/optimizer/join_order/relation_statistics_helper.cpp +351 -0
  927. package/src/duckdb/src/optimizer/optimizer.cpp +49 -14
  928. package/src/duckdb/src/optimizer/pushdown/pushdown_cross_product.cpp +5 -5
  929. package/src/duckdb/src/optimizer/pushdown/pushdown_get.cpp +0 -1
  930. package/src/duckdb/src/optimizer/pushdown/pushdown_projection.cpp +34 -7
  931. package/src/duckdb/src/optimizer/remove_duplicate_groups.cpp +127 -0
  932. package/src/duckdb/src/optimizer/remove_unused_columns.cpp +4 -0
  933. package/src/duckdb/src/optimizer/rule/date_part_simplification.cpp +0 -3
  934. package/src/duckdb/src/optimizer/rule/regex_optimizations.cpp +154 -15
  935. package/src/duckdb/src/optimizer/statistics/expression/propagate_cast.cpp +14 -0
  936. package/src/duckdb/src/optimizer/statistics/operator/propagate_join.cpp +65 -8
  937. package/src/duckdb/src/optimizer/statistics/operator/propagate_order.cpp +1 -1
  938. package/src/duckdb/src/optimizer/statistics/operator/propagate_projection.cpp +0 -1
  939. package/src/duckdb/src/optimizer/statistics_propagator.cpp +7 -5
  940. package/src/duckdb/src/optimizer/topn_optimizer.cpp +27 -10
  941. package/src/duckdb/src/optimizer/unnest_rewriter.cpp +3 -5
  942. package/src/duckdb/src/parallel/executor.cpp +25 -1
  943. package/src/duckdb/src/parallel/pipeline.cpp +0 -17
  944. package/src/duckdb/src/parallel/pipeline_executor.cpp +26 -7
  945. package/src/duckdb/src/parallel/pipeline_finish_event.cpp +55 -1
  946. package/src/duckdb/src/parallel/task_scheduler.cpp +18 -2
  947. package/src/duckdb/src/parser/column_definition.cpp +18 -55
  948. package/src/duckdb/src/parser/column_list.cpp +8 -13
  949. package/src/duckdb/src/parser/constraint.cpp +0 -33
  950. package/src/duckdb/src/parser/constraints/check_constraint.cpp +0 -11
  951. package/src/duckdb/src/parser/constraints/foreign_key_constraint.cpp +3 -27
  952. package/src/duckdb/src/parser/constraints/not_null_constraint.cpp +0 -11
  953. package/src/duckdb/src/parser/constraints/unique_constraint.cpp +3 -24
  954. package/src/duckdb/src/parser/expression/between_expression.cpp +5 -31
  955. package/src/duckdb/src/parser/expression/case_expression.cpp +2 -54
  956. package/src/duckdb/src/parser/expression/cast_expression.cpp +5 -30
  957. package/src/duckdb/src/parser/expression/collate_expression.cpp +5 -27
  958. package/src/duckdb/src/parser/expression/columnref_expression.cpp +5 -25
  959. package/src/duckdb/src/parser/expression/comparison_expression.cpp +5 -27
  960. package/src/duckdb/src/parser/expression/conjunction_expression.cpp +2 -25
  961. package/src/duckdb/src/parser/expression/constant_expression.cpp +5 -23
  962. package/src/duckdb/src/parser/expression/default_expression.cpp +2 -13
  963. package/src/duckdb/src/parser/expression/function_expression.cpp +5 -64
  964. package/src/duckdb/src/parser/expression/lambda_expression.cpp +6 -28
  965. package/src/duckdb/src/parser/expression/operator_expression.cpp +2 -25
  966. package/src/duckdb/src/parser/expression/parameter_expression.cpp +7 -30
  967. package/src/duckdb/src/parser/expression/positional_reference_expression.cpp +6 -23
  968. package/src/duckdb/src/parser/expression/star_expression.cpp +2 -62
  969. package/src/duckdb/src/parser/expression/subquery_expression.cpp +2 -46
  970. package/src/duckdb/src/parser/expression/window_expression.cpp +5 -92
  971. package/src/duckdb/src/parser/keyword_helper.cpp +1 -1
  972. package/src/duckdb/src/parser/parsed_data/alter_info.cpp +4 -36
  973. package/src/duckdb/src/parser/parsed_data/alter_scalar_function_info.cpp +0 -18
  974. package/src/duckdb/src/parser/parsed_data/alter_table_function_info.cpp +0 -13
  975. package/src/duckdb/src/parser/parsed_data/alter_table_info.cpp +38 -188
  976. package/src/duckdb/src/parser/parsed_data/attach_info.cpp +0 -29
  977. package/src/duckdb/src/parser/parsed_data/create_collation_info.cpp +0 -4
  978. package/src/duckdb/src/parser/parsed_data/create_copy_function_info.cpp +0 -4
  979. package/src/duckdb/src/parser/parsed_data/create_index_info.cpp +2 -34
  980. package/src/duckdb/src/parser/parsed_data/create_info.cpp +0 -47
  981. package/src/duckdb/src/parser/parsed_data/create_macro_info.cpp +1 -28
  982. package/src/duckdb/src/parser/parsed_data/create_sequence_info.cpp +0 -29
  983. package/src/duckdb/src/parser/parsed_data/create_table_info.cpp +0 -23
  984. package/src/duckdb/src/parser/parsed_data/create_type_info.cpp +0 -22
  985. package/src/duckdb/src/parser/parsed_data/create_view_info.cpp +0 -23
  986. package/src/duckdb/src/parser/parsed_data/detach_info.cpp +1 -20
  987. package/src/duckdb/src/parser/parsed_data/drop_info.cpp +1 -27
  988. package/src/duckdb/src/parser/parsed_data/sample_options.cpp +2 -43
  989. package/src/duckdb/src/parser/parsed_data/transaction_info.cpp +2 -15
  990. package/src/duckdb/src/parser/parsed_data/vacuum_info.cpp +1 -21
  991. package/src/duckdb/src/parser/parsed_expression.cpp +2 -149
  992. package/src/duckdb/src/parser/parsed_expression_iterator.cpp +7 -0
  993. package/src/duckdb/src/parser/parser.cpp +62 -36
  994. package/src/duckdb/src/parser/query_node/cte_node.cpp +41 -0
  995. package/src/duckdb/src/parser/query_node/recursive_cte_node.cpp +2 -40
  996. package/src/duckdb/src/parser/query_node/select_node.cpp +2 -76
  997. package/src/duckdb/src/parser/query_node/set_operation_node.cpp +2 -32
  998. package/src/duckdb/src/parser/query_node.cpp +12 -104
  999. package/src/duckdb/src/parser/result_modifier.cpp +2 -183
  1000. package/src/duckdb/src/parser/statement/execute_statement.cpp +2 -2
  1001. package/src/duckdb/src/parser/statement/select_statement.cpp +2 -23
  1002. package/src/duckdb/src/parser/tableref/basetableref.cpp +2 -40
  1003. package/src/duckdb/src/parser/tableref/emptytableref.cpp +0 -13
  1004. package/src/duckdb/src/parser/tableref/expressionlistref.cpp +2 -43
  1005. package/src/duckdb/src/parser/tableref/joinref.cpp +5 -46
  1006. package/src/duckdb/src/parser/tableref/pivotref.cpp +0 -116
  1007. package/src/duckdb/src/parser/tableref/subqueryref.cpp +5 -28
  1008. package/src/duckdb/src/parser/tableref/table_function.cpp +2 -32
  1009. package/src/duckdb/src/parser/tableref.cpp +2 -96
  1010. package/src/duckdb/src/parser/transform/constraint/transform_constraint.cpp +55 -38
  1011. package/src/duckdb/src/parser/transform/expression/transform_array_access.cpp +13 -4
  1012. package/src/duckdb/src/parser/transform/expression/transform_constant.cpp +55 -3
  1013. package/src/duckdb/src/parser/transform/expression/transform_expression.cpp +2 -0
  1014. package/src/duckdb/src/parser/transform/expression/transform_function.cpp +3 -0
  1015. package/src/duckdb/src/parser/transform/expression/transform_multi_assign_reference.cpp +44 -0
  1016. package/src/duckdb/src/parser/transform/expression/transform_param_ref.cpp +45 -26
  1017. package/src/duckdb/src/parser/transform/helpers/transform_cte.cpp +19 -1
  1018. package/src/duckdb/src/parser/transform/helpers/transform_typename.cpp +16 -1
  1019. package/src/duckdb/src/parser/transform/statement/transform_copy.cpp +13 -0
  1020. package/src/duckdb/src/parser/transform/statement/transform_create_index.cpp +32 -17
  1021. package/src/duckdb/src/parser/transform/statement/transform_create_type.cpp +1 -1
  1022. package/src/duckdb/src/parser/transform/statement/transform_delete.cpp +6 -1
  1023. package/src/duckdb/src/parser/transform/statement/transform_insert.cpp +6 -1
  1024. package/src/duckdb/src/parser/transform/statement/transform_load.cpp +1 -0
  1025. package/src/duckdb/src/parser/transform/statement/transform_pivot_stmt.cpp +7 -2
  1026. package/src/duckdb/src/parser/transform/statement/transform_pragma.cpp +14 -11
  1027. package/src/duckdb/src/parser/transform/statement/transform_prepare.cpp +28 -6
  1028. package/src/duckdb/src/parser/transform/statement/transform_select_node.cpp +11 -2
  1029. package/src/duckdb/src/parser/transform/statement/transform_update.cpp +6 -1
  1030. package/src/duckdb/src/parser/transformer.cpp +44 -25
  1031. package/src/duckdb/src/planner/binder/expression/bind_aggregate_expression.cpp +1 -4
  1032. package/src/duckdb/src/planner/binder/expression/bind_comparison_expression.cpp +1 -0
  1033. package/src/duckdb/src/planner/binder/expression/bind_function_expression.cpp +14 -5
  1034. package/src/duckdb/src/planner/binder/expression/bind_macro_expression.cpp +5 -3
  1035. package/src/duckdb/src/planner/binder/expression/bind_parameter_expression.cpp +12 -22
  1036. package/src/duckdb/src/planner/binder/expression/bind_subquery_expression.cpp +1 -4
  1037. package/src/duckdb/src/planner/binder/query_node/bind_cte_node.cpp +64 -0
  1038. package/src/duckdb/src/planner/binder/query_node/bind_select_node.cpp +28 -6
  1039. package/src/duckdb/src/planner/binder/query_node/plan_cte_node.cpp +26 -0
  1040. package/src/duckdb/src/planner/binder/query_node/plan_recursive_cte_node.cpp +5 -5
  1041. package/src/duckdb/src/planner/binder/query_node/plan_setop.cpp +4 -4
  1042. package/src/duckdb/src/planner/binder/query_node/plan_subquery.cpp +36 -33
  1043. package/src/duckdb/src/planner/binder/statement/bind_copy.cpp +1 -1
  1044. package/src/duckdb/src/planner/binder/statement/bind_create.cpp +14 -52
  1045. package/src/duckdb/src/planner/binder/statement/bind_create_table.cpp +0 -23
  1046. package/src/duckdb/src/planner/binder/statement/bind_drop.cpp +3 -0
  1047. package/src/duckdb/src/planner/binder/statement/bind_execute.cpp +13 -7
  1048. package/src/duckdb/src/planner/binder/statement/bind_export.cpp +168 -30
  1049. package/src/duckdb/src/planner/binder/tableref/bind_basetableref.cpp +93 -28
  1050. package/src/duckdb/src/planner/binder/tableref/bind_joinref.cpp +32 -5
  1051. package/src/duckdb/src/planner/binder/tableref/bind_pivot.cpp +116 -49
  1052. package/src/duckdb/src/planner/binder/tableref/plan_cteref.cpp +1 -1
  1053. package/src/duckdb/src/planner/binder/tableref/plan_joinref.cpp +79 -35
  1054. package/src/duckdb/src/planner/binder/tableref/plan_subqueryref.cpp +3 -3
  1055. package/src/duckdb/src/planner/binder.cpp +49 -32
  1056. package/src/duckdb/src/planner/bound_parameter_map.cpp +67 -0
  1057. package/src/duckdb/src/planner/bound_result_modifier.cpp +0 -33
  1058. package/src/duckdb/src/planner/expression/bound_aggregate_expression.cpp +20 -21
  1059. package/src/duckdb/src/planner/expression/bound_between_expression.cpp +4 -19
  1060. package/src/duckdb/src/planner/expression/bound_case_expression.cpp +0 -36
  1061. package/src/duckdb/src/planner/expression/bound_cast_expression.cpp +13 -23
  1062. package/src/duckdb/src/planner/expression/bound_columnref_expression.cpp +0 -20
  1063. package/src/duckdb/src/planner/expression/bound_comparison_expression.cpp +0 -13
  1064. package/src/duckdb/src/planner/expression/bound_conjunction_expression.cpp +0 -13
  1065. package/src/duckdb/src/planner/expression/bound_constant_expression.cpp +0 -11
  1066. package/src/duckdb/src/planner/expression/bound_expression.cpp +1 -5
  1067. package/src/duckdb/src/planner/expression/bound_function_expression.cpp +18 -16
  1068. package/src/duckdb/src/planner/expression/bound_lambda_expression.cpp +0 -4
  1069. package/src/duckdb/src/planner/expression/bound_lambdaref_expression.cpp +0 -23
  1070. package/src/duckdb/src/planner/expression/bound_operator_expression.cpp +0 -16
  1071. package/src/duckdb/src/planner/expression/bound_parameter_expression.cpp +23 -33
  1072. package/src/duckdb/src/planner/expression/bound_reference_expression.cpp +0 -16
  1073. package/src/duckdb/src/planner/expression/bound_subquery_expression.cpp +0 -4
  1074. package/src/duckdb/src/planner/expression/bound_unnest_expression.cpp +0 -15
  1075. package/src/duckdb/src/planner/expression/bound_window_expression.cpp +41 -50
  1076. package/src/duckdb/src/planner/expression.cpp +0 -74
  1077. package/src/duckdb/src/planner/expression_binder/base_select_binder.cpp +14 -6
  1078. package/src/duckdb/src/planner/expression_binder/lateral_binder.cpp +4 -31
  1079. package/src/duckdb/src/planner/expression_binder/order_binder.cpp +5 -4
  1080. package/src/duckdb/src/planner/expression_binder.cpp +23 -0
  1081. package/src/duckdb/src/planner/expression_iterator.cpp +6 -0
  1082. package/src/duckdb/src/planner/filter/conjunction_filter.cpp +0 -21
  1083. package/src/duckdb/src/planner/filter/constant_filter.cpp +0 -12
  1084. package/src/duckdb/src/planner/filter/null_filter.cpp +0 -14
  1085. package/src/duckdb/src/planner/joinside.cpp +0 -24
  1086. package/src/duckdb/src/planner/logical_operator.cpp +18 -206
  1087. package/src/duckdb/src/planner/logical_operator_visitor.cpp +5 -6
  1088. package/src/duckdb/src/planner/operator/logical_aggregate.cpp +0 -46
  1089. package/src/duckdb/src/planner/operator/logical_any_join.cpp +0 -14
  1090. package/src/duckdb/src/planner/operator/logical_column_data_get.cpp +0 -23
  1091. package/src/duckdb/src/planner/operator/logical_comparison_join.cpp +0 -21
  1092. package/src/duckdb/src/planner/operator/logical_copy_to_file.cpp +4 -50
  1093. package/src/duckdb/src/planner/operator/logical_create.cpp +10 -8
  1094. package/src/duckdb/src/planner/operator/logical_create_index.cpp +30 -26
  1095. package/src/duckdb/src/planner/operator/logical_create_table.cpp +12 -6
  1096. package/src/duckdb/src/planner/operator/logical_cross_product.cpp +0 -9
  1097. package/src/duckdb/src/planner/operator/logical_cteref.cpp +0 -16
  1098. package/src/duckdb/src/planner/operator/logical_delete.cpp +4 -18
  1099. package/src/duckdb/src/planner/operator/logical_delim_get.cpp +0 -12
  1100. package/src/duckdb/src/planner/operator/logical_dependent_join.cpp +22 -0
  1101. package/src/duckdb/src/planner/operator/logical_distinct.cpp +10 -13
  1102. package/src/duckdb/src/planner/operator/logical_dummy_scan.cpp +0 -10
  1103. package/src/duckdb/src/planner/operator/logical_empty_result.cpp +0 -15
  1104. package/src/duckdb/src/planner/operator/logical_expression_get.cpp +0 -24
  1105. package/src/duckdb/src/planner/operator/logical_extension_operator.cpp +32 -6
  1106. package/src/duckdb/src/planner/operator/logical_filter.cpp +0 -15
  1107. package/src/duckdb/src/planner/operator/logical_get.cpp +63 -70
  1108. package/src/duckdb/src/planner/operator/logical_insert.cpp +4 -74
  1109. package/src/duckdb/src/planner/operator/logical_join.cpp +0 -17
  1110. package/src/duckdb/src/planner/operator/logical_limit.cpp +0 -16
  1111. package/src/duckdb/src/planner/operator/logical_limit_percent.cpp +0 -16
  1112. package/src/duckdb/src/planner/operator/logical_materialized_cte.cpp +9 -0
  1113. package/src/duckdb/src/planner/operator/logical_order.cpp +35 -11
  1114. package/src/duckdb/src/planner/operator/logical_pivot.cpp +3 -20
  1115. package/src/duckdb/src/planner/operator/logical_positional_join.cpp +0 -10
  1116. package/src/duckdb/src/planner/operator/logical_pragma.cpp +0 -8
  1117. package/src/duckdb/src/planner/operator/logical_prepare.cpp +0 -8
  1118. package/src/duckdb/src/planner/operator/logical_projection.cpp +0 -12
  1119. package/src/duckdb/src/planner/operator/logical_recursive_cte.cpp +0 -15
  1120. package/src/duckdb/src/planner/operator/logical_reset.cpp +0 -12
  1121. package/src/duckdb/src/planner/operator/logical_sample.cpp +3 -11
  1122. package/src/duckdb/src/planner/operator/logical_set.cpp +0 -14
  1123. package/src/duckdb/src/planner/operator/logical_set_operation.cpp +0 -13
  1124. package/src/duckdb/src/planner/operator/logical_simple.cpp +0 -60
  1125. package/src/duckdb/src/planner/operator/logical_top_n.cpp +0 -14
  1126. package/src/duckdb/src/planner/operator/logical_unnest.cpp +0 -14
  1127. package/src/duckdb/src/planner/operator/logical_update.cpp +4 -25
  1128. package/src/duckdb/src/planner/operator/logical_window.cpp +0 -13
  1129. package/src/duckdb/src/planner/planner.cpp +26 -24
  1130. package/src/duckdb/src/planner/subquery/flatten_dependent_join.cpp +90 -38
  1131. package/src/duckdb/src/planner/subquery/has_correlated_expressions.cpp +22 -7
  1132. package/src/duckdb/src/planner/subquery/rewrite_correlated_expressions.cpp +65 -7
  1133. package/src/duckdb/src/planner/table_filter.cpp +0 -59
  1134. package/src/duckdb/src/storage/arena_allocator.cpp +13 -2
  1135. package/src/duckdb/src/storage/buffer/block_manager.cpp +13 -9
  1136. package/src/duckdb/src/storage/checkpoint/row_group_writer.cpp +5 -18
  1137. package/src/duckdb/src/storage/checkpoint/table_data_reader.cpp +10 -8
  1138. package/src/duckdb/src/storage/checkpoint/table_data_writer.cpp +25 -25
  1139. package/src/duckdb/src/storage/checkpoint/write_overflow_strings_to_disk.cpp +53 -26
  1140. package/src/duckdb/src/storage/checkpoint_manager.cpp +314 -267
  1141. package/src/duckdb/src/storage/compression/bitpacking.cpp +138 -107
  1142. package/src/duckdb/src/storage/compression/bitpacking_hugeint.cpp +295 -0
  1143. package/src/duckdb/src/storage/compression/fixed_size_uncompressed.cpp +6 -1
  1144. package/src/duckdb/src/storage/compression/fsst.cpp +1 -1
  1145. package/src/duckdb/src/storage/compression/rle.cpp +66 -16
  1146. package/src/duckdb/src/storage/compression/string_uncompressed.cpp +84 -51
  1147. package/src/duckdb/src/storage/compression/validity_uncompressed.cpp +4 -4
  1148. package/src/duckdb/src/storage/data_pointer.cpp +20 -0
  1149. package/src/duckdb/src/storage/data_table.cpp +60 -39
  1150. package/src/duckdb/src/storage/index.cpp +11 -27
  1151. package/src/duckdb/src/storage/local_storage.cpp +17 -16
  1152. package/src/duckdb/src/storage/metadata/metadata_manager.cpp +322 -0
  1153. package/src/duckdb/src/storage/metadata/metadata_reader.cpp +89 -0
  1154. package/src/duckdb/src/storage/metadata/metadata_writer.cpp +92 -0
  1155. package/src/duckdb/src/storage/partial_block_manager.cpp +42 -15
  1156. package/src/duckdb/src/storage/serialization/serialize_constraint.cpp +98 -0
  1157. package/src/duckdb/src/storage/serialization/serialize_create_info.cpp +198 -0
  1158. package/src/duckdb/src/storage/serialization/serialize_expression.cpp +283 -0
  1159. package/src/duckdb/src/storage/serialization/serialize_logical_operator.cpp +748 -0
  1160. package/src/duckdb/src/storage/serialization/serialize_macro_function.cpp +62 -0
  1161. package/src/duckdb/src/storage/serialization/serialize_nodes.cpp +459 -0
  1162. package/src/duckdb/src/storage/serialization/serialize_parse_info.cpp +421 -0
  1163. package/src/duckdb/src/storage/serialization/serialize_parsed_expression.cpp +342 -0
  1164. package/src/duckdb/src/storage/serialization/serialize_query_node.cpp +122 -0
  1165. package/src/duckdb/src/storage/serialization/serialize_result_modifier.cpp +97 -0
  1166. package/src/duckdb/src/storage/serialization/serialize_statement.cpp +22 -0
  1167. package/src/duckdb/src/storage/serialization/serialize_storage.cpp +78 -0
  1168. package/src/duckdb/src/storage/serialization/serialize_table_filter.cpp +97 -0
  1169. package/src/duckdb/src/storage/serialization/serialize_tableref.cpp +164 -0
  1170. package/src/duckdb/src/storage/serialization/serialize_types.cpp +127 -0
  1171. package/src/duckdb/src/storage/single_file_block_manager.cpp +93 -64
  1172. package/src/duckdb/src/storage/standard_buffer_manager.cpp +10 -16
  1173. package/src/duckdb/src/storage/statistics/base_statistics.cpp +62 -53
  1174. package/src/duckdb/src/storage/statistics/column_statistics.cpp +10 -7
  1175. package/src/duckdb/src/storage/statistics/distinct_statistics.cpp +0 -26
  1176. package/src/duckdb/src/storage/statistics/list_stats.cpp +12 -7
  1177. package/src/duckdb/src/storage/statistics/numeric_stats.cpp +53 -43
  1178. package/src/duckdb/src/storage/statistics/string_stats.cpp +37 -18
  1179. package/src/duckdb/src/storage/statistics/struct_stats.cpp +17 -12
  1180. package/src/duckdb/src/storage/storage_info.cpp +3 -2
  1181. package/src/duckdb/src/storage/storage_manager.cpp +22 -7
  1182. package/src/duckdb/src/storage/table/chunk_info.cpp +59 -36
  1183. package/src/duckdb/src/storage/table/column_checkpoint_state.cpp +31 -34
  1184. package/src/duckdb/src/storage/table/column_data.cpp +44 -41
  1185. package/src/duckdb/src/storage/table/column_data_checkpointer.cpp +7 -11
  1186. package/src/duckdb/src/storage/table/column_segment.cpp +21 -12
  1187. package/src/duckdb/src/storage/table/list_column_data.cpp +24 -14
  1188. package/src/duckdb/src/storage/table/persistent_table_data.cpp +1 -2
  1189. package/src/duckdb/src/storage/table/row_group.cpp +131 -209
  1190. package/src/duckdb/src/storage/table/row_group_collection.cpp +34 -25
  1191. package/src/duckdb/src/storage/table/row_version_manager.cpp +228 -0
  1192. package/src/duckdb/src/storage/table/standard_column_data.cpp +10 -7
  1193. package/src/duckdb/src/storage/table/struct_column_data.cpp +22 -11
  1194. package/src/duckdb/src/storage/table/table_statistics.cpp +19 -9
  1195. package/src/duckdb/src/storage/table/update_segment.cpp +3 -3
  1196. package/src/duckdb/src/storage/table_index_list.cpp +2 -2
  1197. package/src/duckdb/src/storage/wal_replay.cpp +135 -127
  1198. package/src/duckdb/src/storage/write_ahead_log.cpp +136 -64
  1199. package/src/duckdb/src/transaction/cleanup_state.cpp +2 -1
  1200. package/src/duckdb/src/transaction/commit_state.cpp +32 -13
  1201. package/src/duckdb/src/transaction/duck_transaction.cpp +5 -2
  1202. package/src/duckdb/src/transaction/rollback_state.cpp +2 -1
  1203. package/src/duckdb/src/transaction/undo_buffer.cpp +3 -5
  1204. package/src/duckdb/src/verification/deserialized_statement_verifier.cpp +11 -6
  1205. package/src/duckdb/src/verification/prepared_statement_verifier.cpp +16 -11
  1206. package/src/duckdb/src/verification/statement_verifier.cpp +0 -3
  1207. package/src/duckdb/third_party/concurrentqueue/concurrentqueue.h +2 -2
  1208. package/src/duckdb/third_party/concurrentqueue/lightweightsemaphore.h +5 -2
  1209. package/src/duckdb/third_party/fast_float/fast_float/fast_float.h +2 -0
  1210. package/src/duckdb/third_party/httplib/httplib.hpp +10 -1
  1211. package/src/duckdb/third_party/libpg_query/include/nodes/parsenodes.hpp +11 -0
  1212. package/src/duckdb/third_party/libpg_query/include/parser/gram.hpp +2 -1
  1213. package/src/duckdb/third_party/libpg_query/pg_functions.cpp +13 -0
  1214. package/src/duckdb/third_party/libpg_query/src_backend_parser_gram.cpp +11019 -10364
  1215. package/src/duckdb/third_party/libpg_query/src_backend_parser_scansup.cpp +9 -0
  1216. package/src/duckdb/third_party/mbedtls/include/mbedtls_wrapper.hpp +14 -1
  1217. package/src/duckdb/third_party/mbedtls/mbedtls_wrapper.cpp +54 -2
  1218. package/src/duckdb/third_party/parquet/parquet_types.h +1 -1
  1219. package/src/duckdb/third_party/utf8proc/include/utf8proc_wrapper.hpp +8 -0
  1220. package/src/duckdb/ub_extension_icu_third_party_icu_i18n.cpp +6 -6
  1221. package/src/duckdb/ub_src_common.cpp +3 -3
  1222. package/src/duckdb/ub_src_common_adbc_nanoarrow.cpp +8 -0
  1223. package/src/duckdb/ub_src_common_arrow_appender.cpp +10 -0
  1224. package/src/duckdb/ub_src_common_row_operations.cpp +1 -1
  1225. package/src/duckdb/ub_src_common_serializer.cpp +3 -3
  1226. package/src/duckdb/ub_src_core_functions_scalar_debug.cpp +2 -0
  1227. package/src/duckdb/ub_src_core_functions_scalar_list.cpp +6 -0
  1228. package/src/duckdb/ub_src_core_functions_scalar_string.cpp +4 -0
  1229. package/src/duckdb/ub_src_execution.cpp +2 -2
  1230. package/src/duckdb/ub_src_execution_index.cpp +4 -0
  1231. package/src/duckdb/ub_src_execution_index_art.cpp +0 -8
  1232. package/src/duckdb/ub_src_execution_operator_csv_scanner.cpp +18 -0
  1233. package/src/duckdb/ub_src_execution_operator_csv_scanner_sniffer.cpp +12 -0
  1234. package/src/duckdb/ub_src_execution_operator_persistent.cpp +1 -11
  1235. package/src/duckdb/ub_src_execution_operator_schema.cpp +1 -1
  1236. package/src/duckdb/ub_src_execution_operator_set.cpp +2 -0
  1237. package/src/duckdb/ub_src_execution_physical_plan.cpp +2 -0
  1238. package/src/duckdb/ub_src_function_cast_union.cpp +2 -0
  1239. package/src/duckdb/ub_src_function_scalar.cpp +2 -0
  1240. package/src/duckdb/ub_src_function_scalar_compressed_materialization.cpp +4 -0
  1241. package/src/duckdb/ub_src_function_scalar_list.cpp +2 -0
  1242. package/src/duckdb/ub_src_function_table_arrow.cpp +2 -0
  1243. package/src/duckdb/ub_src_function_table_system.cpp +2 -0
  1244. package/src/duckdb/ub_src_main.cpp +2 -0
  1245. package/src/duckdb/ub_src_main_chunk_scan_state.cpp +2 -0
  1246. package/src/duckdb/ub_src_optimizer.cpp +6 -0
  1247. package/src/duckdb/ub_src_optimizer_compressed_materialization.cpp +6 -0
  1248. package/src/duckdb/ub_src_optimizer_join_order.cpp +10 -0
  1249. package/src/duckdb/ub_src_optimizer_statistics_expression.cpp +0 -2
  1250. package/src/duckdb/ub_src_parser.cpp +0 -2
  1251. package/src/duckdb/ub_src_parser_query_node.cpp +2 -0
  1252. package/src/duckdb/ub_src_parser_transform_expression.cpp +2 -0
  1253. package/src/duckdb/ub_src_planner.cpp +2 -2
  1254. package/src/duckdb/ub_src_planner_binder_query_node.cpp +4 -0
  1255. package/src/duckdb/ub_src_planner_expression.cpp +0 -2
  1256. package/src/duckdb/ub_src_planner_operator.cpp +3 -11
  1257. package/src/duckdb/ub_src_storage.cpp +2 -4
  1258. package/src/duckdb/ub_src_storage_compression.cpp +2 -0
  1259. package/src/duckdb/ub_src_storage_metadata.cpp +6 -0
  1260. package/src/duckdb/ub_src_storage_serialization.cpp +30 -0
  1261. package/src/duckdb/ub_src_storage_table.cpp +2 -0
  1262. package/src/duckdb_node.cpp +6 -12
  1263. package/src/duckdb_node.hpp +25 -13
  1264. package/src/statement.cpp +27 -25
  1265. package/test/close_hang.test.ts +39 -0
  1266. package/test/columns.test.ts +25 -3
  1267. package/test/database_fail.test.ts +26 -0
  1268. package/test/extension.test.ts +1 -1
  1269. package/test/prepare.test.ts +10 -1
  1270. package/test/test_all_types.test.ts +234 -0
  1271. package/test/worker.js +7 -0
  1272. package/tsconfig.json +1 -0
  1273. package/src/duckdb/src/common/field_writer.cpp +0 -97
  1274. package/src/duckdb/src/common/row_operations/row_match.cpp +0 -359
  1275. package/src/duckdb/src/common/serializer/buffered_deserializer.cpp +0 -27
  1276. package/src/duckdb/src/common/serializer/buffered_serializer.cpp +0 -36
  1277. package/src/duckdb/src/common/serializer.cpp +0 -24
  1278. package/src/duckdb/src/execution/index/art/fixed_size_allocator.cpp +0 -242
  1279. package/src/duckdb/src/execution/index/art/leaf_segment.cpp +0 -52
  1280. package/src/duckdb/src/execution/index/art/prefix_segment.cpp +0 -42
  1281. package/src/duckdb/src/execution/index/art/swizzleable_pointer.cpp +0 -22
  1282. package/src/duckdb/src/execution/operator/persistent/buffered_csv_reader.cpp +0 -1487
  1283. package/src/duckdb/src/execution/operator/persistent/csv_buffer.cpp +0 -72
  1284. package/src/duckdb/src/execution/operator/persistent/csv_file_handle.cpp +0 -158
  1285. package/src/duckdb/src/execution/operator/persistent/csv_reader_options.cpp +0 -260
  1286. package/src/duckdb/src/execution/operator/schema/physical_create_index.cpp +0 -193
  1287. package/src/duckdb/src/execution/partitionable_hashtable.cpp +0 -172
  1288. package/src/duckdb/src/include/duckdb/common/arrow/arrow_options.hpp +0 -25
  1289. package/src/duckdb/src/include/duckdb/common/field_writer.hpp +0 -383
  1290. package/src/duckdb/src/include/duckdb/common/serializer/buffered_deserializer.hpp +0 -41
  1291. package/src/duckdb/src/include/duckdb/common/serializer/buffered_serializer.hpp +0 -50
  1292. package/src/duckdb/src/include/duckdb/common/serializer/format_serializer.hpp +0 -248
  1293. package/src/duckdb/src/include/duckdb/common/serializer.hpp +0 -158
  1294. package/src/duckdb/src/include/duckdb/execution/index/art/fixed_size_allocator.hpp +0 -117
  1295. package/src/duckdb/src/include/duckdb/execution/index/art/leaf_segment.hpp +0 -38
  1296. package/src/duckdb/src/include/duckdb/execution/index/art/prefix_segment.hpp +0 -40
  1297. package/src/duckdb/src/include/duckdb/execution/index/art/swizzleable_pointer.hpp +0 -58
  1298. package/src/duckdb/src/include/duckdb/execution/operator/persistent/buffered_csv_reader.hpp +0 -133
  1299. package/src/duckdb/src/include/duckdb/execution/operator/persistent/csv_buffer.hpp +0 -74
  1300. package/src/duckdb/src/include/duckdb/execution/partitionable_hashtable.hpp +0 -69
  1301. package/src/duckdb/src/include/duckdb/planner/operator/logical_asof_join.hpp +0 -27
  1302. package/src/duckdb/src/include/duckdb/planner/operator/logical_delim_join.hpp +0 -32
  1303. package/src/duckdb/src/include/duckdb/planner/plan_serialization.hpp +0 -44
  1304. package/src/duckdb/src/include/duckdb/storage/meta_block_reader.hpp +0 -49
  1305. package/src/duckdb/src/include/duckdb/storage/meta_block_writer.hpp +0 -50
  1306. package/src/duckdb/src/include/duckdb/verification/deserialized_statement_verifier_v2.hpp +0 -26
  1307. package/src/duckdb/src/optimizer/statistics/expression/propagate_and_compress.cpp +0 -118
  1308. package/src/duckdb/src/parser/common_table_expression_info.cpp +0 -19
  1309. package/src/duckdb/src/planner/expression/bound_default_expression.cpp +0 -16
  1310. package/src/duckdb/src/planner/operator/logical_asof_join.cpp +0 -14
  1311. package/src/duckdb/src/planner/operator/logical_delim_join.cpp +0 -27
  1312. package/src/duckdb/src/planner/operator/logical_execute.cpp +0 -12
  1313. package/src/duckdb/src/planner/operator/logical_explain.cpp +0 -22
  1314. package/src/duckdb/src/planner/operator/logical_export.cpp +0 -13
  1315. package/src/duckdb/src/planner/operator/logical_show.cpp +0 -21
  1316. package/src/duckdb/src/planner/parsed_data/bound_create_table_info.cpp +0 -20
  1317. package/src/duckdb/src/planner/plan_serialization.cpp +0 -20
  1318. package/src/duckdb/src/storage/meta_block_reader.cpp +0 -78
  1319. package/src/duckdb/src/storage/meta_block_writer.cpp +0 -80
  1320. package/src/duckdb/src/verification/deserialized_statement_verifier_v2.cpp +0 -20
  1321. package/src/duckdb/ub_src_planner_parsed_data.cpp +0 -2
@@ -1,46 +1,28 @@
1
1
  #include "duckdb/execution/radix_partitioned_hashtable.hpp"
2
2
 
3
+ #include "duckdb/common/radix_partitioning.hpp"
4
+ #include "duckdb/common/row_operations/row_operations.hpp"
5
+ #include "duckdb/common/types/row/tuple_data_collection.hpp"
6
+ #include "duckdb/common/types/row/tuple_data_iterator.hpp"
7
+ #include "duckdb/execution/aggregate_hashtable.hpp"
8
+ #include "duckdb/execution/executor.hpp"
3
9
  #include "duckdb/execution/operator/aggregate/physical_hash_aggregate.hpp"
10
+ #include "duckdb/main/config.hpp"
4
11
  #include "duckdb/parallel/event.hpp"
5
- #include "duckdb/parallel/task_scheduler.hpp"
6
12
  #include "duckdb/planner/expression/bound_reference_expression.hpp"
7
13
 
8
14
  namespace duckdb {
9
15
 
10
- // compute the GROUPING values
11
- // for each parameter to the GROUPING clause, we check if the hash table groups on this particular group
12
- // if it does, we return 0, otherwise we return 1
13
- // we then use bitshifts to combine these values
14
- void RadixPartitionedHashTable::SetGroupingValues() {
15
- auto &grouping_functions = op.GetGroupingFunctions();
16
- for (auto &grouping : grouping_functions) {
17
- int64_t grouping_value = 0;
18
- D_ASSERT(grouping.size() < sizeof(int64_t) * 8);
19
- for (idx_t i = 0; i < grouping.size(); i++) {
20
- if (grouping_set.find(grouping[i]) == grouping_set.end()) {
21
- // we don't group on this value!
22
- grouping_value += (int64_t)1 << (grouping.size() - (i + 1));
23
- }
24
- }
25
- grouping_values.push_back(Value::BIGINT(grouping_value));
26
- }
27
- }
28
-
29
16
  RadixPartitionedHashTable::RadixPartitionedHashTable(GroupingSet &grouping_set_p, const GroupedAggregateData &op_p)
30
17
  : grouping_set(grouping_set_p), op(op_p) {
31
-
32
18
  auto groups_count = op.GroupCount();
33
19
  for (idx_t i = 0; i < groups_count; i++) {
34
20
  if (grouping_set.find(i) == grouping_set.end()) {
35
21
  null_groups.push_back(i);
36
22
  }
37
23
  }
38
-
39
- // 10000 seems like a good compromise here
40
- radix_limit = 10000;
41
-
42
24
  if (grouping_set.empty()) {
43
- // fake a single group with a constant value for aggregation without groups
25
+ // Fake a single group with a constant value for aggregation without groups
44
26
  group_types.emplace_back(LogicalType::TINYINT);
45
27
  }
46
28
  for (auto &entry : grouping_set) {
@@ -48,71 +30,279 @@ RadixPartitionedHashTable::RadixPartitionedHashTable(GroupingSet &grouping_set_p
48
30
  group_types.push_back(op.group_types[entry]);
49
31
  }
50
32
  SetGroupingValues();
33
+
34
+ auto group_types_copy = group_types;
35
+ group_types_copy.emplace_back(LogicalType::HASH);
36
+ layout.Initialize(std::move(group_types_copy), AggregateObject::CreateAggregateObjects(op.bindings));
37
+ }
38
+
39
+ void RadixPartitionedHashTable::SetGroupingValues() {
40
+ // Compute the GROUPING values:
41
+ // For each parameter to the GROUPING clause, we check if the hash table groups on this particular group
42
+ // If it does, we return 0, otherwise we return 1
43
+ // We then use bitshifts to combine these values
44
+ auto &grouping_functions = op.GetGroupingFunctions();
45
+ for (auto &grouping : grouping_functions) {
46
+ int64_t grouping_value = 0;
47
+ D_ASSERT(grouping.size() < sizeof(int64_t) * 8);
48
+ for (idx_t i = 0; i < grouping.size(); i++) {
49
+ if (grouping_set.find(grouping[i]) == grouping_set.end()) {
50
+ // We don't group on this value!
51
+ grouping_value += (int64_t)1 << (grouping.size() - (i + 1));
52
+ }
53
+ }
54
+ grouping_values.push_back(Value::BIGINT(grouping_value));
55
+ }
56
+ }
57
+
58
+ const TupleDataLayout &RadixPartitionedHashTable::GetLayout() const {
59
+ return layout;
60
+ }
61
+
62
+ unique_ptr<GroupedAggregateHashTable> RadixPartitionedHashTable::CreateHT(ClientContext &context, const idx_t capacity,
63
+ const idx_t radix_bits) const {
64
+ return make_uniq<GroupedAggregateHashTable>(context, BufferAllocator::Get(context), group_types, op.payload_types,
65
+ op.bindings, capacity, radix_bits);
51
66
  }
52
67
 
53
68
  //===--------------------------------------------------------------------===//
54
69
  // Sink
55
70
  //===--------------------------------------------------------------------===//
56
- class RadixHTGlobalState : public GlobalSinkState {
57
- constexpr const static idx_t MAX_RADIX_PARTITIONS = 32;
71
+ struct AggregatePartition {
72
+ explicit AggregatePartition(unique_ptr<TupleDataCollection> data_p) : data(std::move(data_p)), finalized(false) {
73
+ }
74
+ unique_ptr<TupleDataCollection> data;
75
+ atomic<bool> finalized;
76
+ };
58
77
 
78
+ class RadixHTGlobalSinkState;
79
+
80
+ struct RadixHTConfig {
59
81
  public:
60
- explicit RadixHTGlobalState(ClientContext &context)
61
- : is_empty(true), multi_scan(true), partitioned(false),
62
- partition_info(
63
- MinValue<idx_t>(MAX_RADIX_PARTITIONS, TaskScheduler::GetScheduler(context).NumberOfThreads())) {
64
- }
82
+ explicit RadixHTConfig(ClientContext &context, RadixHTGlobalSinkState &sink);
65
83
 
66
- vector<unique_ptr<PartitionableHashTable>> intermediate_hts;
67
- vector<shared_ptr<GroupedAggregateHashTable>> finalized_hts;
84
+ void SetRadixBits(idx_t radix_bits_p);
85
+ bool SetRadixBitsToExternal();
86
+ idx_t GetRadixBits() const;
68
87
 
69
- //! Whether or not any tuples were added to the HT
70
- bool is_empty;
71
- //! Whether or not the hash table should be scannable multiple times
72
- bool multi_scan;
73
- //! The lock for updating the global aggregate state
74
- mutex lock;
75
- //! Whether or not any thread has crossed the partitioning threshold
76
- atomic<bool> partitioned;
88
+ private:
89
+ void SetRadixBitsInternal(const idx_t radix_bits_p, bool external);
90
+ static idx_t InitialSinkRadixBits(ClientContext &context);
91
+ static idx_t MaximumSinkRadixBits(ClientContext &context);
92
+ static idx_t ExternalRadixBits(const idx_t &maximum_sink_radix_bits_p);
93
+ static idx_t SinkCapacity(ClientContext &context);
77
94
 
78
- bool is_finalized = false;
79
- bool is_partitioned = false;
95
+ private:
96
+ //! Assume (1 << 15) = 32KB L1 cache per core, divided by two because hyperthreading
97
+ static constexpr const idx_t L1_CACHE_SIZE = 32768 / 2;
98
+ //! Assume (1 << 20) = 1MB L2 cache per core, divided by two because hyperthreading
99
+ static constexpr const idx_t L2_CACHE_SIZE = 1048576 / 2;
100
+ //! Assume (1 << 20) + (1 << 19) = 1.5MB L3 cache per core (shared), divided by two because hyperthreading
101
+ static constexpr const idx_t L3_CACHE_SIZE = 1572864 / 2;
102
+
103
+ //! Sink radix bits to initialize with
104
+ static constexpr const idx_t MAXIMUM_INITIAL_SINK_RADIX_BITS = 3;
105
+ //! Maximum Sink radix bits (independent of threads)
106
+ static constexpr const idx_t MAXIMUM_FINAL_SINK_RADIX_BITS = 7;
107
+ //! By how many radix bits to increment if we go external
108
+ static constexpr const idx_t EXTERNAL_RADIX_BITS_INCREMENT = 3;
109
+
110
+ //! The global sink state
111
+ RadixHTGlobalSinkState &sink;
112
+ //! Current thread-global sink radix bits
113
+ atomic<idx_t> sink_radix_bits;
114
+ //! Maximum Sink radix bits (set based on number of threads)
115
+ const idx_t maximum_sink_radix_bits;
116
+ //! Radix bits if we go external
117
+ const idx_t external_radix_bits;
118
+
119
+ public:
120
+ //! Capacity of HTs during the Sink
121
+ const idx_t sink_capacity;
80
122
 
81
- RadixPartitionInfo partition_info;
82
- AggregateHTAppendState append_state;
123
+ //! If we fill this many blocks per partition, we trigger a repartition
124
+ static constexpr const double BLOCK_FILL_FACTOR = 1.8;
125
+ //! By how many bits to repartition if a repartition is triggered
126
+ static constexpr const idx_t REPARTITION_RADIX_BITS = 2;
83
127
  };
84
128
 
85
- class RadixHTLocalState : public LocalSinkState {
129
+ class RadixHTGlobalSinkState : public GlobalSinkState {
86
130
  public:
87
- explicit RadixHTLocalState(const RadixPartitionedHashTable &ht) : total_groups(0), is_empty(true) {
88
- // if there are no groups we create a fake group so everything has the same group
89
- group_chunk.InitializeEmpty(ht.group_types);
90
- if (ht.grouping_set.empty()) {
91
- group_chunk.data[0].Reference(Value::TINYINT(42));
131
+ RadixHTGlobalSinkState(ClientContext &context, const RadixPartitionedHashTable &radix_ht);
132
+
133
+ //! Destroys aggregate states (if multi-scan)
134
+ ~RadixHTGlobalSinkState() override;
135
+ void Destroy();
136
+
137
+ public:
138
+ //! The radix HT
139
+ const RadixPartitionedHashTable &radix_ht;
140
+ //! Config for partitioning
141
+ RadixHTConfig config;
142
+
143
+ //! Whether we've called Finalize
144
+ bool finalized;
145
+ //! Whether we are doing an external aggregation
146
+ atomic<bool> external;
147
+ //! Threads that have called Sink
148
+ atomic<idx_t> active_threads;
149
+ //! If any thread has called combine
150
+ atomic<bool> any_combined;
151
+
152
+ //! Lock for uncombined_data/stored_allocators
153
+ mutex lock;
154
+ //! Uncombined partitioned data that will be put into the AggregatePartitions
155
+ unique_ptr<PartitionedTupleData> uncombined_data;
156
+ //! Allocators used during the Sink/Finalize
157
+ vector<shared_ptr<ArenaAllocator>> stored_allocators;
158
+
159
+ //! Partitions that are finalized during GetData
160
+ vector<unique_ptr<AggregatePartition>> partitions;
161
+
162
+ //! For synchronizing finalize tasks
163
+ atomic<idx_t> finalize_idx;
164
+
165
+ //! Pin properties when scanning
166
+ TupleDataPinProperties scan_pin_properties;
167
+ //! Total count before combining
168
+ idx_t count_before_combining;
169
+ };
170
+
171
+ RadixHTGlobalSinkState::RadixHTGlobalSinkState(ClientContext &context, const RadixPartitionedHashTable &radix_ht_p)
172
+ : radix_ht(radix_ht_p), config(context, *this), finalized(false), external(false), active_threads(0),
173
+ any_combined(false), finalize_idx(0), scan_pin_properties(TupleDataPinProperties::DESTROY_AFTER_DONE),
174
+ count_before_combining(0) {
175
+ }
176
+
177
+ RadixHTGlobalSinkState::~RadixHTGlobalSinkState() {
178
+ Destroy();
179
+ }
180
+
181
+ // LCOV_EXCL_START
182
+ void RadixHTGlobalSinkState::Destroy() {
183
+ if (scan_pin_properties == TupleDataPinProperties::DESTROY_AFTER_DONE || count_before_combining == 0 ||
184
+ partitions.empty()) {
185
+ // Already destroyed / empty
186
+ return;
187
+ }
188
+
189
+ TupleDataLayout layout = partitions[0]->data->GetLayout().Copy();
190
+ if (!layout.HasDestructor()) {
191
+ return; // No destructors, exit
192
+ }
193
+
194
+ // There are aggregates with destructors: Call the destructor for each of the aggregates
195
+ RowOperationsState row_state(*stored_allocators.back());
196
+ for (auto &partition : partitions) {
197
+ auto &data_collection = *partition->data;
198
+ if (data_collection.Count() == 0) {
199
+ continue;
92
200
  }
201
+ TupleDataChunkIterator iterator(data_collection, TupleDataPinProperties::DESTROY_AFTER_DONE, false);
202
+ auto &row_locations = iterator.GetChunkState().row_locations;
203
+ do {
204
+ RowOperations::DestroyStates(row_state, layout, row_locations, iterator.GetCurrentChunkCount());
205
+ } while (iterator.Next());
206
+ data_collection.Reset();
207
+ }
208
+ }
209
+ // LCOV_EXCL_STOP
210
+
211
+ RadixHTConfig::RadixHTConfig(ClientContext &context, RadixHTGlobalSinkState &sink_p)
212
+ : sink(sink_p), sink_radix_bits(InitialSinkRadixBits(context)),
213
+ maximum_sink_radix_bits(MaximumSinkRadixBits(context)),
214
+ external_radix_bits(ExternalRadixBits(maximum_sink_radix_bits)), sink_capacity(SinkCapacity(context)) {
215
+ }
216
+
217
+ void RadixHTConfig::SetRadixBits(idx_t radix_bits_p) {
218
+ SetRadixBitsInternal(MinValue(radix_bits_p, maximum_sink_radix_bits), false);
219
+ }
220
+
221
+ bool RadixHTConfig::SetRadixBitsToExternal() {
222
+ SetRadixBitsInternal(external_radix_bits, true);
223
+ return sink.external;
224
+ }
225
+
226
+ idx_t RadixHTConfig::GetRadixBits() const {
227
+ return sink_radix_bits;
228
+ }
229
+
230
+ void RadixHTConfig::SetRadixBitsInternal(const idx_t radix_bits_p, bool external) {
231
+ if (sink_radix_bits >= radix_bits_p || sink.any_combined) {
232
+ return;
93
233
  }
94
234
 
235
+ lock_guard<mutex> guard(sink.lock);
236
+ if (sink_radix_bits >= radix_bits_p || sink.any_combined) {
237
+ return;
238
+ }
239
+
240
+ if (external) {
241
+ sink.external = true;
242
+ }
243
+ sink_radix_bits = radix_bits_p;
244
+ return;
245
+ }
246
+
247
+ idx_t RadixHTConfig::InitialSinkRadixBits(ClientContext &context) {
248
+ const idx_t active_threads = TaskScheduler::GetScheduler(context).NumberOfThreads();
249
+ return MinValue(RadixPartitioning::RadixBits(NextPowerOfTwo(active_threads)), MAXIMUM_INITIAL_SINK_RADIX_BITS);
250
+ }
251
+
252
+ idx_t RadixHTConfig::MaximumSinkRadixBits(ClientContext &context) {
253
+ const idx_t active_threads = TaskScheduler::GetScheduler(context).NumberOfThreads();
254
+ return MinValue(RadixPartitioning::RadixBits(NextPowerOfTwo(active_threads)), MAXIMUM_FINAL_SINK_RADIX_BITS);
255
+ }
256
+
257
+ idx_t RadixHTConfig::ExternalRadixBits(const idx_t &maximum_sink_radix_bits_p) {
258
+ return MinValue(maximum_sink_radix_bits_p + EXTERNAL_RADIX_BITS_INCREMENT, MAXIMUM_FINAL_SINK_RADIX_BITS);
259
+ }
260
+
261
+ idx_t RadixHTConfig::SinkCapacity(ClientContext &context) {
262
+ // Get active and maximum number of threads
263
+ const idx_t active_threads = TaskScheduler::GetScheduler(context).NumberOfThreads();
264
+ const auto max_threads = DBConfig::GetSystemMaxThreads(FileSystem::GetFileSystem(context));
265
+
266
+ // Compute cache size per active thread (assuming cache is shared)
267
+ const auto total_shared_cache_size = max_threads * L3_CACHE_SIZE;
268
+ const auto cache_per_active_thread = L1_CACHE_SIZE + L2_CACHE_SIZE + total_shared_cache_size / active_threads;
269
+
270
+ // Divide cache per active thread by entry size, round up to next power of two, to get capacity
271
+ const auto size_per_entry = sizeof(aggr_ht_entry_t) * GroupedAggregateHashTable::LOAD_FACTOR;
272
+ const auto capacity = NextPowerOfTwo(cache_per_active_thread / size_per_entry);
273
+
274
+ // Capacity must be at least the minimum capacity
275
+ return MaxValue<idx_t>(capacity, GroupedAggregateHashTable::InitialCapacity());
276
+ }
277
+
278
+ class RadixHTLocalSinkState : public LocalSinkState {
279
+ public:
280
+ RadixHTLocalSinkState(ClientContext &context, const RadixPartitionedHashTable &radix_ht);
281
+
282
+ public:
283
+ //! Thread-local HT that is re-used after abandoning
284
+ unique_ptr<GroupedAggregateHashTable> ht;
285
+ //! Chunk with group columns
95
286
  DataChunk group_chunk;
96
- //! The aggregate HT
97
- unique_ptr<PartitionableHashTable> ht;
98
- //! The total number of groups found by this thread
99
- idx_t total_groups;
100
287
 
101
- //! Whether or not any tuples were added to the HT
102
- bool is_empty;
288
+ //! Data that is abandoned ends up here (only if we're doing external aggregation)
289
+ unique_ptr<PartitionedTupleData> abandoned_data;
103
290
  };
104
291
 
105
- void RadixPartitionedHashTable::SetMultiScan(GlobalSinkState &state) {
106
- auto &gstate = state.Cast<RadixHTGlobalState>();
107
- gstate.multi_scan = true;
292
+ RadixHTLocalSinkState::RadixHTLocalSinkState(ClientContext &, const RadixPartitionedHashTable &radix_ht) {
293
+ // If there are no groups we create a fake group so everything has the same group
294
+ group_chunk.InitializeEmpty(radix_ht.group_types);
295
+ if (radix_ht.grouping_set.empty()) {
296
+ group_chunk.data[0].Reference(Value::TINYINT(42));
297
+ }
108
298
  }
109
299
 
110
300
  unique_ptr<GlobalSinkState> RadixPartitionedHashTable::GetGlobalSinkState(ClientContext &context) const {
111
- return make_uniq<RadixHTGlobalState>(context);
301
+ return make_uniq<RadixHTGlobalSinkState>(context, *this);
112
302
  }
113
303
 
114
304
  unique_ptr<LocalSinkState> RadixPartitionedHashTable::GetLocalSinkState(ExecutionContext &context) const {
115
- return make_uniq<RadixHTLocalState>(*this);
305
+ return make_uniq<RadixHTLocalSinkState>(context.client, *this);
116
306
  }
117
307
 
118
308
  void RadixPartitionedHashTable::PopulateGroupChunk(DataChunk &group_chunk, DataChunk &input_chunk) const {
@@ -130,386 +320,491 @@ void RadixPartitionedHashTable::PopulateGroupChunk(DataChunk &group_chunk, DataC
130
320
  group_chunk.Verify();
131
321
  }
132
322
 
133
- void RadixPartitionedHashTable::Sink(ExecutionContext &context, DataChunk &chunk, OperatorSinkInput &input,
134
- DataChunk &payload_input, const unsafe_vector<idx_t> &filter) const {
135
- auto &llstate = input.local_state.Cast<RadixHTLocalState>();
136
- auto &gstate = input.global_state.Cast<RadixHTGlobalState>();
137
- D_ASSERT(!gstate.is_finalized);
138
-
139
- DataChunk &group_chunk = llstate.group_chunk;
140
- PopulateGroupChunk(group_chunk, chunk);
323
+ bool MaybeRepartition(ClientContext &context, RadixHTGlobalSinkState &gstate, RadixHTLocalSinkState &lstate) {
324
+ auto &config = gstate.config;
325
+ auto &ht = *lstate.ht;
326
+ auto &partitioned_data = ht.GetPartitionedData();
327
+
328
+ // Check if we're approaching the memory limit
329
+ const idx_t n_threads = TaskScheduler::GetScheduler(context).NumberOfThreads();
330
+ const idx_t limit = BufferManager::GetBufferManager(context).GetMaxMemory();
331
+ const idx_t thread_limit = 0.6 * limit / n_threads;
332
+ if (ht.GetPartitionedData()->SizeInBytes() > thread_limit || context.config.force_external) {
333
+ if (gstate.config.SetRadixBitsToExternal()) {
334
+ // We're approaching the memory limit, unpin the data
335
+ if (!lstate.abandoned_data) {
336
+ lstate.abandoned_data = make_uniq<RadixPartitionedTupleData>(
337
+ BufferManager::GetBufferManager(context), gstate.radix_ht.GetLayout(), config.GetRadixBits(),
338
+ gstate.radix_ht.GetLayout().ColumnCount() - 1);
339
+ }
141
340
 
142
- // if we have non-combinable aggregates (e.g. string_agg) we cannot keep parallel hash
143
- // tables
144
- if (ForceSingleHT(input.global_state)) {
145
- lock_guard<mutex> glock(gstate.lock);
146
- gstate.is_empty = gstate.is_empty && group_chunk.size() == 0;
147
- if (gstate.finalized_hts.empty()) {
148
- // Create a finalized ht in the global state, that we can populate
149
- gstate.finalized_hts.push_back(
150
- make_shared<GroupedAggregateHashTable>(context.client, Allocator::Get(context.client), group_types,
151
- op.payload_types, op.bindings, HtEntryType::HT_WIDTH_64));
341
+ ht.UnpinData();
342
+ partitioned_data->Repartition(*lstate.abandoned_data);
343
+ ht.SetRadixBits(gstate.config.GetRadixBits());
344
+ ht.InitializePartitionedData();
345
+ return true;
152
346
  }
153
- D_ASSERT(gstate.finalized_hts.size() == 1);
154
- D_ASSERT(gstate.finalized_hts[0]);
155
- llstate.total_groups +=
156
- gstate.finalized_hts[0]->AddChunk(gstate.append_state, group_chunk, payload_input, filter);
157
- return;
158
347
  }
159
348
 
160
- if (group_chunk.size() > 0) {
161
- llstate.is_empty = false;
162
- }
349
+ const auto partition_count = partitioned_data->PartitionCount();
350
+ const auto current_radix_bits = RadixPartitioning::RadixBits(partition_count);
351
+ D_ASSERT(current_radix_bits <= config.GetRadixBits());
163
352
 
164
- if (!llstate.ht) {
165
- llstate.ht =
166
- make_uniq<PartitionableHashTable>(context.client, Allocator::Get(context.client), gstate.partition_info,
167
- group_types, op.payload_types, op.bindings);
353
+ const auto row_size_per_partition =
354
+ partitioned_data->Count() * partitioned_data->GetLayout().GetRowWidth() / partition_count;
355
+ if (row_size_per_partition > config.BLOCK_FILL_FACTOR * Storage::BLOCK_SIZE) {
356
+ // We crossed our block filling threshold, try to increment radix bits
357
+ config.SetRadixBits(current_radix_bits + config.REPARTITION_RADIX_BITS);
168
358
  }
169
359
 
170
- llstate.total_groups += llstate.ht->AddChunk(group_chunk, payload_input,
171
- gstate.partitioned && gstate.partition_info.n_partitions > 1, filter);
172
- if (llstate.total_groups >= radix_limit) {
173
- gstate.partitioned = true;
360
+ const auto global_radix_bits = config.GetRadixBits();
361
+ if (current_radix_bits == global_radix_bits) {
362
+ return false; // We're already on the right number of radix bits
174
363
  }
364
+
365
+ // We're out-of-sync with the global radix bits, repartition
366
+ ht.UnpinData();
367
+ auto old_partitioned_data = std::move(partitioned_data);
368
+ ht.SetRadixBits(global_radix_bits);
369
+ ht.InitializePartitionedData();
370
+ old_partitioned_data->Repartition(*ht.GetPartitionedData());
371
+ return true;
175
372
  }
176
373
 
177
- void RadixPartitionedHashTable::Combine(ExecutionContext &context, GlobalSinkState &state,
178
- LocalSinkState &lstate) const {
179
- auto &llstate = lstate.Cast<RadixHTLocalState>();
180
- auto &gstate = state.Cast<RadixHTGlobalState>();
181
- D_ASSERT(!gstate.is_finalized);
374
+ void RadixPartitionedHashTable::Sink(ExecutionContext &context, DataChunk &chunk, OperatorSinkInput &input,
375
+ DataChunk &payload_input, const unsafe_vector<idx_t> &filter) const {
376
+ auto &gstate = input.global_state.Cast<RadixHTGlobalSinkState>();
377
+ auto &lstate = input.local_state.Cast<RadixHTLocalSinkState>();
378
+ if (!lstate.ht) {
379
+ lstate.ht = CreateHT(context.client, gstate.config.sink_capacity, gstate.config.GetRadixBits());
380
+ gstate.active_threads++;
381
+ }
382
+
383
+ auto &group_chunk = lstate.group_chunk;
384
+ PopulateGroupChunk(group_chunk, chunk);
182
385
 
183
- // this actually does not do a lot but just pushes the local HTs into the global state so we can later combine them
184
- // in parallel
386
+ auto &ht = *lstate.ht;
387
+ ht.AddChunk(group_chunk, payload_input, filter);
185
388
 
186
- if (ForceSingleHT(state)) {
187
- D_ASSERT(gstate.finalized_hts.size() <= 1);
188
- return;
389
+ if (ht.Count() + STANDARD_VECTOR_SIZE < ht.ResizeThreshold()) {
390
+ return; // We can fit another chunk
189
391
  }
190
392
 
191
- if (!llstate.ht) {
192
- return; // no data
393
+ if (gstate.active_threads > 2) {
394
+ // 'Reset' the HT without taking its data, we can just keep appending to the same collection
395
+ // This only works because we never resize the HT
396
+ ht.ClearPointerTable();
397
+ ht.ResetCount();
398
+ // We don't do this when running with 1 or 2 threads, it only makes sense when there's many threads
193
399
  }
194
400
 
195
- if (!llstate.ht->IsPartitioned() && gstate.partition_info.n_partitions > 1 && gstate.partitioned) {
196
- llstate.ht->Partition();
401
+ // Check if we need to repartition
402
+ auto repartitioned = MaybeRepartition(context.client, gstate, lstate);
403
+
404
+ if (repartitioned && ht.Count() != 0) {
405
+ // We repartitioned, but we didn't clear the pointer table / reset the count because we're on 1 or 2 threads
406
+ ht.ClearPointerTable();
407
+ ht.ResetCount();
197
408
  }
198
409
 
199
- // we will never add new values to these HTs so we can drop the first part of the HT
200
- llstate.ht->Finalize();
410
+ // TODO: combine early and often
411
+ }
201
412
 
202
- lock_guard<mutex> glock(gstate.lock);
203
- if (!llstate.is_empty) {
204
- gstate.is_empty = false;
413
+ void RadixPartitionedHashTable::Combine(ExecutionContext &context, GlobalSinkState &gstate_p,
414
+ LocalSinkState &lstate_p) const {
415
+ auto &gstate = gstate_p.Cast<RadixHTGlobalSinkState>();
416
+ auto &lstate = lstate_p.Cast<RadixHTLocalSinkState>();
417
+ if (!lstate.ht) {
418
+ return;
205
419
  }
206
- // at this point we just collect them the PhysicalHashAggregateFinalizeTask (below) will merge them in parallel
207
- gstate.intermediate_hts.push_back(std::move(llstate.ht));
208
- }
209
420
 
210
- bool RadixPartitionedHashTable::Finalize(ClientContext &context, GlobalSinkState &gstate_p) const {
211
- auto &gstate = gstate_p.Cast<RadixHTGlobalState>();
212
- D_ASSERT(!gstate.is_finalized);
213
- gstate.is_finalized = true;
421
+ // Set any_combined, then check one last time whether we need to repartition
422
+ gstate.any_combined = true;
423
+ MaybeRepartition(context.client, gstate, lstate);
214
424
 
215
- // special case if we have non-combinable aggregates
216
- // we have already aggreagted into a global shared HT that does not require any additional finalization steps
217
- if (ForceSingleHT(gstate)) {
218
- D_ASSERT(gstate.finalized_hts.size() <= 1);
219
- D_ASSERT(gstate.finalized_hts.empty() || gstate.finalized_hts[0]);
220
- return false;
221
- }
425
+ auto &ht = *lstate.ht;
426
+ ht.UnpinData();
222
427
 
223
- // we can have two cases now, non-partitioned for few groups and radix-partitioned for very many groups.
224
- // go through all of the child hts and see if we ever called partition() on any of them
225
- // if we did, its the latter case.
226
- bool any_partitioned = false;
227
- for (auto &pht : gstate.intermediate_hts) {
228
- if (pht->IsPartitioned()) {
229
- any_partitioned = true;
230
- break;
231
- }
428
+ if (lstate.abandoned_data) {
429
+ D_ASSERT(gstate.external);
430
+ D_ASSERT(lstate.abandoned_data->PartitionCount() == lstate.ht->GetPartitionedData()->PartitionCount());
431
+ D_ASSERT(lstate.abandoned_data->PartitionCount() ==
432
+ RadixPartitioning::NumberOfPartitions(gstate.config.GetRadixBits()));
433
+ lstate.abandoned_data->Combine(*lstate.ht->GetPartitionedData());
434
+ } else {
435
+ lstate.abandoned_data = std::move(ht.GetPartitionedData());
232
436
  }
233
437
 
234
- auto &allocator = Allocator::Get(context);
235
- if (any_partitioned) {
236
- // if one is partitioned, all have to be
237
- // this should mostly have already happened in Combine, but if not we do it here
238
- for (auto &pht : gstate.intermediate_hts) {
239
- if (!pht->IsPartitioned()) {
240
- pht->Partition();
241
- }
242
- }
243
- // schedule additional tasks to combine the partial HTs
244
- gstate.finalized_hts.resize(gstate.partition_info.n_partitions);
245
- for (idx_t r = 0; r < gstate.partition_info.n_partitions; r++) {
246
- gstate.finalized_hts[r] = make_shared<GroupedAggregateHashTable>(
247
- context, allocator, group_types, op.payload_types, op.bindings, HtEntryType::HT_WIDTH_64);
248
- }
249
- gstate.is_partitioned = true;
250
- return true;
251
- } else { // in the non-partitioned case we immediately combine all the unpartitioned hts created by the threads.
252
- // TODO possible optimization, if total count < limit for 32 bit ht, use that one
253
- // create this ht here so finalize needs no lock on gstate
254
-
255
- gstate.finalized_hts.push_back(make_shared<GroupedAggregateHashTable>(
256
- context, allocator, group_types, op.payload_types, op.bindings, HtEntryType::HT_WIDTH_64));
257
- for (auto &pht : gstate.intermediate_hts) {
258
- auto unpartitioned = pht->GetUnpartitioned();
259
- for (auto &unpartitioned_ht : unpartitioned) {
260
- D_ASSERT(unpartitioned_ht);
261
- gstate.finalized_hts[0]->Combine(*unpartitioned_ht);
262
- unpartitioned_ht.reset();
263
- }
264
- unpartitioned.clear();
265
- }
266
- D_ASSERT(gstate.finalized_hts[0]);
267
- gstate.finalized_hts[0]->Finalize();
268
- return false;
438
+ lock_guard<mutex> guard(gstate.lock);
439
+ if (gstate.uncombined_data) {
440
+ gstate.uncombined_data->Combine(*lstate.abandoned_data);
441
+ } else {
442
+ gstate.uncombined_data = std::move(lstate.abandoned_data);
269
443
  }
444
+ gstate.stored_allocators.emplace_back(ht.GetAggregateAllocator());
270
445
  }
271
446
 
272
- // this task is run in multiple threads and combines the radix-partitioned hash tables into a single onen and then
273
- // folds them into the global ht finally.
274
- class RadixAggregateFinalizeTask : public ExecutorTask {
275
- public:
276
- RadixAggregateFinalizeTask(Executor &executor, shared_ptr<Event> event_p, RadixHTGlobalState &state_p,
277
- idx_t radix_p)
278
- : ExecutorTask(executor), event(std::move(event_p)), state(state_p), radix(radix_p) {
279
- }
280
-
281
- static void FinalizeHT(RadixHTGlobalState &gstate, idx_t radix) {
282
- D_ASSERT(gstate.partition_info.n_partitions <= gstate.finalized_hts.size());
283
- D_ASSERT(gstate.finalized_hts[radix]);
284
- for (auto &pht : gstate.intermediate_hts) {
285
- for (auto &ht : pht->GetPartition(radix)) {
286
- gstate.finalized_hts[radix]->Combine(*ht);
287
- ht.reset();
288
- }
289
- }
290
- gstate.finalized_hts[radix]->Finalize();
291
- }
447
+ void RadixPartitionedHashTable::Finalize(ClientContext &, GlobalSinkState &gstate_p) const {
448
+ auto &gstate = gstate_p.Cast<RadixHTGlobalSinkState>();
292
449
 
293
- TaskExecutionResult ExecuteTask(TaskExecutionMode mode) override {
294
- FinalizeHT(state, radix);
295
- event->FinishTask();
296
- return TaskExecutionResult::TASK_FINISHED;
297
- }
450
+ if (gstate.uncombined_data) {
451
+ auto &uncombined_data = *gstate.uncombined_data;
452
+ gstate.count_before_combining = uncombined_data.Count();
298
453
 
299
- private:
300
- shared_ptr<Event> event;
301
- RadixHTGlobalState &state;
302
- idx_t radix;
303
- };
454
+ // If true there is no need to combine, it was all done by a single thread in a single HT
455
+ const auto single_ht = !gstate.external && gstate.active_threads == 1;
304
456
 
305
- void RadixPartitionedHashTable::ScheduleTasks(Executor &executor, const shared_ptr<Event> &event,
306
- GlobalSinkState &state, vector<shared_ptr<Task>> &tasks) const {
307
- auto &gstate = state.Cast<RadixHTGlobalState>();
308
- if (!gstate.is_partitioned) {
309
- return;
310
- }
311
- for (idx_t r = 0; r < gstate.partition_info.n_partitions; r++) {
312
- D_ASSERT(gstate.partition_info.n_partitions <= gstate.finalized_hts.size());
313
- D_ASSERT(gstate.finalized_hts[r]);
314
- tasks.push_back(make_uniq<RadixAggregateFinalizeTask>(executor, event, gstate, r));
457
+ auto &uncombined_partition_data = uncombined_data.GetPartitions();
458
+ const auto n_partitions = uncombined_partition_data.size();
459
+ gstate.partitions.reserve(n_partitions);
460
+ for (idx_t i = 0; i < n_partitions; i++) {
461
+ gstate.partitions.emplace_back(make_uniq<AggregatePartition>(std::move(uncombined_partition_data[i])));
462
+ if (single_ht) {
463
+ gstate.finalize_idx++;
464
+ gstate.partitions.back()->finalized = true;
465
+ }
466
+ }
467
+ } else {
468
+ gstate.count_before_combining = 0;
315
469
  }
316
- }
317
470
 
318
- bool RadixPartitionedHashTable::ForceSingleHT(GlobalSinkState &state) const {
319
- auto &gstate = state.Cast<RadixHTGlobalState>();
320
- return gstate.partition_info.n_partitions < 2;
471
+ gstate.finalized = true;
321
472
  }
322
473
 
323
474
  //===--------------------------------------------------------------------===//
324
475
  // Source
325
476
  //===--------------------------------------------------------------------===//
477
+ idx_t RadixPartitionedHashTable::NumberOfPartitions(GlobalSinkState &sink_p) const {
478
+ auto &sink = sink_p.Cast<RadixHTGlobalSinkState>();
479
+ return sink.partitions.size();
480
+ }
481
+
482
+ void RadixPartitionedHashTable::SetMultiScan(GlobalSinkState &sink_p) {
483
+ auto &sink = sink_p.Cast<RadixHTGlobalSinkState>();
484
+ sink.scan_pin_properties = TupleDataPinProperties::UNPIN_AFTER_DONE;
485
+ }
486
+
487
+ enum class RadixHTSourceTaskType : uint8_t { NO_TASK, FINALIZE, SCAN };
488
+
489
+ class RadixHTLocalSourceState;
490
+
326
491
  class RadixHTGlobalSourceState : public GlobalSourceState {
327
492
  public:
328
- explicit RadixHTGlobalSourceState(Allocator &allocator, const RadixPartitionedHashTable &ht)
329
- : ht_index(0), initialized(false), finished(false) {
330
- }
493
+ RadixHTGlobalSourceState(ClientContext &context, const RadixPartitionedHashTable &radix_ht);
331
494
 
332
- //! Heavy handed for now.
333
- mutex lock;
334
- //! The current position to scan the HT for output tuples
335
- idx_t ht_index;
336
- //! The set of aggregate scan states
337
- unsafe_unique_array<TupleDataParallelScanState> ht_scan_states;
338
- atomic<bool> initialized;
495
+ //! Assigns a task to a local source state
496
+ bool AssignTask(RadixHTGlobalSinkState &sink, RadixHTLocalSourceState &lstate);
497
+
498
+ public:
499
+ //! The client context
500
+ ClientContext &context;
501
+ //! For synchronizing the source phase
339
502
  atomic<bool> finished;
503
+
504
+ //! Column ids for scanning
505
+ vector<column_t> column_ids;
506
+
507
+ //! For synchronizing scan tasks
508
+ atomic<idx_t> scan_idx;
509
+ atomic<idx_t> scan_done;
340
510
  };
341
511
 
512
+ enum class RadixHTScanStatus : uint8_t { INIT, IN_PROGRESS, DONE };
513
+
342
514
  class RadixHTLocalSourceState : public LocalSourceState {
343
515
  public:
344
- explicit RadixHTLocalSourceState(ExecutionContext &context, const RadixPartitionedHashTable &ht) {
345
- auto &allocator = Allocator::Get(context.client);
346
- auto scan_chunk_types = ht.group_types;
347
- for (auto &aggr_type : ht.op.aggregate_return_types) {
348
- scan_chunk_types.push_back(aggr_type);
349
- }
350
- scan_chunk.Initialize(allocator, scan_chunk_types);
351
- }
516
+ explicit RadixHTLocalSourceState(ExecutionContext &context, const RadixPartitionedHashTable &radix_ht);
517
+
518
+ public:
519
+ //! Do the work this thread has been assigned
520
+ void ExecuteTask(RadixHTGlobalSinkState &sink, RadixHTGlobalSourceState &gstate, DataChunk &chunk);
521
+ //! Whether this thread has finished the work it has been assigned
522
+ bool TaskFinished();
523
+
524
+ private:
525
+ //! Execute the finalize or scan task
526
+ void Finalize(RadixHTGlobalSinkState &sink, RadixHTGlobalSourceState &gstate);
527
+ void Scan(RadixHTGlobalSinkState &sink, RadixHTGlobalSourceState &gstate, DataChunk &chunk);
528
+
529
+ public:
530
+ //! Current task and index
531
+ RadixHTSourceTaskType task;
532
+ idx_t task_idx;
352
533
 
353
- //! Materialized GROUP BY expressions & aggregates
534
+ //! Thread-local HT that is re-used to Finalize
535
+ unique_ptr<GroupedAggregateHashTable> ht;
536
+ //! Current status of a Scan
537
+ RadixHTScanStatus scan_status;
538
+
539
+ private:
540
+ //! Allocator and layout for finalizing state
541
+ TupleDataLayout layout;
542
+ ArenaAllocator aggregate_allocator;
543
+
544
+ //! State and chunk for scanning
545
+ TupleDataScanState scan_state;
354
546
  DataChunk scan_chunk;
355
- //! HT index
356
- idx_t ht_index = DConstants::INVALID_INDEX;
357
- //! A reference to the current HT that we are scanning
358
- shared_ptr<GroupedAggregateHashTable> ht;
359
- //! Scan state for the current HT
360
- TupleDataLocalScanState scan_state;
361
547
  };
362
548
 
363
549
  unique_ptr<GlobalSourceState> RadixPartitionedHashTable::GetGlobalSourceState(ClientContext &context) const {
364
- return make_uniq<RadixHTGlobalSourceState>(Allocator::Get(context), *this);
550
+ return make_uniq<RadixHTGlobalSourceState>(context, *this);
365
551
  }
366
552
 
367
553
  unique_ptr<LocalSourceState> RadixPartitionedHashTable::GetLocalSourceState(ExecutionContext &context) const {
368
554
  return make_uniq<RadixHTLocalSourceState>(context, *this);
369
555
  }
370
556
 
371
- idx_t RadixPartitionedHashTable::Size(GlobalSinkState &sink_state) const {
372
- auto &gstate = sink_state.Cast<RadixHTGlobalState>();
373
- if (gstate.is_empty && grouping_set.empty()) {
374
- return 1;
557
+ RadixHTGlobalSourceState::RadixHTGlobalSourceState(ClientContext &context_p, const RadixPartitionedHashTable &radix_ht)
558
+ : context(context_p), finished(false), scan_idx(0), scan_done(0) {
559
+ for (column_t column_id = 0; column_id < radix_ht.group_types.size(); column_id++) {
560
+ column_ids.push_back(column_id);
561
+ }
562
+ }
563
+
564
+ bool RadixHTGlobalSourceState::AssignTask(RadixHTGlobalSinkState &sink, RadixHTLocalSourceState &lstate) {
565
+ D_ASSERT(lstate.scan_status != RadixHTScanStatus::IN_PROGRESS);
566
+
567
+ const auto n_partitions = sink.partitions.size();
568
+ if (finished) {
569
+ return false;
570
+ }
571
+ // We first try to assign a Scan task, then a Finalize task if that didn't work, without using any locks
572
+
573
+ // We need an atomic compare-and-swap to assign a Scan task, because we need to only increment
574
+ // the 'scan_idx' atomic if the 'finalize' of that partition is true, i.e., ready to be scanned
575
+ bool scan_assigned = true;
576
+ do {
577
+ lstate.task_idx = scan_idx.load();
578
+ if (lstate.task_idx >= n_partitions || !sink.partitions[lstate.task_idx]->finalized) {
579
+ scan_assigned = false;
580
+ break;
581
+ }
582
+ } while (!std::atomic_compare_exchange_weak(&scan_idx, &lstate.task_idx, lstate.task_idx + 1));
583
+
584
+ if (scan_assigned) {
585
+ // We successfully assigned a Scan task
586
+ D_ASSERT(lstate.task_idx < n_partitions && sink.partitions[lstate.task_idx]->finalized);
587
+ lstate.task = RadixHTSourceTaskType::SCAN;
588
+ lstate.scan_status = RadixHTScanStatus::INIT;
589
+ return true;
375
590
  }
376
591
 
377
- idx_t count = 0;
378
- for (const auto &ht : gstate.finalized_hts) {
379
- count += ht->Count();
592
+ // We didn't assign a Scan task
593
+ if (sink.finalize_idx >= n_partitions) {
594
+ return false; // No finalize tasks left
380
595
  }
381
- return count;
596
+
597
+ // We can just increment the atomic here, much simpler than assigning the scan task
598
+ lstate.task_idx = sink.finalize_idx++;
599
+ if (lstate.task_idx < n_partitions) {
600
+ // We successfully assigned a Finalize task
601
+ lstate.task = RadixHTSourceTaskType::FINALIZE;
602
+ return true;
603
+ }
604
+
605
+ // We didn't manage to assign a Finalize task
606
+ return false;
382
607
  }
383
608
 
384
- SourceResultType RadixPartitionedHashTable::GetData(ExecutionContext &context, DataChunk &chunk,
385
- GlobalSinkState &sink_state, OperatorSourceInput &input) const {
386
- auto &gstate = sink_state.Cast<RadixHTGlobalState>();
387
- auto &state = input.global_state.Cast<RadixHTGlobalSourceState>();
388
- auto &lstate = input.local_state.Cast<RadixHTLocalSourceState>();
389
- D_ASSERT(gstate.is_finalized);
390
- if (state.finished) {
391
- return SourceResultType::FINISHED;
609
+ RadixHTLocalSourceState::RadixHTLocalSourceState(ExecutionContext &context, const RadixPartitionedHashTable &radix_ht)
610
+ : task(RadixHTSourceTaskType::NO_TASK), scan_status(RadixHTScanStatus::DONE), layout(radix_ht.GetLayout().Copy()),
611
+ aggregate_allocator(BufferAllocator::Get(context.client)) {
612
+ auto &allocator = BufferAllocator::Get(context.client);
613
+ auto scan_chunk_types = radix_ht.group_types;
614
+ for (auto &aggr_type : radix_ht.op.aggregate_return_types) {
615
+ scan_chunk_types.push_back(aggr_type);
616
+ }
617
+ scan_chunk.Initialize(allocator, scan_chunk_types);
618
+ }
619
+
620
+ void RadixHTLocalSourceState::ExecuteTask(RadixHTGlobalSinkState &sink, RadixHTGlobalSourceState &gstate,
621
+ DataChunk &chunk) {
622
+ switch (task) {
623
+ case RadixHTSourceTaskType::FINALIZE:
624
+ Finalize(sink, gstate);
625
+ break;
626
+ case RadixHTSourceTaskType::SCAN:
627
+ Scan(sink, gstate, chunk);
628
+ break;
629
+ default:
630
+ throw InternalException("Unexpected RadixHTSourceTaskType in ExecuteTask!");
392
631
  }
632
+ }
393
633
 
394
- // special case hack to sort out aggregating from empty intermediates
395
- // for aggregations without groups
396
- if (gstate.is_empty && grouping_set.empty()) {
397
- D_ASSERT(chunk.ColumnCount() == null_groups.size() + op.aggregates.size() + op.grouping_functions.size());
398
- // for each column in the aggregates, set to initial state
399
- chunk.SetCardinality(1);
400
- for (auto null_group : null_groups) {
401
- chunk.data[null_group].SetVectorType(VectorType::CONSTANT_VECTOR);
402
- ConstantVector::SetNull(chunk.data[null_group], true);
403
- }
404
- for (idx_t i = 0; i < op.aggregates.size(); i++) {
405
- D_ASSERT(op.aggregates[i]->GetExpressionClass() == ExpressionClass::BOUND_AGGREGATE);
406
- auto &aggr = op.aggregates[i]->Cast<BoundAggregateExpression>();
407
- auto aggr_state = make_unsafe_uniq_array<data_t>(aggr.function.state_size());
408
- aggr.function.initialize(aggr_state.get());
409
-
410
- AggregateInputData aggr_input_data(aggr.bind_info.get(), Allocator::DefaultAllocator());
411
- Vector state_vector(Value::POINTER(CastPointerToValue(aggr_state.get())));
412
- aggr.function.finalize(state_vector, aggr_input_data, chunk.data[null_groups.size() + i], 1, 0);
413
- if (aggr.function.destructor) {
414
- aggr.function.destructor(state_vector, aggr_input_data, 1);
415
- }
416
- }
417
- // Place the grouping values (all the groups of the grouping_set condensed into a single value)
418
- // Behind the null groups + aggregates
419
- for (idx_t i = 0; i < op.grouping_functions.size(); i++) {
420
- chunk.data[null_groups.size() + op.aggregates.size() + i].Reference(grouping_values[i]);
421
- }
422
- state.finished = true;
423
- return chunk.size() == 0 ? SourceResultType::FINISHED : SourceResultType::HAVE_MORE_OUTPUT;
424
- }
425
- if (gstate.is_empty) {
426
- state.finished = true;
427
- return chunk.size() == 0 ? SourceResultType::FINISHED : SourceResultType::HAVE_MORE_OUTPUT;
428
- }
429
- idx_t elements_found = 0;
430
-
431
- lstate.scan_chunk.Reset();
432
- if (!state.initialized) {
433
- lock_guard<mutex> l(state.lock);
434
- if (!state.initialized) {
435
- auto &finalized_hts = gstate.finalized_hts;
436
- state.ht_scan_states = make_unsafe_uniq_array<TupleDataParallelScanState>(finalized_hts.size());
437
-
438
- const auto &layout = gstate.finalized_hts[0]->GetDataCollection().GetLayout();
439
- vector<column_t> column_ids;
440
- column_ids.reserve(layout.ColumnCount() - 1);
441
- for (idx_t col_idx = 0; col_idx < layout.ColumnCount() - 1; col_idx++) {
442
- column_ids.emplace_back(col_idx);
443
- }
634
+ void RadixHTLocalSourceState::Finalize(RadixHTGlobalSinkState &sink, RadixHTGlobalSourceState &gstate) {
635
+ D_ASSERT(task == RadixHTSourceTaskType::FINALIZE);
636
+ D_ASSERT(scan_status != RadixHTScanStatus::IN_PROGRESS);
444
637
 
445
- for (idx_t ht_idx = 0; ht_idx < finalized_hts.size(); ht_idx++) {
446
- gstate.finalized_hts[ht_idx]->GetDataCollection().InitializeScan(
447
- state.ht_scan_states.get()[ht_idx].scan_state, column_ids);
448
- }
449
- state.initialized = true;
450
- }
638
+ auto &partition = *sink.partitions[task_idx];
639
+ if (partition.data->Count() == 0) {
640
+ partition.finalized = true;
641
+ return;
451
642
  }
452
643
 
453
- auto &local_scan_state = lstate.scan_state;
454
- while (true) {
455
- D_ASSERT(state.ht_scan_states);
456
- idx_t ht_index;
457
- {
458
- lock_guard<mutex> l(state.lock);
459
- ht_index = state.ht_index;
460
- if (ht_index >= gstate.finalized_hts.size()) {
461
- state.finished = true;
462
- return chunk.size() == 0 ? SourceResultType::FINISHED : SourceResultType::HAVE_MORE_OUTPUT;
463
- }
464
- }
465
- D_ASSERT(ht_index < gstate.finalized_hts.size());
466
- if (lstate.ht_index != DConstants::INVALID_INDEX && ht_index != lstate.ht_index) {
467
- lstate.ht->GetDataCollection().FinalizePinState(local_scan_state.pin_state);
644
+ if (!ht) {
645
+ // Create a HT with sufficient capacity
646
+ const auto capacity = GroupedAggregateHashTable::GetCapacityForCount(partition.data->Count());
647
+ ht = sink.radix_ht.CreateHT(gstate.context, capacity, 0);
648
+ } else {
649
+ // We may want to resize here to the size of this partition, but for now we just assume uniform partition sizes
650
+ ht->InitializePartitionedData();
651
+ ht->ClearPointerTable();
652
+ ht->ResetCount();
653
+ }
654
+
655
+ // Now combine the uncombined data using this thread's HT
656
+ ht->Combine(*partition.data);
657
+ ht->UnpinData();
658
+
659
+ // Move the combined data back to the partition
660
+ partition.data =
661
+ make_uniq<TupleDataCollection>(BufferManager::GetBufferManager(gstate.context), sink.radix_ht.GetLayout());
662
+ partition.data->Combine(*ht->GetPartitionedData()->GetPartitions()[0]);
663
+
664
+ // Mark partition as ready to scan
665
+ partition.finalized = true;
666
+
667
+ // Make sure this thread's aggregate allocator does not get lost
668
+ lock_guard<mutex> guard(sink.lock);
669
+ sink.stored_allocators.emplace_back(ht->GetAggregateAllocator());
670
+ }
671
+
672
+ void RadixHTLocalSourceState::Scan(RadixHTGlobalSinkState &sink, RadixHTGlobalSourceState &gstate, DataChunk &chunk) {
673
+ D_ASSERT(task == RadixHTSourceTaskType::SCAN);
674
+ D_ASSERT(scan_status != RadixHTScanStatus::DONE);
675
+
676
+ auto &partition = *sink.partitions[task_idx];
677
+ D_ASSERT(partition.finalized);
678
+ auto &data_collection = *partition.data;
679
+
680
+ if (data_collection.Count() == 0) {
681
+ scan_status = RadixHTScanStatus::DONE;
682
+ if (++gstate.scan_done == sink.partitions.size()) {
683
+ gstate.finished = true;
468
684
  }
469
- lstate.ht_index = ht_index;
470
- lstate.ht = gstate.finalized_hts[ht_index];
471
- D_ASSERT(lstate.ht);
685
+ return;
686
+ }
472
687
 
473
- auto &global_scan_state = state.ht_scan_states[ht_index];
474
- elements_found = lstate.ht->Scan(global_scan_state, local_scan_state, lstate.scan_chunk);
475
- if (elements_found > 0) {
476
- break;
688
+ if (scan_status == RadixHTScanStatus::INIT) {
689
+ data_collection.InitializeScan(scan_state, gstate.column_ids, sink.scan_pin_properties);
690
+ scan_status = RadixHTScanStatus::IN_PROGRESS;
691
+ }
692
+
693
+ if (!data_collection.Scan(scan_state, scan_chunk)) {
694
+ scan_status = RadixHTScanStatus::DONE;
695
+ if (sink.scan_pin_properties == TupleDataPinProperties::DESTROY_AFTER_DONE) {
696
+ data_collection.Reset();
477
697
  }
478
- lstate.ht->GetDataCollection().FinalizePinState(local_scan_state.pin_state);
479
-
480
- // move to the next hash table
481
- lock_guard<mutex> l(state.lock);
482
- ht_index++;
483
- if (ht_index > state.ht_index) {
484
- // we have not yet worked on the table
485
- // move the global index forwards
486
- if (!gstate.multi_scan) {
487
- gstate.finalized_hts[state.ht_index].reset();
488
- }
489
- state.ht_index = ht_index;
698
+ return;
699
+ }
700
+
701
+ if (data_collection.ScanComplete(scan_state)) {
702
+ if (++gstate.scan_done == sink.partitions.size()) {
703
+ gstate.finished = true;
490
704
  }
491
705
  }
492
706
 
493
- // compute the final projection list
494
- chunk.SetCardinality(elements_found);
707
+ RowOperationsState row_state(aggregate_allocator);
708
+ const auto group_cols = layout.ColumnCount() - 1;
709
+ RowOperations::FinalizeStates(row_state, layout, scan_state.chunk_state.row_locations, scan_chunk, group_cols);
710
+
711
+ if (sink.scan_pin_properties == TupleDataPinProperties::DESTROY_AFTER_DONE && layout.HasDestructor()) {
712
+ RowOperations::DestroyStates(row_state, layout, scan_state.chunk_state.row_locations, scan_chunk.size());
713
+ }
495
714
 
715
+ auto &radix_ht = sink.radix_ht;
496
716
  idx_t chunk_index = 0;
497
- for (auto &entry : grouping_set) {
498
- chunk.data[entry].Reference(lstate.scan_chunk.data[chunk_index++]);
717
+ for (auto &entry : radix_ht.grouping_set) {
718
+ chunk.data[entry].Reference(scan_chunk.data[chunk_index++]);
499
719
  }
500
- for (auto null_group : null_groups) {
720
+ for (auto null_group : radix_ht.null_groups) {
501
721
  chunk.data[null_group].SetVectorType(VectorType::CONSTANT_VECTOR);
502
722
  ConstantVector::SetNull(chunk.data[null_group], true);
503
723
  }
504
- D_ASSERT(grouping_set.size() + null_groups.size() == op.GroupCount());
505
- for (idx_t col_idx = 0; col_idx < op.aggregates.size(); col_idx++) {
506
- chunk.data[op.GroupCount() + col_idx].Reference(lstate.scan_chunk.data[group_types.size() + col_idx]);
724
+ D_ASSERT(radix_ht.grouping_set.size() + radix_ht.null_groups.size() == radix_ht.op.GroupCount());
725
+ for (idx_t col_idx = 0; col_idx < radix_ht.op.aggregates.size(); col_idx++) {
726
+ chunk.data[radix_ht.op.GroupCount() + col_idx].Reference(
727
+ scan_chunk.data[radix_ht.group_types.size() + col_idx]);
728
+ }
729
+ D_ASSERT(radix_ht.op.grouping_functions.size() == radix_ht.grouping_values.size());
730
+ for (idx_t i = 0; i < radix_ht.op.grouping_functions.size(); i++) {
731
+ chunk.data[radix_ht.op.GroupCount() + radix_ht.op.aggregates.size() + i].Reference(radix_ht.grouping_values[i]);
732
+ }
733
+ chunk.SetCardinality(scan_chunk);
734
+ D_ASSERT(chunk.size() != 0);
735
+ }
736
+
737
+ bool RadixHTLocalSourceState::TaskFinished() {
738
+ switch (task) {
739
+ case RadixHTSourceTaskType::FINALIZE:
740
+ return true;
741
+ case RadixHTSourceTaskType::SCAN:
742
+ return scan_status == RadixHTScanStatus::DONE;
743
+ default:
744
+ D_ASSERT(task == RadixHTSourceTaskType::NO_TASK);
745
+ return true;
746
+ }
747
+ }
748
+
749
+ SourceResultType RadixPartitionedHashTable::GetData(ExecutionContext &context, DataChunk &chunk,
750
+ GlobalSinkState &sink_p, OperatorSourceInput &input) const {
751
+ auto &sink = sink_p.Cast<RadixHTGlobalSinkState>();
752
+ D_ASSERT(sink.finalized);
753
+
754
+ auto &gstate = input.global_state.Cast<RadixHTGlobalSourceState>();
755
+ auto &lstate = input.local_state.Cast<RadixHTLocalSourceState>();
756
+ D_ASSERT(sink.scan_pin_properties == TupleDataPinProperties::UNPIN_AFTER_DONE ||
757
+ sink.scan_pin_properties == TupleDataPinProperties::DESTROY_AFTER_DONE);
758
+
759
+ if (gstate.finished) {
760
+ return SourceResultType::FINISHED;
761
+ }
762
+
763
+ if (sink.count_before_combining == 0) {
764
+ if (grouping_set.empty()) {
765
+ // Special case hack to sort out aggregating from empty intermediates for aggregations without groups
766
+ D_ASSERT(chunk.ColumnCount() == null_groups.size() + op.aggregates.size() + op.grouping_functions.size());
767
+ // For each column in the aggregates, set to initial state
768
+ chunk.SetCardinality(1);
769
+ for (auto null_group : null_groups) {
770
+ chunk.data[null_group].SetVectorType(VectorType::CONSTANT_VECTOR);
771
+ ConstantVector::SetNull(chunk.data[null_group], true);
772
+ }
773
+ ArenaAllocator allocator(BufferAllocator::Get(context.client));
774
+ for (idx_t i = 0; i < op.aggregates.size(); i++) {
775
+ D_ASSERT(op.aggregates[i]->GetExpressionClass() == ExpressionClass::BOUND_AGGREGATE);
776
+ auto &aggr = op.aggregates[i]->Cast<BoundAggregateExpression>();
777
+ auto aggr_state = make_unsafe_uniq_array<data_t>(aggr.function.state_size());
778
+ aggr.function.initialize(aggr_state.get());
779
+
780
+ AggregateInputData aggr_input_data(aggr.bind_info.get(), allocator);
781
+ Vector state_vector(Value::POINTER(CastPointerToValue(aggr_state.get())));
782
+ aggr.function.finalize(state_vector, aggr_input_data, chunk.data[null_groups.size() + i], 1, 0);
783
+ if (aggr.function.destructor) {
784
+ aggr.function.destructor(state_vector, aggr_input_data, 1);
785
+ }
786
+ }
787
+ // Place the grouping values (all the groups of the grouping_set condensed into a single value)
788
+ // Behind the null groups + aggregates
789
+ for (idx_t i = 0; i < op.grouping_functions.size(); i++) {
790
+ chunk.data[null_groups.size() + op.aggregates.size() + i].Reference(grouping_values[i]);
791
+ }
792
+ }
793
+ gstate.finished = true;
794
+ return SourceResultType::FINISHED;
507
795
  }
508
- D_ASSERT(op.grouping_functions.size() == grouping_values.size());
509
- for (idx_t i = 0; i < op.grouping_functions.size(); i++) {
510
- chunk.data[op.GroupCount() + op.aggregates.size() + i].Reference(grouping_values[i]);
796
+
797
+ while (!gstate.finished && chunk.size() == 0) {
798
+ if (!lstate.TaskFinished() || gstate.AssignTask(sink, lstate)) {
799
+ lstate.ExecuteTask(sink, gstate, chunk);
800
+ }
801
+ }
802
+
803
+ if (chunk.size() != 0) {
804
+ return SourceResultType::HAVE_MORE_OUTPUT;
805
+ } else {
806
+ return SourceResultType::FINISHED;
511
807
  }
512
- return chunk.size() == 0 ? SourceResultType::FINISHED : SourceResultType::HAVE_MORE_OUTPUT;
513
808
  }
514
809
 
515
810
  } // namespace duckdb