From e8a8937fc1f510bd04fa0d84cd335796aaa7a555 Mon Sep 17 00:00:00 2001
From: servertood <servertood@40cadd2d-89a2-4482-8f20-0232d845d7e3>
Date: Mon, 10 Nov 2008 03:09:22 +0000
Subject: [PATCH] add cancel handler

---
 extra/jheidi/runUserQueriesWorker.ajl | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/extra/jheidi/runUserQueriesWorker.ajl b/extra/jheidi/runUserQueriesWorker.ajl
index 47913883..4808c06a 100644
--- a/extra/jheidi/runUserQueriesWorker.ajl
+++ b/extra/jheidi/runUserQueriesWorker.ajl
@@ -8,7 +8,14 @@
   {
 
     final DB db = new SQLLoggingDB((SQLLoggingDB)context.get("db"));
-    w.setProperty("db", db);
+    
+    w.onCancelDo( new Runnable() {
+      public void run()
+      {
+        db.cancel();
+      }
+    });
+    
     db.useCache(false);
     
     DB.AutoCleanupTX tx = new DB.AutoCleanupTX() {
@@ -56,6 +63,16 @@
     
     ToolbarButton btn = context.get("/mainTabs/Query/toolbar/Stop");
     btn.setEnabled(false);
+    
+    btn = context.get("/mainTabs/Query/toolbar/Run");
+    btn.setEnabled(true);
+    
+    Editor editor = context.get("/mainTabs/Query/editor");
+    if (editor.getSelectedText() != null && editor.getSelectedText().length() > 0)
+    {
+       btn = context.get("/mainTabs/Query/toolbar/Run Selection");
+       btn.setEnabled(true);
+    }
   }