driver.d.ts 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. import { type Connection as CallbackConnection, type Pool as CallbackPool, type PoolOptions } from 'mysql2';
  2. import type { Connection, Pool } from 'mysql2/promise';
  3. import type { Cache } from "../cache/core/index.js";
  4. import { entityKind } from "../entity.js";
  5. import type { Logger } from "../logger.js";
  6. import { MySqlDatabase } from "../mysql-core/db.js";
  7. import { MySqlDialect } from "../mysql-core/dialect.js";
  8. import type { Mode } from "../mysql-core/session.js";
  9. import { type RelationalSchemaConfig, type TablesRelationalConfig } from "../relations.js";
  10. import { type DrizzleConfig } from "../utils.js";
  11. import type { MySql2Client, MySql2PreparedQueryHKT, MySql2QueryResultHKT } from "./session.js";
  12. import { MySql2Session } from "./session.js";
  13. export interface MySqlDriverOptions {
  14. logger?: Logger;
  15. cache?: Cache;
  16. }
  17. export declare class MySql2Driver {
  18. private client;
  19. private dialect;
  20. private options;
  21. static readonly [entityKind]: string;
  22. constructor(client: MySql2Client, dialect: MySqlDialect, options?: MySqlDriverOptions);
  23. createSession(schema: RelationalSchemaConfig<TablesRelationalConfig> | undefined, mode: Mode): MySql2Session<Record<string, unknown>, TablesRelationalConfig>;
  24. }
  25. export { MySqlDatabase } from "../mysql-core/db.js";
  26. export declare class MySql2Database<TSchema extends Record<string, unknown> = Record<string, never>> extends MySqlDatabase<MySql2QueryResultHKT, MySql2PreparedQueryHKT, TSchema> {
  27. static readonly [entityKind]: string;
  28. }
  29. export type MySql2DrizzleConfig<TSchema extends Record<string, unknown> = Record<string, never>> = Omit<DrizzleConfig<TSchema>, 'schema'> & ({
  30. schema: TSchema;
  31. mode: Mode;
  32. } | {
  33. schema?: undefined;
  34. mode?: Mode;
  35. });
  36. export type AnyMySql2Connection = Pool | Connection | CallbackPool | CallbackConnection;
  37. export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>, TClient extends AnyMySql2Connection = CallbackPool>(...params: [
  38. TClient | string
  39. ] | [
  40. TClient | string,
  41. MySql2DrizzleConfig<TSchema>
  42. ] | [
  43. (MySql2DrizzleConfig<TSchema> & ({
  44. connection: string | PoolOptions;
  45. } | {
  46. client: TClient;
  47. }))
  48. ]): MySql2Database<TSchema> & {
  49. $client: AnyMySql2Connection extends TClient ? CallbackPool : TClient;
  50. };
  51. export declare namespace drizzle {
  52. function mock<TSchema extends Record<string, unknown> = Record<string, never>>(config?: MySql2DrizzleConfig<TSchema>): MySql2Database<TSchema> & {
  53. $client: '$client is not available on drizzle.mock()';
  54. };
  55. }