SNOW-692968: Async queries support

include/snowflake/client.h

@@ -294,6 +294,25 @@ typedef enum SF_STMT_ATTRIBUTE {
     SF_STMT_USER_REALLOC_FUNC
 } SF_STMT_ATTRIBUTE;
 
+/**
+ * The query status
+ */
+typedef enum SF_QUERY_STATUS {
+    SF_QUERY_STATUS_RUNNING,
+    SF_QUERY_STATUS_ABORTING,
+    SF_QUERY_STATUS_SUCCESS,
+    SF_QUERY_STATUS_FAILED_WITH_ERROR,
+    SF_QUERY_STATUS_ABORTED,
+    SF_QUERY_STATUS_QUEUED,
+    SF_QUERY_STATUS_FAILED_WITH_INCIDENT,
+    SF_QUERY_STATUS_DISCONNECTED,
+    SF_QUERY_STATUS_RESUMING_WAREHOUSE,
+    SF_QUERY_STATUS_QUEUED_REPAIRING_WAREHOUSE,
+    SF_QUERY_STATUS_RESTARTED,
+    SF_QUERY_STATUS_BLOCKED,
+    SF_QUERY_STATUS_NO_DATA
+} SF_QUERY_STATUS;
+
 /**
  * Snowflake Error
  */
@@ -476,6 +495,8 @@ typedef struct SF_STMT {
     SF_STATS *stats;
     void *stmt_attrs;
     sf_bool is_dml;
+    sf_bool is_async;
+    sf_bool is_async_initialized;
 
     /**
      * User realloc function used in snowflake_fetch
@@ -613,6 +634,25 @@ SF_STATUS STDCALL snowflake_get_attribute(
  */
 SF_STMT *STDCALL snowflake_stmt(SF_CONNECT *sf);
 
+/**
+ * Creates sf SNOWFLAKE_STMT context for async queries.
+ *
+ * @param sf The SF_CONNECT context.
+ * @param query_id the query id of the async query.
+ *
+ * @return sfstmt SNOWFLAKE_STMT context for async queries.
+ */
+SF_STMT* STDCALL snowflake_create_async_query_result(SF_CONNECT *sf, const char *query_id);
+
+/**
+ * Get the status of a query
+ * 
+ * @param sfstmt The SF_STMT context.
+ * 
+ * @return The query status.
+ */
+SF_QUERY_STATUS STDCALL snowflake_get_query_status(SF_STMT *sfstmt);
+
 /**
  * Frees the memory used by a SF_QUERY_RESULT_CAPTURE struct.
  * Note that this only frees the struct itself, and *not* the underlying
@@ -775,6 +815,14 @@ snowflake_stmt_get_attr(SF_STMT *sfstmt, SF_STMT_ATTRIBUTE type, void **value);
  */
 SF_STATUS STDCALL snowflake_execute(SF_STMT *sfstmt);
 
+/**
+ * Executes a statement asynchronously.
+ * @param sfstmt SNOWFLAKE_STMT context.
+ *
+ * @return 0 if success, otherwise an errno is returned.
+ */
+SF_STATUS STDCALL snowflake_async_execute(SF_STMT *sfstmt);
+
 /**
  * Executes a statement with capture.
  * @param sfstmt SNOWFLAKE_STMT context.

include/snowflake/platform.h

@@ -142,6 +142,9 @@ void STDCALL sf_memory_error_handler();
 // this should be called by application before any calls of sfclient
 void STDCALL sf_exception_on_memory_failure();
 
+void STDCALL sf_sleep_ms(int sleep_ms);
+
+
 #ifdef __cplusplus
 }
 #endif