driver.js 1.3 KB

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