pqb 0.23.2 → 0.23.4

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 import * as orchid_core from 'orchid-core';
-import { QueryResultRow, AdapterBase, QueryInput, Sql, RecordKeyTrue, EmptyObject, QueryBaseCommon, QueryColumns, QueryInternal, QueryMetaBase, Expression, QueryColumn, MaybeArray, TemplateLiteralArgs, ColumnSchemaConfig, Code, DateColumnData, TimeInterval, ColumnTypeBase, ArrayMethodsData, ParseColumn, EncodeColumn, OperatorToSQL, ColumnsShapeBase, ColumnsParsers, getValueKey, EmptyTuple, QueryColumnToNullable, StringKey, QueryThen, QueryCatch, QueryColumnsInit, ColumnShapeOutput, DefaultSelectColumns, CoreQueryScopes, DbBase, TransactionState, SQLQueryArgs, BaseOperators as BaseOperators$1, SetOptional, MergeObjects, RawSQLBase, RawSQLValues, ExpressionTypeMethod, DynamicSQLArg, StaticSQLArgs, Spread, BaseNumberData, ColumnWithDefault, StringTypeData, PrimaryKeyColumn, ForeignKeyTable, TimestampHelpers, ColumnNameOfTable, ColumnDataBase } from 'orchid-core';
+import { QueryResultRow, AdapterConfigBase, AdapterBase, QueryInput, Sql, RecordKeyTrue, EmptyObject, QueryBaseCommon, QueryColumns, QueryInternal, QueryMetaBase, Expression, QueryColumn, MaybeArray, TemplateLiteralArgs, ColumnSchemaConfig, Code, DateColumnData, TimeInterval, ColumnTypeBase, ArrayMethodsData, ParseColumn, EncodeColumn, OperatorToSQL, ColumnsShapeBase, ColumnsParsers, getValueKey, EmptyTuple, QueryColumnToNullable, StringKey, QueryThen, QueryCatch, QueryColumnsInit, ColumnShapeOutput, DefaultSelectColumns, CoreQueryScopes, DbBase, TransactionState, SQLQueryArgs, BaseOperators as BaseOperators$1, SetOptional, MergeObjects, RawSQLBase, RawSQLValues, ExpressionTypeMethod, DynamicSQLArg, StaticSQLArgs, Spread, BaseNumberData, ColumnWithDefault, StringTypeData, PrimaryKeyColumn, ForeignKeyTable, TimestampHelpers, ColumnNameOfTable, ColumnDataBase } from 'orchid-core';
 import { PoolConfig, Pool, PoolClient } from 'pg';
 import { inspect } from 'node:util';
 import { AsyncLocalStorage } from 'node:async_hooks';
@@ -19,10 +19,10 @@ type QueryArraysResult<R extends any[] = any[]> = {
         name: string;
     }[];
 };
-type AdapterConfig = Omit<PoolConfig, 'types' | 'connectionString'> & {
+interface AdapterConfig extends AdapterConfigBase, Omit<PoolConfig, 'types' | 'connectionString'> {
     schema?: string;
     databaseURL?: string;
-};
+}
 type AdapterOptions = AdapterConfig & {
     types?: TypeParsers;
 };
