parallelised service endpoint discovery and ensured that old threads are destroyed...
[jalview.git] / src / jalview / ws / jws2 / Jws2Discoverer.java
index fdbb740..8cc39d7 100644 (file)
@@ -1,3 +1,20 @@
+/*
+ * Jalview - A Sequence Alignment Editor and Viewer (Version 2.6)
+ * Copyright (C) 2010 J Procter, AM Waterhouse, G Barton, M Clamp, S Searle
+ * 
+ * This file is part of Jalview.
+ * 
+ * Jalview is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License 
+ * as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
+ * 
+ * Jalview is distributed in the hope that it will be useful, but 
+ * WITHOUT ANY WARRANTY; without even the implied warranty 
+ * of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
+ * PURPOSE.  See the GNU General Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Public License along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
+ */
 package jalview.ws.jws2;
 
 import java.awt.Color;
@@ -8,8 +25,10 @@ import java.beans.PropertyChangeListener;
 import java.io.Closeable;
 import java.net.ConnectException;
 import java.net.URL;
+import java.util.ArrayList;
 import java.util.HashSet;
 import java.util.Hashtable;
+import java.util.List;
 import java.util.StringTokenizer;
 import java.util.Vector;
 
@@ -24,6 +43,7 @@ import jalview.bin.Cache;
 import jalview.datamodel.AlignmentView;
 import jalview.gui.AlignFrame;
 import jalview.gui.Desktop;
+import jalview.gui.JalviewChangeSupport;
 import jalview.ws.WSMenuEntryProviderI;
 import jalview.ws.params.ParamDatastoreI;
 import compbio.data.msa.MsaWS;
@@ -70,7 +90,16 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
     changeSupport.removePropertyChangeListener(listener);
   }
 
-  boolean running = false;
+  boolean running = false, aborted = false;
+
+  /**
+   * @param aborted
+   *          the aborted to set
+   */
+  public void setAborted(boolean aborted)
+  {
+    this.aborted = aborted;
+  }
 
   Thread oldthread = null;
 
@@ -78,7 +107,23 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
   {
     if (running && oldthread != null && oldthread.isAlive())
     {
-      return;
+      if (!aborted)
+      {
+        return;
+      }
+      while (running)
+      {
+        try
+        {
+          Cache.log
+                  .debug("Waiting around for old discovery thread to finish.");
+          // wait around until old discoverer dies
+          Thread.sleep(100);
+        } catch (Exception e)
+        {
+        }
+      }
+      Cache.log.debug("Old discovery thread has finished.");
     }
     running = true;
     oldthread = Thread.currentThread();
@@ -98,53 +143,43 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
     {
       services.removeAllElements();
     }
-    for (String jwsservers : getServiceUrls())
+    List<JabaWsServerQuery> qrys = new ArrayList<JabaWsServerQuery>();
+    for (final String jwsservers : getServiceUrls())
+    {
+      JabaWsServerQuery squery = new JabaWsServerQuery(this, jwsservers);
+      qrys.add(squery);
+      new Thread(squery).start();
+    }
+    boolean finished = false;
+    do
     {
       try
       {
-        if (Jws2Client.validURL(jwsservers))
-        {
-          // look for services
-          for (Services srv : Services.values())
-          {
-            MsaWS service = null;
-            try
-            {
-              service = Jws2Client.connect(jwsservers, srv);
-            } catch (Exception e)
-            {
-              System.err.println("Jws2 Discoverer: Problem on "
-                      + jwsservers + " with service " + srv + ":\n"
-                      + e.getMessage());
-              if (!(e instanceof javax.xml.ws.WebServiceException))
-              {
-                e.printStackTrace();
-              }
-            }
-            ;
-            if (service != null)
-            {
-              addService(jwsservers, srv, service);
-            }
-          }
-
-        }
-        else
-        {
-          Cache.log.info("Ignoring invalid Jws2 service url " + jwsservers);
-        }
+        Thread.sleep(100);
       } catch (Exception e)
       {
-        e.printStackTrace();
-        Cache.log.warn("Exception when discovering Jws2 services.", e);
-      } catch (Error e)
+      }
+      ;
+      for (JabaWsServerQuery squery : qrys)
       {
-        Cache.log.error("Exception when discovering Jws2 services.", e);
+        finished |= !squery.isRunning();
       }
-    }
+      if (aborted)
+      {
+        Cache.log.debug("Aborting " + qrys.size()
+                + " JABAWS discovery threads.");
+        for (JabaWsServerQuery squery : qrys)
+        {
+          squery.setQuit(true);
+        }
+      }
+    } while (!aborted && !finished);
     oldthread = null;
     running = false;
-    changeSupport.firePropertyChange("services", new Vector(), services);
+    if (!aborted)
+    {
+      changeSupport.firePropertyChange("services", new Vector(), services);
+    }
   }
 
   /**
@@ -154,7 +189,8 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
    * @param srv
    * @param service2
    */
