Skip to content

Commit 1403d14

Browse files
committed
fix: connection reuse
Signed-off-by: Dennis Zhuang <killme2008@gmail.com>
1 parent 3655ec5 commit 1403d14

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

src/greptimedb_mcp_server/server.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -241,6 +241,10 @@ def _execute_query(state: AppState, query: str, limit: int) -> dict:
241241
columns = [desc[0] for desc in cursor.description]
242242
rows = cursor.fetchmany(limit)
243243
has_more = cursor.fetchone() is not None
244+
if has_more:
245+
# MySQL connector requires all results consumed before connection reuse
246+
while cursor.fetchone():
247+
pass
244248
return {
245249
"type": "query",
246250
"columns": columns,

0 commit comments

Comments
 (0)