@@ -32,6 +32,7 @@ declare class Adapter implements AdapterBase {
     config: PoolConfig;
     schema?: string;
     constructor({ types, ...config }: AdapterOptions);
+    connect(): Promise<PoolClient>;
     query<T extends QueryResultRow = QueryResultRow>(query: QueryInput, types?: TypeParsers): Promise<QueryResult<T>>;
     arrays<R extends any[] = any[]>(query: QueryInput, types?: TypeParsers): Promise<QueryArraysResult<R>>;
     transaction<Result>(begin: Sql, cb: (adapter: TransactionAdapter) => Promise<Result>): Promise<Result>;
@@ -44,6 +45,7 @@ declare class TransactionAdapter implements Adapter {
     pool: Pool;
     config: PoolConfig;
     constructor(adapter: Adapter, client: PoolClient, types: TypeParsers);
+    connect(): Promise<PoolClient>;
     query<T extends QueryResultRow = any>(query: QueryInput, types?: TypeParsers): Promise<QueryResult<T>>;
     arrays<R extends any[] = any[]>(query: QueryInput, types?: TypeParsers): Promise<QueryArraysResult<R>>;
     transaction<Result>(_: Sql, cb: (adapter: TransactionAdapter) => Promise<Result>): Promise<Result>;
@@ -1475,65 +1477,12 @@ declare class OnQueryBuilder<S extends QueryBase = QueryBase, J extends Pick<Que
     onJsonPathEquals<T extends OnQueryBuilder>(this: T, ...args: OnJsonPathEqualsArgs<T>): T;
 }
 
-type AggregateOptions<T extends Query> = {
-    distinct?: boolean;
-    order?: OrderArg<T> | OrderArg<T>[];
-    filter?: WhereArg<T>;
-    filterOr?: WhereArg<T>[];
-    withinGroup?: boolean;
-    over?: Over<T>;
-};
-type Over<T extends Query> = keyof T['windows'] | WindowArgDeclaration<T>;
-type FnExpressionArgs<Q extends Query> = (SelectableOrExpression<Q> | {
-    pairs: Record<string, SelectableOrExpression<Q>>;
-} | {
-    value: unknown;
-})[];
-declare class FnExpression<Q extends Query = Query, T extends QueryColumn = QueryColumn> extends Expression<T> {
-    q: Q;
-    fn: string;
-    args: FnExpressionArgs<Q>;
-    options: AggregateOptions<Q>;
-    _type: T;
-    /**
-     * @param q - query object.
-     * @param fn - SQL function name.
-     * @param args - arguments of the function.
-     * @param options - aggregate options.
-     * @param _type - column type of the function result.
-     */
-    constructor(q: Q, fn: string, args: FnExpressionArgs<Q>, options: AggregateOptions<Q>, _type: T);
-    makeSQL(ctx: ToSQLCtx, quotedAs?: string): string;
-}
-type ColumnExpression<C extends QueryColumn, Ops extends BaseOperators = C['operators']> = Expression<C> & {
-    [K in keyof Ops]: (arg: Ops[K]['_opType']) => ColumnExpression<QueryColumn<boolean | null>>;
-};
-declare const makeExpression: <T extends Query, C extends ColumnTypeBase<{
-    type: unknown;
-    nullable: <T_1 extends ColumnTypeBase<Pick, unknown, any, orchid_core.BaseOperators, unknown, unknown, any, unknown, any, orchid_core.ColumnDataBase>>(this: T_1) => orchid_core.NullableColumn<T_1, unknown, unknown, unknown>;
-    encode: unknown;
-    parse: unknown;
-    asType: unknown;
-    errors?: (<T_2 extends ColumnTypeBase<Pick, unknown, any, orchid_core.BaseOperators, unknown, unknown, any, unknown, any, orchid_core.ColumnDataBase>>(this: T_2, errors: orchid_core.ErrorMessages) => void) | undefined;
-}, unknown, any, orchid_core.BaseOperators, unknown, unknown, any, unknown, any, orchid_core.ColumnDataBase>>(self: T, expr: Expression) => Omit<T, "then" | "result" | "returnType" | "catch"> & {
-    meta: {
-        hasSelect: true;
-    };
-    result: {
-        value: C;
-    };
-    returnType: "valueOrThrow";
-    then: orchid_core.QueryThen<C["outputType"]>;
-    catch: orchid_core.QueryCatch<C["outputType"]>;
-} & C["operators"];
-declare function makeFnExpression<T extends Query, C extends QueryColumn>(self: T, type: C, fn: string, args: FnExpressionArgs<Query>, options?: AggregateOptions<T>): SetQueryReturnsColumn<T, C> & C['operators'];
-
 type WhereArg<T extends WhereQueryBase> = {
     [K in keyof T['selectable'] | 'NOT' | 'OR' | 'IN' | 'EXISTS']?: K extends 'NOT' ? MaybeArray<WhereArg<T>> : K extends 'OR' ? MaybeArray<WhereArg<T>>[] : K extends 'IN' ? MaybeArray<{
         columns: (keyof T['selectable'])[];
         values: unknown[][] | Query | Expression;
     }> : K extends keyof T['selectable'] ? T['selectable'][K]['column']['queryType'] | null | ColumnOperators<T['selectable'], K> | Expression | Query : never;
-} | QueryBase | Expression | ((q: WhereQueryBuilder<T>) => QueryBase | ColumnExpression<QueryColumn<boolean | null>>);
+} | QueryBase | Expression<QueryColumn<boolean | null>> | ((q: WhereQueryBuilder<T>) => ExpressionOrQueryReturning<boolean | null> | WhereQueryBuilder<T>);
 /**
  * Callback argument of `where`.
  * It has `where` methods (`where`, `whereNot`, `whereExists`, etc.),
@@ -2573,6 +2522,9 @@ declare function pushLimitSQL(sql: string[], values: unknown[], q: SelectQueryDa
 type AliasOrTable<T extends Pick<Query, 'table' | 'meta'>> = T['meta']['as'] extends string ? T['meta']['as'] : T['table'] extends string ? T['table'] : never;
 type SelectableOrExpression<T extends Pick<QueryBase, 'selectable'> = QueryBase, C extends QueryColumn = QueryColumn> = '*' | StringKey<keyof T['selectable']> | Expression<C>;
 type ExpressionOutput<T extends QueryBase, Expr extends SelectableOrExpression<T>> = Expr extends keyof T['selectable'] ? T['selectable'][Expr]['column'] : Expr extends Expression ? Expr['_type'] : never;
+type ExpressionOrQueryReturning<T> = Expression<QueryColumn<T>> | (QueryBase & {
+    then: QueryThen<T>;
+});
 declare const getClonedQueryData: (query: QueryData) => QueryData;
 declare const getQueryAs: (q: {
     table?: string | undefined;
@@ -2600,6 +2552,59 @@ declare const resolveSubQueryCallback: (q: ToSQLQuery, cb: (q: ToSQLQuery) => To
  */
 declare const joinSubQuery: (q: ToSQLQuery, sub: ToSQLQuery) => Query;
 