-  private void addService(String jwsservers, Services srv, MsaWS service2)
+  synchronized void addService(String jwsservers, Services srv,
+          MsaWS service2)
   {
     if (services == null)
     {
@@ -162,9 +198,10 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
     }
     System.out.println("Discovered service: " + jwsservers + " "
             + srv.toString());
-    Jws2Instance service = new Jws2Instance(jwsservers, srv.toString(), service2);
+    Jws2Instance service = new Jws2Instance(jwsservers, srv.toString(),
+            service2);
 
-    services.add(service); 
+    services.add(service);
     // retrieve the presets and parameter set and cache now
     service.getParamStore().getPresets();
     service.hasParameters();
@@ -282,22 +319,25 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
     public String getUri()
     {
       // this is only valid for Jaba 1.0 - this formula might have to change!
-      return hosturl+(hosturl.lastIndexOf("/")==(hosturl.length()-1) ? "/" : "") +serviceType;
+      return hosturl
+              + (hosturl.lastIndexOf("/") == (hosturl.length() - 1) ? "/"
+                      : "") + serviceType;
     }
-    private boolean hasParams=false,lookedForParams=false;
+
+    private boolean hasParams = false, lookedForParams = false;
 
     public boolean hasParameters()
     {
       if (!lookedForParams)
       {
-        lookedForParams=true;
-      try
-      {
-        hasParams = (getRunnerConfig().getArguments().size() > 0);
-      } catch (Exception e)
-      {
+        lookedForParams = true;
+        try
+        {
+          hasParams = (getRunnerConfig().getArguments().size() > 0);
+        } catch (Exception e)
+        {
 
-      }
+        }
       }
       return hasParams;
     }
@@ -366,7 +406,7 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
 
     });
     wsmenu.add(jws2al);
-    
+
   }
 
   private void populateWSMenuEntry(JMenu jws2al, final AlignFrame alignFrame)
@@ -385,7 +425,7 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
     MsaWSClient msacl = new MsaWSClient();
     Vector hostLabels = new Vector();
     jws2al.removeAll();
-    String lasthost=null;
+    String lasthost = null;
     for (final Jws2Instance service : services)
     {
       atpoint = jws2al;
@@ -407,22 +447,27 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
           atpoint.setToolTipText(service.getActionText());
         }
       }
-      if (!byhost && !hostLabels.contains(host + service.serviceType+service.getActionText()))
-              //!hostLabels.contains(host + (bytype ? service.serviceType+service.getActionText() : "")))
+      if (!byhost
+              && !hostLabels.contains(host + service.serviceType
+                      + service.getActionText()))
+      // !hostLabels.contains(host + (bytype ?
+      // service.serviceType+service.getActionText() : "")))
       {
         // add a marker indicating where this service is hosted
         // relies on services from the same host being listed in a contiguous
         // group
         JMenuItem hitm;
         atpoint.addSeparator();
-        if (lasthost==null || !lasthost.equals(host))
+        if (lasthost == null || !lasthost.equals(host))
         {
           atpoint.add(hitm = new JMenuItem(host));
           hitm.setForeground(Color.blue);
           lasthost = host;
         }
-        hostLabels.addElement(host + service.serviceType+service.getActionText() );
-        // hostLabels.addElement(host + (bytype ? service.serviceType+service.getActionText() : ""));
+        hostLabels.addElement(host + service.serviceType
+                + service.getActionText());
+        // hostLabels.addElement(host + (bytype ?
+        // service.serviceType+service.getActionText() : ""));
       }
       msacl.attachWSMenuEntry(atpoint, service, alignFrame);
       /*
@@ -441,17 +486,18 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
 
   public static void main(String[] args)
   {
-    Thread runner = new Thread(getDiscoverer());
-    getDiscoverer().addPropertyChangeListener(new PropertyChangeListener()
+    Thread runner = getDiscoverer().startDiscoverer(new PropertyChangeListener()
     {
 
       public void propertyChange(PropertyChangeEvent evt)
       {
-        System.out.println("Changesupport: There are now "
+        if (getDiscoverer().services!=null)
+          {System.out.println("Changesupport: There are now "
                 + getDiscoverer().services.size() + " services");
+          
+          }
       }
     });
-    runner.start();
     while (runner.isAlive())
     {
       try
@@ -591,4 +637,21 @@ public class Jws2Discoverer implements Runnable, WSMenuEntryProviderI
     return true;
   }
 
+  /**
+   * Start a fresh discovery thread and notify the given object when we're finished. Any known existing threads will be killed before this one is started. 
+   * @param changeSupport2
+   * @return new thread
+   */
+  public Thread startDiscoverer(PropertyChangeListener changeSupport2)
+  {
+    if (isRunning())
+    {
+      setAborted(true);
+    }
+    addPropertyChangeListener(changeSupport2);
+    Thread thr = new Thread(this);
+    thr.start();
+    return thr;
+  }
+
 }