Support DuckDB v1.4

Author: dpxcc

.github/workflows/MainDistributionPipeline.yml

@@ -15,11 +15,11 @@ jobs:
   duckdb-next-build:
     name: Build extension binaries
     uses: duckdb/extension-ci-tools/.github/workflows/_extension_distribution.yml@main
-    if: false   # extension-template is currently not compatible with main
     with:
       duckdb_version: main
       ci_tools_version: main
       enable_rust: true
+      exclude_archs: windows_amd64;windows_amd64_mingw;wasm_mvp;wasm_eh;wasm_threads
       extension_name: mooncake
 
   duckdb-stable-build:

duckdb

@@ -6,7 +6,7 @@ namespace duckdb {
 
 class MooncakeExtension : public Extension {
 public:
-	void Load(DuckDB &db) override;
+	void Load(ExtensionLoader &loader) override;
 
 	string Name() override;
 

extension-ci-tools

@@ -8,8 +8,9 @@
 
 namespace duckdb {
 
-void MooncakeExtension::Load(DuckDB &db) {
-	auto &config = DBConfig::GetConfig(*db.instance);
+void MooncakeExtension::Load(ExtensionLoader &loader) {
+	auto &db = loader.GetDatabaseInstance();
+	auto &config = DBConfig::GetConfig(db);
 	config.storage_extensions["mooncake"] = make_uniq<MooncakeStorageExtension>();
 
 	string init_query = Pgmooncake::GetInitQuery();
@@ -33,12 +34,7 @@ string MooncakeExtension::Version() const {
 } // namespace duckdb
 
 extern "C" {
-DUCKDB_EXTENSION_API void mooncake_init(duckdb::DatabaseInstance &db) {
-	duckdb::DuckDB db_wrapper(db);
-	db_wrapper.LoadExtension<duckdb::MooncakeExtension>();
-}
-
-DUCKDB_EXTENSION_API const char *mooncake_version() {
-	return duckdb::DuckDB::LibraryVersion();
+DUCKDB_CPP_EXTENSION_ENTRY(mooncake, loader) {
+	duckdb::MooncakeExtension().Load(loader);
 }
 }

src/include/mooncake_extension.hpp

@@ -1,6 +1,6 @@
 #include "duckdb/catalog/catalog_entry/table_function_catalog_entry.hpp"
 #include "duckdb/common/multi_file/multi_file_reader.hpp"
-#include "duckdb/main/extension_util.hpp"
+#include "duckdb/main/extension_helper.hpp"
 #include "duckdb/parser/tableref/table_function_ref.hpp"
 #include "parquet_reader.hpp"
 #include "storage/mooncake_table.hpp"
@@ -106,7 +106,7 @@ struct MooncakeMultiFileReader : public MultiFileReader {
 		return make_uniq<MooncakeMultiFileReader>(table_function.function_info->Cast<MooncakeFunctionInfo>().table);
 	}
 
-	shared_ptr<MultiFileList> CreateFileList(ClientContext &, const vector<string> &, FileGlobOptions) override {
+	shared_ptr<MultiFileList> CreateFileList(ClientContext &, const vector<string> &, const FileGlobInput &) override {
 		return make_shared_ptr<MooncakeMultiFileList>(table);
 	}
 
@@ -135,7 +135,9 @@ struct MooncakeMultiFileReader : public MultiFileReader {
 };
 
 static TableFunction &GetParquetScan(ClientContext &context) {
-	return ExtensionUtil::GetTableFunction(*context.db, "parquet_scan").functions.GetFunctionReferenceByOffset(0);
+	ExtensionHelper::AutoLoadExtension(*context.db, "parquet");
+	ExtensionLoader loader(*context.db, "mooncake");
+	return loader.GetTableFunction("parquet_scan").functions.GetFunctionReferenceByOffset(0);
 }
 
 static unique_ptr<GlobalTableFunctionState> MooncakeScanInitGlobal(ClientContext &context,

src/mooncake_extension.cpp

@@ -86,15 +86,15 @@ optional_ptr<CatalogEntry> MooncakeSchema::LookupEntry(CatalogTransaction transa
 		throw InternalException("Error decoding schema: %s", ArrowErrorMessage(&error));
 	}
 
-	ArrowTableType arrow_table;
-	vector<string> names;
-	vector<LogicalType> return_types;
-	ArrowTableFunction::PopulateArrowTableType(transaction.db->config, arrow_table, schema, names, return_types);
+	ArrowTableSchema arrow_table;
+	ArrowTableFunction::PopulateArrowTableSchema(transaction.db->config, arrow_table, schema.arrow_schema);
+	auto &names = arrow_table.GetNames();
+	auto &types = arrow_table.GetTypes();
 
 	CreateTableInfo table_info;
 	table_info.table = table_name;
 	for (idx_t i = 0; i < names.size(); i++) {
-		table_info.columns.AddColumn(ColumnDefinition(names[i], return_types[i]));
+		table_info.columns.AddColumn(ColumnDefinition(names[i], types[i]));
 	}
 	tables[table_name] = make_uniq<MooncakeTable>(catalog, *this, table_info, lsn, moonlink);
 	return *tables[table_name];

src/storage/mooncake_scan.cpp

@@ -4,8 +4,8 @@
 
 namespace duckdb {
 
-unique_ptr<Catalog> MooncakeAttach(StorageExtensionInfo *storage_info, ClientContext &context, AttachedDatabase &db,
-                                   const string &name, AttachInfo &info, AccessMode access_mode) {
+unique_ptr<Catalog> MooncakeAttach(optional_ptr<StorageExtensionInfo> storage_info, ClientContext &context,
+                                   AttachedDatabase &db, const string &name, AttachInfo &info, AttachOptions &options) {
 	string uri;
 	string database;
 	for (auto &entry : info.options) {
@@ -29,7 +29,7 @@ unique_ptr<Catalog> MooncakeAttach(StorageExtensionInfo *storage_info, ClientCon
 	return make_uniq<MooncakeCatalog>(db, std::move(uri), std::move(database));
 }
 
-unique_ptr<TransactionManager> MooncakeCreateTransactionManager(StorageExtensionInfo *storage_info,
+unique_ptr<TransactionManager> MooncakeCreateTransactionManager(optional_ptr<StorageExtensionInfo> storage_info,
                                                                 AttachedDatabase &db, Catalog &catalog) {
 	return make_uniq<MooncakeTransactionManager>(db, catalog);
 }