Spaces:
Sleeping
Sleeping
File size: 4,997 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 | import { entityKind } from "../entity.js";
import { NoopLogger } from "../logger.js";
import { fillPlaceholders, sql } from "../sql/sql.js";
import { SQLiteTransaction } from "../sqlite-core/index.js";
import { SQLitePreparedQuery as PreparedQueryBase, SQLiteSession } from "../sqlite-core/session.js";
import { mapResultRow } from "../utils.js";
class SQLJsSession extends SQLiteSession {
constructor(client, dialect, schema, options = {}) {
super(dialect);
this.client = client;
this.schema = schema;
this.logger = options.logger ?? new NoopLogger();
}
static [entityKind] = "SQLJsSession";
logger;
prepareQuery(query, fields, executeMethod, isResponseInArrayMode) {
return new PreparedQuery(this.client, query, this.logger, fields, executeMethod, isResponseInArrayMode);
}
transaction(transaction, config = {}) {
const tx = new SQLJsTransaction("sync", this.dialect, this, this.schema);
this.run(sql.raw(`begin${config.behavior ? ` ${config.behavior}` : ""}`));
try {
const result = transaction(tx);
this.run(sql`commit`);
return result;
} catch (err) {
this.run(sql`rollback`);
throw err;
}
}
}
class SQLJsTransaction extends SQLiteTransaction {
static [entityKind] = "SQLJsTransaction";
transaction(transaction) {
const savepointName = `sp${this.nestedIndex + 1}`;
const tx = new SQLJsTransaction("sync", this.dialect, this.session, this.schema, this.nestedIndex + 1);
tx.run(sql.raw(`savepoint ${savepointName}`));
try {
const result = transaction(tx);
tx.run(sql.raw(`release savepoint ${savepointName}`));
return result;
} catch (err) {
tx.run(sql.raw(`rollback to savepoint ${savepointName}`));
throw err;
}
}
}
class PreparedQuery extends PreparedQueryBase {
constructor(client, query, logger, fields, executeMethod, _isResponseInArrayMode, customResultMapper) {
super("sync", executeMethod, query);
this.client = client;
this.logger = logger;
this.fields = fields;
this._isResponseInArrayMode = _isResponseInArrayMode;
this.customResultMapper = customResultMapper;
}
static [entityKind] = "SQLJsPreparedQuery";
run(placeholderValues) {
const stmt = this.client.prepare(this.query.sql);
const params = fillPlaceholders(this.query.params, placeholderValues ?? {});
this.logger.logQuery(this.query.sql, params);
const result = stmt.run(params);
stmt.free();
return result;
}
all(placeholderValues) {
const stmt = this.client.prepare(this.query.sql);
const { fields, joinsNotNullableMap, logger, query, customResultMapper } = this;
if (!fields && !customResultMapper) {
const params = fillPlaceholders(query.params, placeholderValues ?? {});
logger.logQuery(query.sql, params);
stmt.bind(params);
const rows2 = [];
while (stmt.step()) {
rows2.push(stmt.getAsObject());
}
stmt.free();
return rows2;
}
const rows = this.values(placeholderValues);
if (customResultMapper) {
return customResultMapper(rows, normalizeFieldValue);
}
return rows.map((row) => mapResultRow(fields, row.map((v) => normalizeFieldValue(v)), joinsNotNullableMap));
}
get(placeholderValues) {
const stmt = this.client.prepare(this.query.sql);
const params = fillPlaceholders(this.query.params, placeholderValues ?? {});
this.logger.logQuery(this.query.sql, params);
const { fields, joinsNotNullableMap, customResultMapper } = this;
if (!fields && !customResultMapper) {
const result = stmt.getAsObject(params);
stmt.free();
return result;
}
const row = stmt.get(params);
stmt.free();
if (!row || row.length === 0 && fields.length > 0) {
return void 0;
}
if (customResultMapper) {
return customResultMapper([row], normalizeFieldValue);
}
return mapResultRow(fields, row.map((v) => normalizeFieldValue(v)), joinsNotNullableMap);
}
values(placeholderValues) {
const stmt = this.client.prepare(this.query.sql);
const params = fillPlaceholders(this.query.params, placeholderValues ?? {});
this.logger.logQuery(this.query.sql, params);
stmt.bind(params);
const rows = [];
while (stmt.step()) {
rows.push(stmt.get());
}
stmt.free();
return rows;
}
/** @internal */
isResponseInArrayMode() {
return this._isResponseInArrayMode;
}
}
function normalizeFieldValue(value) {
if (value instanceof Uint8Array) {
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;
}
export {
PreparedQuery,
SQLJsSession,
SQLJsTransaction
};
//# sourceMappingURL=session.js.map |