feat: partition scans in DuckDB Vortex extension (#3204)

Co-authored-by: Alexander Droste <[email protected]>
Co-authored-by: Nicholas Gates <[email protected]>

Author: 3 people

duckdb-vortex/CMakeLists.txt

@@ -6,6 +6,9 @@ project(${TARGET_NAME}_project)
 set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
 set(CMAKE_CXX_STANDARD 17)
 
+# Allow C++20 designator syntax in C++17
+add_compile_options(-Wno-c++20-designator)
+
 include(FetchContent)
 FetchContent_Declare(
     Corrosion

duckdb-vortex/src/include/vortex.hpp

@@ -1,9 +1,8 @@
 #pragma once
 
-
 // Include Vortex FFI, with the DuckDB FFI feature
 #ifndef ENABLE_DUCKDB_FFI
 #define ENABLE_DUCKDB_FFI
 #endif
 
-#include "vortex.h"
+#include "vortex.h"

duckdb-vortex/src/include/vortex_error.hpp

@@ -9,3 +9,11 @@ inline void HandleError(vx_error *error) {
 		throw duckdb::InvalidInputException(msg);
 	}
 }
+
+template <typename Func>
+auto Try(Func func) {
+	vx_error *error = nullptr;
+	auto result = func(&error);
+	HandleError(error);
+	return result;
+}

duckdb-vortex/src/include/vortex_layout_reader.hpp

@@ -0,0 +1,25 @@
+#pragma once
+
+#include "vortex_common.hpp"
+#include "expr/expr.hpp"
+
+class VortexLayoutReader {
+public:
+	explicit VortexLayoutReader(vx_layout_reader *reader) : reader(reader) {
+	}
+
+	~VortexLayoutReader() {
+		vx_layout_reader_free(reader);
+	}
+
+	static std::shared_ptr<VortexLayoutReader> CreateFromFile(VortexFileReader *file) {
+		auto reader = Try([&](auto err) { return vx_layout_reader_create(file->file, err); });
+		return std::make_shared<VortexLayoutReader>(reader);
+	}
+
+	vx_array_stream *Scan(const vx_file_scan_options *options) {
+		return Try([&](auto err) { return vx_layout_reader_scan(this->reader, options, err); });
+	}
+
+	vx_layout_reader *reader;
+};