duckdb 0.6.2-dev2091.0 → 0.6.2-dev2096.0

package/package.json

@@ -2,7 +2,7 @@
   "name": "duckdb",
   "main": "./lib/duckdb.js",
   "types": "./lib/duckdb.d.ts",
-  "version": "0.6.2-dev2091.0",
+  "version": "0.6.2-dev2096.0",
   "description": "DuckDB node.js API",
   "gypfile": true,
   "dependencies": {

package/src/duckdb/src/catalog/catalog_entry/duck_table_entry.cpp

@@ -300,37 +300,13 @@ unique_ptr<CatalogEntry> DuckTableEntry::AddColumn(ClientContext &context, AddCo
 	return make_unique<DuckTableEntry>(catalog, schema, (BoundCreateTableInfo *)bound_create_info.get(), new_storage);
 }
 
-unique_ptr<CatalogEntry> DuckTableEntry::RemoveColumn(ClientContext &context, RemoveColumnInfo &info) {
-	auto removed_index = GetColumnIndex(info.removed_column, info.if_column_exists);
-	if (!removed_index.IsValid()) {
-		if (!info.if_column_exists) {
-			throw CatalogException("Cannot drop column: rowid column cannot be dropped");
-		}
-		return nullptr;
-	}
-
-	auto create_info = make_unique<CreateTableInfo>(schema, name);
-	create_info->temporary = temporary;
-
-	logical_index_set_t removed_columns;
-	if (column_dependency_manager.HasDependents(removed_index)) {
-		removed_columns = column_dependency_manager.GetDependents(removed_index);
-	}
-	if (!removed_columns.empty() && !info.cascade) {
-		throw CatalogException("Cannot drop column: column is a dependency of 1 or more generated column(s)");
-	}
-	for (auto &col : columns.Logical()) {
-		if (col.Logical() == removed_index || removed_columns.count(col.Logical())) {
-			continue;
-		}
-		create_info->columns.AddColumn(col.Copy());
-	}
-	if (create_info->columns.empty()) {
-		throw CatalogException("Cannot drop column: table only has one column remaining!");
-	}
-	auto adjusted_indices = column_dependency_manager.RemoveColumn(removed_index, columns.LogicalColumnCount());
+void DuckTableEntry::UpdateConstraintsOnColumnDrop(const LogicalIndex &removed_index,
+                                                   const vector<LogicalIndex> &adjusted_indices,
+                                                   const RemoveColumnInfo &info, CreateTableInfo &create_info,
+                                                   bool is_generated) {
 	// handle constraints for the new table
 	D_ASSERT(constraints.size() == bound_constraints.size());
+
 	for (idx_t constr_idx = 0; constr_idx < constraints.size(); constr_idx++) {
 		auto &constraint = constraints[constr_idx];
 		auto &bound_constraint = bound_constraints[constr_idx];
@@ -342,14 +318,20 @@ unique_ptr<CatalogEntry> DuckTableEntry::RemoveColumn(ClientContext &context, Re
 				// the constraint is not about this column: we need to copy it
 				// we might need to shift the index back by one though, to account for the removed column
 				auto new_index = adjusted_indices[not_null_index.index];
-				create_info->constraints.push_back(make_unique<NotNullConstraint>(new_index));
+				create_info.constraints.push_back(make_unique<NotNullConstraint>(new_index));
 			}
 			break;
 		}
 		case ConstraintType::CHECK: {
+			// Generated columns can not be part of an index
 			// CHECK constraint
 			auto &bound_check = (BoundCheckConstraint &)*bound_constraint;
 			// check if the removed column is part of the check constraint
+			if (is_generated) {
+				// generated columns can not be referenced by constraints, we can just add the constraint back
+				create_info.constraints.push_back(constraint->Copy());
+				break;
+			}
 			auto physical_index = columns.LogicalToPhysical(removed_index);
 			if (bound_check.bound_columns.find(physical_index) != bound_check.bound_columns.end()) {
 				if (bound_check.bound_columns.size() > 1) {
@@ -362,7 +344,7 @@ unique_ptr<CatalogEntry> DuckTableEntry::RemoveColumn(ClientContext &context, Re
 				}
 			} else {
 				// check constraint does not concern the removed column: simply re-add it
-				create_info->constraints.push_back(constraint->Copy());
+				create_info.constraints.push_back(constraint->Copy());
 			}
 			break;
 		}
@@ -377,7 +359,7 @@ unique_ptr<CatalogEntry> DuckTableEntry::RemoveColumn(ClientContext &context, Re
 				}
 				unique.index = adjusted_indices[unique.index.index];
 			}
-			create_info->constraints.push_back(std::move(copy));
+			create_info.constraints.push_back(std::move(copy));
 			break;
 		}
 		case ConstraintType::FOREIGN_KEY: {
@@ -398,13 +380,50 @@ unique_ptr<CatalogEntry> DuckTableEntry::RemoveColumn(ClientContext &context, Re
 					    info.removed_column);
 				}
 			}
-			create_info->constraints.push_back(std::move(copy));
+			create_info.constraints.push_back(std::move(copy));
 			break;
 		}
 		default:
 			throw InternalException("Unsupported constraint for entry!");
 		}
 	}
