drizzle-orm
Version:
Drizzle ORM package for SQL databases
196 lines • 6.3 kB
JavaScript
"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, {
PgPreparedQuery: () => PgPreparedQuery,
PgSession: () => PgSession,
PgTransaction: () => PgTransaction
});
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_sql = require("../sql/index.cjs");
var import_tracing = require("../tracing.cjs");
var import_db = require("./db.cjs");
class PgPreparedQuery {
constructor(query, cache, queryMetadata, cacheConfig) {
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;
}
}
authToken;
getQuery() {
return this.query;
}
mapResult(response, _isFromBatch) {
return response;
}
/** @internal */
setToken(token) {
this.authToken = token;
return this;
}
static [import_entity.entityKind] = "PgPreparedQuery";
/** @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);
}
}
}
class PgSession {
constructor(dialect) {
this.dialect = dialect;
}
static [import_entity.entityKind] = "PgSession";
/** @internal */
execute(query, token) {
return import_tracing.tracer.startActiveSpan("drizzle.operation", () => {
const prepared = import_tracing.tracer.startActiveSpan("drizzle.prepareQuery", () => {
return this.prepareQuery(
this.dialect.sqlToQuery(query),
void 0,
void 0,
false
);
});
return prepared.setToken(token).execute(void 0, token);
});
}
all(query) {
return this.prepareQuery(
this.dialect.sqlToQuery(query),
void 0,
void 0,
false
).all();
}
/** @internal */
async count(sql2, token) {
const res = await this.execute(sql2, token);
return Number(
res[0]["count"]
);
}
}
class PgTransaction extends import_db.PgDatabase {
constructor(dialect, session, schema, nestedIndex = 0) {
super(dialect, session, schema);
this.schema = schema;
this.nestedIndex = nestedIndex;
}
static [import_entity.entityKind] = "PgTransaction";
rollback() {
throw new import_errors.TransactionRollbackError();
}
/** @internal */
getTransactionConfigSQL(config) {
const chunks = [];
if (config.isolationLevel) {
chunks.push(`isolation level ${config.isolationLevel}`);
}
if (config.accessMode) {
chunks.push(config.accessMode);
}
if (typeof config.deferrable === "boolean") {
chunks.push(config.deferrable ? "deferrable" : "not deferrable");
}
return import_sql.sql.raw(chunks.join(" "));
}
setTransaction(config) {
return this.session.execute(import_sql.sql`set transaction ${this.getTransactionConfigSQL(config)}`);
}
}
// Annotate the CommonJS export names for ESM import in node:
0 && (module.exports = {
PgPreparedQuery,
PgSession,
PgTransaction
});
//# sourceMappingURL=session.cjs.map