drizzle-orm
Version:
Drizzle ORM package for SQL databases
131 lines • 4.33 kB
JavaScript
import { entityKind, is } from "../../entity.js";
import { QueryPromise } from "../../query-promise.js";
import { Param, SQL, sql } from "../../sql/sql.js";
import { Columns, Table } from "../../table.js";
import { haveSameKeys, mapUpdateSet } from "../../utils.js";
import { QueryBuilder } from "./query-builder.js";
class MySqlInsertBuilder {
constructor(table, session, dialect) {
this.table = table;
this.session = session;
this.dialect = dialect;
}
static [entityKind] = "MySqlInsertBuilder";
shouldIgnore = false;
ignore() {
this.shouldIgnore = true;
return this;
}
values(values) {
values = Array.isArray(values) ? values : [values];
if (values.length === 0) {
throw new Error("values() must be called with at least one value");
}
const mappedValues = values.map((entry) => {
const result = {};
const cols = this.table[Table.Symbol.Columns];
for (const colKey of Object.keys(entry)) {
const colValue = entry[colKey];
result[colKey] = is(colValue, SQL) ? colValue : new Param(colValue, cols[colKey]);
}
return result;
});
return new MySqlInsertBase(this.table, mappedValues, this.shouldIgnore, this.session, this.dialect);
}
select(selectQuery) {
const select = typeof selectQuery === "function" ? selectQuery(new QueryBuilder()) : selectQuery;
if (!is(select, SQL) && !haveSameKeys(this.table[Columns], select._.selectedFields)) {
throw new Error(
"Insert select error: selected fields are not the same or are in a different order compared to the table definition"
);
}
return new MySqlInsertBase(this.table, select, this.shouldIgnore, this.session, this.dialect, true);
}
}
class MySqlInsertBase extends QueryPromise {
constructor(table, values, ignore, session, dialect, select) {
super();
this.session = session;
this.dialect = dialect;
this.config = { table, values, select, ignore };
}
static [entityKind] = "MySqlInsert";
config;
/**
* Adds an `on duplicate key update` clause to the query.
*
* Calling this method will update the row if any unique index conflicts. MySQL will automatically determine the conflict target based on the primary key and unique indexes.
*
* See docs: {@link https://orm.drizzle.team/docs/insert#on-duplicate-key-update}
*
* @param config The `set` clause
*
* @example
* ```ts
* await db.insert(cars)
* .values({ id: 1, brand: 'BMW'})
* .onDuplicateKeyUpdate({ set: { brand: 'Porsche' }});
* ```
*
* While MySQL does not directly support doing nothing on conflict, you can perform a no-op by setting any column's value to itself and achieve the same effect:
*
* ```ts
* import { sql } from 'drizzle-orm';
*
* await db.insert(cars)
* .values({ id: 1, brand: 'BMW' })
* .onDuplicateKeyUpdate({ set: { id: sql`id` } });
* ```
*/
onDuplicateKeyUpdate(config) {
const setSql = this.dialect.buildUpdateSet(this.config.table, mapUpdateSet(this.config.table, config.set));
this.config.onConflict = sql`update ${setSql}`;
return this;
}
$returningId() {
const returning = [];
for (const [key, value] of Object.entries(this.config.table[Table.Symbol.Columns])) {
if (value.primary) {
returning.push({ field: value, path: [key] });
}
}
this.config.returning = returning;
return this;
}
/** @internal */
getSQL() {
return this.dialect.buildInsertQuery(this.config).sql;
}
toSQL() {
const { typings: _typings, ...rest } = this.dialect.sqlToQuery(this.getSQL());
return rest;
}
prepare() {
const { sql: sql2, generatedIds } = this.dialect.buildInsertQuery(this.config);
return this.session.prepareQuery(
this.dialect.sqlToQuery(sql2),
void 0,
void 0,
generatedIds,
this.config.returning
);
}
execute = (placeholderValues) => {
return this.prepare().execute(placeholderValues);
};
createIterator = () => {
const self = this;
return async function* (placeholderValues) {
yield* self.prepare().iterator(placeholderValues);
};
};
iterator = this.createIterator();
$dynamic() {
return this;
}
}
export {
MySqlInsertBase,
MySqlInsertBuilder
};
//# sourceMappingURL=insert.js.map