Initial commit
This commit is contained in:
113
node_modules/drizzle-orm/singlestore/driver.cjs
generated
vendored
Normal file
113
node_modules/drizzle-orm/singlestore/driver.cjs
generated
vendored
Normal file
@ -0,0 +1,113 @@
|
||||
"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 driver_exports = {};
|
||||
__export(driver_exports, {
|
||||
SingleStoreDatabase: () => import_db2.SingleStoreDatabase,
|
||||
SingleStoreDriverDatabase: () => SingleStoreDriverDatabase,
|
||||
SingleStoreDriverDriver: () => SingleStoreDriverDriver,
|
||||
drizzle: () => drizzle
|
||||
});
|
||||
module.exports = __toCommonJS(driver_exports);
|
||||
var import_mysql2 = require("mysql2");
|
||||
var import_entity = require("../entity.cjs");
|
||||
var import_logger = require("../logger.cjs");
|
||||
var import_relations = require("../relations.cjs");
|
||||
var import_db = require("../singlestore-core/db.cjs");
|
||||
var import_dialect = require("../singlestore-core/dialect.cjs");
|
||||
var import_utils = require("../utils.cjs");
|
||||
var import_session = require("./session.cjs");
|
||||
var import_db2 = require("../singlestore-core/db.cjs");
|
||||
class SingleStoreDriverDriver {
|
||||
constructor(client, dialect, options = {}) {
|
||||
this.client = client;
|
||||
this.dialect = dialect;
|
||||
this.options = options;
|
||||
}
|
||||
static [import_entity.entityKind] = "SingleStoreDriverDriver";
|
||||
createSession(schema) {
|
||||
return new import_session.SingleStoreDriverSession(this.client, this.dialect, schema, { logger: this.options.logger });
|
||||
}
|
||||
}
|
||||
class SingleStoreDriverDatabase extends import_db.SingleStoreDatabase {
|
||||
static [import_entity.entityKind] = "SingleStoreDriverDatabase";
|
||||
}
|
||||
function construct(client, config = {}) {
|
||||
const dialect = new import_dialect.SingleStoreDialect({ casing: config.casing });
|
||||
let logger;
|
||||
if (config.logger === true) {
|
||||
logger = new import_logger.DefaultLogger();
|
||||
} else if (config.logger !== false) {
|
||||
logger = config.logger;
|
||||
}
|
||||
const clientForInstance = isCallbackClient(client) ? client.promise() : client;
|
||||
let schema;
|
||||
if (config.schema) {
|
||||
const tablesConfig = (0, import_relations.extractTablesRelationalConfig)(
|
||||
config.schema,
|
||||
import_relations.createTableRelationsHelpers
|
||||
);
|
||||
schema = {
|
||||
fullSchema: config.schema,
|
||||
schema: tablesConfig.tables,
|
||||
tableNamesMap: tablesConfig.tableNamesMap
|
||||
};
|
||||
}
|
||||
const driver = new SingleStoreDriverDriver(clientForInstance, dialect, { logger });
|
||||
const session = driver.createSession(schema);
|
||||
const db = new SingleStoreDriverDatabase(dialect, session, schema);
|
||||
db.$client = client;
|
||||
return db;
|
||||
}
|
||||
function isCallbackClient(client) {
|
||||
return typeof client.promise === "function";
|
||||
}
|
||||
function drizzle(...params) {
|
||||
if (typeof params[0] === "string") {
|
||||
const connectionString = params[0];
|
||||
const instance = (0, import_mysql2.createPool)({
|
||||
uri: connectionString
|
||||
});
|
||||
return construct(instance, params[1]);
|
||||
}
|
||||
if ((0, import_utils.isConfig)(params[0])) {
|
||||
const { connection, client, ...drizzleConfig } = params[0];
|
||||
if (client)
|
||||
return construct(client, drizzleConfig);
|
||||
const instance = typeof connection === "string" ? (0, import_mysql2.createPool)({
|
||||
uri: connection
|
||||
}) : (0, import_mysql2.createPool)(connection);
|
||||
const db = construct(instance, drizzleConfig);
|
||||
return db;
|
||||
}
|
||||
return construct(params[0], params[1]);
|
||||
}
|
||||
((drizzle2) => {
|
||||
function mock(config) {
|
||||
return construct({}, config);
|
||||
}
|
||||
drizzle2.mock = mock;
|
||||
})(drizzle || (drizzle = {}));
|
||||
// Annotate the CommonJS export names for ESM import in node:
|
||||
0 && (module.exports = {
|
||||
SingleStoreDatabase,
|
||||
SingleStoreDriverDatabase,
|
||||
SingleStoreDriverDriver,
|
||||
drizzle
|
||||
});
|
||||
//# sourceMappingURL=driver.cjs.map
|
||||
1
node_modules/drizzle-orm/singlestore/driver.cjs.map
generated
vendored
Normal file
1
node_modules/drizzle-orm/singlestore/driver.cjs.map
generated
vendored
Normal file
File diff suppressed because one or more lines are too long
52
node_modules/drizzle-orm/singlestore/driver.d.cts
generated
vendored
Normal file
52
node_modules/drizzle-orm/singlestore/driver.d.cts
generated
vendored
Normal file
@ -0,0 +1,52 @@
|
||||
import { type Connection as CallbackConnection, type Pool as CallbackPool, type PoolOptions } from 'mysql2';
|
||||
import type { Connection, Pool } from 'mysql2/promise';
|
||||
import { entityKind } from "../entity.cjs";
|
||||
import type { Logger } from "../logger.cjs";
|
||||
import { type RelationalSchemaConfig, type TablesRelationalConfig } from "../relations.cjs";
|
||||
import { SingleStoreDatabase } from "../singlestore-core/db.cjs";
|
||||
import { SingleStoreDialect } from "../singlestore-core/dialect.cjs";
|
||||
import { type DrizzleConfig, type IfNotImported, type ImportTypeError } from "../utils.cjs";
|
||||
import type { SingleStoreDriverClient, SingleStoreDriverPreparedQueryHKT, SingleStoreDriverQueryResultHKT } from "./session.cjs";
|
||||
import { SingleStoreDriverSession } from "./session.cjs";
|
||||
export interface SingleStoreDriverOptions {
|
||||
logger?: Logger;
|
||||
}
|
||||
export declare class SingleStoreDriverDriver {
|
||||
private client;
|
||||
private dialect;
|
||||
private options;
|
||||
static readonly [entityKind]: string;
|
||||
constructor(client: SingleStoreDriverClient, dialect: SingleStoreDialect, options?: SingleStoreDriverOptions);
|
||||
createSession(schema: RelationalSchemaConfig<TablesRelationalConfig> | undefined): SingleStoreDriverSession<Record<string, unknown>, TablesRelationalConfig>;
|
||||
}
|
||||
export { SingleStoreDatabase } from "../singlestore-core/db.cjs";
|
||||
export declare class SingleStoreDriverDatabase<TSchema extends Record<string, unknown> = Record<string, never>> extends SingleStoreDatabase<SingleStoreDriverQueryResultHKT, SingleStoreDriverPreparedQueryHKT, TSchema> {
|
||||
static readonly [entityKind]: string;
|
||||
}
|
||||
export type SingleStoreDriverDrizzleConfig<TSchema extends Record<string, unknown> = Record<string, never>> = Omit<DrizzleConfig<TSchema>, 'schema'> & ({
|
||||
schema: TSchema;
|
||||
} | {
|
||||
schema?: undefined;
|
||||
});
|
||||
export type AnySingleStoreDriverConnection = Pool | Connection | CallbackPool | CallbackConnection;
|
||||
export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>, TClient extends AnySingleStoreDriverConnection = CallbackPool>(...params: IfNotImported<CallbackPool, [
|
||||
ImportTypeError<'singlestore'>
|
||||
], [
|
||||
TClient | string
|
||||
] | [
|
||||
TClient | string,
|
||||
SingleStoreDriverDrizzleConfig<TSchema>
|
||||
] | [
|
||||
(SingleStoreDriverDrizzleConfig<TSchema> & ({
|
||||
connection: string | PoolOptions;
|
||||
} | {
|
||||
client: TClient;
|
||||
}))
|
||||
]>): SingleStoreDriverDatabase<TSchema> & {
|
||||
$client: TClient;
|
||||
};
|
||||
export declare namespace drizzle {
|
||||
function mock<TSchema extends Record<string, unknown> = Record<string, never>>(config?: SingleStoreDriverDrizzleConfig<TSchema>): SingleStoreDriverDatabase<TSchema> & {
|
||||
$client: '$client is not available on drizzle.mock()';
|
||||
};
|
||||
}
|
||||
52
node_modules/drizzle-orm/singlestore/driver.d.ts
generated
vendored
Normal file
52
node_modules/drizzle-orm/singlestore/driver.d.ts
generated
vendored
Normal file
@ -0,0 +1,52 @@
|
||||
import { type Connection as CallbackConnection, type Pool as CallbackPool, type PoolOptions } from 'mysql2';
|
||||
import type { Connection, Pool } from 'mysql2/promise';
|
||||
import { entityKind } from "../entity.js";
|
||||
import type { Logger } from "../logger.js";
|
||||
import { type RelationalSchemaConfig, type TablesRelationalConfig } from "../relations.js";
|
||||
import { SingleStoreDatabase } from "../singlestore-core/db.js";
|
||||
import { SingleStoreDialect } from "../singlestore-core/dialect.js";
|
||||
import { type DrizzleConfig, type IfNotImported, type ImportTypeError } from "../utils.js";
|
||||
import type { SingleStoreDriverClient, SingleStoreDriverPreparedQueryHKT, SingleStoreDriverQueryResultHKT } from "./session.js";
|
||||
import { SingleStoreDriverSession } from "./session.js";
|
||||
export interface SingleStoreDriverOptions {
|
||||
logger?: Logger;
|
||||
}
|
||||
export declare class SingleStoreDriverDriver {
|
||||
private client;
|
||||
private dialect;
|
||||
private options;
|
||||
static readonly [entityKind]: string;
|
||||
constructor(client: SingleStoreDriverClient, dialect: SingleStoreDialect, options?: SingleStoreDriverOptions);
|
||||
createSession(schema: RelationalSchemaConfig<TablesRelationalConfig> | undefined): SingleStoreDriverSession<Record<string, unknown>, TablesRelationalConfig>;
|
||||
}
|
||||
export { SingleStoreDatabase } from "../singlestore-core/db.js";
|
||||
export declare class SingleStoreDriverDatabase<TSchema extends Record<string, unknown> = Record<string, never>> extends SingleStoreDatabase<SingleStoreDriverQueryResultHKT, SingleStoreDriverPreparedQueryHKT, TSchema> {
|
||||
static readonly [entityKind]: string;
|
||||
}
|
||||
export type SingleStoreDriverDrizzleConfig<TSchema extends Record<string, unknown> = Record<string, never>> = Omit<DrizzleConfig<TSchema>, 'schema'> & ({
|
||||
schema: TSchema;
|
||||
} | {
|
||||
schema?: undefined;
|
||||
});
|
||||
export type AnySingleStoreDriverConnection = Pool | Connection | CallbackPool | CallbackConnection;
|
||||
export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>, TClient extends AnySingleStoreDriverConnection = CallbackPool>(...params: IfNotImported<CallbackPool, [
|
||||
ImportTypeError<'singlestore'>
|
||||
], [
|
||||
TClient | string
|
||||
] | [
|
||||
TClient | string,
|
||||
SingleStoreDriverDrizzleConfig<TSchema>
|
||||
] | [
|
||||
(SingleStoreDriverDrizzleConfig<TSchema> & ({
|
||||
connection: string | PoolOptions;
|
||||
} | {
|
||||
client: TClient;
|
||||
}))
|
||||
]>): SingleStoreDriverDatabase<TSchema> & {
|
||||
$client: TClient;
|
||||
};
|
||||
export declare namespace drizzle {
|
||||
function mock<TSchema extends Record<string, unknown> = Record<string, never>>(config?: SingleStoreDriverDrizzleConfig<TSchema>): SingleStoreDriverDatabase<TSchema> & {
|
||||
$client: '$client is not available on drizzle.mock()';
|
||||
};
|
||||
}
|
||||
89
node_modules/drizzle-orm/singlestore/driver.js
generated
vendored
Normal file
89
node_modules/drizzle-orm/singlestore/driver.js
generated
vendored
Normal file
@ -0,0 +1,89 @@
|
||||
import { createPool } from "mysql2";
|
||||
import { entityKind } from "../entity.js";
|
||||
import { DefaultLogger } from "../logger.js";
|
||||
import {
|
||||
createTableRelationsHelpers,
|
||||
extractTablesRelationalConfig
|
||||
} from "../relations.js";
|
||||
import { SingleStoreDatabase } from "../singlestore-core/db.js";
|
||||
import { SingleStoreDialect } from "../singlestore-core/dialect.js";
|
||||
import { isConfig } from "../utils.js";
|
||||
import { SingleStoreDriverSession } from "./session.js";
|
||||
class SingleStoreDriverDriver {
|
||||
constructor(client, dialect, options = {}) {
|
||||
this.client = client;
|
||||
this.dialect = dialect;
|
||||
this.options = options;
|
||||
}
|
||||
static [entityKind] = "SingleStoreDriverDriver";
|
||||
createSession(schema) {
|
||||
return new SingleStoreDriverSession(this.client, this.dialect, schema, { logger: this.options.logger });
|
||||
}
|
||||
}
|
||||
import { SingleStoreDatabase as SingleStoreDatabase2 } from "../singlestore-core/db.js";
|
||||
class SingleStoreDriverDatabase extends SingleStoreDatabase {
|
||||
static [entityKind] = "SingleStoreDriverDatabase";
|
||||
}
|
||||
function construct(client, config = {}) {
|
||||
const dialect = new SingleStoreDialect({ casing: config.casing });
|
||||
let logger;
|
||||
if (config.logger === true) {
|
||||
logger = new DefaultLogger();
|
||||
} else if (config.logger !== false) {
|
||||
logger = config.logger;
|
||||
}
|
||||
const clientForInstance = isCallbackClient(client) ? client.promise() : client;
|
||||
let schema;
|
||||
if (config.schema) {
|
||||
const tablesConfig = extractTablesRelationalConfig(
|
||||
config.schema,
|
||||
createTableRelationsHelpers
|
||||
);
|
||||
schema = {
|
||||
fullSchema: config.schema,
|
||||
schema: tablesConfig.tables,
|
||||
tableNamesMap: tablesConfig.tableNamesMap
|
||||
};
|
||||
}
|
||||
const driver = new SingleStoreDriverDriver(clientForInstance, dialect, { logger });
|
||||
const session = driver.createSession(schema);
|
||||
const db = new SingleStoreDriverDatabase(dialect, session, schema);
|
||||
db.$client = client;
|
||||
return db;
|
||||
}
|
||||
function isCallbackClient(client) {
|
||||
return typeof client.promise === "function";
|
||||
}
|
||||
function drizzle(...params) {
|
||||
if (typeof params[0] === "string") {
|
||||
const connectionString = params[0];
|
||||
const instance = createPool({
|
||||
uri: connectionString
|
||||
});
|
||||
return construct(instance, params[1]);
|
||||
}
|
||||
if (isConfig(params[0])) {
|
||||
const { connection, client, ...drizzleConfig } = params[0];
|
||||
if (client)
|
||||
return construct(client, drizzleConfig);
|
||||
const instance = typeof connection === "string" ? createPool({
|
||||
uri: connection
|
||||
}) : createPool(connection);
|
||||
const db = construct(instance, drizzleConfig);
|
||||
return db;
|
||||
}
|
||||
return construct(params[0], params[1]);
|
||||
}
|
||||
((drizzle2) => {
|
||||
function mock(config) {
|
||||
return construct({}, config);
|
||||
}
|
||||
drizzle2.mock = mock;
|
||||
})(drizzle || (drizzle = {}));
|
||||
export {
|
||||
SingleStoreDatabase2 as SingleStoreDatabase,
|
||||
SingleStoreDriverDatabase,
|
||||
SingleStoreDriverDriver,
|
||||
drizzle
|
||||
};
|
||||
//# sourceMappingURL=driver.js.map
|
||||
1
node_modules/drizzle-orm/singlestore/driver.js.map
generated
vendored
Normal file
1
node_modules/drizzle-orm/singlestore/driver.js.map
generated
vendored
Normal file
File diff suppressed because one or more lines are too long
25
node_modules/drizzle-orm/singlestore/index.cjs
generated
vendored
Normal file
25
node_modules/drizzle-orm/singlestore/index.cjs
generated
vendored
Normal file
@ -0,0 +1,25 @@
|
||||
"use strict";
|
||||
var __defProp = Object.defineProperty;
|
||||
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
||||
var __getOwnPropNames = Object.getOwnPropertyNames;
|
||||
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
||||
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 __reExport = (target, mod, secondTarget) => (__copyProps(target, mod, "default"), secondTarget && __copyProps(secondTarget, mod, "default"));
|
||||
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
||||
var singlestore_exports = {};
|
||||
module.exports = __toCommonJS(singlestore_exports);
|
||||
__reExport(singlestore_exports, require("./driver.cjs"), module.exports);
|
||||
__reExport(singlestore_exports, require("./session.cjs"), module.exports);
|
||||
// Annotate the CommonJS export names for ESM import in node:
|
||||
0 && (module.exports = {
|
||||
...require("./driver.cjs"),
|
||||
...require("./session.cjs")
|
||||
});
|
||||
//# sourceMappingURL=index.cjs.map
|
||||
1
node_modules/drizzle-orm/singlestore/index.cjs.map
generated
vendored
Normal file
1
node_modules/drizzle-orm/singlestore/index.cjs.map
generated
vendored
Normal file
@ -0,0 +1 @@
|
||||
{"version":3,"sources":["../../src/singlestore/index.ts"],"sourcesContent":["export * from './driver.ts';\nexport * from './session.ts';\n"],"mappings":";;;;;;;;;;;;;;;AAAA;AAAA;AAAA,gCAAc,wBAAd;AACA,gCAAc,yBADd;","names":[]}
|
||||
2
node_modules/drizzle-orm/singlestore/index.d.cts
generated
vendored
Normal file
2
node_modules/drizzle-orm/singlestore/index.d.cts
generated
vendored
Normal file
@ -0,0 +1,2 @@
|
||||
export * from "./driver.cjs";
|
||||
export * from "./session.cjs";
|
||||
2
node_modules/drizzle-orm/singlestore/index.d.ts
generated
vendored
Normal file
2
node_modules/drizzle-orm/singlestore/index.d.ts
generated
vendored
Normal file
@ -0,0 +1,2 @@
|
||||
export * from "./driver.js";
|
||||
export * from "./session.js";
|
||||
3
node_modules/drizzle-orm/singlestore/index.js
generated
vendored
Normal file
3
node_modules/drizzle-orm/singlestore/index.js
generated
vendored
Normal file
@ -0,0 +1,3 @@
|
||||
export * from "./driver.js";
|
||||
export * from "./session.js";
|
||||
//# sourceMappingURL=index.js.map
|
||||
1
node_modules/drizzle-orm/singlestore/index.js.map
generated
vendored
Normal file
1
node_modules/drizzle-orm/singlestore/index.js.map
generated
vendored
Normal file
@ -0,0 +1 @@
|
||||
{"version":3,"sources":["../../src/singlestore/index.ts"],"sourcesContent":["export * from './driver.ts';\nexport * from './session.ts';\n"],"mappings":"AAAA,cAAc;AACd,cAAc;","names":[]}
|
||||
33
node_modules/drizzle-orm/singlestore/migrator.cjs
generated
vendored
Normal file
33
node_modules/drizzle-orm/singlestore/migrator.cjs
generated
vendored
Normal file
@ -0,0 +1,33 @@
|
||||
"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 migrator_exports = {};
|
||||
__export(migrator_exports, {
|
||||
migrate: () => migrate
|
||||
});
|
||||
module.exports = __toCommonJS(migrator_exports);
|
||||
var import_migrator = require("../migrator.cjs");
|
||||
async function migrate(db, config) {
|
||||
const migrations = (0, import_migrator.readMigrationFiles)(config);
|
||||
await db.dialect.migrate(migrations, db.session, config);
|
||||
}
|
||||
// Annotate the CommonJS export names for ESM import in node:
|
||||
0 && (module.exports = {
|
||||
migrate
|
||||
});
|
||||
//# sourceMappingURL=migrator.cjs.map
|
||||
1
node_modules/drizzle-orm/singlestore/migrator.cjs.map
generated
vendored
Normal file
1
node_modules/drizzle-orm/singlestore/migrator.cjs.map
generated
vendored
Normal file
@ -0,0 +1 @@
|
||||
{"version":3,"sources":["../../src/singlestore/migrator.ts"],"sourcesContent":["import type { MigrationConfig } from '~/migrator.ts';\nimport { readMigrationFiles } from '~/migrator.ts';\nimport type { SingleStoreDriverDatabase } from './driver.ts';\n\nexport async function migrate<TSchema extends Record<string, unknown>>(\n\tdb: SingleStoreDriverDatabase<TSchema>,\n\tconfig: MigrationConfig,\n) {\n\tconst migrations = readMigrationFiles(config);\n\tawait db.dialect.migrate(migrations, db.session, config);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AACA,sBAAmC;AAGnC,eAAsB,QACrB,IACA,QACC;AACD,QAAM,iBAAa,oCAAmB,MAAM;AAC5C,QAAM,GAAG,QAAQ,QAAQ,YAAY,GAAG,SAAS,MAAM;AACxD;","names":[]}
|
||||
3
node_modules/drizzle-orm/singlestore/migrator.d.cts
generated
vendored
Normal file
3
node_modules/drizzle-orm/singlestore/migrator.d.cts
generated
vendored
Normal file
@ -0,0 +1,3 @@
|
||||
import type { MigrationConfig } from "../migrator.cjs";
|
||||
import type { SingleStoreDriverDatabase } from "./driver.cjs";
|
||||
export declare function migrate<TSchema extends Record<string, unknown>>(db: SingleStoreDriverDatabase<TSchema>, config: MigrationConfig): Promise<void>;
|
||||
3
node_modules/drizzle-orm/singlestore/migrator.d.ts
generated
vendored
Normal file
3
node_modules/drizzle-orm/singlestore/migrator.d.ts
generated
vendored
Normal file
@ -0,0 +1,3 @@
|
||||
import type { MigrationConfig } from "../migrator.js";
|
||||
import type { SingleStoreDriverDatabase } from "./driver.js";
|
||||
export declare function migrate<TSchema extends Record<string, unknown>>(db: SingleStoreDriverDatabase<TSchema>, config: MigrationConfig): Promise<void>;
|
||||
9
node_modules/drizzle-orm/singlestore/migrator.js
generated
vendored
Normal file
9
node_modules/drizzle-orm/singlestore/migrator.js
generated
vendored
Normal file
@ -0,0 +1,9 @@
|
||||
import { readMigrationFiles } from "../migrator.js";
|
||||
async function migrate(db, config) {
|
||||
const migrations = readMigrationFiles(config);
|
||||
await db.dialect.migrate(migrations, db.session, config);
|
||||
}
|
||||
export {
|
||||
migrate
|
||||
};
|
||||
//# sourceMappingURL=migrator.js.map
|
||||
1
node_modules/drizzle-orm/singlestore/migrator.js.map
generated
vendored
Normal file
1
node_modules/drizzle-orm/singlestore/migrator.js.map
generated
vendored
Normal file
@ -0,0 +1 @@
|
||||
{"version":3,"sources":["../../src/singlestore/migrator.ts"],"sourcesContent":["import type { MigrationConfig } from '~/migrator.ts';\nimport { readMigrationFiles } from '~/migrator.ts';\nimport type { SingleStoreDriverDatabase } from './driver.ts';\n\nexport async function migrate<TSchema extends Record<string, unknown>>(\n\tdb: SingleStoreDriverDatabase<TSchema>,\n\tconfig: MigrationConfig,\n) {\n\tconst migrations = readMigrationFiles(config);\n\tawait db.dialect.migrate(migrations, db.session, config);\n}\n"],"mappings":"AACA,SAAS,0BAA0B;AAGnC,eAAsB,QACrB,IACA,QACC;AACD,QAAM,aAAa,mBAAmB,MAAM;AAC5C,QAAM,GAAG,QAAQ,QAAQ,YAAY,GAAG,SAAS,MAAM;AACxD;","names":[]}
|
||||
258
node_modules/drizzle-orm/singlestore/session.cjs
generated
vendored
Normal file
258
node_modules/drizzle-orm/singlestore/session.cjs
generated
vendored
Normal file
@ -0,0 +1,258 @@
|
||||
"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 session_exports = {};
|
||||
__export(session_exports, {
|
||||
SingleStoreDriverPreparedQuery: () => SingleStoreDriverPreparedQuery,
|
||||
SingleStoreDriverSession: () => SingleStoreDriverSession,
|
||||
SingleStoreDriverTransaction: () => SingleStoreDriverTransaction
|
||||
});
|
||||
module.exports = __toCommonJS(session_exports);
|
||||
var import_node_events = require("node:events");
|
||||
var import_column = require("../column.cjs");
|
||||
var import_entity = require("../entity.cjs");
|
||||
var import_logger = require("../logger.cjs");
|
||||
var import_session = require("../singlestore-core/session.cjs");
|
||||
var import_sql = require("../sql/sql.cjs");
|
||||
var import_utils = require("../utils.cjs");
|
||||
class SingleStoreDriverPreparedQuery extends import_session.SingleStorePreparedQuery {
|
||||
constructor(client, queryString, params, logger, fields, customResultMapper, generatedIds, returningIds) {
|
||||
super();
|
||||
this.client = client;
|
||||
this.params = params;
|
||||
this.logger = logger;
|
||||
this.fields = fields;
|
||||
this.customResultMapper = customResultMapper;
|
||||
this.generatedIds = generatedIds;
|
||||
this.returningIds = returningIds;
|
||||
this.rawQuery = {
|
||||
sql: queryString,
|
||||
// rowsAsArray: true,
|
||||
typeCast: function(field, next) {
|
||||
if (field.type === "TIMESTAMP" || field.type === "DATETIME" || field.type === "DATE") {
|
||||
return field.string();
|
||||
}
|
||||
return next();
|
||||
}
|
||||
};
|
||||
this.query = {
|
||||
sql: queryString,
|
||||
rowsAsArray: true,
|
||||
typeCast: function(field, next) {
|
||||
if (field.type === "TIMESTAMP" || field.type === "DATETIME" || field.type === "DATE") {
|
||||
return field.string();
|
||||
}
|
||||
return next();
|
||||
}
|
||||
};
|
||||
}
|
||||
static [import_entity.entityKind] = "SingleStoreDriverPreparedQuery";
|
||||
rawQuery;
|
||||
query;
|
||||
async execute(placeholderValues = {}) {
|
||||
const params = (0, import_sql.fillPlaceholders)(this.params, placeholderValues);
|
||||
this.logger.logQuery(this.rawQuery.sql, params);
|
||||
const { fields, client, rawQuery, query, joinsNotNullableMap, customResultMapper, returningIds, generatedIds } = this;
|
||||
if (!fields && !customResultMapper) {
|
||||
const res = await client.query(rawQuery, params);
|
||||
const insertId = res[0].insertId;
|
||||
const affectedRows = res[0].affectedRows;
|
||||
if (returningIds) {
|
||||
const returningResponse = [];
|
||||
let j = 0;
|
||||
for (let i = insertId; i < insertId + affectedRows; i++) {
|
||||
for (const column of returningIds) {
|
||||
const key = returningIds[0].path[0];
|
||||
if ((0, import_entity.is)(column.field, import_column.Column)) {
|
||||
if (column.field.primary && column.field.autoIncrement) {
|
||||
returningResponse.push({ [key]: i });
|
||||
}
|
||||
if (column.field.defaultFn && generatedIds) {
|
||||
returningResponse.push({ [key]: generatedIds[j][key] });
|
||||
}
|
||||
}
|
||||
}
|
||||
j++;
|
||||
}
|
||||
return returningResponse;
|
||||
}
|
||||
return res;
|
||||
}
|
||||
const result = await client.query(query, params);
|
||||
const rows = result[0];
|
||||
if (customResultMapper) {
|
||||
return customResultMapper(rows);
|
||||
}
|
||||
return rows.map((row) => (0, import_utils.mapResultRow)(fields, row, joinsNotNullableMap));
|
||||
}
|
||||
async *iterator(placeholderValues = {}) {
|
||||
const params = (0, import_sql.fillPlaceholders)(this.params, placeholderValues);
|
||||
const conn = (isPool(this.client) ? await this.client.getConnection() : this.client).connection;
|
||||
const { fields, query, rawQuery, joinsNotNullableMap, client, customResultMapper } = this;
|
||||
const hasRowsMapper = Boolean(fields || customResultMapper);
|
||||
const driverQuery = hasRowsMapper ? conn.query(query, params) : conn.query(rawQuery, params);
|
||||
const stream = driverQuery.stream();
|
||||
function dataListener() {
|
||||
stream.pause();
|
||||
}
|
||||
stream.on("data", dataListener);
|
||||
try {
|
||||
const onEnd = (0, import_node_events.once)(stream, "end");
|
||||
const onError = (0, import_node_events.once)(stream, "error");
|
||||
while (true) {
|
||||
stream.resume();
|
||||
const row = await Promise.race([onEnd, onError, new Promise((resolve) => stream.once("data", resolve))]);
|
||||
if (row === void 0 || Array.isArray(row) && row.length === 0) {
|
||||
break;
|
||||
} else if (row instanceof Error) {
|
||||
throw row;
|
||||
} else {
|
||||
if (hasRowsMapper) {
|
||||
if (customResultMapper) {
|
||||
const mappedRow = customResultMapper([row]);
|
||||
yield Array.isArray(mappedRow) ? mappedRow[0] : mappedRow;
|
||||
} else {
|
||||
yield (0, import_utils.mapResultRow)(fields, row, joinsNotNullableMap);
|
||||
}
|
||||
} else {
|
||||
yield row;
|
||||
}
|
||||
}
|
||||
}
|
||||
} finally {
|
||||
stream.off("data", dataListener);
|
||||
if (isPool(client)) {
|
||||
conn.end();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
class SingleStoreDriverSession extends import_session.SingleStoreSession {
|
||||
constructor(client, dialect, schema, options) {
|
||||
super(dialect);
|
||||
this.client = client;
|
||||
this.schema = schema;
|
||||
this.options = options;
|
||||
this.logger = options.logger ?? new import_logger.NoopLogger();
|
||||
}
|
||||
static [import_entity.entityKind] = "SingleStoreDriverSession";
|
||||
logger;
|
||||
prepareQuery(query, fields, customResultMapper, generatedIds, returningIds) {
|
||||
return new SingleStoreDriverPreparedQuery(
|
||||
this.client,
|
||||
query.sql,
|
||||
query.params,
|
||||
this.logger,
|
||||
fields,
|
||||
customResultMapper,
|
||||
generatedIds,
|
||||
returningIds
|
||||
);
|
||||
}
|
||||
/**
|
||||
* @internal
|
||||
* What is its purpose?
|
||||
*/
|
||||
async query(query, params) {
|
||||
this.logger.logQuery(query, params);
|
||||
const result = await this.client.query({
|
||||
sql: query,
|
||||
values: params,
|
||||
rowsAsArray: true,
|
||||
typeCast: function(field, next) {
|
||||
if (field.type === "TIMESTAMP" || field.type === "DATETIME" || field.type === "DATE") {
|
||||
return field.string();
|
||||
}
|
||||
return next();
|
||||
}
|
||||
});
|
||||
return result;
|
||||
}
|
||||
all(query) {
|
||||
const querySql = this.dialect.sqlToQuery(query);
|
||||
this.logger.logQuery(querySql.sql, querySql.params);
|
||||
return this.client.execute(querySql.sql, querySql.params).then((result) => result[0]);
|
||||
}
|
||||
async transaction(transaction, config) {
|
||||
const session = isPool(this.client) ? new SingleStoreDriverSession(
|
||||
await this.client.getConnection(),
|
||||
this.dialect,
|
||||
this.schema,
|
||||
this.options
|
||||
) : this;
|
||||
const tx = new SingleStoreDriverTransaction(
|
||||
this.dialect,
|
||||
session,
|
||||
this.schema,
|
||||
0
|
||||
);
|
||||
if (config) {
|
||||
const setTransactionConfigSql = this.getSetTransactionSQL(config);
|
||||
if (setTransactionConfigSql) {
|
||||
await tx.execute(setTransactionConfigSql);
|
||||
}
|
||||
const startTransactionSql = this.getStartTransactionSQL(config);
|
||||
await (startTransactionSql ? tx.execute(startTransactionSql) : tx.execute(import_sql.sql`begin`));
|
||||
} else {
|
||||
await tx.execute(import_sql.sql`begin`);
|
||||
}
|
||||
try {
|
||||
const result = await transaction(tx);
|
||||
await tx.execute(import_sql.sql`commit`);
|
||||
return result;
|
||||
} catch (err) {
|
||||
await tx.execute(import_sql.sql`rollback`);
|
||||
throw err;
|
||||
} finally {
|
||||
if (isPool(this.client)) {
|
||||
session.client.release();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
class SingleStoreDriverTransaction extends import_session.SingleStoreTransaction {
|
||||
static [import_entity.entityKind] = "SingleStoreDriverTransaction";
|
||||
async transaction(transaction) {
|
||||
const savepointName = `sp${this.nestedIndex + 1}`;
|
||||
const tx = new SingleStoreDriverTransaction(
|
||||
this.dialect,
|
||||
this.session,
|
||||
this.schema,
|
||||
this.nestedIndex + 1
|
||||
);
|
||||
await tx.execute(import_sql.sql.raw(`savepoint ${savepointName}`));
|
||||
try {
|
||||
const result = await transaction(tx);
|
||||
await tx.execute(import_sql.sql.raw(`release savepoint ${savepointName}`));
|
||||
return result;
|
||||
} catch (err) {
|
||||
await tx.execute(import_sql.sql.raw(`rollback to savepoint ${savepointName}`));
|
||||
throw err;
|
||||
}
|
||||
}
|
||||
}
|
||||
function isPool(client) {
|
||||
return "getConnection" in client;
|
||||
}
|
||||
// Annotate the CommonJS export names for ESM import in node:
|
||||
0 && (module.exports = {
|
||||
SingleStoreDriverPreparedQuery,
|
||||
SingleStoreDriverSession,
|
||||
SingleStoreDriverTransaction
|
||||
});
|
||||
//# sourceMappingURL=session.cjs.map
|
||||
1
node_modules/drizzle-orm/singlestore/session.cjs.map
generated
vendored
Normal file
1
node_modules/drizzle-orm/singlestore/session.cjs.map
generated
vendored
Normal file
File diff suppressed because one or more lines are too long
52
node_modules/drizzle-orm/singlestore/session.d.cts
generated
vendored
Normal file
52
node_modules/drizzle-orm/singlestore/session.d.cts
generated
vendored
Normal file
@ -0,0 +1,52 @@
|
||||
import type { Connection, FieldPacket, OkPacket, Pool, ResultSetHeader, RowDataPacket } from 'mysql2/promise';
|
||||
import { entityKind } from "../entity.cjs";
|
||||
import type { Logger } from "../logger.cjs";
|
||||
import type { RelationalSchemaConfig, TablesRelationalConfig } from "../relations.cjs";
|
||||
import type { SingleStoreDialect } from "../singlestore-core/dialect.cjs";
|
||||
import type { SelectedFieldsOrdered } from "../singlestore-core/query-builders/select.types.cjs";
|
||||
import { type PreparedQueryKind, SingleStorePreparedQuery, type SingleStorePreparedQueryConfig, type SingleStorePreparedQueryHKT, type SingleStoreQueryResultHKT, SingleStoreSession, SingleStoreTransaction, type SingleStoreTransactionConfig } from "../singlestore-core/session.cjs";
|
||||
import type { Query, SQL } from "../sql/sql.cjs";
|
||||
import { type Assume } from "../utils.cjs";
|
||||
export type SingleStoreDriverClient = Pool | Connection;
|
||||
export type SingleStoreRawQueryResult = [ResultSetHeader, FieldPacket[]];
|
||||
export type SingleStoreQueryResultType = RowDataPacket[][] | RowDataPacket[] | OkPacket | OkPacket[] | ResultSetHeader;
|
||||
export type SingleStoreQueryResult<T = any> = [T extends ResultSetHeader ? T : T[], FieldPacket[]];
|
||||
export declare class SingleStoreDriverPreparedQuery<T extends SingleStorePreparedQueryConfig> extends SingleStorePreparedQuery<T> {
|
||||
private client;
|
||||
private params;
|
||||
private logger;
|
||||
private fields;
|
||||
private customResultMapper?;
|
||||
private generatedIds?;
|
||||
private returningIds?;
|
||||
static readonly [entityKind]: string;
|
||||
private rawQuery;
|
||||
private query;
|
||||
constructor(client: SingleStoreDriverClient, queryString: string, params: unknown[], logger: Logger, fields: SelectedFieldsOrdered | undefined, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined, generatedIds?: Record<string, unknown>[] | undefined, returningIds?: SelectedFieldsOrdered | undefined);
|
||||
execute(placeholderValues?: Record<string, unknown>): Promise<T['execute']>;
|
||||
iterator(placeholderValues?: Record<string, unknown>): AsyncGenerator<T['execute'] extends any[] ? T['execute'][number] : T['execute']>;
|
||||
}
|
||||
export interface SingleStoreDriverSessionOptions {
|
||||
logger?: Logger;
|
||||
}
|
||||
export declare class SingleStoreDriverSession<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends SingleStoreSession<SingleStoreQueryResultHKT, SingleStoreDriverPreparedQueryHKT, TFullSchema, TSchema> {
|
||||
private client;
|
||||
private schema;
|
||||
private options;
|
||||
static readonly [entityKind]: string;
|
||||
private logger;
|
||||
constructor(client: SingleStoreDriverClient, dialect: SingleStoreDialect, schema: RelationalSchemaConfig<TSchema> | undefined, options: SingleStoreDriverSessionOptions);
|
||||
prepareQuery<T extends SingleStorePreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, customResultMapper?: (rows: unknown[][]) => T['execute'], generatedIds?: Record<string, unknown>[], returningIds?: SelectedFieldsOrdered): PreparedQueryKind<SingleStoreDriverPreparedQueryHKT, T>;
|
||||
all<T = unknown>(query: SQL): Promise<T[]>;
|
||||
transaction<T>(transaction: (tx: SingleStoreDriverTransaction<TFullSchema, TSchema>) => Promise<T>, config?: SingleStoreTransactionConfig): Promise<T>;
|
||||
}
|
||||
export declare class SingleStoreDriverTransaction<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends SingleStoreTransaction<SingleStoreDriverQueryResultHKT, SingleStoreDriverPreparedQueryHKT, TFullSchema, TSchema> {
|
||||
static readonly [entityKind]: string;
|
||||
transaction<T>(transaction: (tx: SingleStoreDriverTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
|
||||
}
|
||||
export interface SingleStoreDriverQueryResultHKT extends SingleStoreQueryResultHKT {
|
||||
type: SingleStoreRawQueryResult;
|
||||
}
|
||||
export interface SingleStoreDriverPreparedQueryHKT extends SingleStorePreparedQueryHKT {
|
||||
type: SingleStoreDriverPreparedQuery<Assume<this['config'], SingleStorePreparedQueryConfig>>;
|
||||
}
|
||||
52
node_modules/drizzle-orm/singlestore/session.d.ts
generated
vendored
Normal file
52
node_modules/drizzle-orm/singlestore/session.d.ts
generated
vendored
Normal file
@ -0,0 +1,52 @@
|
||||
import type { Connection, FieldPacket, OkPacket, Pool, ResultSetHeader, RowDataPacket } from 'mysql2/promise';
|
||||
import { entityKind } from "../entity.js";
|
||||
import type { Logger } from "../logger.js";
|
||||
import type { RelationalSchemaConfig, TablesRelationalConfig } from "../relations.js";
|
||||
import type { SingleStoreDialect } from "../singlestore-core/dialect.js";
|
||||
import type { SelectedFieldsOrdered } from "../singlestore-core/query-builders/select.types.js";
|
||||
import { type PreparedQueryKind, SingleStorePreparedQuery, type SingleStorePreparedQueryConfig, type SingleStorePreparedQueryHKT, type SingleStoreQueryResultHKT, SingleStoreSession, SingleStoreTransaction, type SingleStoreTransactionConfig } from "../singlestore-core/session.js";
|
||||
import type { Query, SQL } from "../sql/sql.js";
|
||||
import { type Assume } from "../utils.js";
|
||||
export type SingleStoreDriverClient = Pool | Connection;
|
||||
export type SingleStoreRawQueryResult = [ResultSetHeader, FieldPacket[]];
|
||||
export type SingleStoreQueryResultType = RowDataPacket[][] | RowDataPacket[] | OkPacket | OkPacket[] | ResultSetHeader;
|
||||
export type SingleStoreQueryResult<T = any> = [T extends ResultSetHeader ? T : T[], FieldPacket[]];
|
||||
export declare class SingleStoreDriverPreparedQuery<T extends SingleStorePreparedQueryConfig> extends SingleStorePreparedQuery<T> {
|
||||
private client;
|
||||
private params;
|
||||
private logger;
|
||||
private fields;
|
||||
private customResultMapper?;
|
||||
private generatedIds?;
|
||||
private returningIds?;
|
||||
static readonly [entityKind]: string;
|
||||
private rawQuery;
|
||||
private query;
|
||||
constructor(client: SingleStoreDriverClient, queryString: string, params: unknown[], logger: Logger, fields: SelectedFieldsOrdered | undefined, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined, generatedIds?: Record<string, unknown>[] | undefined, returningIds?: SelectedFieldsOrdered | undefined);
|
||||
execute(placeholderValues?: Record<string, unknown>): Promise<T['execute']>;
|
||||
iterator(placeholderValues?: Record<string, unknown>): AsyncGenerator<T['execute'] extends any[] ? T['execute'][number] : T['execute']>;
|
||||
}
|
||||
export interface SingleStoreDriverSessionOptions {
|
||||
logger?: Logger;
|
||||
}
|
||||
export declare class SingleStoreDriverSession<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends SingleStoreSession<SingleStoreQueryResultHKT, SingleStoreDriverPreparedQueryHKT, TFullSchema, TSchema> {
|
||||
private client;
|
||||
private schema;
|
||||
private options;
|
||||
static readonly [entityKind]: string;
|
||||
private logger;
|
||||
constructor(client: SingleStoreDriverClient, dialect: SingleStoreDialect, schema: RelationalSchemaConfig<TSchema> | undefined, options: SingleStoreDriverSessionOptions);
|
||||
prepareQuery<T extends SingleStorePreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, customResultMapper?: (rows: unknown[][]) => T['execute'], generatedIds?: Record<string, unknown>[], returningIds?: SelectedFieldsOrdered): PreparedQueryKind<SingleStoreDriverPreparedQueryHKT, T>;
|
||||
all<T = unknown>(query: SQL): Promise<T[]>;
|
||||
transaction<T>(transaction: (tx: SingleStoreDriverTransaction<TFullSchema, TSchema>) => Promise<T>, config?: SingleStoreTransactionConfig): Promise<T>;
|
||||
}
|
||||
export declare class SingleStoreDriverTransaction<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends SingleStoreTransaction<SingleStoreDriverQueryResultHKT, SingleStoreDriverPreparedQueryHKT, TFullSchema, TSchema> {
|
||||
static readonly [entityKind]: string;
|
||||
transaction<T>(transaction: (tx: SingleStoreDriverTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
|
||||
}
|
||||
export interface SingleStoreDriverQueryResultHKT extends SingleStoreQueryResultHKT {
|
||||
type: SingleStoreRawQueryResult;
|
||||
}
|
||||
export interface SingleStoreDriverPreparedQueryHKT extends SingleStorePreparedQueryHKT {
|
||||
type: SingleStoreDriverPreparedQuery<Assume<this['config'], SingleStorePreparedQueryConfig>>;
|
||||
}
|
||||
236
node_modules/drizzle-orm/singlestore/session.js
generated
vendored
Normal file
236
node_modules/drizzle-orm/singlestore/session.js
generated
vendored
Normal file
@ -0,0 +1,236 @@
|
||||
import { once } from "node:events";
|
||||
import { Column } from "../column.js";
|
||||
import { entityKind, is } from "../entity.js";
|
||||
import { NoopLogger } from "../logger.js";
|
||||
import {
|
||||
SingleStorePreparedQuery,
|
||||
SingleStoreSession,
|
||||
SingleStoreTransaction
|
||||
} from "../singlestore-core/session.js";
|
||||
import { fillPlaceholders, sql } from "../sql/sql.js";
|
||||
import { mapResultRow } from "../utils.js";
|
||||
class SingleStoreDriverPreparedQuery extends SingleStorePreparedQuery {
|
||||
constructor(client, queryString, params, logger, fields, customResultMapper, generatedIds, returningIds) {
|
||||
super();
|
||||
this.client = client;
|
||||
this.params = params;
|
||||
this.logger = logger;
|
||||
this.fields = fields;
|
||||
this.customResultMapper = customResultMapper;
|
||||
this.generatedIds = generatedIds;
|
||||
this.returningIds = returningIds;
|
||||
this.rawQuery = {
|
||||
sql: queryString,
|
||||
// rowsAsArray: true,
|
||||
typeCast: function(field, next) {
|
||||
if (field.type === "TIMESTAMP" || field.type === "DATETIME" || field.type === "DATE") {
|
||||
return field.string();
|
||||
}
|
||||
return next();
|
||||
}
|
||||
};
|
||||
this.query = {
|
||||
sql: queryString,
|
||||
rowsAsArray: true,
|
||||
typeCast: function(field, next) {
|
||||
if (field.type === "TIMESTAMP" || field.type === "DATETIME" || field.type === "DATE") {
|
||||
return field.string();
|
||||
}
|
||||
return next();
|
||||
}
|
||||
};
|
||||
}
|
||||
static [entityKind] = "SingleStoreDriverPreparedQuery";
|
||||
rawQuery;
|
||||
query;
|
||||
async execute(placeholderValues = {}) {
|
||||
const params = fillPlaceholders(this.params, placeholderValues);
|
||||
this.logger.logQuery(this.rawQuery.sql, params);
|
||||
const { fields, client, rawQuery, query, joinsNotNullableMap, customResultMapper, returningIds, generatedIds } = this;
|
||||
if (!fields && !customResultMapper) {
|
||||
const res = await client.query(rawQuery, params);
|
||||
const insertId = res[0].insertId;
|
||||
const affectedRows = res[0].affectedRows;
|
||||
if (returningIds) {
|
||||
const returningResponse = [];
|
||||
let j = 0;
|
||||
for (let i = insertId; i < insertId + affectedRows; i++) {
|
||||
for (const column of returningIds) {
|
||||
const key = returningIds[0].path[0];
|
||||
if (is(column.field, Column)) {
|
||||
if (column.field.primary && column.field.autoIncrement) {
|
||||
returningResponse.push({ [key]: i });
|
||||
}
|
||||
if (column.field.defaultFn && generatedIds) {
|
||||
returningResponse.push({ [key]: generatedIds[j][key] });
|
||||
}
|
||||
}
|
||||
}
|
||||
j++;
|
||||
}
|
||||
return returningResponse;
|
||||
}
|
||||
return res;
|
||||
}
|
||||
const result = await client.query(query, params);
|
||||
const rows = result[0];
|
||||
if (customResultMapper) {
|
||||
return customResultMapper(rows);
|
||||
}
|
||||
return rows.map((row) => mapResultRow(fields, row, joinsNotNullableMap));
|
||||
}
|
||||
async *iterator(placeholderValues = {}) {
|
||||
const params = fillPlaceholders(this.params, placeholderValues);
|
||||
const conn = (isPool(this.client) ? await this.client.getConnection() : this.client).connection;
|
||||
const { fields, query, rawQuery, joinsNotNullableMap, client, customResultMapper } = this;
|
||||
const hasRowsMapper = Boolean(fields || customResultMapper);
|
||||
const driverQuery = hasRowsMapper ? conn.query(query, params) : conn.query(rawQuery, params);
|
||||
const stream = driverQuery.stream();
|
||||
function dataListener() {
|
||||
stream.pause();
|
||||
}
|
||||
stream.on("data", dataListener);
|
||||
try {
|
||||
const onEnd = once(stream, "end");
|
||||
const onError = once(stream, "error");
|
||||
while (true) {
|
||||
stream.resume();
|
||||
const row = await Promise.race([onEnd, onError, new Promise((resolve) => stream.once("data", resolve))]);
|
||||
if (row === void 0 || Array.isArray(row) && row.length === 0) {
|
||||
break;
|
||||
} else if (row instanceof Error) {
|
||||
throw row;
|
||||
} else {
|
||||
if (hasRowsMapper) {
|
||||
if (customResultMapper) {
|
||||
const mappedRow = customResultMapper([row]);
|
||||
yield Array.isArray(mappedRow) ? mappedRow[0] : mappedRow;
|
||||
} else {
|
||||
yield mapResultRow(fields, row, joinsNotNullableMap);
|
||||
}
|
||||
} else {
|
||||
yield row;
|
||||
}
|
||||
}
|
||||
}
|
||||
} finally {
|
||||
stream.off("data", dataListener);
|
||||
if (isPool(client)) {
|
||||
conn.end();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
class SingleStoreDriverSession extends SingleStoreSession {
|
||||
constructor(client, dialect, schema, options) {
|
||||
super(dialect);
|
||||
this.client = client;
|
||||
this.schema = schema;
|
||||
this.options = options;
|
||||
this.logger = options.logger ?? new NoopLogger();
|
||||
}
|
||||
static [entityKind] = "SingleStoreDriverSession";
|
||||
logger;
|
||||
prepareQuery(query, fields, customResultMapper, generatedIds, returningIds) {
|
||||
return new SingleStoreDriverPreparedQuery(
|
||||
this.client,
|
||||
query.sql,
|
||||
query.params,
|
||||
this.logger,
|
||||
fields,
|
||||
customResultMapper,
|
||||
generatedIds,
|
||||
returningIds
|
||||
);
|
||||
}
|
||||
/**
|
||||
* @internal
|
||||
* What is its purpose?
|
||||
*/
|
||||
async query(query, params) {
|
||||
this.logger.logQuery(query, params);
|
||||
const result = await this.client.query({
|
||||
sql: query,
|
||||
values: params,
|
||||
rowsAsArray: true,
|
||||
typeCast: function(field, next) {
|
||||
if (field.type === "TIMESTAMP" || field.type === "DATETIME" || field.type === "DATE") {
|
||||
return field.string();
|
||||
}
|
||||
return next();
|
||||
}
|
||||
});
|
||||
return result;
|
||||
}
|
||||
all(query) {
|
||||
const querySql = this.dialect.sqlToQuery(query);
|
||||
this.logger.logQuery(querySql.sql, querySql.params);
|
||||
return this.client.execute(querySql.sql, querySql.params).then((result) => result[0]);
|
||||
}
|
||||
async transaction(transaction, config) {
|
||||
const session = isPool(this.client) ? new SingleStoreDriverSession(
|
||||
await this.client.getConnection(),
|
||||
this.dialect,
|
||||
this.schema,
|
||||
this.options
|
||||
) : this;
|
||||
const tx = new SingleStoreDriverTransaction(
|
||||
this.dialect,
|
||||
session,
|
||||
this.schema,
|
||||
0
|
||||
);
|
||||
if (config) {
|
||||
const setTransactionConfigSql = this.getSetTransactionSQL(config);
|
||||
if (setTransactionConfigSql) {
|
||||
await tx.execute(setTransactionConfigSql);
|
||||
}
|
||||
const startTransactionSql = this.getStartTransactionSQL(config);
|
||||
await (startTransactionSql ? tx.execute(startTransactionSql) : tx.execute(sql`begin`));
|
||||
} else {
|
||||
await tx.execute(sql`begin`);
|
||||
}
|
||||
try {
|
||||
const result = await transaction(tx);
|
||||
await tx.execute(sql`commit`);
|
||||
return result;
|
||||
} catch (err) {
|
||||
await tx.execute(sql`rollback`);
|
||||
throw err;
|
||||
} finally {
|
||||
if (isPool(this.client)) {
|
||||
session.client.release();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
class SingleStoreDriverTransaction extends SingleStoreTransaction {
|
||||
static [entityKind] = "SingleStoreDriverTransaction";
|
||||
async transaction(transaction) {
|
||||
const savepointName = `sp${this.nestedIndex + 1}`;
|
||||
const tx = new SingleStoreDriverTransaction(
|
||||
this.dialect,
|
||||
this.session,
|
||||
this.schema,
|
||||
this.nestedIndex + 1
|
||||
);
|
||||
await tx.execute(sql.raw(`savepoint ${savepointName}`));
|
||||
try {
|
||||
const result = await transaction(tx);
|
||||
await tx.execute(sql.raw(`release savepoint ${savepointName}`));
|
||||
return result;
|
||||
} catch (err) {
|
||||
await tx.execute(sql.raw(`rollback to savepoint ${savepointName}`));
|
||||
throw err;
|
||||
}
|
||||
}
|
||||
}
|
||||
function isPool(client) {
|
||||
return "getConnection" in client;
|
||||
}
|
||||
export {
|
||||
SingleStoreDriverPreparedQuery,
|
||||
SingleStoreDriverSession,
|
||||
SingleStoreDriverTransaction
|
||||
};
|
||||
//# sourceMappingURL=session.js.map
|
||||
1
node_modules/drizzle-orm/singlestore/session.js.map
generated
vendored
Normal file
1
node_modules/drizzle-orm/singlestore/session.js.map
generated
vendored
Normal file
File diff suppressed because one or more lines are too long
Reference in New Issue
Block a user