Reused input types for reduced memory usage

package.json

@@ -2,7 +2,7 @@
 	"name": "drizzle-graphql",
 	"type": "module",
 	"author": "Drizzle Team",
-	"version": "0.7.2",
+	"version": "0.7.3",
 	"description": "Automatically generate GraphQL schema or customizable schema config fields from Drizzle ORM schema",
 	"scripts": {
 		"build": "pnpm tsx scripts/build.ts",

src/util/builders/common.ts

@@ -277,37 +277,35 @@ const generateTableSelectTypeFieldsCached = (table: Table, tableName: string): R
 	return remapped;
 };
 
-const generateSelectFields = <TWithOrder extends boolean>(
-	tables: Record<string, Table>,
-	tableName: string,
-	relationMap: Record<string, Record<string, TableNamedRelations>>,
-	typeName: string,
-	withOrder: TWithOrder,
-	usedTables: Set<string> = new Set(),
-): SelectData<TWithOrder> => {
-	const relations = relationMap[tableName];
-	const relationEntries: [string, TableNamedRelations][] = relations ? Object.entries(relations) : [];
-
-	const table = tables[tableName]!;
+const orderTypeMap = new WeakMap<Object, GraphQLInputObjectType>();
+const generateTableOrderTypeCached = (table: Table, tableName: string) => {
+	if (orderTypeMap.has(table)) return orderTypeMap.get(table)!;
 
 	const orderColumns = generateTableOrderCached(table, tableName);
-	const order = withOrder
-		? new GraphQLInputObjectType({
-			name: `${typeName}OrderBy`,
-			fields: orderColumns,
-		})
-		: undefined;
+	const order = new GraphQLInputObjectType({
+		name: `${tableName}OrderBy`,
+		fields: orderColumns,
+	});
+
+	orderTypeMap.set(table, order);
+
+	return order;
+};
+
+const filterTypeMap = new WeakMap<Object, GraphQLInputObjectType>();
+const generateTableFilterTypeCached = (table: Table, tableName: string) => {
+	if (filterTypeMap.has(table)) return filterTypeMap.get(table)!;
 
 	const filterColumns = generateTableFilterValuesCached(table, tableName);
 	const filters: GraphQLInputObjectType = new GraphQLInputObjectType({
-		name: `${typeName}Filters`,
+		name: `${tableName}Filters`,
 		fields: {
 			...filterColumns,
 			OR: {
 				type: new GraphQLList(
 					new GraphQLNonNull(
 						new GraphQLInputObjectType({
-							name: `${typeName}FiltersOr`,
+							name: `${tableName}FiltersOr`,
 							fields: filterColumns,
 						}),
 					),
@@ -316,6 +314,30 @@ const generateSelectFields = <TWithOrder extends boolean>(
 		},
 	});
 
+	filterTypeMap.set(table, filters);
+
+	return filters;
+};
+
+const generateSelectFields = <TWithOrder extends boolean>(
+	tables: Record<string, Table>,
+	tableName: string,
+	relationMap: Record<string, Record<string, TableNamedRelations>>,
+	typeName: string,
+	withOrder: TWithOrder,
+	usedTables: Set<string> = new Set(),
+): SelectData<TWithOrder> => {
+	const relations = relationMap[tableName];
+	const relationEntries: [string, TableNamedRelations][] = relations ? Object.entries(relations) : [];
+
+	const table = tables[tableName]!;
+
+	const order = withOrder
+		? generateTableOrderTypeCached(table, tableName)
+		: undefined;
+
+	const filters = generateTableFilterTypeCached(table, tableName);
+
 	const tableFields = generateTableSelectTypeFieldsCached(table, tableName);
 
 	if (usedTables.has(tableName) || !relationEntries.length) {