drizzle-orm 1.0.0-beta.1-7722e6a → 1.0.0-beta.2-e93475f

package/cockroach-core/columns/varchar.d.ts

@@ -0,0 +1,30 @@
+import type { ColumnBaseConfig } from "../../column.js";
+import { entityKind } from "../../entity.js";
+import { type Equal, type Writable } from "../../utils.js";
+import { CockroachColumn, CockroachColumnWithArrayBuilder } from "./common.js";
+export declare class CockroachVarcharBuilder<TEnum extends [string, ...string[]]> extends CockroachColumnWithArrayBuilder<{
+    dataType: Equal<TEnum, [string, ...string[]]> extends true ? 'string' : 'string enum';
+    data: TEnum[number];
+    driverParam: string;
+    enumValues: TEnum;
+}, {
+    length: number | undefined;
+    enumValues: TEnum | undefined;
+}> {
+    static readonly [entityKind]: string;
+    constructor(name: string, config: CockroachVarcharConfig<TEnum>);
+}
+export declare class CockroachVarchar<T extends ColumnBaseConfig<'string' | 'string enum'>> extends CockroachColumn<T, {
+    length: number | undefined;
+    enumValues: T['enumValues'];
+}> {
+    static readonly [entityKind]: string;
+    readonly enumValues: T["enumValues"];
+    getSQLType(): string;
+}
+export interface CockroachVarcharConfig<TEnum extends readonly string[] | string[] | undefined = readonly string[] | string[] | undefined> {
+    enum?: TEnum;
+    length?: number | undefined;
+}
+export declare function varchar<U extends string, T extends Readonly<[U, ...U[]]>>(config?: CockroachVarcharConfig<T | Writable<T>>): CockroachVarcharBuilder<Writable<T>>;
+export declare function varchar<U extends string, T extends Readonly<[U, ...U[]]>>(name: string, config?: CockroachVarcharConfig<T | Writable<T>>): CockroachVarcharBuilder<Writable<T>>;

package/cockroach-core/columns/varchar.js

@@ -0,0 +1,35 @@
+import { entityKind } from "../../entity.js";
+import { getColumnNameAndConfig } from "../../utils.js";
+import { CockroachColumn, CockroachColumnWithArrayBuilder } from "./common.js";
+class CockroachVarcharBuilder extends CockroachColumnWithArrayBuilder {
+  static [entityKind] = "CockroachVarcharBuilder";
+  constructor(name, config) {
+    super(name, config.enum?.length ? "string enum" : "string", "CockroachVarchar");
+    this.config.length = config.length;
+    this.config.enumValues = config.enum;
+  }
+  /** @internal */
+  build(table) {
+    return new CockroachVarchar(
+      table,
+      this.config
+    );
+  }
+}
+class CockroachVarchar extends CockroachColumn {
+  static [entityKind] = "CockroachVarchar";
+  enumValues = this.config.enumValues;
+  getSQLType() {
+    return this.length === void 0 ? `varchar` : `varchar(${this.length})`;
+  }
+}
+function varchar(a, b = {}) {
+  const { name, config } = getColumnNameAndConfig(a, b);
+  return new CockroachVarcharBuilder(name, config);
+}
+export {
+  CockroachVarchar,
+  CockroachVarcharBuilder,
+  varchar
+};
+//# sourceMappingURL=varchar.js.map

