fix(core): should call handle_session() when query fail. (#359)

databendlabs · Mar 8, 2024 · 9290c2a · 9290c2a
1 parent 4c3041b
commit 9290c2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/src/client.rs b/core/src/client.rs
@@ -323,10 +323,10 @@ impl APIClient {
         }
 
         let resp: QueryResponse = resp.json().await?;
+        self.handle_session(&resp.session).await;
         if let Some(err) = resp.error {
             return Err(Error::InvalidResponse(err));
         }
-        self.handle_session(&resp.session).await;
         self.handle_warnings(&resp);
         Ok(resp)
     }