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

文章由LinuxBoy分享于2019-04-01 01:04:40热评（34）

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

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

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

推荐文章：

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

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

最新Linux教程教程

Linux头条

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

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

相关内容

最新Linux教程教程

Linux头条