Add query cancellation controls (#41)

Co-authored-by: SuLe <sule@SuLedeMacBook-Air.local>
Co-authored-by: skyler <77960507+t8y2@users.noreply.github.com>

src/lib/queryExecutionState.ts

@@ -0,0 +1,13 @@
+export interface QueryExecutionStateLike {
+  isExecuting: boolean;
+  isCancelling?: boolean;
+  executionId?: string;
+}
+
+export function canCancelQueryExecution(state: QueryExecutionStateLike): boolean {
+  return state.isExecuting && !!state.executionId && !state.isCancelling;
+}
+
+export function queryExecutionLabelKey(state: Pick<QueryExecutionStateLike, "isCancelling">): "common.loading" | "common.stopping" {
+  return state.isCancelling ? "common.stopping" : "common.loading";
+}

src/lib/tauri.ts

@@ -95,8 +95,12 @@ export async function getColumns(
   return invoke("get_columns", { connectionId, database, schema, table });
 }
 
-export async function executeQuery(connectionId: string, database: string, sql: string): Promise<QueryResult> {
-  return invoke("execute_query", { connectionId, database, sql });
+export async function executeQuery(connectionId: string, database: string, sql: string, executionId?: string): Promise<QueryResult> {
+  return invoke("execute_query", { connectionId, database, sql, executionId });
+}
+
+export async function cancelQuery(executionId: string): Promise<boolean> {
+  return invoke("cancel_query", { executionId });
 }
 
 export async function executeBatch(connectionId: string, database: string, statements: string[]): Promise<QueryResult> {