UNPKG

drizzle-orm

Version:

Drizzle ORM package for SQL databases

99 lines 2.8 kB
import { entityKind } from "../../entity.js"; import { QueryPromise } from "../../query-promise.js"; import { Table } from "../../table.js"; import { mapUpdateSet, orderSelectedFields } from "../../utils.js"; class PgUpdateBuilder { constructor(table, session, dialect, withList) { this.table = table; this.session = session; this.dialect = dialect; this.withList = withList; } static [entityKind] = "PgUpdateBuilder"; set(values) { return new PgUpdateBase( this.table, mapUpdateSet(this.table, values), this.session, this.dialect, this.withList ); } } class PgUpdateBase extends QueryPromise { constructor(table, set, session, dialect, withList) { super(); this.session = session; this.dialect = dialect; this.config = { set, table, withList }; } static [entityKind] = "PgUpdate"; config; /** * Adds a 'where' clause to the query. * * Calling this method will update only those rows that fulfill a specified condition. * * See docs: {@link https://orm.drizzle.team/docs/update} * * @param where the 'where' clause. * * @example * You can use conditional operators and `sql function` to filter the rows to be updated. * * ```ts * // Update all cars with green color * await db.update(cars).set({ color: 'red' }) * .where(eq(cars.color, 'green')); * // or * await db.update(cars).set({ color: 'red' }) * .where(sql`${cars.color} = 'green'`) * ``` * * You can logically combine conditional operators with `and()` and `or()` operators: * * ```ts * // Update all BMW cars with a green color * await db.update(cars).set({ color: 'red' }) * .where(and(eq(cars.color, 'green'), eq(cars.brand, 'BMW'))); * * // Update all cars with the green or blue color * await db.update(cars).set({ color: 'red' }) * .where(or(eq(cars.color, 'green'), eq(cars.color, 'blue'))); * ``` */ where(where) { this.config.where = where; return this; } returning(fields = this.config.table[Table.Symbol.Columns]) { this.config.returning = orderSelectedFields(fields); return this; } /** @internal */ getSQL() { return this.dialect.buildUpdateQuery(this.config); } toSQL() { const { typings: _typings, ...rest } = this.dialect.sqlToQuery(this.getSQL()); return rest; } /** @internal */ _prepare(name) { return this.session.prepareQuery(this.dialect.sqlToQuery(this.getSQL()), this.config.returning, name, true); } prepare(name) { return this._prepare(name); } execute = (placeholderValues) => { return this._prepare().execute(placeholderValues); }; $dynamic() { return this; } } export { PgUpdateBase, PgUpdateBuilder }; //# sourceMappingURL=update.js.map