AI_DATA/doris-mcp-server
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[bugfix]Release connection after executing query (#34)

Browse Source
This commit is contained in:
ivin
2025-07-29 14:05:44 +08:00
committed by GitHub
parent fb5e864a24
commit 6247d49192
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
doris_mcp_server/utils/query_executor.py
View File

@@ -432,21 +432,17 @@ class DorisQueryExecutor:
) )
# Execute query # Execute query
connection = await self.connection_manager.get_connection(
query_request.session_id
)
# Set timeout if specified # Set timeout if specified
if query_request.timeout: if query_request.timeout:
try: try:
result = await asyncio.wait_for( result = await asyncio.wait_for(
connection.execute(optimized_sql, query_request.parameters, auth_context), self.connection_manager.execute_query(query_request.session_id, optimized_sql, query_request.parameters, auth_context),
timeout=query_request.timeout timeout=query_request.timeout
) )
except asyncio.TimeoutError: except asyncio.TimeoutError:
raise Exception(f"Query timeout after {query_request.timeout} seconds") raise Exception(f"Query timeout after {query_request.timeout} seconds")
else: else:
result = await connection.execute(optimized_sql, query_request.parameters, auth_context) result = await self.connection_manager.execute_query(query_request.session_id, optimized_sql, query_request.parameters, auth_context)
return result return result