Spaces:
Sleeping
Sleeping
File size: 6,138 Bytes
6491ad4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 | "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, {
GelDbPreparedQuery: () => GelDbPreparedQuery,
GelDbSession: () => GelDbSession,
GelDbTransaction: () => GelDbTransaction
});
module.exports = __toCommonJS(session_exports);
var import_core = require("../cache/core/index.cjs");
var import_entity = require("../entity.cjs");
var import_session = require("../gel-core/session.cjs");
var import_logger = require("../logger.cjs");
var import_sql = require("../sql/sql.cjs");
var import_tracing = require("../tracing.cjs");
var import_utils = require("../utils.cjs");
class GelDbPreparedQuery extends import_session.GelPreparedQuery {
constructor(client, queryString, params, logger, cache, queryMetadata, cacheConfig, fields, _isResponseInArrayMode, customResultMapper, transaction = false) {
super({ sql: queryString, params }, cache, queryMetadata, cacheConfig);
this.client = client;
this.queryString = queryString;
this.params = params;
this.logger = logger;
this.fields = fields;
this._isResponseInArrayMode = _isResponseInArrayMode;
this.customResultMapper = customResultMapper;
this.transaction = transaction;
}
static [import_entity.entityKind] = "GelPreparedQuery";
async execute(placeholderValues = {}) {
return import_tracing.tracer.startActiveSpan("drizzle.execute", async () => {
const params = (0, import_sql.fillPlaceholders)(this.params, placeholderValues);
this.logger.logQuery(this.queryString, params);
const { fields, queryString: query, client, joinsNotNullableMap, customResultMapper } = this;
if (!fields && !customResultMapper) {
return import_tracing.tracer.startActiveSpan("drizzle.driver.execute", async (span) => {
span?.setAttributes({
"drizzle.query.text": query,
"drizzle.query.params": JSON.stringify(params)
});
return await this.queryWithCache(query, params, async () => {
return await client.querySQL(query, params.length ? params : void 0);
});
});
}
const result = await import_tracing.tracer.startActiveSpan("drizzle.driver.execute", async (span) => {
span?.setAttributes({
"drizzle.query.text": query,
"drizzle.query.params": JSON.stringify(params)
});
return await this.queryWithCache(query, params, async () => {
return await client.withSQLRowMode("array").querySQL(query, params.length ? params : void 0);
});
});
return import_tracing.tracer.startActiveSpan("drizzle.mapResponse", () => {
return customResultMapper ? customResultMapper(result) : result.map((row) => (0, import_utils.mapResultRow)(fields, row, joinsNotNullableMap));
});
});
}
async all(placeholderValues = {}) {
return await import_tracing.tracer.startActiveSpan("drizzle.execute", async () => {
const params = (0, import_sql.fillPlaceholders)(this.params, placeholderValues);
this.logger.logQuery(this.queryString, params);
return await import_tracing.tracer.startActiveSpan("drizzle.driver.execute", async (span) => {
span?.setAttributes({
"drizzle.query.text": this.queryString,
"drizzle.query.params": JSON.stringify(params)
});
return await this.queryWithCache(this.queryString, params, async () => {
return await this.client.withSQLRowMode("array").querySQL(
this.queryString,
params.length ? params : void 0
).then((result) => result);
});
});
});
}
/** @internal */
isResponseInArrayMode() {
return this._isResponseInArrayMode;
}
}
class GelDbSession extends import_session.GelSession {
constructor(client, dialect, schema, options = {}) {
super(dialect);
this.client = client;
this.schema = schema;
this.options = options;
this.logger = options.logger ?? new import_logger.NoopLogger();
this.cache = options.cache ?? new import_core.NoopCache();
}
static [import_entity.entityKind] = "GelDbSession";
logger;
cache;
prepareQuery(query, fields, name, isResponseInArrayMode, customResultMapper, queryMetadata, cacheConfig) {
return new GelDbPreparedQuery(
this.client,
query.sql,
query.params,
this.logger,
this.cache,
queryMetadata,
cacheConfig,
fields,
isResponseInArrayMode,
customResultMapper
);
}
async transaction(transaction) {
return await this.client.transaction(async (clientTx) => {
const session = new GelDbSession(clientTx, this.dialect, this.schema, this.options);
const tx = new GelDbTransaction(this.dialect, session, this.schema);
return await transaction(tx);
});
}
async count(sql) {
const res = await this.execute(sql);
return Number(res[0]["count"]);
}
}
class GelDbTransaction extends import_session.GelTransaction {
static [import_entity.entityKind] = "GelDbTransaction";
async transaction(transaction) {
const tx = new GelDbTransaction(
this.dialect,
this.session,
this.schema
);
return await transaction(tx);
}
}
// Annotate the CommonJS export names for ESM import in node:
0 && (module.exports = {
GelDbPreparedQuery,
GelDbSession,
GelDbTransaction
});
//# sourceMappingURL=session.cjs.map |