hysteria-orm 10.3.3 → 10.3.5

package/lib/index.d.cts

@@ -1107,10 +1107,46 @@ type ConnectionPolicies = {
     };
 };
 type SqlDataSourceModel = typeof Model;
+/**
+ * @description Base migration configuration options available for all databases
+ */
+type MigrationConfigBase = {
+    /**
+     * @description The path to the migrations folder, can be overridden in the cli command
+     * @default "database/migrations"
+     */
+    path?: string;
+    /**
+     * @description Path to the tsconfig.json file for TypeScript migration files, can be overridden in the cli command
+     * @default "./tsconfig.json"
+     */
+    tsconfig?: string;
+    /**
+     * @description Acquire advisory lock before running migrations to prevent concurrent execution, can be overridden in the cli command
+     * @default true
+     */
+    lock?: boolean;
+};
+/**
+ * @description Migration configuration with transactional support for PostgreSQL and CockroachDB
+ */
+type MigrationConfigWithTransactional = MigrationConfigBase & {
+    /**
+     * @description Runs all pending migrations in a single transaction, can be overridden in the cli command
+     * @default true
+     * @note Only available for PostgreSQL and CockroachDB
+     */
+    transactional?: boolean;
+};
+/**
+ * @description Migration configuration type based on database type
+ * @description Adds transactional option only for PostgreSQL and CockroachDB
+ */
+type MigrationConfig<D extends SqlDataSourceType = SqlDataSourceType> = D extends "postgres" | "cockroachdb" ? MigrationConfigWithTransactional : MigrationConfigBase;
 /**
  * @description Common input properties shared across all SqlDataSource types
  */
