多线程下载工具（练手项目十一）

package com.nyohh.flashget.ui; import java.awt.BorderLayout; import java.awt.Dimension; import java.awt.Insets; import java.awt.MenuItem; import java.awt.PopupMenu; import java.awt.SystemTray; import java.awt.Toolkit; import java.awt.TrayIcon; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; import java.awt.event.MouseAdapter; import java.awt.event.MouseEvent; import java.awt.event.WindowAdapter; import java.awt.event.WindowEvent; import java.awt.image.BufferedImage; import java.io.File; import java.io.FileInputStream; import java.io.FileOutputStream; import java.io.ObjectInputStream; import java.io.ObjectOutputStream; import java.util.ArrayList; import java.util.Iterator; import javax.swing.AbstractAction; import javax.swing.Action; import javax.swing.ImageIcon; import javax.swing.JFrame; import javax.swing.JList; import javax.swing.JScrollPane; import javax.swing.JSplitPane; import javax.swing.JToolBar; import javax.swing.Timer; import javax.swing.tree.DefaultMutableTreeNode; import javax.swing.tree.TreePath; import com.nyohh.flashget.ContextHolder; import com.nyohh.flashget.DownloadContext; import com.nyohh.flashget.info.Info; import com.nyohh.flashget.navigation.DownloadNode; import com.nyohh.flashget.navigation.DownloadingNode; import com.nyohh.flashget.navigation.FailNode; import com.nyohh.flashget.navigation.FinishNode; import com.nyohh.flashget.navigation.TaskNode; import com.nyohh.flashget.object.Resource; import com.nyohh.flashget.state.Downloading; import com.nyohh.flashget.state.Failed; import com.nyohh.flashget.state.Finished; import com.nyohh.flashget.state.Pause; import com.nyohh.flashget.util.DateUtil; import com.nyohh.flashget.util.FileUtil; import com.nyohh.flashget.util.ImageUtil; /** * 主界面 * * @author yulei * @Date: 2013年11月13日 * @Copyright: 2013 www.nyohh.com Inc. All rights reserved. */ public class MainFrame extends JFrame { private static final long serialVersionUID = -7385308668156588694L; private Dimension screen; // 得到屏幕大小 private static final String FILE_SIZE_TEXT = "文件大小: "; private static final String FILE_PATH_TEXT = "文件路径: "; private static final String DOWNLOAD_DATE_TEXT = "下载时间: "; private static final String RESOURCE_INFO_TEXT = "资源信息: "; private ArrayList<Info> infoList; // 信息对象数据 // 信息列表信息 private Info fileSize, info, downloadDate, filePath; private TaskNode taskNode; // 所有任务节点 private NavigationTree navTree; // 导航树 private DownloadTable downloadTable; // 下载列表 private JList<Object> infoJList; // 信息列表 private JToolBar toolBar; // 工具栏 private TrayIcon trayIcon; // 任务栏图标 // 任务栏菜单 private MenuItem quitItem; private MenuItem removeItem; private MenuItem pauseItem; private MenuItem startItem; private MenuItem newItem; private MenuItem openItem; private NewTaskFrame taskFrame; // 新建任务界面 private DownloadNode currentNode = taskNode; // 当前用户浏览的节点 private Action newTask = new AbstractAction("新任务", new ImageIcon("images/tool/new-download.gif")) { private static final long serialVersionUID = 1L; @Override public void actionPerformed(ActionEvent e) { newTask(); } }; private void newTask() { taskFrame.setVisible(true); } private Action start = new AbstractAction("开始", new ImageIcon("images/tool/do-download.gif")) { private static final long serialVersionUID = 1L; @Override public void actionPerformed(ActionEvent e) { start(); } }; private void start() { Resource r = getResource(); if (r == null) { return; } if (r.getState() instanceof Pause || r.getState() instanceof Failed) { ContextHolder.dh.resumeDownload(r); } } /** * 得到用户在列表中所选择的资源 */ private Resource getResource() { int row = this.downloadTable.getSelectedRow(); int column = this.downloadTable.getColumn(DownloadTableModel.ID_COLUMN).getModelIndex(); if (row == -1) { return null; } String id = (String) this.downloadTable.getValueAt(row, column); return ContextHolder.ctx.getResource(id); } private Action pause = new AbstractAction("暂停", new ImageIcon("images/tool/pause.gif")) { private static final long serialVersionUID = 1L; @Override public void actionPerformed(ActionEvent e) { pause(); } }; private void pause() { Resource r = getResource(); if (r == null) { return; } // 判断状态 if (!(r.getState() instanceof Downloading)) { return; } r.setState(DownloadContext.PAUSE); } private Action delete = new AbstractAction("删除任务", new ImageIcon("images/tool/delete.gif")) { private static final long serialVersionUID = 1L; @Override public void actionPerformed(ActionEvent e) { delete(); } }; /** * 删除资源 */ private void delete() { Resource r = getResource(); if (r == null) { return; } // 先将任务停止 r.setState(DownloadContext.PAUSE); // 删除所有的.part文件 FileUtil.deletePartFiles(r); // 从上下文集合中删除资源 ContextHolder.ctx.resources.remove(r); } private Action deleteFinished = new AbstractAction("删除任务", new ImageIcon("images/tool/remove-finished.gif")) { private static final long serialVersionUID = 1L; @Override public void actionPerformed(ActionEvent e) { deleteFinished(); } }; /** * 删除已下载完成的资源 */ public void deleteFinished() { for (Iterator<Resource> it = ContextHolder.ctx.resources.iterator(); it.hasNext();) { Resource r = it.next(); if (r.getState() instanceof Finished) { it.remove(); } } } public MainFrame() { screen = Toolkit.getDefaultToolkit().getScreenSize(); // 得到屏幕大小 taskFrame = new NewTaskFrame(); // 新建任务界面 initialization(); // 创建任务栏图标 createTrayIcon(); // 创建悬浮窗口 new SuspendWindow(this); initlisteners(); // 创建定时器 Timer timer = new Timer(1000, new ActionListener() { @Override public void actionPerformed(ActionEvent e) { downloadTable.updateUI(); // 刷新列表 } }); timer.start(); // 读取序列化文件 reverseSer(); } /** * 反序列化 */ private void reverseSer() { File serFile = FileUtil.SERIALIZABLE_FILE; if (!serFile.exists()) { return; } try { // 得到文件输入流 FileInputStream fis = new FileInputStream(serFile); ObjectInputStream ois = new ObjectInputStream(fis); // 设置ContextHolder的DownloadContext ContextHolder.ctx = (DownloadContext) ois.readObject(); ois.close(); fis.close(); } catch (Exception e) { e.printStackTrace(); } // 设置列表 addTableData(); } private void addTableData() { DownloadTableModel model = (DownloadTableModel) this.downloadTable.getModel(); // 将保存的资源设置到列表中 model.setResources(ContextHolder.ctx.resources); // 刷新列表 this.downloadTable.refresh(); } private void initlisteners() { // 任务栏图标监听器 this.trayIcon.addMouseListener(new MouseAdapter() { @Override public void mouseClicked(MouseEvent e) { if (e.getClickCount() == 2) { setVisible(true); } } }); this.openItem.addActionListener(new ActionListener() { @Override public void actionPerformed(ActionEvent arg0) { if (isVisible()) { setVisible(false); } else { setVisible(true); } } }); this.newItem.addActionListener(new