prisma-sql 1.66.1 → 1.68.0

package/dist/index.d.mts

@@ -16,6 +16,7 @@ interface SqlResult {
     requiresReduction?: boolean;
     includeSpec?: Record<string, any>;
     supportsStreaming?: boolean;
+    isArrayAgg?: boolean;
 }
 type PrismaMethod = 'findUnique' | 'findFirst' | 'findMany' | 'create' | 'createMany' | 'update' | 'updateMany' | 'upsert' | 'delete' | 'deleteMany' | 'count' | 'aggregate' | 'groupBy';
 interface PrismaSQLConfig<TClient> {
@@ -41,6 +42,7 @@ interface SQLDirective {
     paramMappings: readonly ParamMap[];
     requiresReduction: boolean;
     includeSpec: Record<string, any>;
+    isArrayAgg: boolean;
     originalDirective: DirectiveProps;
 }
 
@@ -65,7 +67,7 @@ declare function buildBatchSql(queries: Record<string, BatchQuery>, modelMap: Ma
     aliases: string[];
 };
 declare function buildBatchCountSql(queries: BatchCountQuery[], modelMap: Map<string, Model>, models: Model[], dialect: SqlDialect): BatchResult;
-declare function parseBatchCountResults(row: Record<string, unknown>, count: number): number[];
+declare function parseBatchCountResults(row: Record<string, unknown>, queryCount: number): number[];
 declare function parseBatchResults(row: Record<string, unknown>, keys: string[], queries: Record<string, BatchQuery>, aliases?: string[], modelMap?: Map<string, Model>): Record<string, unknown>;
 
 interface TransactionQuery {
@@ -90,14 +92,6 @@ declare function createTransactionExecutor(deps: {
 
 declare function transformQueryResults(method: string, results: unknown): unknown;
 
-type RelStats = {
-    avg: number;
-    p95: number;
-    p99: number;
-    max: number;
-    coverage: number;
-};
-
 interface WhereInSegment {
     relationName: string;
     relArgs: unknown;
@@ -109,14 +103,13 @@ interface WhereInSegment {
 interface QueryPlan {
     filteredArgs: any;
     whereInSegments: WhereInSegment[];
+    injectedParentKeys: string[];
 }
-type RelationStatsLike = Record<string, Record<string, RelStats>>;
 declare function planQueryStrategy(params: {
     model: Model;
     method: string;
     args: any;
     allModels: readonly Model[];
-    relationStats?: RelationStatsLike;
     dialect: 'postgres' | 'sqlite';
 }): QueryPlan;
 
@@ -128,10 +121,55 @@ interface ExecuteWhereInParams {
     modelMap: Map<string, Model>;
     dialect: 'postgres' | 'sqlite';
     execute: (sql: string, params: unknown[]) => Promise<any[]>;
+    originalArgs?: any;
+    method?: string;
 }
 
 declare function executeWhereInSegments(params: ExecuteWhereInParams): Promise<void>;
 
+interface ArrayAggRelationMeta {
+    name: string;
+    isList: boolean;
+    scalarFields: Array<{
+        fieldName: string;
+        colName: string;
+        isJson: boolean;
+    }>;
+    pkFieldName: string;
+}
+interface ArrayAggReducerConfig {
+    parentModel: Model;
+    parentScalarFields: string[];
+    parentJsonSet: ReadonlySet<string>;
+    relations: ArrayAggRelationMeta[];
+}
+declare function buildArrayAggReducerConfig(parentModel: Model, includeSpec: Record<string, any>, allModels: readonly Model[]): ArrayAggReducerConfig;
+declare function reduceArrayAggRows(rows: any[], config: ArrayAggReducerConfig): any[];
+
+interface StreamingWhereInParams {
+    segments: WhereInSegment[];
+    parentSql: string;
+    parentParams: unknown[];
+    parentModel: Model;
+    allModels: readonly Model[];
+    modelMap: Map<string, Model>;
+    dialect: 'postgres' | 'sqlite';
+    execute: (sql: string, params: unknown[]) => Promise<any[]>;
+}
+interface PreFetchedWhereInParams {
+    segments: WhereInSegment[];
+    parentRows: any[];
+    parentModel: Model;
+    allModels: readonly Model[];
+    modelMap: Map<string, Model>;
+    dialect: 'postgres' | 'sqlite';
+    execute: (sql: string, params: unknown[]) => Promise<any[]>;
+}
+declare function executeWhereInSegmentsStreaming(params: StreamingWhereInParams): Promise<any[]>;
+declare function executeWithPreFetchedParents(params: PreFetchedWhereInParams): Promise<any[]>;
+
+declare function getPrimaryKeyField(model: Model): string;
+
 interface ReducerConfig {
     parentModel: Model;
     includedRelations: RelationMetadata[];
@@ -150,6 +188,7 @@ interface RelationMetadata {
     includeAllScalars: boolean;
     selectedScalarFields: string[];
     nestedIncludes?: ReducerConfig | null;
+    childLimit?: number;
     path: string;
     keyCols: string[];
     scalarCols: ScalarColSpec[];
@@ -162,7 +201,7 @@ declare function normalizeValue(value: unknown, seen?: WeakSet<object>, depth?:
 declare function createStreamingReducer(config: ReducerConfig): {
     processRow(row: any): void;
     getResults(): any[];
-    getParentMap(): Map<string, any>;
+    getParentMap(): Map<unknown, any>;
 };
 
 interface ProgressiveReducer {
@@ -173,20 +212,6 @@ interface ProgressiveReducer {
 }
 declare function createProgressiveReducer(config: ReducerConfig): ProgressiveReducer;
 
-interface StreamingWhereInParams {
-    segments: WhereInSegment[];
-    parentSql: string;
-    parentParams: unknown[];
-    parentModel: Model;
-    allModels: readonly Model[];
-    modelMap: Map<string, Model>;
-    dialect: 'postgres' | 'sqlite';
-    execute: (sql: string, params: unknown[]) => Promise<any[]>;
-    batchSize?: number;
-    maxConcurrency?: number;
-}
-declare function executeWhereInSegmentsStreaming(params: StreamingWhereInParams): Promise<any[]>;
-
 declare function transformAggregateRow(row: any): any;
 declare function extractCountValue(row: any): number | bigint;
 declare function getRowTransformer(method: string): ((row: any) => any) | null;
@@ -194,8 +219,8 @@ declare function getRowTransformer(method: string): ((row: any) => any) | null;
 declare function getOrPrepareStatement(client: any, sql: string): any;
 declare function shouldSqliteUseGet(method: string): boolean;
 declare function normalizeParams(params: unknown[]): unknown[];
-declare function executePostgresQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any | undefined, allModels: readonly any[]): Promise<unknown[]>;
-declare function executeSqliteQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any | undefined, allModels: readonly any[]): unknown[];
+declare function executePostgresQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any, allModels: readonly any[], isArrayAgg?: boolean): Promise<unknown[]>;
+declare function executeSqliteQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any, allModels: readonly any[]): unknown[];
 declare function executeRaw(client: any, sql: string, params: unknown[] | undefined, dialect: string): Promise<unknown[]>;
 
 declare function buildSQL(model: Model, models: Model[], method: PrismaMethod, args: Record<string, unknown>, dialect: SqlDialect): SqlResult;
@@ -205,4 +230,4 @@ declare function generateSQL(directive: DirectiveProps): SQLDirective;
 declare function generateAllSQL(directives: DirectiveProps[]): SQLDirective[];
 declare function generateSQLByModel(directives: DirectiveProps[]): Map<string, SQLDirective[]>;
 
-export { type BatchCountQuery, type BatchQuery, type Model, type PrismaMethod, type PrismaSQLConfig, type PrismaSQLResult, type ReducerConfig, type SqlResult, type TransactionOptions, type TransactionQuery, buildBatchCountSql, buildBatchSql, buildReducerConfig, buildSQL, createPrismaSQL, createProgressiveReducer, createStreamingReducer, createToSQL, createTransactionExecutor, executePostgresQuery, executeRaw, executeSqliteQuery, executeWhereInSegments, executeWhereInSegmentsStreaming, extractCountValue, generateAllSQL, generateSQL, generateSQLByModel, getOrPrepareStatement, getRowTransformer, normalizeParams, normalizeValue, parseBatchCountResults, parseBatchResults, planQueryStrategy, reduceFlatRows, shouldSqliteUseGet, transformAggregateRow, transformQueryResults };
+export { type BatchCountQuery, type BatchQuery, type Model, type PrismaMethod, type PrismaSQLConfig, type PrismaSQLResult, type ReducerConfig, type SqlResult, type TransactionOptions, type TransactionQuery, buildArrayAggReducerConfig, buildBatchCountSql, buildBatchSql, buildReducerConfig, buildSQL, createPrismaSQL, createProgressiveReducer, createStreamingReducer, createToSQL, createTransactionExecutor, executePostgresQuery, executeRaw, executeSqliteQuery, executeWhereInSegments, executeWhereInSegmentsStreaming, executeWithPreFetchedParents, extractCountValue, generateAllSQL, generateSQL, generateSQLByModel, getOrPrepareStatement, getPrimaryKeyField, getRowTransformer, normalizeParams, normalizeValue, parseBatchCountResults, parseBatchResults, planQueryStrategy, reduceArrayAggRows, reduceFlatRows, shouldSqliteUseGet, transformAggregateRow, transformQueryResults };

package/dist/index.d.ts

@@ -16,6 +16,7 @@ interface SqlResult {
     requiresReduction?: boolean;
     includeSpec?: Record<string, any>;
     supportsStreaming?: boolean;
+    isArrayAgg?: boolean;
 }
 type PrismaMethod = 'findUnique' | 'findFirst' | 'findMany' | 'create' | 'createMany' | 'update' | 'updateMany' | 'upsert' | 'delete' | 'deleteMany' | 'count' | 'aggregate' | 'groupBy';
 interface PrismaSQLConfig<TClient> {
@@ -41,6 +42,7 @@ interface SQLDirective {
     paramMappings: readonly ParamMap[];
     requiresReduction: boolean;
     includeSpec: Record<string, any>;
+    isArrayAgg: boolean;
     originalDirective: DirectiveProps;
 }
 
@@ -65,7 +67,7 @@ declare function buildBatchSql(queries: Record<string, BatchQuery>, modelMap: Ma
     aliases: string[];
 };
 declare function buildBatchCountSql(queries: BatchCountQuery[], modelMap: Map<string, Model>, models: Model[], dialect: SqlDialect): BatchResult;
-declare function parseBatchCountResults(row: Record<string, unknown>, count: number): number[];
+declare function parseBatchCountResults(row: Record<string, unknown>, queryCount: number): number[];
 declare function parseBatchResults(row: Record<string, unknown>, keys: string[], queries: Record<string, BatchQuery>, aliases?: string[], modelMap?: Map<string, Model>): Record<string, unknown>;
 
 interface TransactionQuery {
@@ -90,14 +92,6 @@ declare function createTransactionExecutor(deps: {
 
 declare function transformQueryResults(method: string, results: unknown): unknown;
 
-type RelStats = {
-    avg: number;
-    p95: number;
-    p99: number;
-    max: number;
-    coverage: number;
-};
-
 interface WhereInSegment {
     relationName: string;
     relArgs: unknown;
@@ -109,14 +103,13 @@ interface WhereInSegment {
 interface QueryPlan {
     filteredArgs: any;
     whereInSegments: WhereInSegment[];
+    injectedParentKeys: string[];
 }
-type RelationStatsLike = Record<string, Record<string, RelStats>>;
 declare function planQueryStrategy(params: {
     model: Model;
     method: string;
     args: any;
     allModels: readonly Model[];
-    relationStats?: RelationStatsLike;
     dialect: 'postgres' | 'sqlite';
 }): QueryPlan;
 
@@ -128,10 +121,55 @@ interface ExecuteWhereInParams {
     modelMap: Map<string, Model>;
     dialect: 'postgres' | 'sqlite';
     execute: (sql: string, params: unknown[]) => Promise<any[]>;
+    originalArgs?: any;
+    method?: string;
 }
 
 declare function executeWhereInSegments(params: ExecuteWhereInParams): Promise<void>;
 
+interface ArrayAggRelationMeta {
+    name: string;
+    isList: boolean;
+    scalarFields: Array<{
+        fieldName: string;
+        colName: string;
+        isJson: boolean;
+    }>;
+    pkFieldName: string;
+}
+interface ArrayAggReducerConfig {
+    parentModel: Model;
+    parentScalarFields: string[];
+    parentJsonSet: ReadonlySet<string>;
+    relations: ArrayAggRelationMeta[];
+}
+declare function buildArrayAggReducerConfig(parentModel: Model, includeSpec: Record<string, any>, allModels: readonly Model[]): ArrayAggReducerConfig;
+declare function reduceArrayAggRows(rows: any[], config: ArrayAggReducerConfig): any[];
+
+interface StreamingWhereInParams {
+    segments: WhereInSegment[];
+    parentSql: string;
+    parentParams: unknown[];
+    parentModel: Model;
+    allModels: readonly Model[];
+    modelMap: Map<string, Model>;
+    dialect: 'postgres' | 'sqlite';
+    execute: (sql: string, params: unknown[]) => Promise<any[]>;
+}
+interface PreFetchedWhereInParams {
+    segments: WhereInSegment[];
+    parentRows: any[];
+    parentModel: Model;
+    allModels: readonly Model[];
+    modelMap: Map<string, Model>;
+    dialect: 'postgres' | 'sqlite';
+    execute: (sql: string, params: unknown[]) => Promise<any[]>;
+}
+declare function executeWhereInSegmentsStreaming(params: StreamingWhereInParams): Promise<any[]>;
+declare function executeWithPreFetchedParents(params: PreFetchedWhereInParams): Promise<any[]>;
+
+declare function getPrimaryKeyField(model: Model): string;
+
 interface ReducerConfig {
     parentModel: Model;
     includedRelations: RelationMetadata[];
@@ -150,6 +188,7 @@ interface RelationMetadata {
     includeAllScalars: boolean;
     selectedScalarFields: string[];
     nestedIncludes?: ReducerConfig | null;
+    childLimit?: number;
     path: string;
     keyCols: string[];
     scalarCols: ScalarColSpec[];
@@ -162,7 +201,7 @@ declare function normalizeValue(value: unknown, seen?: WeakSet<object>, depth?:
 declare function createStreamingReducer(config: ReducerConfig): {
     processRow(row: any): void;
     getResults(): any[];
-    getParentMap(): Map<string, any>;
+    getParentMap(): Map<unknown, any>;
 };
 
 interface ProgressiveReducer {
@@ -173,20 +212,6 @@ interface ProgressiveReducer {
 }
 declare function createProgressiveReducer(config: ReducerConfig): ProgressiveReducer;
 
-interface StreamingWhereInParams {
-    segments: WhereInSegment[];
-    parentSql: string;
-    parentParams: unknown[];
-    parentModel: Model;
-    allModels: readonly Model[];
-    modelMap: Map<string, Model>;
-    dialect: 'postgres' | 'sqlite';
-    execute: (sql: string, params: unknown[]) => Promise<any[]>;
-    batchSize?: number;
-    maxConcurrency?: number;
-}
-declare function executeWhereInSegmentsStreaming(params: StreamingWhereInParams): Promise<any[]>;
-
 declare function transformAggregateRow(row: any): any;
 declare function extractCountValue(row: any): number | bigint;
 declare function getRowTransformer(method: string): ((row: any) => any) | null;
@@ -194,8 +219,8 @@ declare function getRowTransformer(method: string): ((row: any) => any) | null;
 declare function getOrPrepareStatement(client: any, sql: string): any;
 declare function shouldSqliteUseGet(method: string): boolean;
 declare function normalizeParams(params: unknown[]): unknown[];
-declare function executePostgresQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any | undefined, allModels: readonly any[]): Promise<unknown[]>;
-declare function executeSqliteQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any | undefined, allModels: readonly any[]): unknown[];
+declare function executePostgresQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any, allModels: readonly any[], isArrayAgg?: boolean): Promise<unknown[]>;
+declare function executeSqliteQuery(client: any, sql: string, params: unknown[], method: string, requiresReduction: boolean, includeSpec: Record<string, any> | undefined, model: any, allModels: readonly any[]): unknown[];
 declare function executeRaw(client: any, sql: string, params: unknown[] | undefined, dialect: string): Promise<unknown[]>;
 
 declare function buildSQL(model: Model, models: Model[], method: PrismaMethod, args: Record<string, unknown>, dialect: SqlDialect): SqlResult;
@@ -205,4 +230,4 @@ declare function generateSQL(directive: DirectiveProps): SQLDirective;
 declare function generateAllSQL(directives: DirectiveProps[]): SQLDirective[];
 declare function generateSQLByModel(directives: DirectiveProps[]): Map<string, SQLDirective[]>;
 
-export { type BatchCountQuery, type BatchQuery, type Model, type PrismaMethod, type PrismaSQLConfig, type PrismaSQLResult, type ReducerConfig, type SqlResult, type TransactionOptions, type TransactionQuery, buildBatchCountSql, buildBatchSql, buildReducerConfig, buildSQL, createPrismaSQL, createProgressiveReducer, createStreamingReducer, createToSQL, createTransactionExecutor, executePostgresQuery, executeRaw, executeSqliteQuery, executeWhereInSegments, executeWhereInSegmentsStreaming, extractCountValue, generateAllSQL, generateSQL, generateSQLByModel, getOrPrepareStatement, getRowTransformer, normalizeParams, normalizeValue, parseBatchCountResults, parseBatchResults, planQueryStrategy, reduceFlatRows, shouldSqliteUseGet, transformAggregateRow, transformQueryResults };
+export { type BatchCountQuery, type BatchQuery, type Model, type PrismaMethod, type PrismaSQLConfig, type PrismaSQLResult, type ReducerConfig, type SqlResult, type TransactionOptions, type TransactionQuery, buildArrayAggReducerConfig, buildBatchCountSql, buildBatchSql, buildReducerConfig, buildSQL, createPrismaSQL, createProgressiveReducer, createStreamingReducer, createToSQL, createTransactionExecutor, executePostgresQuery, executeRaw, executeSqliteQuery, executeWhereInSegments, executeWhereInSegmentsStreaming, executeWithPreFetchedParents, extractCountValue, generateAllSQL, generateSQL, generateSQLByModel, getOrPrepareStatement, getPrimaryKeyField, getRowTransformer, normalizeParams, normalizeValue, parseBatchCountResults, parseBatchResults, planQueryStrategy, reduceArrayAggRows, reduceFlatRows, shouldSqliteUseGet, transformAggregateRow, transformQueryResults };