session.d.cts 4.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. import type { Client, Connection, ExecutedQuery, Transaction } from '@planetscale/database';
  2. import { type Cache } from "../cache/core/index.cjs";
  3. import type { WithCacheConfig } from "../cache/core/types.cjs";
  4. import { entityKind } from "../entity.cjs";
  5. import type { Logger } from "../logger.cjs";
  6. import type { MySqlDialect } from "../mysql-core/dialect.cjs";
  7. import type { SelectedFieldsOrdered } from "../mysql-core/query-builders/select.types.cjs";
  8. import { MySqlPreparedQuery, type MySqlPreparedQueryConfig, type MySqlPreparedQueryHKT, type MySqlQueryResultHKT, MySqlSession, MySqlTransaction } from "../mysql-core/session.cjs";
  9. import type { RelationalSchemaConfig, TablesRelationalConfig } from "../relations.cjs";
  10. import { type Query, type SQL } from "../sql/sql.cjs";
  11. import { type Assume } from "../utils.cjs";
  12. export declare class PlanetScalePreparedQuery<T extends MySqlPreparedQueryConfig> extends MySqlPreparedQuery<T> {
  13. private client;
  14. private queryString;
  15. private params;
  16. private logger;
  17. private fields;
  18. private customResultMapper?;
  19. private generatedIds?;
  20. private returningIds?;
  21. static readonly [entityKind]: string;
  22. private rawQuery;
  23. private query;
  24. constructor(client: Client | Transaction | Connection, queryString: string, params: unknown[], logger: Logger, cache: Cache, queryMetadata: {
  25. type: 'select' | 'update' | 'delete' | 'insert';
  26. tables: string[];
  27. } | undefined, cacheConfig: WithCacheConfig | undefined, fields: SelectedFieldsOrdered | undefined, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined, generatedIds?: Record<string, unknown>[] | undefined, returningIds?: SelectedFieldsOrdered | undefined);
  28. execute(placeholderValues?: Record<string, unknown> | undefined): Promise<T['execute']>;
  29. iterator(_placeholderValues?: Record<string, unknown>): AsyncGenerator<T['iterator']>;
  30. }
  31. export interface PlanetscaleSessionOptions {
  32. logger?: Logger;
  33. cache?: Cache;
  34. }
  35. export declare class PlanetscaleSession<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends MySqlSession<MySqlQueryResultHKT, PlanetScalePreparedQueryHKT, TFullSchema, TSchema> {
  36. private baseClient;
  37. private schema;
  38. private options;
  39. static readonly [entityKind]: string;
  40. private logger;
  41. private client;
  42. private cache;
  43. constructor(baseClient: Client | Connection, dialect: MySqlDialect, tx: Transaction | undefined, schema: RelationalSchemaConfig<TSchema> | undefined, options?: PlanetscaleSessionOptions);
  44. prepareQuery<T extends MySqlPreparedQueryConfig = MySqlPreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, customResultMapper?: (rows: unknown[][]) => T['execute'], generatedIds?: Record<string, unknown>[], returningIds?: SelectedFieldsOrdered, queryMetadata?: {
  45. type: 'select' | 'update' | 'delete' | 'insert';
  46. tables: string[];
  47. }, cacheConfig?: WithCacheConfig): MySqlPreparedQuery<T>;
  48. query(query: string, params: unknown[]): Promise<ExecutedQuery>;
  49. queryObjects(query: string, params: unknown[]): Promise<ExecutedQuery>;
  50. all<T = unknown>(query: SQL): Promise<T[]>;
  51. count(sql: SQL): Promise<number>;
  52. transaction<T>(transaction: (tx: PlanetScaleTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
  53. }
  54. export declare class PlanetScaleTransaction<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends MySqlTransaction<PlanetscaleQueryResultHKT, PlanetScalePreparedQueryHKT, TFullSchema, TSchema> {
  55. static readonly [entityKind]: string;
  56. constructor(dialect: MySqlDialect, session: MySqlSession, schema: RelationalSchemaConfig<TSchema> | undefined, nestedIndex?: number);
  57. transaction<T>(transaction: (tx: PlanetScaleTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
  58. }
  59. export interface PlanetscaleQueryResultHKT extends MySqlQueryResultHKT {
  60. type: ExecutedQuery;
  61. }
  62. export interface PlanetScalePreparedQueryHKT extends MySqlPreparedQueryHKT {
  63. type: PlanetScalePreparedQuery<Assume<this['config'], MySqlPreparedQueryConfig>>;
  64. }