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

at_eof bug fix

Browse Source
This commit is contained in:
FreeOnePlus
2025-07-08 18:08:40 +08:00
parent aa953e9fe1
commit 357bda502c
2 changed files with 478 additions and 720 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1134
doris_mcp_server/utils/db.py
View File

File diff suppressed because it is too large Load Diff

48
doris_mcp_server/utils/schema_extractor.py
View File

@@ -1215,33 +1215,39 @@ class MetadataExtractor:
try: try:
if self.connection_manager: if self.connection_manager:
import asyncio import asyncio
import concurrent.futures
import threading
# Try to run the async query # Always run in a separate thread with new event loop to avoid conflicts
try: def run_in_new_loop():
# Check if there's a running event loop # Create new event loop for this thread
loop = asyncio.get_running_loop() new_loop = asyncio.new_event_loop()
# If we're in an async context, we need to run in a separate thread asyncio.set_event_loop(new_loop)
import concurrent.futures try:
return new_loop.run_until_complete(
def run_in_new_loop(): self._execute_query_async(query, db_name, return_dataframe)
new_loop = asyncio.new_event_loop() )
asyncio.set_event_loop(new_loop) finally:
try: try:
return new_loop.run_until_complete( # Properly close the loop
self._execute_query_async(query, db_name, return_dataframe) pending = asyncio.all_tasks(new_loop)
) if pending:
new_loop.run_until_complete(asyncio.gather(*pending, return_exceptions=True))
finally: finally:
new_loop.close() new_loop.close()
with concurrent.futures.ThreadPoolExecutor() as executor: # Use ThreadPoolExecutor to run in separate thread
future = executor.submit(run_in_new_loop) with concurrent.futures.ThreadPoolExecutor(max_workers=1) as executor:
future = executor.submit(run_in_new_loop)
try:
return future.result(timeout=30) return future.result(timeout=30)
except concurrent.futures.TimeoutError:
except RuntimeError: logger.error("Query execution timed out after 30 seconds")
# No running loop, we can safely create one if return_dataframe:
return asyncio.run( import pandas as pd
self._execute_query_async(query, db_name, return_dataframe) return pd.DataFrame()
) else:
return []
else: else:
# Fallback: Return empty result # Fallback: Return empty result
logger.warning("No connection manager provided, returning empty result") logger.warning("No connection manager provided, returning empty result")