github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-21 17:37:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(datasouce): Add GaussDB data source connector (#2813)

Browse Source 
Co-authored-by: tanwei <tanwei@fulan.com.cn>
This commit is contained in:
Tam
2025-07-09 22:50:22 +08:00
committed by GitHub
parent 3f60470388
commit 0ecfeabd8e
104 changed files with 445 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
packages/dbgpt-serve/src/dbgpt_serve/datasource/manages/connector_manager.py
View File

@@ -53,6 +53,9 @@ class ConnectorManager(BaseComponent):
)
from dbgpt_ext.datasource.rdbms.conn_doris import DorisConnector # noqa: F401
from dbgpt_ext.datasource.rdbms.conn_duckdb import DuckDbConnector # noqa: F401
from dbgpt_ext.datasource.rdbms.conn_gaussdb import ( # noqa: F401
GaussDBConnector,
)
from dbgpt_ext.datasource.rdbms.conn_hive import HiveConnector # noqa: F401
from dbgpt_ext.datasource.rdbms.conn_mssql import MSSQLConnector # noqa: F401
from dbgpt_ext.datasource.rdbms.conn_mysql import MySQLConnector # noqa: F401
@@ -197,8 +200,22 @@ class ConnectorManager(BaseComponent):
db_port = db_config.get("db_port")
db_user = db_config.get("db_user")
db_pwd = db_config.get("db_pwd")
try:
ext_config = db_config.get("ext_config")
db_json = json.loads(ext_config)
schema = db_json.get("schema", None)
except json.JSONDecodeError:
# 处理解码失败的情况
db_json = {}
schema = None
return connect_instance.from_uri_db( # type: ignore
host=db_host, port=db_port, user=db_user, pwd=db_pwd, db_name=db_name
host=db_host,
port=db_port,
user=db_user,
pwd=db_pwd,
db_name=db_name,
schema=schema,
)
def _create_parameters(