229 lines
7.0 KiB
Plaintext
229 lines
7.0 KiB
Plaintext
import { entityKind, is } from "../../entity.js";
|
|
import { PgTable } from "../table.js";
|
|
import { QueryPromise } from "../../query-promise.js";
|
|
import { SelectionProxyHandler } from "../../selection-proxy.js";
|
|
import { SQL } from "../../sql/sql.js";
|
|
import { Subquery } from "../../subquery.js";
|
|
import { getTableName, Table } from "../../table.js";
|
|
import {
|
|
getTableLikeName,
|
|
mapUpdateSet,
|
|
orderSelectedFields
|
|
} from "../../utils.js";
|
|
import { ViewBaseConfig } from "../../view-common.js";
|
|
import { extractUsedTable } 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";
|
|
authToken;
|
|
setToken(token) {
|
|
this.authToken = token;
|
|
return this;
|
|
}
|
|
set(values) {
|
|
return new PgUpdateBase(
|
|
this.table,
|
|
mapUpdateSet(this.table, values),
|
|
this.session,
|
|
this.dialect,
|
|
this.withList
|
|
).setToken(this.authToken);
|
|
}
|
|
}
|
|
class PgUpdateBase extends QueryPromise {
|
|
constructor(table, set, session, dialect, withList) {
|
|
super();
|
|
this.session = session;
|
|
this.dialect = dialect;
|
|
this.config = { set, table, withList, joins: [] };
|
|
this.tableName = getTableLikeName(table);
|
|
this.joinsNotNullableMap = typeof this.tableName === "string" ? { [this.tableName]: true } : {};
|
|
}
|
|
static [entityKind] = "PgUpdate";
|
|
config;
|
|
tableName;
|
|
joinsNotNullableMap;
|
|
cacheConfig;
|
|
from(source) {
|
|
const src = source;
|
|
const tableName = getTableLikeName(src);
|
|
if (typeof tableName === "string") {
|
|
this.joinsNotNullableMap[tableName] = true;
|
|
}
|
|
this.config.from = src;
|
|
return this;
|
|
}
|
|
getTableLikeFields(table) {
|
|
if (is(table, PgTable)) {
|
|
return table[Table.Symbol.Columns];
|
|
} else if (is(table, Subquery)) {
|
|
return table._.selectedFields;
|
|
}
|
|
return table[ViewBaseConfig].selectedFields;
|
|
}
|
|
createJoin(joinType) {
|
|
return (table, on) => {
|
|
const tableName = getTableLikeName(table);
|
|
if (typeof tableName === "string" && this.config.joins.some((join) => join.alias === tableName)) {
|
|
throw new Error(`Alias "${tableName}" is already used in this query`);
|
|
}
|
|
if (typeof on === "function") {
|
|
const from = this.config.from && !is(this.config.from, SQL) ? this.getTableLikeFields(this.config.from) : void 0;
|
|
on = on(
|
|
new Proxy(
|
|
this.config.table[Table.Symbol.Columns],
|
|
new SelectionProxyHandler({ sqlAliasedBehavior: "sql", sqlBehavior: "sql" })
|
|
),
|
|
from && new Proxy(
|
|
from,
|
|
new SelectionProxyHandler({ sqlAliasedBehavior: "sql", sqlBehavior: "sql" })
|
|
)
|
|
);
|
|
}
|
|
this.config.joins.push({ on, table, joinType, alias: tableName });
|
|
if (typeof tableName === "string") {
|
|
switch (joinType) {
|
|
case "left": {
|
|
this.joinsNotNullableMap[tableName] = false;
|
|
break;
|
|
}
|
|
case "right": {
|
|
this.joinsNotNullableMap = Object.fromEntries(
|
|
Object.entries(this.joinsNotNullableMap).map(([key]) => [key, false])
|
|
);
|
|
this.joinsNotNullableMap[tableName] = true;
|
|
break;
|
|
}
|
|
case "inner": {
|
|
this.joinsNotNullableMap[tableName] = true;
|
|
break;
|
|
}
|
|
case "full": {
|
|
this.joinsNotNullableMap = Object.fromEntries(
|
|
Object.entries(this.joinsNotNullableMap).map(([key]) => [key, false])
|
|
);
|
|
this.joinsNotNullableMap[tableName] = false;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return this;
|
|
};
|
|
}
|
|
leftJoin = this.createJoin("left");
|
|
rightJoin = this.createJoin("right");
|
|
innerJoin = this.createJoin("inner");
|
|
fullJoin = this.createJoin("full");
|
|
/**
|
|
* 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) {
|
|
if (!fields) {
|
|
fields = Object.assign({}, this.config.table[Table.Symbol.Columns]);
|
|
if (this.config.from) {
|
|
const tableName = getTableLikeName(this.config.from);
|
|
if (typeof tableName === "string" && this.config.from && !is(this.config.from, SQL)) {
|
|
const fromFields = this.getTableLikeFields(this.config.from);
|
|
fields[tableName] = fromFields;
|
|
}
|
|
for (const join of this.config.joins) {
|
|
const tableName2 = getTableLikeName(join.table);
|
|
if (typeof tableName2 === "string" && !is(join.table, SQL)) {
|
|
const fromFields = this.getTableLikeFields(join.table);
|
|
fields[tableName2] = fromFields;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
this.config.returningFields = fields;
|
|
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) {
|
|
const query = this.session.prepareQuery(this.dialect.sqlToQuery(this.getSQL()), this.config.returning, name, true, void 0, {
|
|
type: "insert",
|
|
tables: extractUsedTable(this.config.table)
|
|
}, this.cacheConfig);
|
|
query.joinsNotNullableMap = this.joinsNotNullableMap;
|
|
return query;
|
|
}
|
|
prepare(name) {
|
|
return this._prepare(name);
|
|
}
|
|
authToken;
|
|
/** @internal */
|
|
setToken(token) {
|
|
this.authToken = token;
|
|
return this;
|
|
}
|
|
execute = (placeholderValues) => {
|
|
return this._prepare().execute(placeholderValues, this.authToken);
|
|
};
|
|
/** @internal */
|
|
getSelectedFields() {
|
|
return this.config.returningFields ? new Proxy(
|
|
this.config.returningFields,
|
|
new SelectionProxyHandler({
|
|
alias: getTableName(this.config.table),
|
|
sqlAliasedBehavior: "alias",
|
|
sqlBehavior: "error"
|
|
})
|
|
) : void 0;
|
|
}
|
|
$dynamic() {
|
|
return this;
|
|
}
|
|
}
|
|
export {
|
|
PgUpdateBase,
|
|
PgUpdateBuilder
|
|
};
|
|
//# sourceMappingURL=update.js.map |