QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-20 00:24:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(mcp): add eager loading to get_info tools to prevent N+1 query timeouts (#38129)

Browse Source 
Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Amin Ghadersohi
2026-02-25 11:28:58 -05:00
committed by GitHub
parent 438a927420
commit c54b21ef98
7 changed files with 68 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
superset/daos/database.py
View File

@@ -70,11 +70,15 @@ class DatabaseDAO(BaseDAO[Database]):
model_id: str | int,
skip_base_filter: bool = False,
id_column: str | None = None,
query_options: list[Any] | None = None,
) -> Database | None:
"""
Find a database by id, eagerly loading the SSH tunnel relationship.
"""
query = db.session.query(cls.model_cls).options(joinedload(Database.ssh_tunnel))
all_options = [joinedload(Database.ssh_tunnel)]
if query_options:
all_options.extend(query_options)
query = db.session.query(cls.model_cls).options(*all_options)
query = cls._apply_base_filter(query, skip_base_filter)
column_name = id_column or cls.id_column_name