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

1、服务器端代码

package demo.net;

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、客户端代码

package demo.net;

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

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

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