116 lines
3.0 KiB
Plaintext
116 lines
3.0 KiB
Plaintext
import { entityKind } from "../../entity.js";
|
|
import { QueryPromise } from "../../query-promise.js";
|
|
import {
|
|
mapRelationalRow
|
|
} from "../../relations.js";
|
|
import { tracer } from "../../tracing.js";
|
|
class RelationalQueryBuilder {
|
|
constructor(fullSchema, schema, tableNamesMap, table, tableConfig, dialect, session) {
|
|
this.fullSchema = fullSchema;
|
|
this.schema = schema;
|
|
this.tableNamesMap = tableNamesMap;
|
|
this.table = table;
|
|
this.tableConfig = tableConfig;
|
|
this.dialect = dialect;
|
|
this.session = session;
|
|
}
|
|
static [entityKind] = "GelRelationalQueryBuilder";
|
|
findMany(config) {
|
|
return new GelRelationalQuery(
|
|
this.fullSchema,
|
|
this.schema,
|
|
this.tableNamesMap,
|
|
this.table,
|
|
this.tableConfig,
|
|
this.dialect,
|
|
this.session,
|
|
config ? config : {},
|
|
"many"
|
|
);
|
|
}
|
|
findFirst(config) {
|
|
return new GelRelationalQuery(
|
|
this.fullSchema,
|
|
this.schema,
|
|
this.tableNamesMap,
|
|
this.table,
|
|
this.tableConfig,
|
|
this.dialect,
|
|
this.session,
|
|
config ? { ...config, limit: 1 } : { limit: 1 },
|
|
"first"
|
|
);
|
|
}
|
|
}
|
|
class GelRelationalQuery extends QueryPromise {
|
|
constructor(fullSchema, schema, tableNamesMap, table, tableConfig, dialect, session, config, mode) {
|
|
super();
|
|
this.fullSchema = fullSchema;
|
|
this.schema = schema;
|
|
this.tableNamesMap = tableNamesMap;
|
|
this.table = table;
|
|
this.tableConfig = tableConfig;
|
|
this.dialect = dialect;
|
|
this.session = session;
|
|
this.config = config;
|
|
this.mode = mode;
|
|
}
|
|
static [entityKind] = "GelRelationalQuery";
|
|
/** @internal */
|
|
_prepare(name) {
|
|
return tracer.startActiveSpan("drizzle.prepareQuery", () => {
|
|
const { query, builtQuery } = this._toSQL();
|
|
return this.session.prepareQuery(
|
|
builtQuery,
|
|
void 0,
|
|
name,
|
|
true,
|
|
(rawRows, mapColumnValue) => {
|
|
const rows = rawRows.map(
|
|
(row) => mapRelationalRow(this.schema, this.tableConfig, row, query.selection, mapColumnValue)
|
|
);
|
|
if (this.mode === "first") {
|
|
return rows[0];
|
|
}
|
|
return rows;
|
|
}
|
|
);
|
|
});
|
|
}
|
|
prepare(name) {
|
|
return this._prepare(name);
|
|
}
|
|
_getQuery() {
|
|
return this.dialect.buildRelationalQueryWithoutPK({
|
|
fullSchema: this.fullSchema,
|
|
schema: this.schema,
|
|
tableNamesMap: this.tableNamesMap,
|
|
table: this.table,
|
|
tableConfig: this.tableConfig,
|
|
queryConfig: this.config,
|
|
tableAlias: this.tableConfig.tsName
|
|
});
|
|
}
|
|
/** @internal */
|
|
getSQL() {
|
|
return this._getQuery().sql;
|
|
}
|
|
_toSQL() {
|
|
const query = this._getQuery();
|
|
const builtQuery = this.dialect.sqlToQuery(query.sql);
|
|
return { query, builtQuery };
|
|
}
|
|
toSQL() {
|
|
return this._toSQL().builtQuery;
|
|
}
|
|
execute() {
|
|
return tracer.startActiveSpan("drizzle.operation", () => {
|
|
return this._prepare().execute(void 0);
|
|
});
|
|
}
|
|
}
|
|
export {
|
|
GelRelationalQuery,
|
|
RelationalQueryBuilder
|
|
};
|
|
//# sourceMappingURL=query.js.map |