From 65a469e8b9f654ded9f0915d57d2507dee2e1860 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jind=C5=99ich=20Moravec?= <xmorave2@fi.muni.cz>
Date: Sun, 28 Jan 2024 22:21:03 +0100
Subject: [PATCH] fix: cancellation before next query

---
 server/src/main.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/server/src/main.rs b/server/src/main.rs
index da7a08a..b9fc5ae 100644
--- a/server/src/main.rs
+++ b/server/src/main.rs
@@ -167,6 +167,9 @@ async fn handle_query<W>(writer: &mut W, state: &SharedDbState, query: String, t
     where
         W: BackendProtoWriter + ProtoFlush + Send,
 {
+    //  Make sure token is reset before next query
+    token.reset();
+
     let operation = {
         let state = state.read().await;
         let db_schema = state.db_schema();