driver.js 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. import { entityKind } from "../entity.js";
  2. import { DefaultLogger } from "../logger.js";
  3. import { PgDatabase } from "../pg-core/db.js";
  4. import { PgDialect } from "../pg-core/dialect.js";
  5. import {
  6. createTableRelationsHelpers,
  7. extractTablesRelationalConfig
  8. } from "../relations.js";
  9. import { PgRemoteSession } from "./session.js";
  10. class PgRemoteDatabase extends PgDatabase {
  11. static [entityKind] = "PgRemoteDatabase";
  12. }
  13. function drizzle(callback, config = {}, _dialect = () => new PgDialect({ casing: config.casing })) {
  14. const dialect = _dialect();
  15. let logger;
  16. if (config.logger === true) {
  17. logger = new DefaultLogger();
  18. } else if (config.logger !== false) {
  19. logger = config.logger;
  20. }
  21. let schema;
  22. if (config.schema) {
  23. const tablesConfig = extractTablesRelationalConfig(
  24. config.schema,
  25. createTableRelationsHelpers
  26. );
  27. schema = {
  28. fullSchema: config.schema,
  29. schema: tablesConfig.tables,
  30. tableNamesMap: tablesConfig.tableNamesMap
  31. };
  32. }
  33. const session = new PgRemoteSession(callback, dialect, schema, { logger, cache: config.cache });
  34. const db = new PgRemoteDatabase(dialect, session, schema);
  35. db.$cache = config.cache;
  36. if (db.$cache) {
  37. db.$cache["invalidate"] = config.cache?.onMutate;
  38. }
  39. return db;
  40. }
  41. export {
  42. PgRemoteDatabase,
  43. drizzle
  44. };
  45. //# sourceMappingURL=driver.js.map