+type AggregateOptions<T extends Query> = {
+    distinct?: boolean;
+    order?: OrderArg<T> | OrderArg<T>[];
+    filter?: WhereArg<T>;
+    filterOr?: WhereArg<T>[];
+    withinGroup?: boolean;
+    over?: Over<T>;
+};
+type Over<T extends Query> = keyof T['windows'] | WindowArgDeclaration<T>;
+type FnExpressionArgs<Q extends Query> = (SelectableOrExpression<Q> | {
+    pairs: Record<string, SelectableOrExpression<Q>>;
+} | {
+    value: unknown;
+})[];
+declare class FnExpression<Q extends Query = Query, T extends QueryColumn = QueryColumn> extends Expression<T> {
+    q: Q;
+    fn: string;
+    args: FnExpressionArgs<Q>;
+    options: AggregateOptions<Q>;
+    _type: T;
+    /**
+     * @param q - query object.
+     * @param fn - SQL function name.
+     * @param args - arguments of the function.
+     * @param options - aggregate options.
+     * @param _type - column type of the function result.
+     */
+    constructor(q: Q, fn: string, args: FnExpressionArgs<Q>, options: AggregateOptions<Q>, _type: T);
+    makeSQL(ctx: ToSQLCtx, quotedAs?: string): string;
+}
+type ColumnExpression<C extends QueryColumn, Ops extends BaseOperators = C['operators']> = Expression<C> & {
+    [K in keyof Ops]: (arg: Ops[K]['_opType']) => ColumnExpression<QueryColumn<boolean | null>>;
+};
+declare const makeExpression: <T extends Query, C extends ColumnTypeBase<{
+    type: unknown;
+    nullable: <T_1 extends ColumnTypeBase<Pick, unknown, any, orchid_core.BaseOperators, unknown, unknown, any, unknown, any, orchid_core.ColumnDataBase>>(this: T_1) => orchid_core.NullableColumn<T_1, unknown, unknown, unknown>;
+    encode: unknown;
+    parse: unknown;
+    asType: unknown;
+    errors?: (<T_2 extends ColumnTypeBase<Pick, unknown, any, orchid_core.BaseOperators, unknown, unknown, any, unknown, any, orchid_core.ColumnDataBase>>(this: T_2, errors: orchid_core.ErrorMessages) => void) | undefined;
+}, unknown, any, orchid_core.BaseOperators, unknown, unknown, any, unknown, any, orchid_core.ColumnDataBase>>(self: T, expr: Expression) => Omit<T, "then" | "result" | "returnType" | "catch"> & {
+    meta: {
+        hasSelect: true;
+    };
+    result: {
+        value: C;
+    };
+    returnType: "valueOrThrow";
+    then: orchid_core.QueryThen<C["outputType"]>;
+    catch: orchid_core.QueryCatch<C["outputType"]>;
+} & C["operators"];
+declare function makeFnExpression<T extends Query, C extends QueryColumn>(self: T, type: C, fn: string, args: FnExpressionArgs<Query>, options?: AggregateOptions<T>): SetQueryReturnsColumn<T, C> & C['operators'];
+
 type QueryGetSelf = Pick<Query, 'selectable' | 'q' | 'table' | 'baseQuery' | 'clone'>;
 type GetArg<T extends QueryGetSelf> = GetStringArg<T> | Expression;
 type GetStringArg<T extends Pick<Query, 'selectable'>> = StringKey<keyof T['selectable']>;
