DB-GPT/dbgpt/datasource/rdbms/dialect/oceanbase/ob_dialect.py
明天 0541d1494c
feat(datasource):add oceanbase support (#1622)
Co-authored-by: csunny <cfqsunny@163.com>
Co-authored-by: aries_ckt <916701291@qq.com>
2024-06-13 15:13:50 +08:00

27 lines
846 B
Python

"""OB Dialect support."""
from sqlalchemy.dialects import registry
from sqlalchemy.dialects.mysql import pymysql
class OBDialect(pymysql.MySQLDialect_pymysql):
"""OBDialect expend."""
def initialize(self, connection):
"""Ob dialect initialize."""
super(OBDialect, self).initialize(connection)
self._server_version_info = (5, 7, 19)
self.server_version_info = (5, 7, 19)
def _server_version_info(self, connection):
"""Ob set fixed version ending compatibility issue."""
return (5, 7, 19)
def get_isolation_level(self, dbapi_connection):
"""Ob set fixed version ending compatibility issue."""
self.server_version_info = (5, 7, 19)
return super(OBDialect, self).get_isolation_level(dbapi_connection)
registry.register("mysql.ob", __name__, "OBDialect")