typeorm
Version:
Data-Mapper ORM for TypeScript and ES2021+. Supports MySQL/MariaDB, PostgreSQL, MS SQL Server, Oracle, SAP HANA, SQLite, MongoDB databases.
104 lines (102 loc) • 4.69 kB
JavaScript
import { QueryFailedError } from "../../error/QueryFailedError";
import { QueryRunnerAlreadyReleasedError } from "../../error/QueryRunnerAlreadyReleasedError";
import { QueryResult } from "../../query-runner/QueryResult";
import { Broadcaster } from "../../subscriber/Broadcaster";
import { BroadcasterResult } from "../../subscriber/BroadcasterResult";
import { AbstractSqliteQueryRunner } from "../sqlite-abstract/AbstractSqliteQueryRunner";
/**
* Runs queries on a single sqlite database connection.
*/
export class ReactNativeQueryRunner extends AbstractSqliteQueryRunner {
// -------------------------------------------------------------------------
// Constructor
// -------------------------------------------------------------------------
constructor(driver) {
super();
this.driver = driver;
this.connection = driver.connection;
this.broadcaster = new Broadcaster(this);
}
/**
* Called before migrations are run.
*/
async beforeMigration() {
await this.query(`PRAGMA foreign_keys = OFF`);
}
/**
* Called after migrations are run.
*/
async afterMigration() {
await this.query(`PRAGMA foreign_keys = ON`);
}
/**
* Executes a given SQL query.
*/
async query(query, parameters, useStructuredResult = false) {
if (this.isReleased)
throw new QueryRunnerAlreadyReleasedError();
const databaseConnection = await this.connect();
this.driver.connection.logger.logQuery(query, parameters, this);
await this.broadcaster.broadcast("BeforeQuery", query, parameters);
const broadcasterResult = new BroadcasterResult();
const queryStartTime = Date.now();
return new Promise(async (ok, fail) => {
try {
databaseConnection.executeSql(query, parameters, async (raw) => {
// log slow queries if maxQueryExecution time is set
const maxQueryExecutionTime = this.driver.options.maxQueryExecutionTime;
const queryEndTime = Date.now();
const queryExecutionTime = queryEndTime - queryStartTime;
this.broadcaster.broadcastAfterQueryEvent(broadcasterResult, query, parameters, true, queryExecutionTime, raw, undefined);
if (maxQueryExecutionTime &&
queryExecutionTime > maxQueryExecutionTime)
this.driver.connection.logger.logQuerySlow(queryExecutionTime, query, parameters, this);
if (broadcasterResult.promises.length > 0)
await Promise.all(broadcasterResult.promises);
const result = new QueryResult();
if (raw?.hasOwnProperty("rowsAffected")) {
result.affected = raw.rowsAffected;
}
if (raw?.hasOwnProperty("rows")) {
const records = [];
for (let i = 0; i < raw.rows.length; i++) {
records.push(raw.rows.item(i));
}
result.raw = records;
result.records = records;
}
// return id of inserted row, if query was insert statement.
if (query.substr(0, 11) === "INSERT INTO") {
result.raw = raw.insertId;
}
if (useStructuredResult) {
ok(result);
}
else {
ok(result.raw);
}
}, async (err) => {
this.driver.connection.logger.logQueryError(err, query, parameters, this);
this.broadcaster.broadcastAfterQueryEvent(broadcasterResult, query, parameters, false, undefined, undefined, err);
fail(new QueryFailedError(query, parameters, err));
});
}
catch (err) {
fail(err);
}
finally {
await broadcasterResult.wait();
}
});
}
// -------------------------------------------------------------------------
// Protected Methods
// -------------------------------------------------------------------------
/**
* Parametrizes given object of values. Used to create column=value queries.
*/
parametrize(objectLiteral, startIndex = 0) {
return Object.keys(objectLiteral).map((key, index) => `"${key}"` + "=?");
}
}
//# sourceMappingURL=ReactNativeQueryRunner.js.map