@@ -3614,7 +3619,7 @@ type UpdateColumn<T extends UpdateSelf, Key extends keyof T['inputType']> = T['i
         kind: 'select';
     };
 }));
-type UpdateRelationData<T extends UpdateSelf, Rel extends RelationConfigBase> = Rel['one'] extends true ? Rel['dataForUpdate'] | (QueryReturnsAll<T['returnType']> extends true ? never : Rel['dataForUpdateOne']) : Rel['dataForUpdate'] & (QueryReturnsAll<T['returnType']> extends true ? EmptyObject : Rel['dataForUpdateOne']);
+type UpdateRelationData<T extends UpdateSelf, Rel extends RelationConfigBase> = QueryReturnsAll<T['returnType']> extends true ? Rel['dataForUpdate'] : Rel['one'] extends true ? Rel['dataForUpdate'] | Rel['dataForUpdateOne'] : Rel['dataForUpdate'] & Rel['dataForUpdateOne'];
 type UpdateArg<T extends UpdateSelf> = T['meta']['hasWhere'] extends true ? UpdateData<T> : never;
 type UpdateRawArgs<T extends UpdateSelf> = T['meta']['hasWhere'] extends true ? [sql: Expression] | TemplateLiteralArgs : never;
 type UpdateResult<T extends UpdateSelf> = T['meta']['hasSelect'] extends true ? SetQueryKind<T, 'update'> : SetQueryReturnsRowCount<SetQueryKind<T, 'update'>>;
@@ -7069,4 +7074,4 @@ type CopyResult<T extends Query> = SetQueryKind<T, 'copy'>;
  */
 declare function copyTableData<T extends Query>(query: T, arg: CopyArg<T>): CopyResult<T>;
 
