fix

Author: youngsofun

src/query/service/src/servers/http/v1/http_query_handlers.rs

@@ -310,6 +310,7 @@ async fn query_final_handler(
                 &ctx.client_session_id,
                 StopReason::Finished,
                 ErrorCode::ClosedQuery("Query closed by client"),
+                true
             )
             .await?
         {
@@ -351,6 +352,7 @@ async fn query_cancel_handler(
                 &ctx.client_session_id,
                 StopReason::Canceled,
                 ErrorCode::AbortedQuery("canceled by client"),
+                true
             )
             .await?
         {

src/query/service/src/servers/http/v1/query/http_query_manager.rs

@@ -189,6 +189,7 @@ impl HttpQueryManager {
                                 &None,
                                 stop_reason,
                                 ErrorCode::AbortedQuery(&msg),
+                                false
                             )
                             .await
                             .ok();
@@ -214,14 +215,15 @@ impl HttpQueryManager {
         client_session_id: &Option<String>,
         reason: StopReason,
         error: ErrorCode,
+        check_session_id: bool,
     ) -> poem::error::Result<Option<Arc<HttpQuery>>> {
         let now = SystemTime::now()
             .duration_since(SystemTime::UNIX_EPOCH)
             .expect("Time went backwards")
             .as_secs();
         let (query, stop_first_run) = self.queries.write().stop(query_id, reason, now);
         if let Some(q) = &query {
-            if reason != StopReason::Timeout {
+            if check_session_id {
                 q.check_client_session_id(client_session_id)?;
             }
             if stop_first_run {