转载自:http://www.cnblogs.com/kinghitomi/archive/2012/01/19/2327449.html
1、服务器端代码
- import java.io.File;
- import java.io.FileInputStream;
- import java.io.FileNotFoundException;
- import java.io.IOException;
- import java.io.InputStream;
- import java.net.DatagramPacket;
- import java.net.DatagramSocket;
- import java.net.SocketException;
- import java.util.concurrent.ExecutorService;
- import java.util.concurrent.Executors;
- /**
- * 下载服务器,采用UDP协议,传送的过程中可能会丢包,导致下载的文件不完整
- */
- public class DownloadServer {
- // 提供服务
- public void service() {
- try {
- // 创建本机指定端口8289的服务器
- DatagramSocket dataSocket = new DatagramSocket(8289);
- // 线程池,固定有十个线程
- ExecutorService ThreadPool = Executors.newFixedThreadPool(10);
- while (true) {// 不断接收来自客户端的请求
- byte[] buff = new byte[101];// 文件名长度不超过50
- DatagramPacket dataPacket = new DatagramPacket(buff, buff.length);
- dataSocket.receive(dataPacket);// 等待接收来自客户端的数据包
- // 接收到数据包,开一个线程为该客户服务
- ThreadPool.execute(new WorkThread(dataPacket));
- }
- } catch (SocketException e) {
- e.printStackTrace();
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- // 内部类,为每个客户提供服务
- private class WorkThread implements Runnable {
- private DatagramPacket packet;
- private DatagramSocket dataSocket;
- public WorkThread(DatagramPacket packet) {
- this.packet = packet;
- try {// 创建本机可以端口的DatagramSocket
- dataSocket = new DatagramSocket();
- } catch (SocketException e) {
- e.printStackTrace();
- }
- }
- // 获取可以下载的文件列表传送给客户端
- private void showFiles() {
- File files = new File("upload_download");
- File[] allFile = files.listFiles();// 获取所有文件
- StringBuffer message = new StringBuffer();
- for (File f : allFile) {
- if (f.isFile()) {
- message.append(f.getName());
- message.append('\n');
- }
- }
- // 构造响应数据包
- byte[] response = message.toString().getBytes();
- DatagramPacket dataPacket = new DatagramPacket(response, response.length, packet.getAddress(), packet.getPort());
- try {// 发送
- dataSocket.send(dataPacket);
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- // 下载指定的文件
- private void download(String fileName) {
- try {
- InputStream in = new FileInputStream("upload_download/" + fileName);
- DatagramPacket dataPacket;
- byte[] response = new byte[60000];// 每次发送60000字节
- while (true) {
- int len = in.read(response, 0, response.length);
- dataPacket = new DatagramPacket(response, len, packet.getAddress(), packet.getPort());
- dataSocket.send(dataPacket);// 发送
- if (in.available() == 0)// 发送完毕
- break;
- }
- in.close();
- } catch (FileNotFoundException e) {
- e.printStackTrace();
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- @Override
- public void run() {
- // 获取客户端传送过来的数据
- byte[] data = packet.getData();
- // 表示客户端点击显示文件按钮,该请求是要得到所有可以下载的文件
- if (data[0] == 0)
- showFiles();
- else if (data[0] == 1)// 表示客户端的请求是下载请求
- download(new String(data, 1, packet.getLength()).trim());
- else
- System.out.println("请求错误");
- }
- }
- public static void main(String[] args) {
- new DownloadServer().service();
- }
- }
说明:DatagramSocket 类是基于UDP协议,服务器端默认提供下载的文件存放在当前目录下的upload_download文件夹中。服务器采用线程池能同时为多个用户服务
2、客户端代码
- import java.awt.BorderLayout;
- import java.awt.GridLayout;
- import java.awt.event.ActionEvent;
- import java.awt.event.ActionListener;
- import java.io.File;
- import java.io.FileOutputStream;
- import java.io.IOException;
- import java.io.OutputStream;
- import java.net.DatagramPacket;
- import java.net.DatagramSocket;
- import java.net.InetAddress;
- import javax.swing.*;
- /**
- * 下载客户端
- */
- public class DownloadClient extends JFrame {
- // 显示可下载的文件
- private JTextArea textArea = new JTextArea();
- private JPanel panel = new JPanel();
- // 下载时保存文件
- private JFileChooser saveFile = new JFileChooser(".");
- private JButton showButton = new JButton("显示文件");
- private JButton downloadButton = new JButton("下载...");
- // 下载时填入要下载的文件名,注意文件名必须是textArea显示的文件名
- private JTextField downloadFile = new JTextField("");
- private DatagramSocket dataSocket=null;
- public DownloadClient() {
- // frame 的基本设置
- this.setTitle("下载客户端");
- this.setVisible(true);
- this.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
- this.setSize(400, 500);
- this.setLayout(new BorderLayout());
- this.setResizable(false);
- // 设置不可编辑
- textArea.setEditable(false);
- panel.setLayout(new GridLayout(3, 2, 5, 5));
- panel.add(new JLabel("点击按钮显示可下载的文件"));
- panel.add(showButton);
- panel.add(downloadFile);
- panel.add(downloadButton);
- // 组件加入frame中
- add(new JScrollPane(textArea));
- add(panel, BorderLayout.SOUTH);
- // saveFile只能打开目录
- saveFile.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
- // 显示文件按钮注册事件
- showButton.addActionListener(new ActionListener() {
- @Override
- public void actionPerformed(ActionEvent e) {
- showButton.setEnabled(false);
- downloadButton.setEnabled(false);
- showFiles();
- showButton.setEnabled(true);
- downloadButton.setEnabled(true);
- }
- });
- // 下载按钮注册事件
- downloadButton.addActionListener(new ActionListener() {
- @Override
- public void actionPerformed(ActionEvent e) {
- showButton.setEnabled(false);
- downloadButton.setEnabled(false);
- downloadFile();
- showButton.setEnabled(true);
- downloadButton.setEnabled(true);
- }
- });
- }
- // 显示文件
- private void showFiles() {
- try {
- if (dataSocket == null)
- dataSocket = new DatagramSocket();
- // 创建发送数据包并发送给服务器
- byte[] request = { 0 };
- DatagramPacket requestPacket = new DatagramPacket(request, request.length, InetAddress.getLocalHost(), 8289);
- dataSocket.send(requestPacket);
- // 接收服务器的数据包,显示在textArea中
- byte[] receive = new byte[1024 * 1024];
- DatagramPacket receivePacket = new DatagramPacket(receive, receive.length);
- dataSocket.receive(receivePacket);
- String str = new String(receivePacket.getData(), 0, receivePacket.getLength());
- textArea.setText(str);
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- // 下载文件
- private void downloadFile() {
- // 获取要下载的文件名
- String fileName = downloadFile.getText().trim();
- // 所有可以下载的文件
- String allFiles = textArea.getText();
- // 文件名为空
- if (fileName == null || "".equals(fileName))
- JOptionPane.showMessageDialog(null, "请选中正确的文件名", "文件名错误", JOptionPane.WARNING_MESSAGE);
- // 文件名是在可以下载的文件中
- else if (allFiles.contains((fileName + '\n'))) {
- saveFile.showSaveDialog(null);
- File f = saveFile.getSelectedFile();// 获取选中的文件夹
- if (f.exists()) {
- // 检测该文件是否已经存在于目录中
- String[] fileNames = f.list();
- boolean exit = false;
- for (String name : fileNames)
- if (name.equals(fileName)) {
- exit = true;
- break;
- }
- if (exit)// 如果要下载的文件已经存在
- JOptionPane.showMessageDialog(null, "此文件已经存在", "请选择另外的文件下载", JOptionPane.WARNING_MESSAGE);
- else {
- // 发送的请求
- byte[] request = (new String(new byte[] { 1 }) + fileName).getBytes();
- try {
- if (dataSocket == null)
- dataSocket = new DatagramSocket();
- // 创建发送数据包并发送给服务器
- DatagramPacket requestPacket = new DatagramPacket(request, request.length, InetAddress.getLocalHost(), 8289);
- dataSocket.send(requestPacket);
- // 接收服务器的数据包,把文件保存在选中的文件夹中
- OutputStream out = new FileOutputStream(f.getAbsolutePath() + "/" + fileName, true);
- byte[] receive = new byte[60000];// 每次接收60000字节
- DatagramPacket receivePacket;
- // 不断接收来自服务器的数据包
- while (true) {
- receivePacket = new DatagramPacket(receive, receive.length);
- dataSocket.receive(receivePacket);
- out.write(receivePacket.getData(), 0, receivePacket.getLength());// 输出流把文件内容输出到文件中
- out.flush();
- if (receivePacket.getLength() != receive.length)
- break;
- }
- out.close();
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- } else
- // 选择的文件夹不存在
- JOptionPane.showMessageDialog(null, "请选择正确的存储路径", "存储路径错误", JOptionPane.WARNING_MESSAGE);
- } else {// 文件名错误
- JOptionPane.showMessageDialog(null, "请选择正确的文件名", "文件名错误", JOptionPane.WARNING_MESSAGE);
- }
- }
- public static void main(String[] args) {
- new DownloadClient();
- }
- }
说明:客户端下载时先点击显示文件,然后输入要下载的文件名,最后点击下载按钮、选择保存路径
3、运行时先开服务器,然后再开客户端。运行结果:
点击显示文件按钮、输入01.jpg后:
点击下载按钮后、选择保存路径
点击保存按钮后,文件下载到指定目录下