drizzle-orm
Version:
Drizzle ORM package for SQL databases
94 lines • 2.52 kB
JavaScript
import { entityKind, is } from "../../entity.js";
import { PgDialect } from "../dialect.js";
import { SelectionProxyHandler } from "../../selection-proxy.js";
import { WithSubquery } from "../../subquery.js";
import { PgSelectBuilder } from "./select.js";
class QueryBuilder {
static [entityKind] = "PgQueryBuilder";
dialect;
dialectConfig;
constructor(dialect) {
this.dialect = is(dialect, PgDialect) ? dialect : void 0;
this.dialectConfig = is(dialect, PgDialect) ? void 0 : dialect;
}
$with = (alias, selection) => {
const queryBuilder = this;
const as = (qb) => {
if (typeof qb === "function") {
qb = qb(queryBuilder);
}
return new Proxy(
new WithSubquery(
qb.getSQL(),
selection ?? ("getSelectedFields" in qb ? qb.getSelectedFields() ?? {} : {}),
alias,
true
),
new SelectionProxyHandler({ alias, sqlAliasedBehavior: "alias", sqlBehavior: "error" })
);
};
return { as };
};
with(...queries) {
const self = this;
function select(fields) {
return new PgSelectBuilder({
fields: fields ?? void 0,
session: void 0,
dialect: self.getDialect(),
withList: queries
});
}
function selectDistinct(fields) {
return new PgSelectBuilder({
fields: fields ?? void 0,
session: void 0,
dialect: self.getDialect(),
distinct: true
});
}
function selectDistinctOn(on, fields) {
return new PgSelectBuilder({
fields: fields ?? void 0,
session: void 0,
dialect: self.getDialect(),
distinct: { on }
});
}
return { select, selectDistinct, selectDistinctOn };
}
select(fields) {
return new PgSelectBuilder({
fields: fields ?? void 0,
session: void 0,
dialect: this.getDialect()
});
}
selectDistinct(fields) {
return new PgSelectBuilder({
fields: fields ?? void 0,
session: void 0,
dialect: this.getDialect(),
distinct: true
});
}
selectDistinctOn(on, fields) {
return new PgSelectBuilder({
fields: fields ?? void 0,
session: void 0,
dialect: this.getDialect(),
distinct: { on }
});
}
// Lazy load dialect to avoid circular dependency
getDialect() {
if (!this.dialect) {
this.dialect = new PgDialect(this.dialectConfig);
}
return this.dialect;
}
}
export {
QueryBuilder
};
//# sourceMappingURL=query-builder.js.map