-export { Adapter, AdapterConfig, AdapterOptions, AddQueryDefaults, AddQuerySelect, AddQueryWith, AfterHook, AggregateMethods, AggregateOptions, AliasOrTable, ArrayColumn, ArrayColumnValue, ArrayData, AsMethods, AsQueryArg, BaseOperators, BigIntColumn, BigSerialColumn, BitColumn, BitVaryingColumn, BooleanColumn$1 as BooleanColumn, BoxColumn, ByteaColumn, CharColumn, CidrColumn, CircleColumn, CitextColumn, Clear, ClearStatement, CloneSelfKeys, ColumnData, ColumnExpression, ColumnFromDbParams, ColumnInfoQueryData, ColumnOperators, ColumnRefExpression, ColumnType, ColumnsByType, ColumnsShape, ColumnsShapeToNullableObject, ColumnsShapeToObject, ColumnsShapeToObjectArray, ColumnsShapeToPluck, CommonQueryData, ComputedColumnsBase, CopyOptions, CopyQueryData, Create, CreateColumn, CreateCtx, CreateData, CreateKind, CreateMethodsNames, CreateRelationDataOmittingFKeys, CreateRelationsData, CreateRelationsDataOmittingFKeys, CreateSelf, CustomTypeColumn, DateBaseColumn, DateColumn, DateColumnInput, DateTimeBaseClass, DateTimeTzBaseClass, Db, DbOptions, DbResult, DbTableConstructor, DbTableOptionScopes, DbTableOptions, DecimalColumn, DefaultColumnTypes, DefaultSchemaConfig, Delete, DeleteArgs, DeleteMethodsNames, DeleteQueryData, DeleteResult, DeleteSelf, DomainColumn, DoublePrecisionColumn, DropMode, DynamicRawSQL, EnumColumn, ExpressionOutput, FnExpression, FnExpressionArgs, For, ForeignKey, ForeignKeyAction, ForeignKeyMatch, ForeignKeyOptions, From, FromArgs, FromQueryArg, FromQuerySelf, FromResult, GetArg, GetColumnInfo, GetQueryResult, GetResult, GetResultOptional, GetStringArg, GroupArg, Having, HavingItem, HookAction, HookSelect, IdentityColumn, IndexColumnOptions, IndexOptions, InetColumn, InsertQueryData, IntegerBaseColumn, IntegerColumn, IntervalColumn, IsolationLevel, JSONColumn, JSONTextColumn, Join, JoinArgs, JoinCallback, JoinFirstArg, JoinItem, JoinLateralCallback, JoinLateralItem, JoinLateralResult, JoinOverrides, JoinResult, JoinedParsers, JoinedShapes, JsonItem, JsonMethods, JsonModifiers, LimitedTextBaseColumn, LineColumn, LsegColumn, MacAddr8Column, MacAddrColumn, MapTableScopesOption, MergeQuery, MergeQueryMethods, MoneyColumn, MoreThanOneRowError, NoPrimaryKeyOption, NotFoundError, NumberAsStringBaseColumn, NumberBaseColumn, NumberColumnData, OnConflictItem, OnConflictMergeUpdate, OnConflictQueryBuilder, OnQueryBuilder, Operator, Operators, OperatorsAny, OperatorsArray, OperatorsBoolean, OperatorsDate, OperatorsJson, OperatorsNumber, OperatorsText, OperatorsTime, OrCreateArg, OrchidOrmError, OrchidOrmInternalError, OrderArg, OrderArgSelf, OrderArgs, OrderItem, OrderTsQueryConfig, Over, PathColumn, PointColumn, PolygonColumn, Query, QueryAfterHook, QueryArraysResult, QueryBase, QueryBeforeHook, QueryData, QueryDataJoinTo, QueryDefaultReturnData, QueryError, QueryErrorName, QueryGet, QueryGetSelf, QueryHelperResult, QueryHookSelect, QueryHooks, QueryLog, QueryLogObject, QueryLogOptions, QueryLogger, QueryMethods, QueryResult, QueryReturnType, QueryReturnsAll, QueryScopeData, QueryScopes, QuerySourceItem, QueryTransform, QueryTransformFn, QueryUpsertOrCreate, QueryWithComputed, QueryWithTable, RawSQL, RawSqlMethods, RealColumn, RelationConfigBase, RelationConfigDataForCreate, RelationJoinQuery, RelationQuery, RelationQueryBase, RelationsBase, RelationsChain, SearchArg, SearchMethods, SearchWeight, Select, SelectArg, SelectAs, SelectItem, SelectQueryData, SelectSubQueryResult, SelectableBase, SelectableFromShape, SelectableOfType, SelectableOrExpression, SelectableOrExpressionOfType, SerialColumn, SerialColumnData, SetQueryKind, SetQueryReturns, SetQueryReturnsAll, SetQueryReturnsColumn, SetQueryReturnsColumnInfo, SetQueryReturnsColumnOptional, SetQueryReturnsOne, SetQueryReturnsOneOptional, SetQueryReturnsPluck, SetQueryReturnsPluckColumn, SetQueryReturnsRowCount, SetQueryReturnsRows, SetQueryReturnsValue, SetQueryReturnsValueOptional, SetQueryReturnsVoid, SetQueryTableAlias, SetQueryWith, SimpleJoinItem, SingleColumnIndexOptions, SmallIntColumn, SmallSerialColumn, SortDir, StringColumn, TableData, TextBaseColumn, TextColumn, TextColumnData, Then, TimeColumn, TimestampColumn, TimestampTZColumn, ToSQLCtx, ToSQLOptions, ToSQLQuery, Transaction, TransactionAdapter, TransactionOptions, TransactionSelf, TransformMethods, TruncateQueryData, TsQueryColumn, TsVectorColumn, TypeParsers, UUIDColumn, UnhandledTypeError, Union, UnionArg, UnionItem, UnionKind, UnknownColumn, Update, UpdateArg, UpdateCtx, UpdateData, UpdateQueryData, UpdateQueryDataItem, UpdateQueryDataObject, UpdateSelf, UpdatedAtDataInjector, UpsertArg, UpsertResult, UpsertThis, VarCharColumn, VirtualColumn, Where, WhereArg, WhereArgs, WhereInArg, WhereInColumn, WhereInItem, WhereInValues, WhereItem, WhereJsonPathEqualsItem, WhereNotArgs, WhereOnItem, WhereOnJoinItem, WhereQueryBase, WhereQueryBuilder, WhereResult, WhereSearchItem, WhereSearchResult, WindowArg, WindowArgDeclaration, WindowDeclaration, WindowItem, With, WithDataBase, WithDataItem, WithItem, WithOptions, WrapQueryArg, WrapQuerySelf, XMLColumn, _queryAfterSaveCommit, _queryAll, _queryAs, _queryChangeCounter, _queryCreate, _queryCreateFrom, _queryCreateMany, _queryCreateManyFrom, _queryCreateManyRaw, _queryCreateRaw, _queryDefaults, _queryDelete, _queryExec, _queryFindBy, _queryFindByOptional, _queryGet, _queryGetOptional, _queryHookAfterCreate, _queryHookAfterCreateCommit, _queryHookAfterDelete, _queryHookAfterDeleteCommit, _queryHookAfterQuery, _queryHookAfterSave, _queryHookAfterUpdate, _queryHookAfterUpdateCommit, _queryHookBeforeCreate, _queryHookBeforeDelete, _queryHookBeforeQuery, _queryHookBeforeSave, _queryHookBeforeUpdate, _queryInsert, _queryInsertFrom, _queryInsertMany, _queryInsertManyFrom, _queryInsertManyRaw, _queryInsertRaw, _queryJoinOn, _queryJoinOnJsonPathEquals, _queryJoinOrOn, _queryOr, _queryOrNot, _queryRows, _querySelect, _queryTake, _queryTakeOptional, _queryUpdate, _queryUpdateOrThrow, _queryUpdateRaw, _queryWhere, _queryWhereIn, _queryWhereNot, addComputedColumns, addParserForRawExpression, addParserForSelectItem, addQueryOn, anyShape, checkIfASimpleQuery, cloneQuery, columnCheckToCode, columnCode, columnForeignKeysToCode, columnIndexesToCode, columnsShapeToCode, constraintPropsToCode, constraintToCode, copyTableData, countSelect, createDb, defaultSchemaConfig, extendQuery, foreignKeyArgumentToCode, getClonedQueryData, getColumnInfo, getColumnTypes, getConstraintKind, getQueryAs, getShapeFromSelect, getTableData, handleResult, identityToCode, indexToCode, instantiateColumn, isQueryReturnsAll, isSelectingCount, joinSubQuery, logColors, logParamToLogObject, makeColumnTypes, makeColumnsByType, makeExpression, makeFnExpression, makeRegexToFindInSql, makeSQL, newTableData, parseRecord, parseResult, primaryKeyToCode, processSelectArg, pushLimitSQL, pushQueryArray, pushQueryOn, pushQueryOrOn, pushQueryValue, queryFrom, queryJson, queryMethodByReturnType, queryTypeWithLimitOne, queryWrap, quote, quoteString, raw, referencesArgsToCode, resetTableData, resolveSubQueryCallback, saveSearchAlias, setParserForSelectedString, setQueryObjectValue, setQueryOperators, simplifyColumnDefault, templateLiteralToSQL, testTransaction, throwIfNoWhere, toSQL, toSQLCacheKey };
+export { Adapter, AdapterConfig, AdapterOptions, AddQueryDefaults, AddQuerySelect, AddQueryWith, AfterHook, AggregateMethods, AggregateOptions, AliasOrTable, ArrayColumn, ArrayColumnValue, ArrayData, AsMethods, AsQueryArg, BaseOperators, BigIntColumn, BigSerialColumn, BitColumn, BitVaryingColumn, BooleanColumn$1 as BooleanColumn, BoxColumn, ByteaColumn, CharColumn, CidrColumn, CircleColumn, CitextColumn, Clear, ClearStatement, CloneSelfKeys, ColumnData, ColumnExpression, ColumnFromDbParams, ColumnInfoQueryData, ColumnOperators, ColumnRefExpression, ColumnType, ColumnsByType, ColumnsShape, ColumnsShapeToNullableObject, ColumnsShapeToObject, ColumnsShapeToObjectArray, ColumnsShapeToPluck, CommonQueryData, ComputedColumnsBase, CopyOptions, CopyQueryData, Create, CreateColumn, CreateCtx, CreateData, CreateKind, CreateMethodsNames, CreateRelationDataOmittingFKeys, CreateRelationsData, CreateRelationsDataOmittingFKeys, CreateSelf, CustomTypeColumn, DateBaseColumn, DateColumn, DateColumnInput, DateTimeBaseClass, DateTimeTzBaseClass, Db, DbOptions, DbResult, DbTableConstructor, DbTableOptionScopes, DbTableOptions, DecimalColumn, DefaultColumnTypes, DefaultSchemaConfig, Delete, DeleteArgs, DeleteMethodsNames, DeleteQueryData, DeleteResult, DeleteSelf, DomainColumn, DoublePrecisionColumn, DropMode, DynamicRawSQL, EnumColumn, ExpressionOrQueryReturning, ExpressionOutput, FnExpression, FnExpressionArgs, For, ForeignKey, ForeignKeyAction, ForeignKeyMatch, ForeignKeyOptions, From, FromArgs, FromQueryArg, FromQuerySelf, FromResult, GetArg, GetColumnInfo, GetQueryResult, GetResult, GetResultOptional, GetStringArg, GroupArg, Having, HavingItem, HookAction, HookSelect, IdentityColumn, IndexColumnOptions, IndexOptions, InetColumn, InsertQueryData, IntegerBaseColumn, IntegerColumn, IntervalColumn, IsolationLevel, JSONColumn, JSONTextColumn, Join, JoinArgs, JoinCallback, JoinFirstArg, JoinItem, JoinLateralCallback, JoinLateralItem, JoinLateralResult, JoinOverrides, JoinResult, JoinedParsers, JoinedShapes, JsonItem, JsonMethods, JsonModifiers, LimitedTextBaseColumn, LineColumn, LsegColumn, MacAddr8Column, MacAddrColumn, MapTableScopesOption, MergeQuery, MergeQueryMethods, MoneyColumn, MoreThanOneRowError, NoPrimaryKeyOption, NotFoundError, NumberAsStringBaseColumn, NumberBaseColumn, NumberColumnData, OnConflictItem, OnConflictMergeUpdate, OnConflictQueryBuilder, OnQueryBuilder, Operator, Operators, OperatorsAny, OperatorsArray, OperatorsBoolean, OperatorsDate, OperatorsJson, OperatorsNumber, OperatorsText, OperatorsTime, OrCreateArg, OrchidOrmError, OrchidOrmInternalError, OrderArg, OrderArgSelf, OrderArgs, OrderItem, OrderTsQueryConfig, Over, PathColumn, PointColumn, PolygonColumn, Query, QueryAfterHook, QueryArraysResult, QueryBase, QueryBeforeHook, QueryData, QueryDataJoinTo, QueryDefaultReturnData, QueryError, QueryErrorName, QueryGet, QueryGetSelf, QueryHelperResult, QueryHookSelect, QueryHooks, QueryLog, QueryLogObject, QueryLogOptions, QueryLogger, QueryMethods, QueryResult, QueryReturnType, QueryReturnsAll, QueryScopeData, QueryScopes, QuerySourceItem, QueryTransform, QueryTransformFn, QueryUpsertOrCreate, QueryWithComputed, QueryWithTable, RawSQL, RawSqlMethods, RealColumn, RelationConfigBase, RelationConfigDataForCreate, RelationJoinQuery, RelationQuery, RelationQueryBase, RelationsBase, RelationsChain, SearchArg, SearchMethods, SearchWeight, Select, SelectArg, SelectAs, SelectItem, SelectQueryData, SelectSubQueryResult, SelectableBase, SelectableFromShape, SelectableOfType, SelectableOrExpression, SelectableOrExpressionOfType, SerialColumn, SerialColumnData, SetQueryKind, SetQueryReturns, SetQueryReturnsAll, SetQueryReturnsColumn, SetQueryReturnsColumnInfo, SetQueryReturnsColumnOptional, SetQueryReturnsOne, SetQueryReturnsOneOptional, SetQueryReturnsPluck, SetQueryReturnsPluckColumn, SetQueryReturnsRowCount, SetQueryReturnsRows, SetQueryReturnsValue, SetQueryReturnsValueOptional, SetQueryReturnsVoid, SetQueryTableAlias, SetQueryWith, SimpleJoinItem, SingleColumnIndexOptions, SmallIntColumn, SmallSerialColumn, SortDir, StringColumn, TableData, TextBaseColumn, TextColumn, TextColumnData, Then, TimeColumn, TimestampColumn, TimestampTZColumn, ToSQLCtx, ToSQLOptions, ToSQLQuery, Transaction, TransactionAdapter, TransactionOptions, TransactionSelf, TransformMethods, TruncateQueryData, TsQueryColumn, TsVectorColumn, TypeParsers, UUIDColumn, UnhandledTypeError, Union, UnionArg, UnionItem, UnionKind, UnknownColumn, Update, UpdateArg, UpdateCtx, UpdateData, UpdateQueryData, UpdateQueryDataItem, UpdateQueryDataObject, UpdateSelf, UpdatedAtDataInjector, UpsertArg, UpsertResult, UpsertThis, VarCharColumn, VirtualColumn, Where, WhereArg, WhereArgs, WhereInArg, WhereInColumn, WhereInItem, WhereInValues, WhereItem, WhereJsonPathEqualsItem, WhereNotArgs, WhereOnItem, WhereOnJoinItem, WhereQueryBase, WhereQueryBuilder, WhereResult, WhereSearchItem, WhereSearchResult, WindowArg, WindowArgDeclaration, WindowDeclaration, WindowItem, With, WithDataBase, WithDataItem, WithItem, WithOptions, WrapQueryArg, WrapQuerySelf, XMLColumn, _queryAfterSaveCommit, _queryAll, _queryAs, _queryChangeCounter, _queryCreate, _queryCreateFrom, _queryCreateMany, _queryCreateManyFrom, _queryCreateManyRaw, _queryCreateRaw, _queryDefaults, _queryDelete, _queryExec, _queryFindBy, _queryFindByOptional, _queryGet, _queryGetOptional, _queryHookAfterCreate, _queryHookAfterCreateCommit, _queryHookAfterDelete, _queryHookAfterDeleteCommit, _queryHookAfterQuery, _queryHookAfterSave, _queryHookAfterUpdate, _queryHookAfterUpdateCommit, _queryHookBeforeCreate, _queryHookBeforeDelete, _queryHookBeforeQuery, _queryHookBeforeSave, _queryHookBeforeUpdate, _queryInsert, _queryInsertFrom, _queryInsertMany, _queryInsertManyFrom, _queryInsertManyRaw, _queryInsertRaw, _queryJoinOn, _queryJoinOnJsonPathEquals, _queryJoinOrOn, _queryOr, _queryOrNot, _queryRows, _querySelect, _queryTake, _queryTakeOptional, _queryUpdate, _queryUpdateOrThrow, _queryUpdateRaw, _queryWhere, _queryWhereIn, _queryWhereNot, addComputedColumns, addParserForRawExpression, addParserForSelectItem, addQueryOn, anyShape, checkIfASimpleQuery, cloneQuery, columnCheckToCode, columnCode, columnForeignKeysToCode, columnIndexesToCode, columnsShapeToCode, constraintPropsToCode, constraintToCode, copyTableData, countSelect, createDb, defaultSchemaConfig, extendQuery, foreignKeyArgumentToCode, getClonedQueryData, getColumnInfo, getColumnTypes, getConstraintKind, getQueryAs, getShapeFromSelect, getTableData, handleResult, identityToCode, indexToCode, instantiateColumn, isQueryReturnsAll, isSelectingCount, joinSubQuery, logColors, logParamToLogObject, makeColumnTypes, makeColumnsByType, makeExpression, makeFnExpression, makeRegexToFindInSql, makeSQL, newTableData, parseRecord, parseResult, primaryKeyToCode, processSelectArg, pushLimitSQL, pushQueryArray, pushQueryOn, pushQueryOrOn, pushQueryValue, queryFrom, queryJson, queryMethodByReturnType, queryTypeWithLimitOne, queryWrap, quote, quoteString, raw, referencesArgsToCode, resetTableData, resolveSubQueryCallback, saveSearchAlias, setParserForSelectedString, setQueryObjectValue, setQueryOperators, simplifyColumnDefault, templateLiteralToSQL, testTransaction, throwIfNoWhere, toSQL, toSQLCacheKey };

