Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/feature/DOP-16999' into feature/…
Browse files Browse the repository at this point in the history
…DOP-16999
  • Loading branch information
maxim-lixakov committed Aug 20, 2024
2 parents 4dc8b56 + 81d86f6 commit 97d49b9
Show file tree
Hide file tree
Showing 4 changed files with 17 additions and 18 deletions.
3 changes: 0 additions & 3 deletions docs/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -120,9 +120,6 @@
{"rel": "icon", "href": "icon.svg", "type": "image/svg+xml"},
]

# TODO: remove after https://github.com/mgeier/sphinx-last-updated-by-git/pull/77
git_exclude_patterns = ["docs/_static/logo_wide.svg"]

# The master toctree document.
master_doc = "index"

Expand Down
4 changes: 2 additions & 2 deletions onetl/connection/db_connection/jdbc_connection/connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ def sql(

query = clear_statement(query)

log.info("|%s| Detected dialect: '%s'", self.__class__.__name__, self.jdbc_dialect)
log.info("|%s| Detected dialect: '%s'", self.__class__.__name__, self._get_spark_dialect_name())
log.info("|%s| Executing SQL query (on executor):", self.__class__.__name__)
log_lines(log, query)

Expand Down Expand Up @@ -196,7 +196,7 @@ def get_df_schema(
columns: list[str] | None = None,
options: JDBCReadOptions | None = None,
) -> StructType:
log.info("|%s| Detected dialect: '%s'", self.__class__.__name__, self.jdbc_dialect)
log.info("|%s| Detected dialect: '%s'", self.__class__.__name__, self._get_spark_dialect_name())
log.info("|%s| Fetching schema of table %r ...", self.__class__.__name__, source)

query = self.dialect.get_sql_query(source, columns=columns, limit=0, compact=True)
Expand Down
25 changes: 13 additions & 12 deletions onetl/connection/db_connection/jdbc_mixin/connection.py
Original file line number Diff line number Diff line change
Expand Up @@ -205,7 +205,7 @@ def fetch(

query = clear_statement(query)

log.info("|%s| Detected dialect: '%s'", self.__class__.__name__, self.jdbc_dialect)
log.info("|%s| Detected dialect: '%s'", self.__class__.__name__, self._get_spark_dialect_name())
log.info("|%s| Executing SQL query (on driver):", self.__class__.__name__)
log_lines(log, query)

Expand Down Expand Up @@ -278,7 +278,7 @@ def execute(

statement = clear_statement(statement)

log.info("|%s| Detected dialect: '%s'", self.__class__.__name__, self.jdbc_dialect)
log.info("|%s| Detected dialect: '%s'", self.__class__.__name__, self._get_spark_dialect_name())
log.info("|%s| Executing statement (on driver):", self.__class__.__name__)
log_lines(log, statement)

Expand Down Expand Up @@ -310,16 +310,6 @@ def execute(
log_lines(log, str(metrics))
return df

@property
def jdbc_dialect(self):
"""
Returns the JDBC dialect associated with the connection URL.
"""
jdbc_dialects_package = self.spark._jvm.org.apache.spark.sql.jdbc
dialect = jdbc_dialects_package.JdbcDialects.get(self.jdbc_url).toString()

return dialect.split("$")[0] if "$" in dialect else dialect

@validator("spark")
def _check_java_class_imported(cls, spark):
try:
Expand Down Expand Up @@ -429,6 +419,17 @@ def _get_jdbc_connection(self, options: JDBCFetchOptions | JDBCExecuteOptions):
self._last_connection_and_options.data = (new_connection, options)
return new_connection

def _get_spark_dialect_name(self) -> str:
"""
Returns the name of the JDBC dialect associated with the connection URL.
"""
dialect = self._get_spark_dialect().toString()
return dialect.split("$")[0] if "$" in dialect else dialect

def _get_spark_dialect(self):
jdbc_dialects_package = self.spark._jvm.org.apache.spark.sql.jdbc
return jdbc_dialects_package.JdbcDialects.get(self.jdbc_url)

def _close_connections(self):
with suppress(Exception):
# connection maybe not opened yet
Expand Down
3 changes: 2 additions & 1 deletion requirements/docs.txt
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,8 @@ sphinx<8
sphinx-copybutton
sphinx-design
sphinx-favicon
sphinx-last-updated-by-git
# https://github.com/mgeier/sphinx-last-updated-by-git/pull/77
sphinx-last-updated-by-git>=0.3.8
# TODO: uncomment after https://github.com/zqmillet/sphinx-plantuml/pull/4
# sphinx-plantuml
sphinx-tabs
Expand Down

0 comments on commit 97d49b9

Please sign in to comment.