UNPKG

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
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