Also record direct dependency table -> server

Author: Y--

src/pgduckdb_background_worker.cpp

@@ -231,6 +231,15 @@ BgwMainLoop() {
 	elog(LOG, "pg_duckdb background worker for database '%s' (%u) has now terminated.", db_name, MyDatabaseId);
 }
 
+void recordDependencyOnMDServer(ObjectAddress *object_address) {
+	ObjectAddress server_address = {
+	    .classId = ForeignServerRelationId,
+	    .objectId = GetMotherduckForeignServerOid(),
+	    .objectSubId = 0,
+	};
+	recordDependencyOn(object_address, &server_address, DEPENDENCY_NORMAL);
+}
+
 } // namespace pgduckdb
 
 extern "C" {
@@ -774,6 +783,25 @@ CreateTable(const char *postgres_schema_name, const char *table_name, const char
 		return false;
 	}
 
+	// Record the dependency on the MotherDuck server
+	{
+		HeapTuple new_table_tuple = SearchSysCache2(RELNAMENSP, CStringGetDatum(table_name), ObjectIdGetDatum(schema_oid));
+		if (HeapTupleIsValid(new_table_tuple)) {
+			Form_pg_class postgres_relation = (Form_pg_class)GETSTRUCT(new_table_tuple);
+			auto oid = postgres_relation->oid;
+			ReleaseSysCache(new_table_tuple);
+			ObjectAddress object_address = {
+			    .classId = RelationRelationId,
+			    .objectId = oid,
+			    .objectSubId = 0,
+			};
+			recordDependencyOnMDServer(&object_address);
+		} else {
+			// Something went really wrong (since `create_table_succeeded`)
+			elog(WARNING, "Failed to find table '%s.%s' so won't record dependency to 'motherduck' SERVER", table_name, postgres_schema_name);
+		}
+	}
+
 	if (did_delete_table) {
 		/*
 		 * Commit the subtransaction that contains both the drop and the create that
@@ -933,12 +961,7 @@ CreateSchemaIfNotExists(const char *postgres_schema_name, bool is_default_db) {
 		    .objectId = schema_oid,
 		    .objectSubId = 0,
 		};
-		ObjectAddress server_address = {
-		    .classId = ForeignServerRelationId,
-		    .objectId = GetMotherduckForeignServerOid(), // pgduckdb::ExtensionOid(),
-		    .objectSubId = 0,
-		};
-		recordDependencyOn(&schema_address, &server_address, DEPENDENCY_NORMAL);
+		recordDependencyOnMDServer(&schema_address);
 	}
 
 	/* Success, so we commit the subtransaction */