Application: gvSIG desktop

/*

 * Created on 01-oct-2005

 */

package org.gvsig.remoteClient.taskplanning.retrieving;

import java.io.BufferedOutputStream;

import java.io.DataOutputStream;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.net.MalformedURLException;

import java.net.URL;

import java.util.Iterator;

import java.util.Vector;

import org.gvsig.remoteClient.taskplanning.IRunnableTask;

/**

 * Clase para bajar ficheros en un thread independiente.

 * La idea (y parte del c?digo) est? tomada de EarthFlicks

 * @author Luis W. Sevilla (sevilla_lui@gva.es)

 * @see http://today.java.net/lpt/a/212

 */

public class URLRetrieveTask  implements IRunnableTask {

        private boolean running, cancelled;

        private URLRequest request;

        private InputStream is;

        /**

         * 

         */

        public URLRetrieveTask(URLRequest request) {

                this.request = request;

                request.setWorker(this);

                running = cancelled = false;

        }

        public void execute() {

                request.setStatus(RetrieveEvent.NOT_STARTED);

                cancelled = false;

                running= true;

                long t = System.currentTimeMillis();

                File f = new File(request.getFileName()+System.currentTimeMillis());

                while (f.exists()) {

                        t++;

                        f = new File(request.getFileName()+t);

                }

                URL url;

                try {

                        request.setStatus(RetrieveEvent.CONNECTING);

                        url = request.getUrl();

                        request.setFileName(f.getAbsolutePath());

                        System.out.println("downloading '"+url+"' to: "+f.getAbsolutePath());

                        DataOutputStream dos = new DataOutputStream( new BufferedOutputStream(new FileOutputStream(f)));

                        byte[] buffer = new byte[1024*256];

                        is = url.openStream();

                        request.setStatus(RetrieveEvent.TRANSFERRING);

                        if (!cancelled) {

                                long readed = 0;

                                for (int i = is.read(buffer); !cancelled && i>0; i = is.read(buffer)){

                                        dos.write(buffer, 0, i);

                                        readed += i;

                                }

                                dos.close();

                        }

                        if (cancelled) {

                                // Bad incomplete file, delete it.

                                System.out.println("download cancelled ("+url+")");

                                f.delete();

                        }

                        running = false;

                        if (cancelled)

                                request.setStatus(RetrieveEvent.REQUEST_CANCELLED);

                        else

                                request.setStatus(RetrieveEvent.REQUEST_FINISHED);

                } catch (MalformedURLException e) {

                        e.printStackTrace();

                        request.setStatus(RetrieveEvent.REQUEST_FAILED);

                } catch (IOException e) {

                        e.printStackTrace();

                        request.setStatus(RetrieveEvent.REQUEST_FAILED);

                }

        }

        public void cancel() {

                cancelled = true;

                try {

                        if (is != null) {

                                is.close();

                                is = null;

                        }

                } catch (IOException e) {

                        e.printStackTrace();

                }

        }

        public boolean isRunning() {

                return running && !cancelled;

        }

        public URLRequest getRequest() {

                return request;

        }

}