243 lines
8.6 KiB
JavaScript
243 lines
8.6 KiB
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, {
|
|
LibSQLPreparedQuery: () => LibSQLPreparedQuery,
|
|
LibSQLSession: () => LibSQLSession,
|
|
LibSQLTransaction: () => LibSQLTransaction
|
|
});
|
|
module.exports = __toCommonJS(session_exports);
|
|
var import_entity = require("../entity.cjs");
|
|
var import_logger = require("../logger.cjs");
|
|
var import_sql = require("../sql/sql.cjs");
|
|
var import_sqlite_core = require("../sqlite-core/index.cjs");
|
|
var import_session = require("../sqlite-core/session.cjs");
|
|
var import_utils = require("../utils.cjs");
|
|
class LibSQLSession extends import_session.SQLiteSession {
|
|
constructor(client, dialect, schema, options, tx) {
|
|
super(dialect);
|
|
this.client = client;
|
|
this.schema = schema;
|
|
this.options = options;
|
|
this.tx = tx;
|
|
this.logger = options.logger ?? new import_logger.NoopLogger();
|
|
}
|
|
static [import_entity.entityKind] = "LibSQLSession";
|
|
logger;
|
|
prepareQuery(query, fields, executeMethod, isResponseInArrayMode, customResultMapper) {
|
|
return new LibSQLPreparedQuery(
|
|
this.client,
|
|
query,
|
|
this.logger,
|
|
fields,
|
|
this.tx,
|
|
executeMethod,
|
|
isResponseInArrayMode,
|
|
customResultMapper
|
|
);
|
|
}
|
|
async batch(queries) {
|
|
const preparedQueries = [];
|
|
const builtQueries = [];
|
|
for (const query of queries) {
|
|
const preparedQuery = query._prepare();
|
|
const builtQuery = preparedQuery.getQuery();
|
|
preparedQueries.push(preparedQuery);
|
|
builtQueries.push({ sql: builtQuery.sql, args: builtQuery.params });
|
|
}
|
|
const batchResults = await this.client.batch(builtQueries);
|
|
return batchResults.map((result, i) => preparedQueries[i].mapResult(result, true));
|
|
}
|
|
async migrate(queries) {
|
|
const preparedQueries = [];
|
|
const builtQueries = [];
|
|
for (const query of queries) {
|
|
const preparedQuery = query._prepare();
|
|
const builtQuery = preparedQuery.getQuery();
|
|
preparedQueries.push(preparedQuery);
|
|
builtQueries.push({ sql: builtQuery.sql, args: builtQuery.params });
|
|
}
|
|
const batchResults = await this.client.migrate(builtQueries);
|
|
return batchResults.map((result, i) => preparedQueries[i].mapResult(result, true));
|
|
}
|
|
async transaction(transaction, _config) {
|
|
const libsqlTx = await this.client.transaction();
|
|
const session = new LibSQLSession(
|
|
this.client,
|
|
this.dialect,
|
|
this.schema,
|
|
this.options,
|
|
libsqlTx
|
|
);
|
|
const tx = new LibSQLTransaction("async", this.dialect, session, this.schema);
|
|
try {
|
|
const result = await transaction(tx);
|
|
await libsqlTx.commit();
|
|
return result;
|
|
} catch (err) {
|
|
await libsqlTx.rollback();
|
|
throw err;
|
|
}
|
|
}
|
|
extractRawAllValueFromBatchResult(result) {
|
|
return result.rows;
|
|
}
|
|
extractRawGetValueFromBatchResult(result) {
|
|
return result.rows[0];
|
|
}
|
|
extractRawValuesValueFromBatchResult(result) {
|
|
return result.rows;
|
|
}
|
|
}
|
|
class LibSQLTransaction extends import_sqlite_core.SQLiteTransaction {
|
|
static [import_entity.entityKind] = "LibSQLTransaction";
|
|
async transaction(transaction) {
|
|
const savepointName = `sp${this.nestedIndex}`;
|
|
const tx = new LibSQLTransaction("async", this.dialect, this.session, this.schema, this.nestedIndex + 1);
|
|
await this.session.run(import_sql.sql.raw(`savepoint ${savepointName}`));
|
|
try {
|
|
const result = await transaction(tx);
|
|
await this.session.run(import_sql.sql.raw(`release savepoint ${savepointName}`));
|
|
return result;
|
|
} catch (err) {
|
|
await this.session.run(import_sql.sql.raw(`rollback to savepoint ${savepointName}`));
|
|
throw err;
|
|
}
|
|
}
|
|
}
|
|
class LibSQLPreparedQuery extends import_session.SQLitePreparedQuery {
|
|
constructor(client, query, logger, fields, tx, executeMethod, _isResponseInArrayMode, customResultMapper) {
|
|
super("async", executeMethod, query);
|
|
this.client = client;
|
|
this.logger = logger;
|
|
this.fields = fields;
|
|
this.tx = tx;
|
|
this._isResponseInArrayMode = _isResponseInArrayMode;
|
|
this.customResultMapper = customResultMapper;
|
|
this.customResultMapper = customResultMapper;
|
|
this.fields = fields;
|
|
}
|
|
static [import_entity.entityKind] = "LibSQLPreparedQuery";
|
|
run(placeholderValues) {
|
|
const params = (0, import_sql.fillPlaceholders)(this.query.params, placeholderValues ?? {});
|
|
this.logger.logQuery(this.query.sql, params);
|
|
const stmt = { sql: this.query.sql, args: params };
|
|
return this.tx ? this.tx.execute(stmt) : this.client.execute(stmt);
|
|
}
|
|
async all(placeholderValues) {
|
|
const { fields, logger, query, tx, client, customResultMapper } = this;
|
|
if (!fields && !customResultMapper) {
|
|
const params = (0, import_sql.fillPlaceholders)(query.params, placeholderValues ?? {});
|
|
logger.logQuery(query.sql, params);
|
|
const stmt = { sql: query.sql, args: params };
|
|
return (tx ? tx.execute(stmt) : client.execute(stmt)).then(({ rows: rows2 }) => this.mapAllResult(rows2));
|
|
}
|
|
const rows = await this.values(placeholderValues);
|
|
return this.mapAllResult(rows);
|
|
}
|
|
mapAllResult(rows, isFromBatch) {
|
|
if (isFromBatch) {
|
|
rows = rows.rows;
|
|
}
|
|
if (!this.fields && !this.customResultMapper) {
|
|
return rows.map((row) => normalizeRow(row));
|
|
}
|
|
if (this.customResultMapper) {
|
|
return this.customResultMapper(rows, normalizeFieldValue);
|
|
}
|
|
return rows.map((row) => {
|
|
return (0, import_utils.mapResultRow)(
|
|
this.fields,
|
|
Array.prototype.slice.call(row).map((v) => normalizeFieldValue(v)),
|
|
this.joinsNotNullableMap
|
|
);
|
|
});
|
|
}
|
|
async get(placeholderValues) {
|
|
const { fields, logger, query, tx, client, customResultMapper } = this;
|
|
if (!fields && !customResultMapper) {
|
|
const params = (0, import_sql.fillPlaceholders)(query.params, placeholderValues ?? {});
|
|
logger.logQuery(query.sql, params);
|
|
const stmt = { sql: query.sql, args: params };
|
|
return (tx ? tx.execute(stmt) : client.execute(stmt)).then(({ rows: rows2 }) => this.mapGetResult(rows2));
|
|
}
|
|
const rows = await this.values(placeholderValues);
|
|
return this.mapGetResult(rows);
|
|
}
|
|
mapGetResult(rows, isFromBatch) {
|
|
if (isFromBatch) {
|
|
rows = rows.rows;
|
|
}
|
|
const row = rows[0];
|
|
if (!this.fields && !this.customResultMapper) {
|
|
return normalizeRow(row);
|
|
}
|
|
if (!row) {
|
|
return void 0;
|
|
}
|
|
if (this.customResultMapper) {
|
|
return this.customResultMapper(rows, normalizeFieldValue);
|
|
}
|
|
return (0, import_utils.mapResultRow)(
|
|
this.fields,
|
|
Array.prototype.slice.call(row).map((v) => normalizeFieldValue(v)),
|
|
this.joinsNotNullableMap
|
|
);
|
|
}
|
|
values(placeholderValues) {
|
|
const params = (0, import_sql.fillPlaceholders)(this.query.params, placeholderValues ?? {});
|
|
this.logger.logQuery(this.query.sql, params);
|
|
const stmt = { sql: this.query.sql, args: params };
|
|
return (this.tx ? this.tx.execute(stmt) : this.client.execute(stmt)).then(({ rows }) => rows);
|
|
}
|
|
/** @internal */
|
|
isResponseInArrayMode() {
|
|
return this._isResponseInArrayMode;
|
|
}
|
|
}
|
|
function normalizeRow(obj) {
|
|
return Object.keys(obj).reduce((acc, key) => {
|
|
if (Object.prototype.propertyIsEnumerable.call(obj, key)) {
|
|
acc[key] = obj[key];
|
|
}
|
|
return acc;
|
|
}, {});
|
|
}
|
|
function normalizeFieldValue(value) {
|
|
if (typeof ArrayBuffer !== "undefined" && value instanceof ArrayBuffer) {
|
|
if (typeof Buffer !== "undefined") {
|
|
if (!(value instanceof Buffer)) {
|
|
return Buffer.from(value);
|
|
}
|
|
return value;
|
|
}
|
|
if (typeof TextDecoder !== "undefined") {
|
|
return new TextDecoder().decode(value);
|
|
}
|
|
throw new Error("TextDecoder is not available. Please provide either Buffer or TextDecoder polyfill.");
|
|
}
|
|
return value;
|
|
}
|
|
// Annotate the CommonJS export names for ESM import in node:
|
|
0 && (module.exports = {
|
|
LibSQLPreparedQuery,
|
|
LibSQLSession,
|
|
LibSQLTransaction
|
|
});
|
|
//# sourceMappingURL=session.cjs.map
|