kysely-mapper
Version:
Flexible Kysely-based utility for mapping between tables and objects
130 lines • 6.33 kB
JavaScript
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
};
var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
if (kind === "m") throw new TypeError("Private method is not writable");
if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
};
var _MappingInsertQuery_returningQB;
/**
* Mapping query for inserting rows into a database table.
*/
export class MappingInsertQuery {
constructor(db, qb, transforms, returnColumns) {
this.db = db;
this.qb = qb;
this.transforms = transforms;
this.returnColumns = returnColumns;
_MappingInsertQuery_returningQB.set(this, null);
}
/**
* Modifies the underlying Kysely query builder.
* @param factory A function that takes the current query builder and
* returns a new query builder.
*/
modify(factory) {
return new MappingInsertQuery(this.db, factory(this.qb), this.transforms, this.returnColumns);
}
returnAll(objs) {
return __awaiter(this, void 0, void 0, function* () {
if (this.returnColumns.length === 0) {
yield this.loadInsertedObjects(this.qb, objs).execute();
return;
}
const returns = yield this.loadInsertedObjects(this.getReturningQB(), objs).execute();
return this.transforms.insertReturnTransform === undefined
? returns
: returns.map((row, i) => this.transforms.insertReturnTransform(objs[i], row));
});
}
returnOne(obj) {
return __awaiter(this, void 0, void 0, function* () {
if (this.returnColumns.length === 0) {
yield this.loadInsertedObjects(this.qb, obj).execute();
return;
}
const result = yield this.loadInsertedObjects(this.getReturningQB(), obj).executeTakeFirst();
return this.transforms.insertReturnTransform === undefined
? result
: this.transforms.insertReturnTransform(obj, result);
});
}
/**
* Runs the query, inserting rows into the table without returning any columns.
* @param objOrObjs The object or objects to be inserted.
* @returns Returns `true`; throws an exception on error.
*/
run(objOrObjs) {
return __awaiter(this, void 0, void 0, function* () {
yield this.loadInsertedObjects(this.qb, objOrObjs).execute();
return true;
});
}
/**
* Returns an array of the columns to be inserted, with
* `['*']` indicating that all columns will be inserted.
* @returns An array of the columns to be inserted.
*/
getInsertColumns() {
return ['*'];
}
/**
* Returns a query builder for inserting rows into the table and
* returning values, caching the query builder for future use.
* @returns A query builder for inserting rows into the table and
* returning values.
*/
getReturningQB() {
if (__classPrivateFieldGet(this, _MappingInsertQuery_returningQB, "f") === null) {
__classPrivateFieldSet(this, _MappingInsertQuery_returningQB, this.returnColumns[0] == '*'
? this.qb.returningAll()
: this.qb.returning(this.returnColumns), "f");
}
return __classPrivateFieldGet(this, _MappingInsertQuery_returningQB, "f");
}
/**
* Loads the objects to be inserted into the query builder.
* @param qb The query builder to load the objects into.
* @param objOrObjs The object or objects to be inserted.
* @returns The query builder with the objects loaded.
*/
loadInsertedObjects(qb, objOrObjs) {
const insertColumns = this.getInsertColumns();
if (Array.isArray(objOrObjs)) {
const transformedObjs = this.transforms.insertTransform === undefined
? objOrObjs
: objOrObjs.map((obj) => this.transforms.insertTransform(obj, insertColumns));
// TS requires separate calls to values() for different arg types.
return this.setColumnValues(qb, transformedObjs);
}
const transformedObj = this.transforms.insertTransform === undefined
? objOrObjs
: this.transforms.insertTransform(objOrObjs, insertColumns);
// TS requires separate calls to values() for different arg types.
return this.setColumnValues(qb, transformedObj);
}
/**
* Sets the values of the inserted columns.
* @param qb The query builder to set the values into.
* @param objOrObjs The object or objects of column-value pairs
* to be inserted.
*/
setColumnValues(qb, objOrObjs) {
return qb.values(objOrObjs);
}
}
_MappingInsertQuery_returningQB = new WeakMap();
//# sourceMappingURL=insert-query.js.map