src/jalview/ws/utils/UrlDownloadClient.java

   1 /*
   2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
   3  * Copyright (C) $$Year-Rel$$ The Jalview Authors
   4  *
   5  * This file is part of Jalview.
   6  *
   7  * Jalview is free software: you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License
   9  * as published by the Free Software Foundation, either version 3
  10  * of the License, or (at your option) any later version.
  11  *
  12  * Jalview is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty
  14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  15  * PURPOSE.  See the GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
  19  * The Jalview Authors are detailed in the 'AUTHORS' file.
  20  */
  21
  22 package jalview.ws.utils;
  23
  24 import jalview.util.Platform;
  25
  26 import java.io.File;
  27 import java.io.FileOutputStream;
  28 import java.io.IOException;
  29 import java.net.URL;
  30 import java.nio.channels.Channels;
  31 import java.nio.channels.ReadableByteChannel;
  32 import java.nio.file.Files;
  33 import java.nio.file.Path;
  34 import java.nio.file.Paths;
  35 import java.nio.file.StandardCopyOption;
  36
  37 public class UrlDownloadClient
  38 {
  39   /**
  40    * Download and save a file from a URL
  41    *
  42    * @param urlstring
  43    *          url to download from, as string
  44    * @param outfile
  45    *          the name of file to save the URLs to
  46    * @throws IOException
  47    */
  48   public static void download(String urlstring, String outfile)
  49           throws IOException
  50   {
  51
  52       FileOutputStream fos = null;
  53       ReadableByteChannel rbc = null;
  54       Path temp = null;
  55       try
  56       {
  57         temp = Files.createTempFile(".jalview_", ".tmp");
  58
  59         URL url = new URL(urlstring);
  60         rbc = Channels.newChannel(url.openStream());
  61         fos = new FileOutputStream(temp.toString());
  62         fos.getChannel().transferFrom(rbc, 0, Long.MAX_VALUE);
  63
  64         // copy tempfile to outfile once our download completes
  65         // incase something goes wrong
  66         Files.copy(temp, Paths.get(outfile),
  67                 StandardCopyOption.REPLACE_EXISTING);
  68       } catch (IOException e)
  69       {
  70         throw e;
  71       } finally
  72       {
  73         try
  74         {
  75           if (fos != null)
  76           {
  77             fos.close();
  78           }
  79         } catch (IOException e)
  80         {
  81           System.out.println(
  82                   "Exception while closing download file output stream: "
  83                           + e.getMessage());
  84         }
  85         try
  86         {
  87           if (rbc != null)
  88           {
  89             rbc.close();
  90           }
  91         } catch (IOException e)
  92         {
  93           System.out.println("Exception while closing download channel: "
  94                   + e.getMessage());
  95         }
  96         try
  97         {
  98           if (temp != null)
  99           {
 100             Files.deleteIfExists(temp);
 101           }
 102         } catch (IOException e)
 103         {
 104           System.out.println("Exception while deleting download temp file: "
 105                   + e.getMessage());
 106         }
 107       }
 108
 109   }
 110
 111   public static void download(String urlstring, File tempFile) throws IOException
 112   {
 113     if (!Platform.setFileBytes(tempFile, urlstring))
 114     {
 115       download(urlstring, tempFile.toString());
 116     }
 117   }
 118 }