Merge pull request #1312 from sqlancer/fix/datafusion-server-memory-leak

mrigger · web-flow · commit d35a70dce1ad · 2026-04-07T23:21:30.000+08:00
Fix DataFusion server memory leak: clear state on database reset
diff --git a/src/sqlancer/datafusion/server/datafusion_server/src/main.rs b/src/sqlancer/datafusion/server/datafusion_server/src/main.rs
@@ -215,6 +215,11 @@ impl FlightSqlService for FlightSqlServiceImpl {
 
             let mut ctx_guard = self.ctx.lock().await; // Use `lock()` for async Mutex
             *ctx_guard = new_ctx;
+
+            // Clear leaked state from previous round
+            self.statements.clear();
+            self.results.clear();
+            self.contexts.clear();
         }
         // no authentication actually takes place here
         // see Ballista implementation for example of basic auth
