X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fws%2Febi%2FEBIFetchClient.java;h=b5a732894aa5e55324f917b8c63fd7bc748b4d97;hb=8fa69554edf6aeb278b4a4afd8e2b60264fdccd8;hp=d6cad6423e837c6dcdb24778778fe5907d905e90;hpb=db93a1adcbe0a4eaaf06e0a70ade0d6c5c1961c3;p=jalview.git

diff --git a/src/jalview/ws/ebi/EBIFetchClient.java b/src/jalview/ws/ebi/EBIFetchClient.java
index d6cad64..b5a7328 100644
--- a/src/jalview/ws/ebi/EBIFetchClient.java
+++ b/src/jalview/ws/ebi/EBIFetchClient.java
@@ -1,6 +1,6 @@
 /*
- * Jalview - A Sequence Alignment Editor and Viewer (Version 2.9.0b2)
- * Copyright (C) 2015 The Jalview Authors
+ * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
+ * Copyright (C) $$Year-Rel$$ The Jalview Authors
  * 
  * This file is part of Jalview.
  * 
@@ -22,13 +22,14 @@ package jalview.ws.ebi;
 
 import jalview.datamodel.DBRefSource;
 import jalview.util.MessageManager;
+import jalview.util.Platform;
 
-import java.io.BufferedInputStream;
 import java.io.BufferedReader;
 import java.io.File;
-import java.io.FileOutputStream;
+import java.io.IOException;
 import java.io.InputStream;
 import java.io.InputStreamReader;
+import java.net.HttpURLConnection;
 import java.net.URL;
 import java.util.ArrayList;
 import java.util.List;
@@ -91,7 +92,7 @@ public class EBIFetchClient
    * @param format
    *          the format wanted
    * @param extension
-   *          for the temporary file to hold response
+   *          for the temporary file to hold response (without separator)
    * @return the file holding the response
    * @throws OutOfMemoryError
    */
@@ -102,7 +103,7 @@ public class EBIFetchClient
     File outFile = null;
     try
     {
-      outFile = File.createTempFile("jalview", ext);
+      outFile = File.createTempFile("jalview", "." + ext);
       outFile.deleteOnExit();
       fetchData(ids, format, outFile);
       if (outFile.length() == 0)
@@ -192,43 +193,39 @@ public class EBIFetchClient
    * @param database
    * @param format
    * @param outFile
-   * @return
+   * @return array of lines from EBI only if outFile is null (which it will not
+   *         be)
    * @throws OutOfMemoryError
    */
   String[] fetchBatch(String ids, String database, String format,
           File outFile) throws OutOfMemoryError
   {
-    // long time = System.currentTimeMillis();
     String url = buildUrl(ids, database, format);
-
+    InputStream is = null;
     try
     {
       URL rcall = new URL(url);
-
-      InputStream is = new BufferedInputStream(rcall.openStream());
-      if (outFile != null)
+      HttpURLConnection conn = (HttpURLConnection) rcall.openConnection();
+      int responseCode = conn.getResponseCode();
+      if (responseCode == 200)
       {
-        FileOutputStream fio = new FileOutputStream(outFile);
-        byte[] bb = new byte[32 * 1024];
-        int l;
-        while ((l = is.read(bb)) > 0)
+        is = conn.getInputStream();
+        if (outFile != null)
         {
-          fio.write(bb, 0, l);
+          Platform.streamToFile(is, outFile);
+          return null;
         }
-        fio.close();
-        is.close();
-      }
-      else
-      {
         BufferedReader br = new BufferedReader(new InputStreamReader(is));
         String rtn;
-        List<String> arl = new ArrayList<String>();
+        List<String> arl = new ArrayList<>();
         while ((rtn = br.readLine()) != null)
         {
           arl.add(rtn);
         }
-        return arl.toArray(new String[arl.size()]);
+        return (String[]) arl.toArray();
       }
+      System.err.println(
+              "Warning: response code " + responseCode + " for " + url);
     } catch (OutOfMemoryError er)
     {
       System.out.println("OUT OF MEMORY DOWNLOADING QUERY FROM " + database
@@ -236,19 +233,24 @@ public class EBIFetchClient
       throw er;
     } catch (Exception ex)
     {
-      if (ex.getMessage().startsWith(
+      if (!ex.getMessage().startsWith(
               "uk.ac.ebi.jdbfetch.exceptions.DbfNoEntryFoundException"))
       {
-        return null;
+        System.err.println("Unexpected exception when retrieving from "
+                + database + "\nQuery was : '" + ids + "'");
+        ex.printStackTrace(System.err);
       }
-      System.err.println("Unexpected exception when retrieving from "
-              + database + "\nQuery was : '" + ids + "'");
-      ex.printStackTrace(System.err);
-      return null;
     } finally
     {
-      // System.err.println("EBIFetch took " + (System.currentTimeMillis() -
-      // time) + " ms");
+      if (is != null)
+      {
+        try
+        {
+          is.close();
+        } catch (IOException e)
+        {
+        }
+      }
     }
     return null;
   }
@@ -267,12 +269,12 @@ public class EBIFetchClient
     if (database.equalsIgnoreCase(DBRefSource.EMBL)
             || database.equalsIgnoreCase(DBRefSource.EMBLCDS))
     {
-      url = "http://www.ebi.ac.uk/ena/data/view/" + ids.toLowerCase()
+      url = "https://www.ebi.ac.uk/ena/data/view/" + ids.toLowerCase()
               + (format != null ? "&" + format : "");
     }
     else
     {
-      url = "http://www.ebi.ac.uk/Tools/dbfetch/dbfetch/"
+      url = "https://www.ebi.ac.uk/Tools/dbfetch/dbfetch/"
               + database.toLowerCase() + "/" + ids.toLowerCase()
               + (format != null ? "/" + format : "");
     }