-type SqlDataSourceInputBase<T extends Record<string, SqlDataSourceModel> = {}, C extends CacheKeys = {}> = {
+type SqlDataSourceInputBase<T extends Record<string, SqlDataSourceModel> = {}, C extends CacheKeys = {}, D extends SqlDataSourceType = SqlDataSourceType> = {
     /**
      * @description Whether to log the sql queries and other debug information
      */
@@ -1129,10 +1165,9 @@ type SqlDataSourceInputBase<T extends Record<string, SqlDataSourceModel> = {}, C
      */
     models?: T;
     /**
-     * @description The path to the migrations folder for the sql data source, it's used to configure the migrations path for the sql data source
-     * @default "database/migrations"
+     * @description Migration configuration for the sql data source
      */
-    migrationsPath?: string;
+    migrations?: MigrationConfig<D>;
     /**
      * @description The cache strategy to use for the sql data source, it's used to configure the cache strategy for the sql data source
      */
@@ -1150,11 +1185,19 @@ type SqlDataSourceInputBase<T extends Record<string, SqlDataSourceModel> = {}, C
  * @description Maps a SqlDataSourceType to its corresponding input interface
  */
 type MapSqlDataSourceTypeToInput<D extends SqlDataSourceType> = D extends "mysql" | "mariadb" ? MysqlSqlDataSourceInput : D extends "postgres" | "cockroachdb" ? PostgresSqlDataSourceInput : D extends "sqlite" ? SqliteDataSourceInput : D extends "mssql" ? MssqlDataSourceInput : D extends "oracledb" ? OracleDBDataSourceInput : never;
+type SlaveContext = {
+    type: SqlDataSourceType;
+    host: string;
+    port: number;
+    username: string;
+    password: string;
+    database: string;
+};
 /**
  * @description The input type for the SqlDataSource constructor
  * @description The connectionPolicies object is used to configure the connection policies for the sql data source
  */
-type SqlDataSourceInput<D extends SqlDataSourceType = SqlDataSourceType, T extends Record<string, SqlDataSourceModel> = {}, C extends CacheKeys = {}> = SqlDataSourceInputBase<T, C> & {
+type SqlDataSourceInput<D extends SqlDataSourceType = SqlDataSourceType, T extends Record<string, SqlDataSourceModel> = {}, C extends CacheKeys = {}> = SqlDataSourceInputBase<T, C, D> & {
     /**
      * @description The type of the database to connect to
      */
@@ -1168,10 +1211,14 @@ type SqlDataSourceInput<D extends SqlDataSourceType = SqlDataSourceType, T exten
      * @description The replication configuration for the sql data source, it's used to configure the replication for the sql data source
      */
     replication?: {
+        /**
+         * @description The function to call when a slave server fails, if not provided an error will be thrown
+         */
+        onSlaveServerFailure?: (error: Error, context: SlaveContext) => void | Promise<void>;
         /**
          * @description The slaves data sources to use for the sql data source, slaves are automatically used for read operations unless specified otherwise
          */
-        slaves?: Omit<UseConnectionInput<D, T, C>, "slaves" | "models" | "cacheStrategy" | "adminJs" | "logs" | "queryFormatOptions" | "migrationsPath">[];
+        slaves?: Omit<UseConnectionInput<D, T, C>, "slaves" | "models" | "cacheStrategy" | "adminJs" | "logs" | "queryFormatOptions" | "migrations">[];
         /**
          * @description The algorithm to use for selecting the slave for read operations
          * @default "roundRobin" - Distributes requests evenly across all slaves in sequence
@@ -1544,6 +1591,14 @@ declare class CreateTableBuilder extends BaseBuilder {
     private sqlType;
     constructor(sqlType: SqlDataSourceType, nodes: QueryNode[], tableName?: string, context?: "alter_table" | "create_table");
     private build;
+    /**
+     * @description Adds a raw statement to an operation that will be executed as is
+     * @example
+     * ```ts
+     * table.varchar("name").default(table.rawStatement("CURRENT_TIMESTAMP"));
+     * ```
+     */
+    rawStatement(value: string): RawNode;
     /**
      * Fixed-length character string
      * @mysql Supported as CHAR(length)
@@ -1825,6 +1880,14 @@ declare class AlterTableBuilder extends BaseBuilder {
     private table;
     private sqlType;
     constructor(table: string, nodes: QueryNode[], sqlType: SqlDataSourceType);
+    /**
+     * @description Adds a raw statement to an operation that will be executed as is
+     * @example
+     * ```ts
+     * table.varchar("name").default(table.rawStatement("CURRENT_TIMESTAMP"));
+     * ```
+     */
+    rawStatement(value: string): RawNode;
     /**
      * @description Adds a column to the table
      * @param cb is the callback that will be used to build the column
@@ -1918,6 +1981,17 @@ declare class Schema {
     queryStatements: string[];
     sqlType: SqlDataSourceType;
     constructor(sqlType?: SqlDataSourceType);
+    /**
+     * @description Adds a raw statement to an operation that will be executed as is
+     * @example
+     * ```ts
+     * schema.rawStatement("CURRENT_TIMESTAMP");
+     * schema.alterTable("users", (table) => {
+     *   table.timestamp("created_at").default(this.schema.rawStatement("CURRENT_TIMESTAMP"));
+     * });
+     * ```
+     */
+    rawStatement(value: string): RawNode;
     /**
      * @description Add raw query to the migration
      */
@@ -2037,7 +2111,7 @@ declare class DryQueryBuilder extends QueryBuilder {
      * @warning This method does not run model or column hooks
      * @returns The query builder
      */
-    update(data: Record<string, any>): this;
+    update(data: Record<string, WriteQueryParam>): this;
     /**
      * @description Builds the delete query statement without executing it, use 'unWrap' or 'toQuery' to get the query statement
      * @warning This method does not run model or column hooks
@@ -3107,6 +3181,7 @@ type UpsertOptionsRawBuilder = {
 };
 type DryQueryBuilderWithoutReadOperations = Omit<DryQueryBuilder, "many" | "one" | "oneOrFail" | "first" | "firstOrFail" | "paginate" | "paginateWithCursor" | "exists" | "pluck" | "increment" | "decrement" | "getSum" | "getAvg" | "getMin" | "getMax" | "getCount" | "stream" | "chunk" | "paginate" | "paginateWithCursor" | "exists">;
 type DryModelQueryBuilderWithoutReadOperations<T extends Model, A extends Record<string, any> = {}, R extends Record<string, any> = {}> = Omit<DryModelQueryBuilder<T, A, R>, "many" | "one" | "oneOrFail" | "first" | "firstOrFail" | "paginate" | "paginateWithCursor" | "exists" | "pluck" | "upsert" | "upsertMany" | "increment" | "decrement" | "getSum" | "getAvg" | "getMin" | "getMax" | "getCount" | "stream" | "chunk" | "paginate" | "paginateWithCursor" | "exists">;
+type WriteQueryParam$1 = string | number | boolean | Date | RawNode | object | null | undefined;
 
 declare class SqlModelManagerUtils<T extends Model> {
     protected dbType: SqlDataSourceType;
@@ -3698,9 +3773,14 @@ declare class SqlDataSource<D extends SqlDataSourceType = SqlDataSourceType, T e
      */
     cacheKeys: C;
     /**
-     * @description The path to the migrations folder for the sql data source, it's used to configure the migrations path for the sql data source
+     * @description Migration configuration for the sql data source
      */
-    migrationsPath: string;
+    migrationConfig: {
+        path: string;
+        tsconfig?: string;
+        lock: boolean;
+        transactional: boolean;
+    };
     /**
      * @description AdminJS configuration options
      */
@@ -3709,6 +3789,14 @@ declare class SqlDataSource<D extends SqlDataSourceType = SqlDataSourceType, T e
      * @description Cached AdminJS instance
      */
     private adminJsInstance?;
+    /**
+     * @description Callback to handle slave server failures
+     */
+    private onSlaveServerFailure?;
+    /**
+     * @description Returns the configured slave failure callback
+     */
+    getOnSlaveServerFailure(): ((error: Error, context: SlaveContext) => void | Promise<void>) | undefined;
     /**
      * @description Returns the primary instance of the SqlDataSource (set via connect with setPrimary: true)
      * All models by default will use this instance to execute queries unless you pass a different connection/transaction in the query options
@@ -3938,10 +4026,16 @@ declare class SqlDataSource<D extends SqlDataSourceType = SqlDataSourceType, T e
      */
     rawQuery<R = RawQueryResponseType<D>>(query: string, params?: any[], options?: RawQueryOptions): Promise<R>;
     /**
-     * @description Adds a raw statement to an operation like where or update
+     * @description Adds a raw statement to an operation that will be executed as is
      * @example
      * ```ts
-     * await User.query().where("name", sql.rawStatement("LOWER(name)"));
+     * await sql.query("users").where("name", sql.rawStatement("LOWER(name)"));
+     * await sql.query("users").update({
+     *   name: sql.rawStatement("LOWER(name)"),
+     * });
+     * await sql.query("users").insert({
+     *   name: sql.rawStatement("LOWER(name)"),
+     * });
      * ```
      */
     rawStatement(value: string): RawNode;
@@ -3995,6 +4089,30 @@ declare class SqlDataSource<D extends SqlDataSourceType = SqlDataSourceType, T e
      * @description Introspects table primary key from the database
      */
     getPrimaryKeyInfo(table: string): Promise<TablePrimaryKeyInfo | undefined>;
+    /**
+     * @description Acquires an advisory lock
+     * @description Useful for preventing concurrent operations from running simultaneously
+     * @param lockKey - The lock identifier (defaults to 'hysteria_lock')
+     * @param timeoutMs - Maximum time to wait for lock in milliseconds (postgres/mssql only)
+     * @returns true if lock was acquired, false otherwise
+     */
+    acquireLock(lockKey?: string, timeoutMs?: number): Promise<boolean>;
+    /**
+     * @description Releases an advisory lock
+     * @param lockKey - The lock identifier (defaults to 'hysteria_lock')
+     * @returns true if lock was released, false otherwise
+     */
+    releaseLock(lockKey?: string): Promise<boolean>;
+    /**
+     * @description Converts a string to a numeric lock ID for databases that require it
+     */
+    private hashStringToLockId;
+    /**
+     * @description Executes an operation on a slave, handling failures with the configured callback
+     * @param operation The operation to execute on the slave
+     * @returns The result of the operation, or falls back to master if slave fails
+     */
+    private executeOnSlave;
     /**
      * @description Internal method to establish connection without setting as primary instance
      * @description Used by connectToSecondarySource and useConnection
@@ -4328,18 +4446,18 @@ declare class QueryBuilder<T extends Model = any> extends JsonQueryBuilder<T> {
      */
     withMaterialized(alias: string, cb: (qb: QueryBuilder<T>) => void): this;
     /**
-     * @description Insert record into a table
+     * @description Insert record into a table, you can use raw statements in the data object for literal references to other columns
      * @param returning - The columns to return from the query, only supported by postgres and cockroachdb - default is "*"
      * @returns raw driver response
      */
-    insert(data: Record<string, any>, returning?: string[]): Promise<T>;
+    insert(data: Record<string, WriteQueryParam$1>, returning?: string[]): Promise<T>;
     /**
      * @description Insert multiple records into a table
      * @param returning - The columns to return from the query, only supported by postgres and cockroachdb - default is "*"
      * @returns raw driver response
      * @oracledb may do multiple inserts with auto-generated identity columns
      */
-    insertMany(data: Record<string, any>[], returning?: string[]): Promise<T[]>;
+    insertMany(data: Record<string, WriteQueryParam$1>[], returning?: string[]): Promise<T[]>;
     /**
      * @description Updates or creates a new record using upsert functionality
      * @param data The data to insert or update
@@ -4361,10 +4479,10 @@ declare class QueryBuilder<T extends Model = any> extends JsonQueryBuilder<T> {
      */
     private executeMssqlMergeRaw;
     /**
-     * @description Updates records from a table
+     * @description Updates records from a table, you can use raw statements in the data object for literal references to other columns
      * @returns the number of affected rows
      */
-    update(data: Record<string, any>): Promise<number>;
+    update(data: Record<string, WriteQueryParam$1>): Promise<number>;
     /**
      * @description Deletes all records from a table
      * @warning This operation does not trigger any hook
@@ -4456,6 +4574,13 @@ declare class QueryBuilder<T extends Model = any> extends JsonQueryBuilder<T> {
      */
     protected executePaginateQueries<M, C>(modelsQuery: () => Promise<M>, countQuery: () => Promise<C>): Promise<[M, C]>;
     protected getSqlDataSource(mode: "read" | "write"): Promise<SqlDataSource>;
+    /**
+     * @description Executes SQL with slave failure handling
+     * @param mode The operation mode (read or write)
+     * @param operation The execSql operation to perform
+     * @returns The result of the operation
+     */
+    protected execSqlWithSlaveHandling<R>(mode: "read" | "write", operation: (dataSource: SqlDataSource) => Promise<R>): Promise<R>;
 }
 
 type UnionCallBack<T extends Model> = (queryBuilder: QueryBuilder<T>) => QueryBuilder<T>;
@@ -5934,4 +6059,4 @@ declare const generateOpenApiModelWithMetadata: <T extends new () => Model>(mode
     $id?: string;
 }>;
 
-export { type AdminJsActionOptions, type AdminJsAssets, type AdminJsBranding, type AdminJsInstance, type AdminJsLocale, type AdminJsOptions, type AdminJsPage, type AdminJsPropertyOptions, type AdminJsResourceOptions, type AdminJsSettings, type AnnotatedModel, type AsymmetricEncryptionOptions, AutogeneratedModel, type BaseModelMethodOptions, type BaseModelRelationType, type CacheAdapter, type CacheKeys, ClientMigrator, Collection, type ColumnDataTypeOption, type ColumnDataTypeOptionSimple, type ColumnDataTypeOptionWithBinary, type ColumnDataTypeOptionWithDatePrecision, type ColumnDataTypeOptionWithEnum, type ColumnDataTypeOptionWithLength, type ColumnDataTypeOptionWithPrecision, type ColumnDataTypeOptionWithScaleAndPrecision, type ColumnDataTypeOptionWithText, type ColumnOptions, type ColumnType, type CommonDataSourceInput, type CommonSqlMethodReturnType, type ConnectionPolicies, type DataSourceInput, type DataSourceType, type DateColumnOptions, DryModelQueryBuilder, DryQueryBuilder, type FetchHooks, type GetConnectionReturnType, HysteriaError, InMemoryAdapter, type IndexType, type LazyRelationType, type ManyOptions, type ManyToManyOptions, type ManyToManyStringOptions, Migration, Model, type ModelInstanceType, ModelQueryBuilder, type ModelWithoutRelations, MongoDataSource, type MongoDataSourceInput$1 as MongoDataSourceInput, type MssqlConnectionInstance, type MssqlDataSourceInput, type MssqlPoolInstance, type MysqlConnectionInstance, type MysqlSqlDataSourceInput, type NotNullableMysqlSqlDataSourceInput, type NotNullableOracleDBDataSourceInput, type NotNullableOracleMssqlDataSourceInput, type NotNullablePostgresSqlDataSourceInput, type NotNullableSqliteDataSourceInput, type NumberModelKey, type OneOptions, type OracleDBDataSourceInput, type OracleDBPoolInstance, type PgPoolClientInstance, type PostgresSqlDataSourceInput, QueryBuilder, type RawModelOptions, RawNode, type RawQueryOptions, RedisCacheAdapter, type RedisFetchable, type RedisStorable, type RelatedInstance, type RelationQueryBuilderType, type ReplicationType, type SlaveAlgorithm, type SqlCloneOptions, SqlDataSource, type SqlDataSourceInput, type SqlDataSourceModel, type SqlDataSourceType, type SqlDriverSpecificOptions, type SqlPoolType, type Sqlite3ConnectionOptions, type SqliteConnectionInstance, type SqliteDataSourceInput, type StartTransactionOptions, type SymmetricEncryptionOptions, type TableFormat, type ThroughModel, TimestampedModel, Transaction, type TransactionExecutionOptions, type UniqueType, type UseCacheReturnType, type UseConnectionInput, UserMixin, UuidModel, belongsTo, column, createModelFactory, defineMigrator, generateOpenApiModel, generateOpenApiModelSchema, generateOpenApiModelWithMetadata, getCollectionProperties, getIndexes, getModelColumns, type getPoolReturnType, getPrimaryKey, getRelations, getRelationsMetadata, getUniques, hasMany, hasOne, index, HysteriaLogger as logger, manyToMany, property, RedisDataSource as redis, unique, view, withPerformance };
+export { type AdminJsActionOptions, type AdminJsAssets, type AdminJsBranding, type AdminJsInstance, type AdminJsLocale, type AdminJsOptions, type AdminJsPage, type AdminJsPropertyOptions, type AdminJsResourceOptions, type AdminJsSettings, type AnnotatedModel, type AsymmetricEncryptionOptions, AutogeneratedModel, type BaseModelMethodOptions, type BaseModelRelationType, type CacheAdapter, type CacheKeys, ClientMigrator, Collection, type ColumnDataTypeOption, type ColumnDataTypeOptionSimple, type ColumnDataTypeOptionWithBinary, type ColumnDataTypeOptionWithDatePrecision, type ColumnDataTypeOptionWithEnum, type ColumnDataTypeOptionWithLength, type ColumnDataTypeOptionWithPrecision, type ColumnDataTypeOptionWithScaleAndPrecision, type ColumnDataTypeOptionWithText, type ColumnOptions, type ColumnType, type CommonDataSourceInput, type CommonSqlMethodReturnType, type ConnectionPolicies, type DataSourceInput, type DataSourceType, type DateColumnOptions, DryModelQueryBuilder, DryQueryBuilder, type FetchHooks, type GetConnectionReturnType, HysteriaError, InMemoryAdapter, type IndexType, type LazyRelationType, type ManyOptions, type ManyToManyOptions, type ManyToManyStringOptions, Migration, type MigrationConfig, type MigrationConfigBase, Model, type ModelInstanceType, ModelQueryBuilder, type ModelWithoutRelations, MongoDataSource, type MongoDataSourceInput$1 as MongoDataSourceInput, type MssqlConnectionInstance, type MssqlDataSourceInput, type MssqlPoolInstance, type MysqlConnectionInstance, type MysqlSqlDataSourceInput, type NotNullableMysqlSqlDataSourceInput, type NotNullableOracleDBDataSourceInput, type NotNullableOracleMssqlDataSourceInput, type NotNullablePostgresSqlDataSourceInput, type NotNullableSqliteDataSourceInput, type NumberModelKey, type OneOptions, type OracleDBDataSourceInput, type OracleDBPoolInstance, type PgPoolClientInstance, type PostgresSqlDataSourceInput, QueryBuilder, type RawModelOptions, RawNode, type RawQueryOptions, RedisCacheAdapter, type RedisFetchable, type RedisStorable, type RelatedInstance, type RelationQueryBuilderType, type ReplicationType, type SlaveAlgorithm, type SlaveContext, type SqlCloneOptions, SqlDataSource, type SqlDataSourceInput, type SqlDataSourceModel, type SqlDataSourceType, type SqlDriverSpecificOptions, type SqlPoolType, type Sqlite3ConnectionOptions, type SqliteConnectionInstance, type SqliteDataSourceInput, type StartTransactionOptions, type SymmetricEncryptionOptions, type TableFormat, type ThroughModel, TimestampedModel, Transaction, type TransactionExecutionOptions, type UniqueType, type UseCacheReturnType, type UseConnectionInput, UserMixin, UuidModel, belongsTo, column, createModelFactory, defineMigrator, generateOpenApiModel, generateOpenApiModelSchema, generateOpenApiModelWithMetadata, getCollectionProperties, getIndexes, getModelColumns, type getPoolReturnType, getPrimaryKey, getRelations, getRelationsMetadata, getUniques, hasMany, hasOne, index, HysteriaLogger as logger, manyToMany, property, RedisDataSource as redis, unique, view, withPerformance };

package/lib/index.d.ts

@@ -1107,10 +1107,46 @@ type ConnectionPolicies = {
     };
 };
 type SqlDataSourceModel = typeof Model;
+/**
+ * @description Base migration configuration options available for all databases
+ */
+type MigrationConfigBase = {
+    /**
+     * @description The path to the migrations folder, can be overridden in the cli command
+     * @default "database/migrations"
+     */
+    path?: string;
+    /**
+     * @description Path to the tsconfig.json file for TypeScript migration files, can be overridden in the cli command
+     * @default "./tsconfig.json"
+     */
+    tsconfig?: string;
+    /**
+     * @description Acquire advisory lock before running migrations to prevent concurrent execution, can be overridden in the cli command
+     * @default true
+     */
+    lock?: boolean;
+};
+/**
+ * @description Migration configuration with transactional support for PostgreSQL and CockroachDB
+ */
+type MigrationConfigWithTransactional = MigrationConfigBase & {
+    /**
+     * @description Runs all pending migrations in a single transaction, can be overridden in the cli command
+     * @default true
+     * @note Only available for PostgreSQL and CockroachDB
+     */
+    transactional?: boolean;
+};
+/**
+ * @description Migration configuration type based on database type
+ * @description Adds transactional option only for PostgreSQL and CockroachDB
+ */
+type MigrationConfig<D extends SqlDataSourceType = SqlDataSourceType> = D extends "postgres" | "cockroachdb" ? MigrationConfigWithTransactional : MigrationConfigBase;
 /**
  * @description Common input properties shared across all SqlDataSource types
  */
-type SqlDataSourceInputBase<T extends Record<string, SqlDataSourceModel> = {}, C extends CacheKeys = {}> = {
+type SqlDataSourceInputBase<T extends Record<string, SqlDataSourceModel> = {}, C extends CacheKeys = {}, D extends SqlDataSourceType = SqlDataSourceType> = {
     /**
      * @description Whether to log the sql queries and other debug information
      */
@@ -1129,10 +1165,9 @@ type SqlDataSourceInputBase<T extends Record<string, SqlDataSourceModel> = {}, C
      */
     models?: T;
     /**
-     * @description The path to the migrations folder for the sql data source, it's used to configure the migrations path for the sql data source
-     * @default "database/migrations"
+     * @description Migration configuration for the sql data source
      */
-    migrationsPath?: string;
+    migrations?: MigrationConfig<D>;
     /**
      * @description The cache strategy to use for the sql data source, it's used to configure the cache strategy for the sql data source
      */
@@ -1150,11 +1185,19 @@ type SqlDataSourceInputBase<T extends Record<string, SqlDataSourceModel> = {}, C
  * @description Maps a SqlDataSourceType to its corresponding input interface
  */
 type MapSqlDataSourceTypeToInput<D extends SqlDataSourceType> = D extends "mysql" | "mariadb" ? MysqlSqlDataSourceInput : D extends "postgres" | "cockroachdb" ? PostgresSqlDataSourceInput : D extends "sqlite" ? SqliteDataSourceInput : D extends "mssql" ? MssqlDataSourceInput : D extends "oracledb" ? OracleDBDataSourceInput : never;
+type SlaveContext = {
+    type: SqlDataSourceType;
+    host: string;
+    port: number;
+    username: string;
+    password: string;
+    database: string;
+};
 /**
  * @description The input type for the SqlDataSource constructor
  * @description The connectionPolicies object is used to configure the connection policies for the sql data source
  */
-type SqlDataSourceInput<D extends SqlDataSourceType = SqlDataSourceType, T extends Record<string, SqlDataSourceModel> = {}, C extends CacheKeys = {}> = SqlDataSourceInputBase<T, C> & {
+type SqlDataSourceInput<D extends SqlDataSourceType = SqlDataSourceType, T extends Record<string, SqlDataSourceModel> = {}, C extends CacheKeys = {}> = SqlDataSourceInputBase<T, C, D> & {
     /**
      * @description The type of the database to connect to
      */
@@ -1168,10 +1211,14 @@ type SqlDataSourceInput<D extends SqlDataSourceType = SqlDataSourceType, T exten
      * @description The replication configuration for the sql data source, it's used to configure the replication for the sql data source
      */
     replication?: {
+        /**
+         * @description The function to call when a slave server fails, if not provided an error will be thrown
+         */
+        onSlaveServerFailure?: (error: Error, context: SlaveContext) => void | Promise<void>;
         /**
          * @description The slaves data sources to use for the sql data source, slaves are automatically used for read operations unless specified otherwise
          */
-        slaves?: Omit<UseConnectionInput<D, T, C>, "slaves" | "models" | "cacheStrategy" | "adminJs" | "logs" | "queryFormatOptions" | "migrationsPath">[];
+        slaves?: Omit<UseConnectionInput<D, T, C>, "slaves" | "models" | "cacheStrategy" | "adminJs" | "logs" | "queryFormatOptions" | "migrations">[];
         /**
          * @description The algorithm to use for selecting the slave for read operations
          * @default "roundRobin" - Distributes requests evenly across all slaves in sequence
@@ -1544,6 +1591,14 @@ declare class CreateTableBuilder extends BaseBuilder {
     private sqlType;
     constructor(sqlType: SqlDataSourceType, nodes: QueryNode[], tableName?: string, context?: "alter_table" | "create_table");
     private build;
+    /**
+     * @description Adds a raw statement to an operation that will be executed as is
+     * @example
+     * ```ts
+     * table.varchar("name").default(table.rawStatement("CURRENT_TIMESTAMP"));
+     * ```
+     */
+    rawStatement(value: string): RawNode;
     /**
      * Fixed-length character string
      * @mysql Supported as CHAR(length)
@@ -1825,6 +1880,14 @@ declare class AlterTableBuilder extends BaseBuilder {
     private table;
     private sqlType;
     constructor(table: string, nodes: QueryNode[], sqlType: SqlDataSourceType);
+    /**
+     * @description Adds a raw statement to an operation that will be executed as is
+     * @example
+     * ```ts
+     * table.varchar("name").default(table.rawStatement("CURRENT_TIMESTAMP"));
+     * ```
+     */
+    rawStatement(value: string): RawNode;
     /**
      * @description Adds a column to the table
      * @param cb is the callback that will be used to build the column
@@ -1918,6 +1981,17 @@ declare class Schema {
     queryStatements: string[];
     sqlType: SqlDataSourceType;
     constructor(sqlType?: SqlDataSourceType);
+    /**
+     * @description Adds a raw statement to an operation that will be executed as is
+     * @example
+     * ```ts
+     * schema.rawStatement("CURRENT_TIMESTAMP");
+     * schema.alterTable("users", (table) => {
+     *   table.timestamp("created_at").default(this.schema.rawStatement("CURRENT_TIMESTAMP"));
+     * });
+     * ```
+     */
+    rawStatement(value: string): RawNode;
     /**
      * @description Add raw query to the migration
      */
@@ -2037,7 +2111,7 @@ declare class DryQueryBuilder extends QueryBuilder {
      * @warning This method does not run model or column hooks
      * @returns The query builder
      */
-    update(data: Record<string, any>): this;
+    update(data: Record<string, WriteQueryParam>): this;
     /**
      * @description Builds the delete query statement without executing it, use 'unWrap' or 'toQuery' to get the query statement
      * @warning This method does not run model or column hooks
@@ -3107,6 +3181,7 @@ type UpsertOptionsRawBuilder = {
 };
 type DryQueryBuilderWithoutReadOperations = Omit<DryQueryBuilder, "many" | "one" | "oneOrFail" | "first" | "firstOrFail" | "paginate" | "paginateWithCursor" | "exists" | "pluck" | "increment" | "decrement" | "getSum" | "getAvg" | "getMin" | "getMax" | "getCount" | "stream" | "chunk" | "paginate" | "paginateWithCursor" | "exists">;
 type DryModelQueryBuilderWithoutReadOperations<T extends Model, A extends Record<string, any> = {}, R extends Record<string, any> = {}> = Omit<DryModelQueryBuilder<T, A, R>, "many" | "one" | "oneOrFail" | "first" | "firstOrFail" | "paginate" | "paginateWithCursor" | "exists" | "pluck" | "upsert" | "upsertMany" | "increment" | "decrement" | "getSum" | "getAvg" | "getMin" | "getMax" | "getCount" | "stream" | "chunk" | "paginate" | "paginateWithCursor" | "exists">;
+type WriteQueryParam$1 = string | number | boolean | Date | RawNode | object | null | undefined;
 
 declare class SqlModelManagerUtils<T extends Model> {
     protected dbType: SqlDataSourceType;
@@ -3698,9 +3773,14 @@ declare class SqlDataSource<D extends SqlDataSourceType = SqlDataSourceType, T e
      */
     cacheKeys: C;
     /**
-     * @description The path to the migrations folder for the sql data source, it's used to configure the migrations path for the sql data source
+     * @description Migration configuration for the sql data source
      */
-    migrationsPath: string;
+    migrationConfig: {
+        path: string;
+        tsconfig?: string;
+        lock: boolean;
+        transactional: boolean;
+    };
     /**
      * @description AdminJS configuration options
      */
@@ -3709,6 +3789,14 @@ declare class SqlDataSource<D extends SqlDataSourceType = SqlDataSourceType, T e
      * @description Cached AdminJS instance
      */
     private adminJsInstance?;
+    /**
+     * @description Callback to handle slave server failures
+     */
+    private onSlaveServerFailure?;
+    /**
+     * @description Returns the configured slave failure callback
+     */
+    getOnSlaveServerFailure(): ((error: Error, context: SlaveContext) => void | Promise<void>) | undefined;
     /**
      * @description Returns the primary instance of the SqlDataSource (set via connect with setPrimary: true)
      * All models by default will use this instance to execute queries unless you pass a different connection/transaction in the query options
@@ -3938,10 +4026,16 @@ declare class SqlDataSource<D extends SqlDataSourceType = SqlDataSourceType, T e
      */
     rawQuery<R = RawQueryResponseType<D>>(query: string, params?: any[], options?: RawQueryOptions): Promise<R>;
     /**
-     * @description Adds a raw statement to an operation like where or update
+     * @description Adds a raw statement to an operation that will be executed as is
      * @example
      * ```ts
-     * await User.query().where("name", sql.rawStatement("LOWER(name)"));
+     * await sql.query("users").where("name", sql.rawStatement("LOWER(name)"));
+     * await sql.query("users").update({
+     *   name: sql.rawStatement("LOWER(name)"),
+     * });
+     * await sql.query("users").insert({
+     *   name: sql.rawStatement("LOWER(name)"),
+     * });
      * ```
      */
     rawStatement(value: string): RawNode;
@@ -3995,6 +4089,30 @@ declare class SqlDataSource<D extends SqlDataSourceType = SqlDataSourceType, T e
      * @description Introspects table primary key from the database
      */
     getPrimaryKeyInfo(table: string): Promise<TablePrimaryKeyInfo | undefined>;
+    /**
+     * @description Acquires an advisory lock
+     * @description Useful for preventing concurrent operations from running simultaneously
+     * @param lockKey - The lock identifier (defaults to 'hysteria_lock')
+     * @param timeoutMs - Maximum time to wait for lock in milliseconds (postgres/mssql only)
+     * @returns true if lock was acquired, false otherwise
+     */
+    acquireLock(lockKey?: string, timeoutMs?: number): Promise<boolean>;
+    /**
+     * @description Releases an advisory lock
+     * @param lockKey - The lock identifier (defaults to 'hysteria_lock')
+     * @returns true if lock was released, false otherwise
+     */
+    releaseLock(lockKey?: string): Promise<boolean>;
+    /**
+     * @description Converts a string to a numeric lock ID for databases that require it
+     */
+    private hashStringToLockId;
+    /**
+     * @description Executes an operation on a slave, handling failures with the configured callback
+     * @param operation The operation to execute on the slave
+     * @returns The result of the operation, or falls back to master if slave fails
+     */
+    private executeOnSlave;
     /**
      * @description Internal method to establish connection without setting as primary instance
      * @description Used by connectToSecondarySource and useConnection
@@ -4328,18 +4446,18 @@ declare class QueryBuilder<T extends Model = any> extends JsonQueryBuilder<T> {
      */
     withMaterialized(alias: string, cb: (qb: QueryBuilder<T>) => void): this;
     /**
-     * @description Insert record into a table
+     * @description Insert record into a table, you can use raw statements in the data object for literal references to other columns
      * @param returning - The columns to return from the query, only supported by postgres and cockroachdb - default is "*"
      * @returns raw driver response
      */
-    insert(data: Record<string, any>, returning?: string[]): Promise<T>;
+    insert(data: Record<string, WriteQueryParam$1>, returning?: string[]): Promise<T>;
     /**
      * @description Insert multiple records into a table
      * @param returning - The columns to return from the query, only supported by postgres and cockroachdb - default is "*"
      * @returns raw driver response
      * @oracledb may do multiple inserts with auto-generated identity columns
      */
-    insertMany(data: Record<string, any>[], returning?: string[]): Promise<T[]>;
+    insertMany(data: Record<string, WriteQueryParam$1>[], returning?: string[]): Promise<T[]>;
     /**
      * @description Updates or creates a new record using upsert functionality
      * @param data The data to insert or update
@@ -4361,10 +4479,10 @@ declare class QueryBuilder<T extends Model = any> extends JsonQueryBuilder<T> {
      */
     private executeMssqlMergeRaw;
     /**
-     * @description Updates records from a table
+     * @description Updates records from a table, you can use raw statements in the data object for literal references to other columns
      * @returns the number of affected rows
      */
-    update(data: Record<string, any>): Promise<number>;
+    update(data: Record<string, WriteQueryParam$1>): Promise<number>;
     /**
      * @description Deletes all records from a table
      * @warning This operation does not trigger any hook
@@ -4456,6 +4574,13 @@ declare class QueryBuilder<T extends Model = any> extends JsonQueryBuilder<T> {
      */
     protected executePaginateQueries<M, C>(modelsQuery: () => Promise<M>, countQuery: () => Promise<C>): Promise<[M, C]>;
     protected getSqlDataSource(mode: "read" | "write"): Promise<SqlDataSource>;
+    /**
+     * @description Executes SQL with slave failure handling
+     * @param mode The operation mode (read or write)
+     * @param operation The execSql operation to perform
+     * @returns The result of the operation
+     */
+    protected execSqlWithSlaveHandling<R>(mode: "read" | "write", operation: (dataSource: SqlDataSource) => Promise<R>): Promise<R>;
 }
 
 type UnionCallBack<T extends Model> = (queryBuilder: QueryBuilder<T>) => QueryBuilder<T>;
@@ -5934,4 +6059,4 @@ declare const generateOpenApiModelWithMetadata: <T extends new () => Model>(mode
     $id?: string;
 }>;
 
-export { type AdminJsActionOptions, type AdminJsAssets, type AdminJsBranding, type AdminJsInstance, type AdminJsLocale, type AdminJsOptions, type AdminJsPage, type AdminJsPropertyOptions, type AdminJsResourceOptions, type AdminJsSettings, type AnnotatedModel, type AsymmetricEncryptionOptions, AutogeneratedModel, type BaseModelMethodOptions, type BaseModelRelationType, type CacheAdapter, type CacheKeys, ClientMigrator, Collection, type ColumnDataTypeOption, type ColumnDataTypeOptionSimple, type ColumnDataTypeOptionWithBinary, type ColumnDataTypeOptionWithDatePrecision, type ColumnDataTypeOptionWithEnum, type ColumnDataTypeOptionWithLength, type ColumnDataTypeOptionWithPrecision, type ColumnDataTypeOptionWithScaleAndPrecision, type ColumnDataTypeOptionWithText, type ColumnOptions, type ColumnType, type CommonDataSourceInput, type CommonSqlMethodReturnType, type ConnectionPolicies, type DataSourceInput, type DataSourceType, type DateColumnOptions, DryModelQueryBuilder, DryQueryBuilder, type FetchHooks, type GetConnectionReturnType, HysteriaError, InMemoryAdapter, type IndexType, type LazyRelationType, type ManyOptions, type ManyToManyOptions, type ManyToManyStringOptions, Migration, Model, type ModelInstanceType, ModelQueryBuilder, type ModelWithoutRelations, MongoDataSource, type MongoDataSourceInput$1 as MongoDataSourceInput, type MssqlConnectionInstance, type MssqlDataSourceInput, type MssqlPoolInstance, type MysqlConnectionInstance, type MysqlSqlDataSourceInput, type NotNullableMysqlSqlDataSourceInput, type NotNullableOracleDBDataSourceInput, type NotNullableOracleMssqlDataSourceInput, type NotNullablePostgresSqlDataSourceInput, type NotNullableSqliteDataSourceInput, type NumberModelKey, type OneOptions, type OracleDBDataSourceInput, type OracleDBPoolInstance, type PgPoolClientInstance, type PostgresSqlDataSourceInput, QueryBuilder, type RawModelOptions, RawNode, type RawQueryOptions, RedisCacheAdapter, type RedisFetchable, type RedisStorable, type RelatedInstance, type RelationQueryBuilderType, type ReplicationType, type SlaveAlgorithm, type SqlCloneOptions, SqlDataSource, type SqlDataSourceInput, type SqlDataSourceModel, type SqlDataSourceType, type SqlDriverSpecificOptions, type SqlPoolType, type Sqlite3ConnectionOptions, type SqliteConnectionInstance, type SqliteDataSourceInput, type StartTransactionOptions, type SymmetricEncryptionOptions, type TableFormat, type ThroughModel, TimestampedModel, Transaction, type TransactionExecutionOptions, type UniqueType, type UseCacheReturnType, type UseConnectionInput, UserMixin, UuidModel, belongsTo, column, createModelFactory, defineMigrator, generateOpenApiModel, generateOpenApiModelSchema, generateOpenApiModelWithMetadata, getCollectionProperties, getIndexes, getModelColumns, type getPoolReturnType, getPrimaryKey, getRelations, getRelationsMetadata, getUniques, hasMany, hasOne, index, HysteriaLogger as logger, manyToMany, property, RedisDataSource as redis, unique, view, withPerformance };
+export { type AdminJsActionOptions, type AdminJsAssets, type AdminJsBranding, type AdminJsInstance, type AdminJsLocale, type AdminJsOptions, type AdminJsPage, type AdminJsPropertyOptions, type AdminJsResourceOptions, type AdminJsSettings, type AnnotatedModel, type AsymmetricEncryptionOptions, AutogeneratedModel, type BaseModelMethodOptions, type BaseModelRelationType, type CacheAdapter, type CacheKeys, ClientMigrator, Collection, type ColumnDataTypeOption, type ColumnDataTypeOptionSimple, type ColumnDataTypeOptionWithBinary, type ColumnDataTypeOptionWithDatePrecision, type ColumnDataTypeOptionWithEnum, type ColumnDataTypeOptionWithLength, type ColumnDataTypeOptionWithPrecision, type ColumnDataTypeOptionWithScaleAndPrecision, type ColumnDataTypeOptionWithText, type ColumnOptions, type ColumnType, type CommonDataSourceInput, type CommonSqlMethodReturnType, type ConnectionPolicies, type DataSourceInput, type DataSourceType, type DateColumnOptions, DryModelQueryBuilder, DryQueryBuilder, type FetchHooks, type GetConnectionReturnType, HysteriaError, InMemoryAdapter, type IndexType, type LazyRelationType, type ManyOptions, type ManyToManyOptions, type ManyToManyStringOptions, Migration, type MigrationConfig, type MigrationConfigBase, Model, type ModelInstanceType, ModelQueryBuilder, type ModelWithoutRelations, MongoDataSource, type MongoDataSourceInput$1 as MongoDataSourceInput, type MssqlConnectionInstance, type MssqlDataSourceInput, type MssqlPoolInstance, type MysqlConnectionInstance, type MysqlSqlDataSourceInput, type NotNullableMysqlSqlDataSourceInput, type NotNullableOracleDBDataSourceInput, type NotNullableOracleMssqlDataSourceInput, type NotNullablePostgresSqlDataSourceInput, type NotNullableSqliteDataSourceInput, type NumberModelKey, type OneOptions, type OracleDBDataSourceInput, type OracleDBPoolInstance, type PgPoolClientInstance, type PostgresSqlDataSourceInput, QueryBuilder, type RawModelOptions, RawNode, type RawQueryOptions, RedisCacheAdapter, type RedisFetchable, type RedisStorable, type RelatedInstance, type RelationQueryBuilderType, type ReplicationType, type SlaveAlgorithm, type SlaveContext, type SqlCloneOptions, SqlDataSource, type SqlDataSourceInput, type SqlDataSourceModel, type SqlDataSourceType, type SqlDriverSpecificOptions, type SqlPoolType, type Sqlite3ConnectionOptions, type SqliteConnectionInstance, type SqliteDataSourceInput, type StartTransactionOptions, type SymmetricEncryptionOptions, type TableFormat, type ThroughModel, TimestampedModel, Transaction, type TransactionExecutionOptions, type UniqueType, type UseCacheReturnType, type UseConnectionInput, UserMixin, UuidModel, belongsTo, column, createModelFactory, defineMigrator, generateOpenApiModel, generateOpenApiModelSchema, generateOpenApiModelWithMetadata, getCollectionProperties, getIndexes, getModelColumns, type getPoolReturnType, getPrimaryKey, getRelations, getRelationsMetadata, getUniques, hasMany, hasOne, index, HysteriaLogger as logger, manyToMany, property, RedisDataSource as redis, unique, view, withPerformance };