@nkzw/fate 1.0.0 → 1.0.2

package/lib/server/drizzle.mjs

@@ -1,9 +1,9 @@
 import { t as isRecord } from "../record-AeZJC9fd.mjs";
-import { O as withConnection, S as isDataView, c as resolveSourceConnection, d as createSourceDefinitions, f as createSourcePlan, g as attachComputedState, i as createSourceRegistry, l as collectDataViewConfigs, o as resolveSourceById, p as getDataViewSourceConfig, s as resolveSourceByIds, t as bindSourceProcedures } from "../sourceRouter-Bwpkr062.mjs";
-import { and, asc, createTableRelationsHelpers, desc, eq, extractTablesRelationalConfig, getTableColumns, gt, inArray, isTable, lt, normalizeRelation, or, sql } from "drizzle-orm";
+import { O as withConnection, S as isDataView, c as resolveSourceConnection, d as createSourceDefinitions, f as createSourcePlan, g as attachComputedState, i as createSourceRegistry, l as collectDataViewConfigs, o as resolveSourceById, p as getDataViewSourceConfig, s as resolveSourceByIds, t as bindSourceProcedures } from "../sourceRouter-C22uFcZg.mjs";
+import { Many, and, asc, createTableRelationsHelpers, desc, eq, extractTablesRelationalConfig, getTableColumns, gt, inArray, is, isTable, lt, normalizeRelation, or, sql } from "drizzle-orm";
 //#region src/server/drizzle.ts
 /**
-* Fate's Drizzle integration.
+* fate's Drizzle integration.
 *
 * @example
 * import { createDrizzleFate } from '@nkzw/fate/server/drizzle';
@@ -29,6 +29,21 @@ const hasPagination = (args) => {
 };
 const getConnectionDirection = (args) => args?.before !== void 0 || typeof args?.last === "number" ? "backward" : "forward";
 const getConnectionSize = (fallback, args) => (typeof args?.first === "number" ? args.first : void 0) ?? (typeof args?.last === "number" ? args.last : void 0) ?? fallback;
+const defaultNestedPaginationConcurrency = 10;
+const getNestedPaginationConcurrency = (value) => typeof value === "number" && Number.isFinite(value) && value > 0 ? Math.floor(value) : defaultNestedPaginationConcurrency;
+const mapWithConcurrency = async (items, concurrency, fn) => {
+	if (items.length === 0) return [];
+	const results = new Array(items.length);
+	let nextIndex = 0;
+	const workers = Array.from({ length: Math.min(concurrency, items.length) }, async () => {
+		while (nextIndex < items.length) {
+			const index = nextIndex++;
+			results[index] = await fn(items[index], index);
+		}
+	});
+	await Promise.all(workers);
+	return results;
+};
 const buildConnection = ({ cursor, direction, items, pageSize }) => {
 	const hasMore = items.length > pageSize;
 	const connectionItems = (direction === "forward" ? items.slice(0, pageSize) : items.slice(-pageSize)).map((node) => ({
@@ -186,6 +201,51 @@ const inferDirectRelation = ({ field, metadata, sourceConfig, targetConfig }) =>
 		localKey: getColumnKey(sourceTableConfig.columns, normalized.fields[0])
 	};
 };
+const getComputedCountRelations = (view) => {
+	const relations = /* @__PURE__ */ new Set();
+	for (const field of Object.values(view.fields)) {
+		if (!isRecord(field) || field.kind !== "computed" || !isRecord(field.select)) continue;
+		for (const selection of Object.values(field.select)) if (isRecord(selection) && selection.kind === "count" && typeof selection.relation === "string") relations.add(selection.relation);
+	}
+	return relations;
+};
+const getTableConfigIfAvailable = ({ config, metadata }) => {
+	try {
+		return getTableConfig({
+			config,
+			metadata
+		});
+	} catch {
+		return null;
+	}
+};
+const inferHiddenCountRelation = ({ field, metadata, sourceConfig, viewConfigs }) => {
+	const sourceTableConfig = getTableConfigIfAvailable({
+		config: sourceConfig,
+		metadata
+	});
+	const relation = sourceTableConfig?.relations[field];
+	if (!sourceTableConfig || !relation || !is(relation, Many)) return null;
+	const normalized = normalizeRelation(metadata.schema, metadata.tableNamesMap, relation);
+	const localColumn = normalized.fields[0];
+	const foreignColumn = normalized.references[0];
+	if (!localColumn || !foreignColumn) return null;
+	for (const [index, targetConfig] of viewConfigs.entries()) {
+		const targetTableConfig = getTableConfigIfAvailable({
+			config: targetConfig,
+			metadata
+		});
+		if (!targetTableConfig || !includesColumn(targetTableConfig.columns, foreignColumn)) continue;
+		return {
+			index,
+			relation: {
+				foreignKey: getColumnKey(targetTableConfig.columns, foreignColumn),
+				localKey: getColumnKey(sourceTableConfig.columns, localColumn)
+			}
+		};
+	}
+	return null;
+};
 const inferManyToManyRelation = ({ field, metadata, sourceConfig, targetConfig }) => {
 	const sourceTableConfig = getTableConfig({
 		config: sourceConfig,
@@ -251,7 +311,8 @@ const resolveManyToManyConfig = ({ field, source, sourceRelation, through: throu
 		table: through.table
 	};
 };
-function createDrizzleSourceAdapter({ db, schema, views }) {
+function createDrizzleSourceAdapter({ db, nestedPaginationConcurrency, schema, views }) {
+	const nestedPaginationLimit = getNestedPaginationConcurrency(nestedPaginationConcurrency);
 	const metadata = getDrizzleSchemaMetadata({
 		db,
 		schema
@@ -282,6 +343,28 @@ function createDrizzleSourceAdapter({ db, schema, views }) {
 			targetConfig
 		}) : void 0) ?? void 0;
 	} });
+	if (metadata) for (const [index, config] of viewConfigs.entries()) {
+		const source = sourceDefinitions[index];
+		for (const field of getComputedCountRelations(config.view)) {
+			if (source.relations?.[field]) continue;
+			const hidden = inferHiddenCountRelation({
+				field,
+				metadata,
+				sourceConfig: config,
+				viewConfigs
+			});
+			const targetSource = hidden && sourceDefinitions[hidden.index];
+			if (!hidden || !targetSource) continue;
+			source.relations = {
+				...source.relations,
+				[field]: {
+					...hidden.relation,
+					kind: "many",
+					source: targetSource
+				}
+			};
+		}
+	}
 	const sourcesByView = new Map(sourceDefinitions.map((source) => [source.view, source]));
 	const sourcesByFields = new Map(sourceDefinitions.map((source) => [source.view.fields, source]));
 	const sourceConfigs = new Map(viewConfigs.map((config, index) => [sourceDefinitions[index], {
@@ -517,7 +600,7 @@ function createDrizzleSourceAdapter({ db, schema, views }) {
 					through: throughConfig
 				});
 				if (hasPagination(relationNode.args)) {
-					const connections = await Promise.all(items.map(async (item) => [item[sourceRelation.localKey], await fetchManyToManyConnection(item[sourceRelation.localKey], relationNode, sourceRelation, through, ctx)]));
+					const connections = await mapWithConcurrency(items, nestedPaginationLimit, async (item) => [item[sourceRelation.localKey], await fetchManyToManyConnection(item[sourceRelation.localKey], relationNode, sourceRelation, through, ctx)]);
 					const connectionByParentKey = new Map(connections);
 					for (const item of items) item[field] = connectionByParentKey.get(item[sourceRelation.localKey]);
 					continue;
@@ -534,7 +617,7 @@ function createDrizzleSourceAdapter({ db, schema, views }) {
 				continue;
 			}
 			if (hasPagination(relationNode.args)) {
-				const connections = await Promise.all(items.map(async (item) => [item[sourceRelation.localKey], await fetchManyConnection(item[sourceRelation.localKey], relationNode, sourceRelation, ctx)]));
+				const connections = await mapWithConcurrency(items, nestedPaginationLimit, async (item) => [item[sourceRelation.localKey], await fetchManyConnection(item[sourceRelation.localKey], relationNode, sourceRelation, ctx)]);
 				const connectionByParentKey = new Map(connections);
 				for (const item of items) item[field] = connectionByParentKey.get(item[sourceRelation.localKey]);
 				continue;

package/lib/server/prisma.mjs

@@ -1,8 +1,8 @@
 import { t as isRecord } from "../record-AeZJC9fd.mjs";
-import { O as withConnection, S as isDataView, T as toPrismaSelect, c as resolveSourceConnection, d as createSourceDefinitions, f as createSourcePlan, g as attachComputedState, i as createSourceRegistry, l as collectDataViewConfigs, o as resolveSourceById, p as getDataViewSourceConfig, s as resolveSourceByIds, t as bindSourceProcedures } from "../sourceRouter-Bwpkr062.mjs";
+import { O as withConnection, S as isDataView, T as toPrismaSelect, c as resolveSourceConnection, d as createSourceDefinitions, f as createSourcePlan, g as attachComputedState, i as createSourceRegistry, l as collectDataViewConfigs, o as resolveSourceById, p as getDataViewSourceConfig, s as resolveSourceByIds, t as bindSourceProcedures } from "../sourceRouter-C22uFcZg.mjs";
 //#region src/server/prisma.ts
 /**
-* Fate's Prisma integration.
+* fate's Prisma integration.
 *
 * @example
 * import { createPrismaFate } from '@nkzw/fate/server/prisma';

package/lib/server.d.mts

@@ -1,5 +1,5 @@
 import { K as AnyProcedure, t as AnyRecord$1 } from "./types-Dz46PXr3.mjs";
-import { C as liveGlobalConnectionTopic, S as liveEntityTopic, _ as LiveEventBus, b as createLiveEventBus, c as FateServer, d as NativeFateAPI, f as createFateFetchHandler, g as LiveConnectionSourceEvent, h as LiveConnectionEventType, l as FateServerManifest, m as createHonoFateHandler, p as createFateServer, s as FateRequestError, t as isRecord, u as InferFateAPI, v as LiveEventType, x as liveConnectionTopic, y as LiveSourceEvent } from "./record-ENh92gyb.mjs";
+import { C as liveGlobalConnectionTopic, S as liveEntityTopic, _ as LiveEventBus, b as createLiveEventBus, c as FateServer, d as NativeFateAPI, f as createFateFetchHandler, g as LiveConnectionSourceEvent, h as LiveConnectionEventType, l as FateServerManifest, m as createHonoFateHandler, p as createFateServer, s as FateRequestError, t as isRecord, u as InferFateAPI, v as LiveEventType, x as liveConnectionTopic, y as LiveSourceEvent } from "./record-CirbfZWX.mjs";
 import { A as DataViewListOptions, B as dataView, C as connectionArgs, D as CountSelection, E as ComputedSelection, F as FieldSelection, H as list$1, I as ViewPlan, L as computed, M as DataViewOrderDirection, N as DataViewResult, O as CountWhere, P as Entity, R as count, S as ConnectionResult, T as ComputedField, U as resolver, V as field, _ as createSourcePlan, a as resolveSourceConnection, b as ConnectionItem, c as SourceConfig, d as SourceOrderField, f as SourcePlan, g as createNestedSourcePlan, h as SourceRelationConfig, i as resolveSourceByIds, j as DataViewOrderBy, l as SourceDefinition, m as SourceRelation, n as refetchSourceById, p as SourcePlanNode, r as resolveSourceById, s as OrderDirection, t as SourceRegistry, u as SourceOrder, v as getNestedSelection, w as withConnection, x as ConnectionPagination, y as hasNestedSelection, z as createResolver } from "./executor-BqwHdN3n.mjs";
 import { z } from "zod";
 
@@ -74,7 +74,7 @@ type SourceProcedureResult<ById extends boolean | undefined, List extends boolea
   list: ConnectionProcedureResult<ConnectionProcedure>;
 });
 /**
- * Creates standard `byId` and `list` procedures for a Fate source.
+ * Creates standard `byId` and `list` procedures for a fate source.
  *
  * Use this when a router also has custom mutations or custom queries and you
  * want to spread the generated source procedures into that router.

package/lib/server.mjs

@@ -1,6 +1,6 @@
 import { t as isRecord } from "./record-AeZJC9fd.mjs";
 import { d as hashArgs, i as FateRequestError, n as liveEntityTopic, o as toProtocolError, r as liveGlobalConnectionTopic, t as liveConnectionTopic, u as filterConnectionArgs } from "./liveTopics-DzNtJaBD.mjs";
-import { C as list, D as resolveConnection, E as connectionArgs, O as withConnection, S as isDataView, T as toPrismaSelect, _ as computed, a as refetchSourceById, b as dataView, c as resolveSourceConnection, f as createSourcePlan, h as hasNestedSelection, k as getScopedArgs, m as getNestedSelection, n as createSourceProcedures, o as resolveSourceById, r as byIdInput, s as resolveSourceByIds, t as bindSourceProcedures, u as createNestedSourcePlan, v as count, w as resolver, x as field, y as createResolver } from "./sourceRouter-Bwpkr062.mjs";
+import { C as list, D as resolveConnection, E as connectionArgs, O as withConnection, S as isDataView, T as toPrismaSelect, _ as computed, a as refetchSourceById, b as dataView, c as resolveSourceConnection, f as createSourcePlan, h as hasNestedSelection, k as getScopedArgs, m as getNestedSelection, n as createSourceProcedures, o as resolveSourceById, r as byIdInput, s as resolveSourceByIds, t as bindSourceProcedures, u as createNestedSourcePlan, v as count, w as resolver, x as field, y as createResolver } from "./sourceRouter-C22uFcZg.mjs";
 import { EventEmitter, on } from "node:events";
 //#region src/server/live.ts
 const eventName = (type, id) => `${type}:${String(id)}`;
@@ -71,7 +71,7 @@ const mergeEvents = (emitter, names, signal) => ({ [Symbol.asyncIterator]() {
 	};
 } });
 /**
-* Creates a small in-memory event bus for Fate live view subscriptions.
+* Creates a small in-memory event bus for fate live view subscriptions.
 *
 * The bus signals that an entity changed and can optionally include changed
 * field paths. The native live handler refetches the selected record, or only
@@ -218,6 +218,27 @@ const getLiveBus = (live) => {
 	if ("subscribe" in live) return live;
 	return live.bus;
 };
+const defaultLiveMaxQueueSize = 1e3;
+const getLiveMaxQueueSize = (live) => {
+	if (!live || "subscribe" in live) return defaultLiveMaxQueueSize;
+	const value = live.maxQueueSize;
+	return typeof value === "number" && Number.isFinite(value) && value >= 0 ? Math.floor(value) : defaultLiveMaxQueueSize;
+};
+const compactLiveQueue = (connection) => {
+	if (connection.queueHead === 0) return;
+	connection.queue = connection.queue.slice(connection.queueHead);
+	connection.queueHead = 0;
+};
+const getLiveQueueSize = (connection) => connection.queue.length - connection.queueHead;
+const dequeueLiveItem = (connection) => {
+	if (connection.queueHead >= connection.queue.length) {
+		compactLiveQueue(connection);
+		return null;
+	}
+	const item = connection.queue[connection.queueHead++];
+	if (connection.queueHead > 64 && connection.queueHead * 2 >= connection.queue.length) compactLiveQueue(connection);
+	return item;
+};
 const getWaitUntil = (adapterContext) => {
 	const context = adapterContext;
 	if (!context) return;
@@ -400,6 +421,7 @@ function createFateServer({ context, lists, live, mutations, queries, roots, sou
 	}
 	for (const name of rootQueries.keys()) if (!queries?.[name]) throw new Error(`Native fate root query '${name}' requires a matching resolver in createFateServer({ queries }).`);
 	const liveBus = getLiveBus(live);
+	const liveMaxQueueSize = getLiveMaxQueueSize(live);
 	const getContext = async (request, adapterContext) => context ? await context({
 		adapterContext,
 		request
@@ -540,13 +562,22 @@ function createFateServer({ context, lists, live, mutations, queries, roots, sou
 			connection.closed = true;
 		}
 	};
+	const enqueueLiveItem = (connection, item) => {
+		if (connection.closed) return;
+		if (getLiveQueueSize(connection) >= liveMaxQueueSize) {
+			cleanupLiveConnection(connection.id);
+			return;
+		}
+		connection.queue.push(item);
+		scheduleLiveDrain(connection);
+	};
 	const drainLiveConnection = async (connection) => {
 		if (connection.closed || connection.draining) return;
 		connection.draining = true;
 		let activeOperationId = null;
 		try {
-			while (!connection.closed && connection.queue.length > 0) {
-				const item = connection.queue.shift();
+			let item;
+			while (!connection.closed && (item = dequeueLiveItem(connection))) {
 				activeOperationId = item.operation.id;
 				if (!item.subscription.active || connection.subscriptions.get(item.operation.id) !== item.subscription) continue;
 				if (item.kind === "entity") {
@@ -657,14 +688,13 @@ function createFateServer({ context, lists, live, mutations, queries, roots, sou
 		};
 		if (liveBus.listen) {
 			const unsubscribe = liveBus.listen(operation.type, operation.entityId, (event) => {
-				connection.queue.push({
+				enqueueLiveItem(connection, {
 					event,
 					kind: "entity",
 					operation,
 					source,
 					subscription
 				});
-				scheduleLiveDrain(connection);
 			}, {
 				lastEventId: operation.lastEventId,
 				signal: subscriptionController.signal
@@ -742,13 +772,12 @@ function createFateServer({ context, lists, live, mutations, queries, roots, sou
 				args: operation.args,
 				procedure: operation.procedure
 			}, (event) => {
-				connection.queue.push({
+				enqueueLiveItem(connection, {
 					event,
 					kind: "connection",
 					operation,
 					subscription
 				});
-				scheduleLiveDrain(connection);
 			}, {
 				lastEventId: operation.lastEventId,
 				signal: subscriptionController.signal
@@ -842,8 +871,10 @@ function createFateServer({ context, lists, live, mutations, queries, roots, sou
 							closed: false,
 							controller,
 							ctx,
+							id: connectionId,
 							lastHeartbeat: Date.now(),
 							queue: [],
+							queueHead: 0,
 							subscriptions: /* @__PURE__ */ new Map()
 						};
 						connection.drainInterval = setInterval(() => void drainLiveConnection(connection), 100);

package/lib/{sourceRouter-Bwpkr062.mjs → sourceRouter-C22uFcZg.mjs}

@@ -926,7 +926,7 @@ const byIdInput = z.object({
 //#region src/server/sourceRouter.ts
 const normalizeSourceOptions = (input) => "source" in input ? input : { source: input };
 /**
-* Creates standard `byId` and `list` procedures for a Fate source.
+* Creates standard `byId` and `list` procedures for a fate source.
 *
 * Use this when a router also has custom mutations or custom queries and you
 * want to spread the generated source procedures into that router.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nkzw/fate",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "fate is a modern data client for React.",
   "homepage": "https://github.com/nkzw-tech/fate",
   "license": "MIT",