UNPKG

drizzle-orm

Version:

Drizzle ORM package for SQL databases

233 lines • 7.61 kB
"use strict"; var __defProp = Object.defineProperty; var __getOwnPropDesc = Object.getOwnPropertyDescriptor; var __getOwnPropNames = Object.getOwnPropertyNames; var __hasOwnProp = Object.prototype.hasOwnProperty; var __export = (target, all) => { for (var name in all) __defProp(target, name, { get: all[name], enumerable: true }); }; var __copyProps = (to, from, except, desc) => { if (from && typeof from === "object" || typeof from === "function") { for (let key of __getOwnPropNames(from)) if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable }); } return to; }; var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod); var session_exports = {}; __export(session_exports, { ExecuteResultSync: () => ExecuteResultSync, SQLitePreparedQuery: () => SQLitePreparedQuery, SQLiteSession: () => SQLiteSession, SQLiteTransaction: () => SQLiteTransaction }); module.exports = __toCommonJS(session_exports); var import_cache = require("../cache/core/cache.cjs"); var import_entity = require("../entity.cjs"); var import_errors = require("../errors.cjs"); var import_query_promise = require("../query-promise.cjs"); var import_db = require("./db.cjs"); class ExecuteResultSync extends import_query_promise.QueryPromise { constructor(resultCb) { super(); this.resultCb = resultCb; } static [import_entity.entityKind] = "ExecuteResultSync"; async execute() { return this.resultCb(); } sync() { return this.resultCb(); } } class SQLitePreparedQuery { constructor(mode, executeMethod, query, cache, queryMetadata, cacheConfig) { this.mode = mode; this.executeMethod = executeMethod; this.query = query; this.cache = cache; this.queryMetadata = queryMetadata; this.cacheConfig = cacheConfig; if (cache && cache.strategy() === "all" && cacheConfig === void 0) { this.cacheConfig = { enable: true, autoInvalidate: true }; } if (!this.cacheConfig?.enable) { this.cacheConfig = void 0; } } static [import_entity.entityKind] = "PreparedQuery"; /** @internal */ joinsNotNullableMap; /** @internal */ async queryWithCache(queryString, params, query) { if (this.cache === void 0 || (0, import_entity.is)(this.cache, import_cache.NoopCache) || this.queryMetadata === void 0) { try { return await query(); } catch (e) { throw new import_errors.DrizzleQueryError(queryString, params, e); } } if (this.cacheConfig && !this.cacheConfig.enable) { try { return await query(); } catch (e) { throw new import_errors.DrizzleQueryError(queryString, params, e); } } if ((this.queryMetadata.type === "insert" || this.queryMetadata.type === "update" || this.queryMetadata.type === "delete") && this.queryMetadata.tables.length > 0) { try { const [res] = await Promise.all([ query(), this.cache.onMutate({ tables: this.queryMetadata.tables }) ]); return res; } catch (e) { throw new import_errors.DrizzleQueryError(queryString, params, e); } } if (!this.cacheConfig) { try { return await query(); } catch (e) { throw new import_errors.DrizzleQueryError(queryString, params, e); } } if (this.queryMetadata.type === "select") { const fromCache = await this.cache.get( this.cacheConfig.tag ?? (await (0, import_cache.hashQuery)(queryString, params)), this.queryMetadata.tables, this.cacheConfig.tag !== void 0, this.cacheConfig.autoInvalidate ); if (fromCache === void 0) { let result; try { result = await query(); } catch (e) { throw new import_errors.DrizzleQueryError(queryString, params, e); } await this.cache.put( this.cacheConfig.tag ?? (await (0, import_cache.hashQuery)(queryString, params)), result, // make sure we send tables that were used in a query only if user wants to invalidate it on each write this.cacheConfig.autoInvalidate ? this.queryMetadata.tables : [], this.cacheConfig.tag !== void 0, this.cacheConfig.config ); return result; } return fromCache; } try { return await query(); } catch (e) { throw new import_errors.DrizzleQueryError(queryString, params, e); } } getQuery() { return this.query; } mapRunResult(result, _isFromBatch) { return result; } mapAllResult(_result, _isFromBatch) { throw new Error("Not implemented"); } mapGetResult(_result, _isFromBatch) { throw new Error("Not implemented"); } execute(placeholderValues) { if (this.mode === "async") { return this[this.executeMethod](placeholderValues); } return new ExecuteResultSync(() => this[this.executeMethod](placeholderValues)); } mapResult(response, isFromBatch) { switch (this.executeMethod) { case "run": { return this.mapRunResult(response, isFromBatch); } case "all": { return this.mapAllResult(response, isFromBatch); } case "get": { return this.mapGetResult(response, isFromBatch); } } } } class SQLiteSession { constructor(dialect) { this.dialect = dialect; } static [import_entity.entityKind] = "SQLiteSession"; prepareOneTimeQuery(query, fields, executeMethod, isResponseInArrayMode, customResultMapper, queryMetadata, cacheConfig) { return this.prepareQuery( query, fields, executeMethod, isResponseInArrayMode, customResultMapper, queryMetadata, cacheConfig ); } run(query) { const staticQuery = this.dialect.sqlToQuery(query); try { return this.prepareOneTimeQuery(staticQuery, void 0, "run", false).run(); } catch (err) { throw new import_errors.DrizzleError({ cause: err, message: `Failed to run the query '${staticQuery.sql}'` }); } } /** @internal */ extractRawRunValueFromBatchResult(result) { return result; } all(query) { return this.prepareOneTimeQuery(this.dialect.sqlToQuery(query), void 0, "run", false).all(); } /** @internal */ extractRawAllValueFromBatchResult(_result) { throw new Error("Not implemented"); } get(query) { return this.prepareOneTimeQuery(this.dialect.sqlToQuery(query), void 0, "run", false).get(); } /** @internal */ extractRawGetValueFromBatchResult(_result) { throw new Error("Not implemented"); } values(query) { return this.prepareOneTimeQuery(this.dialect.sqlToQuery(query), void 0, "run", false).values(); } async count(sql) { const result = await this.values(sql); return result[0][0]; } /** @internal */ extractRawValuesValueFromBatchResult(_result) { throw new Error("Not implemented"); } } class SQLiteTransaction extends import_db.BaseSQLiteDatabase { constructor(resultType, dialect, session, schema, nestedIndex = 0) { super(resultType, dialect, session, schema); this.schema = schema; this.nestedIndex = nestedIndex; } static [import_entity.entityKind] = "SQLiteTransaction"; rollback() { throw new import_errors.TransactionRollbackError(); } } // Annotate the CommonJS export names for ESM import in node: 0 && (module.exports = { ExecuteResultSync, SQLitePreparedQuery, SQLiteSession, SQLiteTransaction }); //# sourceMappingURL=session.cjs.map