package/cockroach-core/columns/varchar.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/cockroach-core/columns/varchar.ts"],"sourcesContent":["import type { AnyCockroachTable } from '~/cockroach-core/table.ts';\nimport type { ColumnBaseConfig } from '~/column.ts';\nimport { entityKind } from '~/entity.ts';\nimport { type Equal, getColumnNameAndConfig, type Writable } from '~/utils.ts';\nimport { CockroachColumn, CockroachColumnWithArrayBuilder } from './common.ts';\n\nexport class CockroachVarcharBuilder<\n\tTEnum extends [string, ...string[]],\n> extends CockroachColumnWithArrayBuilder<\n\t{\n\t\tdataType: Equal<TEnum, [string, ...string[]]> extends true ? 'string' : 'string enum';\n\t\tdata: TEnum[number];\n\t\tdriverParam: string;\n\t\tenumValues: TEnum;\n\t},\n\t{ length: number | undefined; enumValues: TEnum | undefined }\n> {\n\tstatic override readonly [entityKind]: string = 'CockroachVarcharBuilder';\n\n\tconstructor(name: string, config: CockroachVarcharConfig<TEnum>) {\n\t\tsuper(name, config.enum?.length ? 'string enum' : 'string', 'CockroachVarchar');\n\t\tthis.config.length = config.length;\n\t\tthis.config.enumValues = config.enum;\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachVarchar(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachVarchar<\n\tT extends ColumnBaseConfig<'string' | 'string enum'>,\n> extends CockroachColumn<T, { length: number | undefined; enumValues: T['enumValues'] }> {\n\tstatic override readonly [entityKind]: string = 'CockroachVarchar';\n\n\toverride readonly enumValues = this.config.enumValues;\n\n\tgetSQLType(): string {\n\t\treturn this.length === undefined ? `varchar` : `varchar(${this.length})`;\n\t}\n}\n\nexport interface CockroachVarcharConfig<\n\tTEnum extends readonly string[] | string[] | undefined = readonly string[] | string[] | undefined,\n> {\n\tenum?: TEnum;\n\tlength?: number | undefined;\n}\n\nexport function varchar<\n\tU extends string,\n\tT extends Readonly<[U, ...U[]]>,\n>(\n\tconfig?: CockroachVarcharConfig<T | Writable<T>>,\n): CockroachVarcharBuilder<Writable<T>>;\nexport function varchar<\n\tU extends string,\n\tT extends Readonly<[U, ...U[]]>,\n>(\n\tname: string,\n\tconfig?: CockroachVarcharConfig<T | Writable<T>>,\n): CockroachVarcharBuilder<Writable<T>>;\nexport function varchar(a?: string | CockroachVarcharConfig, b: CockroachVarcharConfig = {}): any {\n\tconst { name, config } = getColumnNameAndConfig<CockroachVarcharConfig>(a, b);\n\treturn new CockroachVarcharBuilder(name, config as any);\n}\n"],"mappings":"AAEA,SAAS,kBAAkB;AAC3B,SAAqB,8BAA6C;AAClE,SAAS,iBAAiB,uCAAuC;AAE1D,MAAM,gCAEH,gCAQR;AAAA,EACD,QAA0B,UAAU,IAAY;AAAA,EAEhD,YAAY,MAAc,QAAuC;AAChE,UAAM,MAAM,OAAO,MAAM,SAAS,gBAAgB,UAAU,kBAAkB;AAC9E,SAAK,OAAO,SAAS,OAAO;AAC5B,SAAK,OAAO,aAAa,OAAO;AAAA,EACjC;AAAA;AAAA,EAGS,MACR,OACC;AACD,WAAO,IAAI;AAAA,MACV;AAAA,MACA,KAAK;AAAA,IACN;AAAA,EACD;AACD;AAEO,MAAM,yBAEH,gBAAgF;AAAA,EACzF,QAA0B,UAAU,IAAY;AAAA,EAE9B,aAAa,KAAK,OAAO;AAAA,EAE3C,aAAqB;AACpB,WAAO,KAAK,WAAW,SAAY,YAAY,WAAW,KAAK,MAAM;AAAA,EACtE;AACD;AAsBO,SAAS,QAAQ,GAAqC,IAA4B,CAAC,GAAQ;AACjG,QAAM,EAAE,MAAM,OAAO,IAAI,uBAA+C,GAAG,CAAC;AAC5E,SAAO,IAAI,wBAAwB,MAAM,MAAa;AACvD;","names":[]}

package/cockroach-core/columns/vector.cjs

@@ -0,0 +1,65 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var vector_exports = {};
+__export(vector_exports, {
+  CockroachVector: () => CockroachVector,
+  CockroachVectorBuilder: () => CockroachVectorBuilder,
+  vector: () => vector
+});
+module.exports = __toCommonJS(vector_exports);
+var import_entity = require("../../entity.cjs");
+var import_utils = require("../../utils.cjs");
+var import_common = require("./common.cjs");
+class CockroachVectorBuilder extends import_common.CockroachColumnBuilder {
+  static [import_entity.entityKind] = "CockroachVectorBuilder";
+  constructor(name, config) {
+    super(name, "array vector", "CockroachVector");
+    this.config.length = config.dimensions;
+  }
+  /** @internal */
+  build(table) {
+    return new CockroachVector(
+      table,
+      this.config
+    );
+  }
+}
+class CockroachVector extends import_common.CockroachColumn {
+  static [import_entity.entityKind] = "CockroachVector";
+  getSQLType() {
+    return `vector(${this.config.length})`;
+  }
+  mapToDriverValue(value) {
+    return JSON.stringify(value);
+  }
+  mapFromDriverValue(value) {
+    return value.slice(1, -1).split(",").map((v) => Number.parseFloat(v));
+  }
+}
+function vector(a, b) {
+  const { name, config } = (0, import_utils.getColumnNameAndConfig)(a, b);
+  return new CockroachVectorBuilder(name, config);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  CockroachVector,
+  CockroachVectorBuilder,
+  vector
+});
+//# sourceMappingURL=vector.cjs.map

package/cockroach-core/columns/vector.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/cockroach-core/columns/vector.ts"],"sourcesContent":["import type { AnyCockroachTable } from '~/cockroach-core/table.ts';\nimport type { ColumnBaseConfig } from '~/column.ts';\nimport { entityKind } from '~/entity.ts';\nimport { getColumnNameAndConfig } from '~/utils.ts';\nimport { CockroachColumn, CockroachColumnBuilder } from './common.ts';\n\nexport class CockroachVectorBuilder extends CockroachColumnBuilder<\n\t{\n\t\tdataType: 'array vector';\n\t\tdata: number[];\n\t\tdriverParam: string;\n\t},\n\t{ length: number }\n> {\n\tstatic override readonly [entityKind]: string = 'CockroachVectorBuilder';\n\n\tconstructor(name: string, config: CockroachVectorConfig) {\n\t\tsuper(name, 'array vector', 'CockroachVector');\n\t\tthis.config.length = config.dimensions;\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachVector(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachVector<\n\tT extends ColumnBaseConfig<'array vector'>,\n> extends CockroachColumn<T, { length: number }> {\n\tstatic override readonly [entityKind]: string = 'CockroachVector';\n\n\tgetSQLType(): string {\n\t\treturn `vector(${this.config.length})`;\n\t}\n\n\toverride mapToDriverValue(value: unknown): unknown {\n\t\treturn JSON.stringify(value);\n\t}\n\n\toverride mapFromDriverValue(value: string): unknown {\n\t\treturn value\n\t\t\t.slice(1, -1)\n\t\t\t.split(',')\n\t\t\t.map((v) => Number.parseFloat(v));\n\t}\n}\n\nexport interface CockroachVectorConfig {\n\tdimensions: number;\n}\n\nexport function vector(\n\tconfig: CockroachVectorConfig,\n): CockroachVectorBuilder;\nexport function vector(\n\tname: string,\n\tconfig: CockroachVectorConfig,\n): CockroachVectorBuilder;\nexport function vector(a: string | CockroachVectorConfig, b?: CockroachVectorConfig) {\n\tconst { name, config } = getColumnNameAndConfig<CockroachVectorConfig>(a, b);\n\treturn new CockroachVectorBuilder(name, config);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAEA,oBAA2B;AAC3B,mBAAuC;AACvC,oBAAwD;AAEjD,MAAM,+BAA+B,qCAO1C;AAAA,EACD,QAA0B,wBAAU,IAAY;AAAA,EAEhD,YAAY,MAAc,QAA+B;AACxD,UAAM,MAAM,gBAAgB,iBAAiB;AAC7C,SAAK,OAAO,SAAS,OAAO;AAAA,EAC7B;AAAA;AAAA,EAGS,MACR,OACC;AACD,WAAO,IAAI;AAAA,MACV;AAAA,MACA,KAAK;AAAA,IACN;AAAA,EACD;AACD;AAEO,MAAM,wBAEH,8BAAuC;AAAA,EAChD,QAA0B,wBAAU,IAAY;AAAA,EAEhD,aAAqB;AACpB,WAAO,UAAU,KAAK,OAAO,MAAM;AAAA,EACpC;AAAA,EAES,iBAAiB,OAAyB;AAClD,WAAO,KAAK,UAAU,KAAK;AAAA,EAC5B;AAAA,EAES,mBAAmB,OAAwB;AACnD,WAAO,MACL,MAAM,GAAG,EAAE,EACX,MAAM,GAAG,EACT,IAAI,CAAC,MAAM,OAAO,WAAW,CAAC,CAAC;AAAA,EAClC;AACD;AAaO,SAAS,OAAO,GAAmC,GAA2B;AACpF,QAAM,EAAE,MAAM,OAAO,QAAI,qCAA8C,GAAG,CAAC;AAC3E,SAAO,IAAI,uBAAuB,MAAM,MAAM;AAC/C;","names":[]}

package/cockroach-core/columns/vector.d.cts

@@ -0,0 +1,26 @@
+import type { ColumnBaseConfig } from "../../column.cjs";
+import { entityKind } from "../../entity.cjs";
+import { CockroachColumn, CockroachColumnBuilder } from "./common.cjs";
+export declare class CockroachVectorBuilder extends CockroachColumnBuilder<{
+    dataType: 'array vector';
+    data: number[];
+    driverParam: string;
+}, {
+    length: number;
+}> {
+    static readonly [entityKind]: string;
+    constructor(name: string, config: CockroachVectorConfig);
+}
+export declare class CockroachVector<T extends ColumnBaseConfig<'array vector'>> extends CockroachColumn<T, {
+    length: number;
+}> {
+    static readonly [entityKind]: string;
+    getSQLType(): string;
+    mapToDriverValue(value: unknown): unknown;
+    mapFromDriverValue(value: string): unknown;
+}
+export interface CockroachVectorConfig {
+    dimensions: number;
+}
+export declare function vector(config: CockroachVectorConfig): CockroachVectorBuilder;
+export declare function vector(name: string, config: CockroachVectorConfig): CockroachVectorBuilder;

package/cockroach-core/columns/vector.d.ts

@@ -0,0 +1,26 @@
+import type { ColumnBaseConfig } from "../../column.js";
+import { entityKind } from "../../entity.js";
+import { CockroachColumn, CockroachColumnBuilder } from "./common.js";
+export declare class CockroachVectorBuilder extends CockroachColumnBuilder<{
+    dataType: 'array vector';
+    data: number[];
+    driverParam: string;
+}, {
+    length: number;
+}> {
+    static readonly [entityKind]: string;
+    constructor(name: string, config: CockroachVectorConfig);
+}
+export declare class CockroachVector<T extends ColumnBaseConfig<'array vector'>> extends CockroachColumn<T, {
+    length: number;
+}> {
+    static readonly [entityKind]: string;
+    getSQLType(): string;
+    mapToDriverValue(value: unknown): unknown;
+    mapFromDriverValue(value: string): unknown;
+}
+export interface CockroachVectorConfig {
+    dimensions: number;
+}
+export declare function vector(config: CockroachVectorConfig): CockroachVectorBuilder;
+export declare function vector(name: string, config: CockroachVectorConfig): CockroachVectorBuilder;

package/cockroach-core/columns/vector.js

@@ -0,0 +1,39 @@
+import { entityKind } from "../../entity.js";
+import { getColumnNameAndConfig } from "../../utils.js";
+import { CockroachColumn, CockroachColumnBuilder } from "./common.js";
+class CockroachVectorBuilder extends CockroachColumnBuilder {
+  static [entityKind] = "CockroachVectorBuilder";
+  constructor(name, config) {
+    super(name, "array vector", "CockroachVector");
+    this.config.length = config.dimensions;
+  }
+  /** @internal */
+  build(table) {
+    return new CockroachVector(
+      table,
+      this.config
+    );
+  }
+}
+class CockroachVector extends CockroachColumn {
+  static [entityKind] = "CockroachVector";
+  getSQLType() {
+    return `vector(${this.config.length})`;
+  }
+  mapToDriverValue(value) {
+    return JSON.stringify(value);
+  }
+  mapFromDriverValue(value) {
+    return value.slice(1, -1).split(",").map((v) => Number.parseFloat(v));
+  }
+}
+function vector(a, b) {
+  const { name, config } = getColumnNameAndConfig(a, b);
+  return new CockroachVectorBuilder(name, config);
+}
+export {
+  CockroachVector,
+  CockroachVectorBuilder,
+  vector
+};
+//# sourceMappingURL=vector.js.map

package/cockroach-core/columns/vector.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/cockroach-core/columns/vector.ts"],"sourcesContent":["import type { AnyCockroachTable } from '~/cockroach-core/table.ts';\nimport type { ColumnBaseConfig } from '~/column.ts';\nimport { entityKind } from '~/entity.ts';\nimport { getColumnNameAndConfig } from '~/utils.ts';\nimport { CockroachColumn, CockroachColumnBuilder } from './common.ts';\n\nexport class CockroachVectorBuilder extends CockroachColumnBuilder<\n\t{\n\t\tdataType: 'array vector';\n\t\tdata: number[];\n\t\tdriverParam: string;\n\t},\n\t{ length: number }\n> {\n\tstatic override readonly [entityKind]: string = 'CockroachVectorBuilder';\n\n\tconstructor(name: string, config: CockroachVectorConfig) {\n\t\tsuper(name, 'array vector', 'CockroachVector');\n\t\tthis.config.length = config.dimensions;\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachVector(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachVector<\n\tT extends ColumnBaseConfig<'array vector'>,\n> extends CockroachColumn<T, { length: number }> {\n\tstatic override readonly [entityKind]: string = 'CockroachVector';\n\n\tgetSQLType(): string {\n\t\treturn `vector(${this.config.length})`;\n\t}\n\n\toverride mapToDriverValue(value: unknown): unknown {\n\t\treturn JSON.stringify(value);\n\t}\n\n\toverride mapFromDriverValue(value: string): unknown {\n\t\treturn value\n\t\t\t.slice(1, -1)\n\t\t\t.split(',')\n\t\t\t.map((v) => Number.parseFloat(v));\n\t}\n}\n\nexport interface CockroachVectorConfig {\n\tdimensions: number;\n}\n\nexport function vector(\n\tconfig: CockroachVectorConfig,\n): CockroachVectorBuilder;\nexport function vector(\n\tname: string,\n\tconfig: CockroachVectorConfig,\n): CockroachVectorBuilder;\nexport function vector(a: string | CockroachVectorConfig, b?: CockroachVectorConfig) {\n\tconst { name, config } = getColumnNameAndConfig<CockroachVectorConfig>(a, b);\n\treturn new CockroachVectorBuilder(name, config);\n}\n"],"mappings":"AAEA,SAAS,kBAAkB;AAC3B,SAAS,8BAA8B;AACvC,SAAS,iBAAiB,8BAA8B;AAEjD,MAAM,+BAA+B,uBAO1C;AAAA,EACD,QAA0B,UAAU,IAAY;AAAA,EAEhD,YAAY,MAAc,QAA+B;AACxD,UAAM,MAAM,gBAAgB,iBAAiB;AAC7C,SAAK,OAAO,SAAS,OAAO;AAAA,EAC7B;AAAA;AAAA,EAGS,MACR,OACC;AACD,WAAO,IAAI;AAAA,MACV;AAAA,MACA,KAAK;AAAA,IACN;AAAA,EACD;AACD;AAEO,MAAM,wBAEH,gBAAuC;AAAA,EAChD,QAA0B,UAAU,IAAY;AAAA,EAEhD,aAAqB;AACpB,WAAO,UAAU,KAAK,OAAO,MAAM;AAAA,EACpC;AAAA,EAES,iBAAiB,OAAyB;AAClD,WAAO,KAAK,UAAU,KAAK;AAAA,EAC5B;AAAA,EAES,mBAAmB,OAAwB;AACnD,WAAO,MACL,MAAM,GAAG,EAAE,EACX,MAAM,GAAG,EACT,IAAI,CAAC,MAAM,OAAO,WAAW,CAAC,CAAC;AAAA,EAClC;AACD;AAaO,SAAS,OAAO,GAAmC,GAA2B;AACpF,QAAM,EAAE,MAAM,OAAO,IAAI,uBAA8C,GAAG,CAAC;AAC3E,SAAO,IAAI,uBAAuB,MAAM,MAAM;AAC/C;","names":[]}

package/cockroach-core/db.cjs

@@ -0,0 +1,347 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var db_exports = {};
+__export(db_exports, {
+  CockroachDatabase: () => CockroachDatabase,
+  withReplicas: () => withReplicas
+});
+module.exports = __toCommonJS(db_exports);
+var import_query_builders = require("./query-builders/index.cjs");
+var import_entity = require("../entity.cjs");
+var import_selection_proxy = require("../selection-proxy.cjs");
+var import_sql = require("../sql/sql.cjs");
+var import_subquery = require("../subquery.cjs");
+var import_count = require("./query-builders/count.cjs");
+var import_query = require("./query-builders/query.cjs");
+var import_raw = require("./query-builders/raw.cjs");
+var import_refresh_materialized_view = require("./query-builders/refresh-materialized-view.cjs");
+class CockroachDatabase {
+  constructor(dialect, session, schema) {
+    this.dialect = dialect;
+    this.session = session;
+    this._ = schema ? {
+      schema: schema.schema,
+      fullSchema: schema.fullSchema,
+      tableNamesMap: schema.tableNamesMap,
+      session
+    } : {
+      schema: void 0,
+      fullSchema: {},
+      tableNamesMap: {},
+      session
+    };
+    this._query = {};
+    if (this._.schema) {
+      for (const [tableName, columns] of Object.entries(this._.schema)) {
+        this._query[tableName] = new import_query.RelationalQueryBuilder(
+          schema.fullSchema,
+          this._.schema,
+          this._.tableNamesMap,
+          schema.fullSchema[tableName],
+          columns,
+          dialect,
+          session
+        );
+      }
+    }
+  }
+  static [import_entity.entityKind] = "CockroachDatabase";
+  _query;
+  /**
+   * Creates a subquery that defines a temporary named result set as a CTE.
+   *
+   * It is useful for breaking down complex queries into simpler parts and for reusing the result set in subsequent parts of the query.
+   *
+   * See docs: {@link https://orm.drizzle.team/docs/select#with-clause}
+   *
+   * @param alias The alias for the subquery.
+   *
+   * Failure to provide an alias will result in a DrizzleTypeError, preventing the subquery from being referenced in other queries.
+   *
+   * @example
+   *
+   * ```ts
+   * // Create a subquery with alias 'sq' and use it in the select query
+   * const sq = db.$with('sq').as(db.select().from(users).where(eq(users.id, 42)));
+   *
+   * const result = await db.with(sq).select().from(sq);
+   * ```
+   *
+   * To select arbitrary SQL values as fields in a CTE and reference them in other CTEs or in the main query, you need to add aliases to them:
+   *
+   * ```ts
+   * // Select an arbitrary SQL value as a field in a CTE and reference it in the main query
+   * const sq = db.$with('sq').as(db.select({
+   *   name: sql<string>`upper(${users.name})`.as('name'),
+   * })
+   * .from(users));
+   *
+   * const result = await db.with(sq).select({ name: sq.name }).from(sq);
+   * ```
+   */
+  $with = (alias, selection) => {
+    const self = this;
+    const as = (qb) => {
+      if (typeof qb === "function") {
+        qb = qb(new import_query_builders.QueryBuilder(self.dialect));
+      }
+      return new Proxy(
+        new import_subquery.WithSubquery(
+          qb.getSQL(),
+          selection ?? ("getSelectedFields" in qb ? qb.getSelectedFields() ?? {} : {}),
+          alias,
+          true
+        ),
+        new import_selection_proxy.SelectionProxyHandler({ alias, sqlAliasedBehavior: "alias", sqlBehavior: "error" })
+      );
+    };
+    return { as };
+  };
+  $count(source, filters) {
+    return new import_count.CockroachCountBuilder({ source, filters, session: this.session });
+  }
+  /**
+   * Incorporates a previously defined CTE (using `$with`) into the main query.
+   *
+   * This method allows the main query to reference a temporary named result set.
+   *
+   * See docs: {@link https://orm.drizzle.team/docs/select#with-clause}
+   *
+   * @param queries The CTEs to incorporate into the main query.
+   *
+   * @example
+   *
+   * ```ts
+   * // Define a subquery 'sq' as a CTE using $with
+   * const sq = db.$with('sq').as(db.select().from(users).where(eq(users.id, 42)));
+   *
+   * // Incorporate the CTE 'sq' into the main query and select from it
+   * const result = await db.with(sq).select().from(sq);
+   * ```
+   */
+  with(...queries) {
+    const self = this;
+    function select(fields) {
+      return new import_query_builders.CockroachSelectBuilder({
+        fields: fields ?? void 0,
+        session: self.session,
+        dialect: self.dialect,
+        withList: queries
+      });
+    }
+    function selectDistinct(fields) {
+      return new import_query_builders.CockroachSelectBuilder({
+        fields: fields ?? void 0,
+        session: self.session,
+        dialect: self.dialect,
+        withList: queries,
+        distinct: true
+      });
+    }
+    function selectDistinctOn(on, fields) {
+      return new import_query_builders.CockroachSelectBuilder({
+        fields: fields ?? void 0,
+        session: self.session,
+        dialect: self.dialect,
+        withList: queries,
+        distinct: { on }
+      });
+    }
+    function update(table) {
+      return new import_query_builders.CockroachUpdateBuilder(table, self.session, self.dialect, queries);
+    }
+    function insert(table) {
+      return new import_query_builders.CockroachInsertBuilder(table, self.session, self.dialect, queries);
+    }
+    function delete_(table) {
+      return new import_query_builders.CockroachDeleteBase(table, self.session, self.dialect, queries);
+    }
+    return { select, selectDistinct, selectDistinctOn, update, insert, delete: delete_ };
+  }
+  select(fields) {
+    return new import_query_builders.CockroachSelectBuilder({
+      fields: fields ?? void 0,
+      session: this.session,
+      dialect: this.dialect
+    });
+  }
+  selectDistinct(fields) {
+    return new import_query_builders.CockroachSelectBuilder({
+      fields: fields ?? void 0,
+      session: this.session,
+      dialect: this.dialect,
+      distinct: true
+    });
+  }
+  selectDistinctOn(on, fields) {
+    return new import_query_builders.CockroachSelectBuilder({
+      fields: fields ?? void 0,
+      session: this.session,
+      dialect: this.dialect,
+      distinct: { on }
+    });
+  }
+  /**
+   * Creates an update query.
+   *
+   * Calling this method without `.where()` clause will update all rows in a table. The `.where()` clause specifies which rows should be updated.
+   *
+   * Use `.set()` method to specify which values to update.
+   *
+   * See docs: {@link https://orm.drizzle.team/docs/update}
+   *
+   * @param table The table to update.
+   *
+   * @example
+   *
+   * ```ts
+   * // Update all rows in the 'cars' table
+   * await db.update(cars).set({ color: 'red' });
+   *
+   * // Update rows with filters and conditions
+   * await db.update(cars).set({ color: 'red' }).where(eq(cars.brand, 'BMW'));
+   *
+   * // Update with returning clause
+   * const updatedCar: Car[] = await db.update(cars)
+   *   .set({ color: 'red' })
+   *   .where(eq(cars.id, 1))
+   *   .returning();
+   * ```
+   */
+  update(table) {
+    return new import_query_builders.CockroachUpdateBuilder(table, this.session, this.dialect);
+  }
+  /**
+   * Creates an insert query.
+   *
+   * Calling this method will create new rows in a table. Use `.values()` method to specify which values to insert.
+   *
+   * See docs: {@link https://orm.drizzle.team/docs/insert}
+   *
+   * @param table The table to insert into.
+   *
+   * @example
+   *
+   * ```ts
+   * // Insert one row
+   * await db.insert(cars).values({ brand: 'BMW' });
+   *
+   * // Insert multiple rows
+   * await db.insert(cars).values([{ brand: 'BMW' }, { brand: 'Porsche' }]);
+   *
+   * // Insert with returning clause
+   * const insertedCar: Car[] = await db.insert(cars)
+   *   .values({ brand: 'BMW' })
+   *   .returning();
+   * ```
+   */
+  insert(table) {
+    return new import_query_builders.CockroachInsertBuilder(table, this.session, this.dialect);
+  }
+  /**
+   * Creates a delete query.
+   *
+   * Calling this method without `.where()` clause will delete all rows in a table. The `.where()` clause specifies which rows should be deleted.
+   *
+   * See docs: {@link https://orm.drizzle.team/docs/delete}
+   *
+   * @param table The table to delete from.
+   *
+   * @example
+   *
+   * ```ts
+   * // Delete all rows in the 'cars' table
+   * await db.delete(cars);
+   *
+   * // Delete rows with filters and conditions
+   * await db.delete(cars).where(eq(cars.color, 'green'));
+   *
+   * // Delete with returning clause
+   * const deletedCar: Car[] = await db.delete(cars)
+   *   .where(eq(cars.id, 1))
+   *   .returning();
+   * ```
+   */
+  delete(table) {
+    return new import_query_builders.CockroachDeleteBase(table, this.session, this.dialect);
+  }
+  refreshMaterializedView(view) {
+    return new import_refresh_materialized_view.CockroachRefreshMaterializedView(view, this.session, this.dialect);
+  }
+  authToken;
+  execute(query) {
+    const sequel = typeof query === "string" ? import_sql.sql.raw(query) : query.getSQL();
+    const builtQuery = this.dialect.sqlToQuery(sequel);
+    const prepared = this.session.prepareQuery(
+      builtQuery,
+      void 0,
+      void 0,
+      false
+    );
+    return new import_raw.CockroachRaw(
+      () => prepared.execute(void 0, this.authToken),
+      sequel,
+      builtQuery,
+      (result) => prepared.mapResult(result, true)
+    );
+  }
+  transaction(transaction, config) {
+    return this.session.transaction(transaction, config);
+  }
+}
+const withReplicas = (primary, replicas, getReplica = () => replicas[Math.floor(Math.random() * replicas.length)]) => {
+  const select = (...args) => getReplica(replicas).select(...args);
+  const selectDistinct = (...args) => getReplica(replicas).selectDistinct(...args);
+  const selectDistinctOn = (...args) => getReplica(replicas).selectDistinctOn(...args);
+  const $count = (...args) => getReplica(replicas).$count(...args);
+  const _with = (...args) => getReplica(replicas).with(...args);
+  const $with = (arg) => getReplica(replicas).$with(arg);
+  const update = (...args) => primary.update(...args);
+  const insert = (...args) => primary.insert(...args);
+  const $delete = (...args) => primary.delete(...args);
+  const execute = (...args) => primary.execute(...args);
+  const transaction = (...args) => primary.transaction(...args);
+  const refreshMaterializedView = (...args) => primary.refreshMaterializedView(...args);
+  return {
+    ...primary,
+    update,
+    insert,
+    delete: $delete,
+    execute,
+    transaction,
+    refreshMaterializedView,
+    $primary: primary,
+    $replicas: replicas,
+    select,
+    selectDistinct,
+    selectDistinctOn,
+    $count,
+    $with,
+    with: _with,
+    get _query() {
+      return getReplica(replicas)._query;
+    }
+  };
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  CockroachDatabase,
+  withReplicas
+});
+//# sourceMappingURL=db.cjs.map