package/dist/index.js

@@ -4521,27 +4521,26 @@ class Adapter {
       this.schema = schema === "public" ? void 0 : schema;
     this.config = config;
     this.pool = new pg.Pool(config);
+    if (config.connectRetry) {
+      orchidCore.setAdapterConnectRetry(
+        this,
+        () => this.pool.connect(),
+        config.connectRetry === true ? orchidCore.emptyObject : config.connectRetry
+      );
+    }
+  }
+  connect() {
+    return this.pool.connect();
   }
   query(query, types2) {
-    return performQuery$1(
-      this.pool,
-      query,
-      types2,
-      this.schema
-    );
+    return performQuery$1(this, query, types2);
   }
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   arrays(query, types2) {
-    return performQuery$1(
-      this.pool,
-      query,
-      types2,
-      this.schema,
-      "array"
-    );
+    return performQuery$1(this, query, types2, "array");
   }
   async transaction(begin, cb) {
-    const client = await this.pool.connect();
+    const client = await this.connect();
     try {
       await setSearchPath(client, this.schema);
       await performQueryOnClient(client, begin, this.types);
@@ -4576,10 +4575,10 @@ const setSearchPath = (client, schema) => {
   }
   return;
 };
-const performQuery$1 = async (pool, query, types2, schema, rowMode) => {
-  const client = await pool.connect();
+const performQuery$1 = async (adapter, query, types2, rowMode) => {
+  const client = await adapter.connect();
   try {
-    await setSearchPath(client, schema);
+    await setSearchPath(client, adapter.schema);
     return await performQueryOnClient(client, query, types2, rowMode);
   } finally {
     client.release();
@@ -4606,6 +4605,9 @@ class TransactionAdapter {
     this.pool = adapter.pool;
     this.config = adapter.config;
   }
+  connect() {
+    return Promise.resolve(this.client);
+  }
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   async query(query, types2) {
     return await performQueryOnClient(this.client, query, types2);