java网络编程：DatagramSocket类简单实现文件下载

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后：

 

 

点击下载按钮后、选择保存路径

 

 

点击保存按钮后,文件下载到指定目录下