diff --git a/lib/dl_connector_mysql/dl_connector_mysql/core/connector.py b/lib/dl_connector_mysql/dl_connector_mysql/core/connector.py index 70011262b..9c4a55848 100644 --- a/lib/dl_connector_mysql/dl_connector_mysql/core/connector.py +++ b/lib/dl_connector_mysql/dl_connector_mysql/core/connector.py @@ -24,6 +24,7 @@ MySQLSubselectDataSource, ) from dl_connector_mysql.core.data_source_migration import MySQLDataSourceMigrator +from dl_connector_mysql.core.query_compiler import MySQLQueryCompiler from dl_connector_mysql.core.sa_types import SQLALCHEMY_MYSQL_TYPES from dl_connector_mysql.core.storage_schemas.connection import ConnectionMySQLDataStorageSchema from dl_connector_mysql.core.type_transformer import MySQLTypeTransformer @@ -60,3 +61,4 @@ class MySQLCoreConnector(CoreConnector): ) rqe_adapter_classes = frozenset({MySQLAdapter, AsyncMySQLAdapter}) sa_types = SQLALCHEMY_MYSQL_TYPES + compiler_cls = MySQLQueryCompiler diff --git a/lib/dl_connector_mysql/dl_connector_mysql/core/data_source.py b/lib/dl_connector_mysql/dl_connector_mysql/core/data_source.py index 1ceaf548f..967b3fcce 100644 --- a/lib/dl_connector_mysql/dl_connector_mysql/core/data_source.py +++ b/lib/dl_connector_mysql/dl_connector_mysql/core/data_source.py @@ -37,6 +37,7 @@ class MySQLDataSourceMixin(BaseSQLDataSource): ) conn_type = CONNECTION_TYPE_MYSQL + compiler_cls = MySQLQueryCompiler @classmethod def is_compatible_with_type(cls, source_type: DataSourceType) -> bool: @@ -46,8 +47,6 @@ def is_compatible_with_type(cls, source_type: DataSourceType) -> bool: class MySQLDataSource(MySQLDataSourceMixin, StandardSQLDataSource): """MySQL table""" - compiler_cls = MySQLQueryCompiler - class MySQLSubselectDataSource(MySQLDataSourceMixin, SubselectDataSource): """MySQL table from a subquery"""