UNPKG

@clickup/ent-framework

Version:

A PostgreSQL graph-database-alike library with microsharding and row-level security

48 lines 1.69 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.PgQueryDelete = void 0; const QueryBase_1 = require("../abstract/QueryBase"); const types_1 = require("../types"); const PgRunner_1 = require("./PgRunner"); class PgQueryDelete extends QueryBase_1.QueryBase { constructor() { super(...arguments); /** @ignore */ this.RUNNER_CLASS = PgRunnerDelete; } } exports.PgQueryDelete = PgQueryDelete; class PgRunnerDelete extends PgRunner_1.PgRunner { constructor(schema, client) { super(schema, client); this.op = "DELETE"; this.maxBatchSize = 100; this.default = false; // If no delete happened, returns false. this.builder = { prefix: this.fmt("DELETE FROM %T WHERE "), func: this.createOneOfBuilder(types_1.ID), suffix: this.fmt(` RETURNING %PK AS ${types_1.ID}`), }; } key(input) { return input; } async runSingle(input, annotations) { const sql = this.builder.prefix + this.builder.func([input]) + this.builder.suffix; const rows = await this.clientQuery(sql, annotations, 1); return rows.length > 0 ? true : false; } async runBatch(inputs, annotations) { const sql = this.builder.prefix + this.builder.func(inputs.values()) + this.builder.suffix; const rows = await this.clientQuery(sql, annotations, inputs.size); const outputs = new Map(); for (const row of rows) { outputs.set(row[types_1.ID], true); } return outputs; } } PgRunnerDelete.IS_WRITE = true; //# sourceMappingURL=PgQueryDelete.js.map