summaryrefslogtreecommitdiffstats
path: root/Dozentenmodul/src/main/java/ftp/DownloadTask.java
blob: 5592adba80d74dd91d892680479d743dfc4f4e01 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
package ftp;

import gui.image.FTPEditDownloader_GUI;

import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;

import javax.swing.JOptionPane;
import javax.swing.SwingWorker;

/**
 * Execute file download in a background thread and update the progress.
 * 
 * @author www.codejava.net
 * 
 */
public class DownloadTask extends SwingWorker<Void, Void> {

	private static final int BUFFER_SIZE = 4096;

	private String host;
	private int port;
	private String username;
	private String password;
	private String downloadPath;
	private String saveDir;
	private int percentCompleted;

	public DownloadTask(String host, int port, String username,
			String password, String downloadPath, String saveDir) {
		this.host = host;
		this.port = port;
		this.username = username;
		this.password = password;
		this.downloadPath = downloadPath;
		this.saveDir = saveDir;

	}

	/**
	 * Executed in background thread
	 */
	@Override
	protected Void doInBackground() throws Exception {
		FTPUtility util = new FTPUtility(host, port, username, password);
		try {
			util.connect();

			byte[] buffer = new byte[BUFFER_SIZE];
			int bytesRead = -1;
			long totalBytesRead = 0;
			percentCompleted = 0;
			long start = System.nanoTime();
			final double NANOS_PER_SECOND = 1000000000.0;
			final double BYTES_PER_MIB = 1024 * 1024;
			long fileSize = util.getFileSize(downloadPath);
			// gui.setFileSize(fileSize);

			String fileName = new File(downloadPath).getName();

			File downloadFile = new File(saveDir + File.separator + fileName);
			FileOutputStream outputStream = new FileOutputStream(downloadFile);

			util.downloadFile(downloadPath);
			InputStream inputStream = util.getInputStream();

			while ((bytesRead = inputStream.read(buffer)) != -1
					&& isCancelled() == false) {
				outputStream.write(buffer, 0, bytesRead);
				totalBytesRead += bytesRead;
				// System.out.println(totalBytesRead);
				double speed = NANOS_PER_SECOND / BYTES_PER_MIB
						* totalBytesRead / (System.nanoTime() - start + 1);
				percentCompleted = (int) (totalBytesRead * 100 / fileSize);
				setProgress(percentCompleted);
				firePropertyChange("speed", 0, speed);
				firePropertyChange("filesize", 0, fileSize);
				firePropertyChange("bytesread", 0, totalBytesRead);

			}

			outputStream.close();

			util.finish();
		} catch (FTPException ex) {
			JOptionPane.showMessageDialog(null,
					"Error downloading file: " + ex.getMessage(), "Error",
					JOptionPane.ERROR_MESSAGE);
			ex.printStackTrace();
			setProgress(0);
			cancel(true);
		} finally {
			util.disconnect();
		}

		return null;
	}

	/**
	 * Executed in Swing's event dispatching thread
	 */
	@Override
	protected void done() {
		if (!isCancelled() && percentCompleted==100) {
			System.out.println("Datei erfolgreich heruntergeladen");
			JOptionPane.showMessageDialog(null,
					"Datei erfolgreich heruntergeladen.", "Message",
					JOptionPane.INFORMATION_MESSAGE);
		} else if(!isCancelled() && percentCompleted != 100){
			System.out.println("Datein wurde unvollständig heruntergeladen");
			JOptionPane.showMessageDialog(null,
					"Datei wurde unvollständig heruntergeladen. Bitte wiederholen.", "Message",
					JOptionPane.INFORMATION_MESSAGE);
		}
	}
}