Merge pull request #52 from nickrobinson/main

Adding query in progress response code check to async polling logic
estie-inc · Dec 22, 2024 · 0858c24 · 0858c24
2 parents 210cfb2 + 6e49174
commit 0858c24
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/query.rs b/src/query.rs
@@ -240,7 +240,9 @@ async fn poll_for_async_results(
 
         let response: SnowflakeResponse =
             serde_json::from_str(&body).map_err(|e| Error::Json(e, body))?;
-        if response.code.as_deref() != Some(QUERY_IN_PROGRESS_ASYNC_CODE) {
+        if response.code.as_deref() != Some(QUERY_IN_PROGRESS_ASYNC_CODE)
+            && response.code.as_deref() != Some(QUERY_IN_PROGRESS_CODE)
+        {
             return Ok(response);
         }
     }