+}
+
+unique_ptr<CatalogEntry> DuckTableEntry::RemoveColumn(ClientContext &context, RemoveColumnInfo &info) {
+	auto removed_index = GetColumnIndex(info.removed_column, info.if_column_exists);
+	if (!removed_index.IsValid()) {
+		if (!info.if_column_exists) {
+			throw CatalogException("Cannot drop column: rowid column cannot be dropped");
+		}
+		return nullptr;
+	}
+
+	auto create_info = make_unique<CreateTableInfo>(schema, name);
+	create_info->temporary = temporary;
+
+	logical_index_set_t removed_columns;
+	if (column_dependency_manager.HasDependents(removed_index)) {
+		removed_columns = column_dependency_manager.GetDependents(removed_index);
+	}
+	if (!removed_columns.empty() && !info.cascade) {
+		throw CatalogException("Cannot drop column: column is a dependency of 1 or more generated column(s)");
+	}
+	bool dropped_column_is_generated = false;
+	for (auto &col : columns.Logical()) {
+		if (col.Logical() == removed_index || removed_columns.count(col.Logical())) {
+			if (col.Generated()) {
+				dropped_column_is_generated = true;
+			}
+			continue;
+		}
+		create_info->columns.AddColumn(col.Copy());
+	}
+	if (create_info->columns.empty()) {
+		throw CatalogException("Cannot drop column: table only has one column remaining!");
+	}
+	auto adjusted_indices = column_dependency_manager.RemoveColumn(removed_index, columns.LogicalColumnCount());
+
+	UpdateConstraintsOnColumnDrop(removed_index, adjusted_indices, info, *create_info, dropped_column_is_generated);
 
 	auto binder = Binder::CreateBinder(context);
 	auto bound_create_info = binder->BindCreateTableInfo(std::move(create_info));

package/src/duckdb/src/function/table/version/pragma_version.cpp

@@ -1,8 +1,8 @@
 #ifndef DUCKDB_VERSION
-#define DUCKDB_VERSION "0.6.2-dev2091"
+#define DUCKDB_VERSION "0.6.2-dev2096"
 #endif
 #ifndef DUCKDB_SOURCE_ID
-#define DUCKDB_SOURCE_ID "4d9705ca54"
+#define DUCKDB_SOURCE_ID "f596d7b0fe"
 #endif
 #include "duckdb/function/table/system_functions.hpp"
 #include "duckdb/main/database.hpp"

package/src/duckdb/src/include/duckdb/catalog/catalog_entry/duck_table_entry.hpp

@@ -57,6 +57,9 @@ private:
 	unique_ptr<CatalogEntry> AddForeignKeyConstraint(ClientContext &context, AlterForeignKeyInfo &info);
 	unique_ptr<CatalogEntry> DropForeignKeyConstraint(ClientContext &context, AlterForeignKeyInfo &info);
 
+	void UpdateConstraintsOnColumnDrop(const LogicalIndex &removed_index, const vector<LogicalIndex> &adjusted_indices,
+	                                   const RemoveColumnInfo &info, CreateTableInfo &create_info, bool is_generated);
+
 private:
 	//! A reference to the underlying storage unit used for this table
 	std::shared_ptr<DataTable> storage;

package/src/duckdb/src/main/extension/extension_alias.cpp

@@ -4,6 +4,7 @@ namespace duckdb {
 
 static ExtensionAlias internal_aliases[] = {{"http", "httpfs"}, // httpfs
                                             {"https", "httpfs"},
+                                            {"md", "motherduck"}, // motherduck
                                             {"s3", "httpfs"},
                                             {"postgres", "postgres_scanner"}, // postgres
                                             {"sqlite", "sqlite_scanner"},     // sqlite

package/src/duckdb/src/main/extension/extension_helper.cpp

@@ -97,6 +97,7 @@ static DefaultExtension internal_extensions[] = {
     {"httpfs", "Adds support for reading and writing files over a HTTP(S) connection", HTTPFS_STATICALLY_LOADED},
     {"json", "Adds support for JSON operations", JSON_STATICALLY_LOADED},
     {"jemalloc", "Overwrites system allocator with JEMalloc", JEMALLOC_STATICALLY_LOADED},
+    {"motherduck", "Enables motherduck integration with the system", false},
     {"sqlite_scanner", "Adds support for reading SQLite database files", false},
     {"postgres_scanner", "Adds support for reading from a Postgres database", false},
     {"inet", "Adds support for IP-related data types and functions", false},