Java網絡編程六：DatagramSocket類簡單實現文件下載

1、服務器端代碼

1. package demo.net;
3. import java.io.File;
4. import java.io.FileInputStream;
5. import java.io.FileNotFoundException;
6. import java.io.IOException;
7. import java.io.InputStream;
8. import java.net.DatagramPacket;
9. import java.net.DatagramSocket;
10. import java.net.SocketException;
11. import java.util.concurrent.ExecutorService;
12. import java.util.concurrent.Executors;
14. /**
15. * 下載服務器,采用UDP協議，傳送的過程中可能會丟包，導致下載的文件不完整
16. */
17. public class DownloadServer {
18. // 提供服務
19. public void service() {
20. try {
21. // 創建本機指定端口8289的服務器
22. DatagramSocket dataSocket = new DatagramSocket(8289);
23. // 線程池，固定有十個線程
24. ExecutorService ThreadPool = Executors.newFixedThreadPool(10);
26. while (true) {// 不斷接收來自客戶端的請求
27. byte[] buff = new byte[101];// 文件名長度不超過50
28. DatagramPacket dataPacket = new DatagramPacket(buff, buff.length);
29. dataSocket.receive(dataPacket);// 等待接收來自客戶端的數據包
30. // 接收到數據包，開一個線程為該客戶服務
31. ThreadPool.execute(new WorkThread(dataPacket));
32. }
33. } catch (SocketException e) {
34. e.printStackTrace();
35. } catch (IOException e) {
36. e.printStackTrace();
37. }
38. }
40. // 內部類，為每個客戶提供服務
41. private class WorkThread implements Runnable {
42. private DatagramPacket packet;
43. private DatagramSocket dataSocket;
45. public WorkThread(DatagramPacket packet) {
46. this.packet = packet;
47. try {// 創建本機可以端口的DatagramSocket
48. dataSocket = new DatagramSocket();
49. } catch (SocketException e) {
50. e.printStackTrace();
51. }
52. }
54. // 獲取可以下載的文件列表傳送給客戶端
55. private void showFiles() {
56. File files = new File("upload_download");
57. File[] allFile = files.listFiles();// 獲取所有文件
58. StringBuffer message = new StringBuffer();
59. for (File f : allFile) {
60. if (f.isFile()) {
61. message.append(f.getName());
62. message.append('\n');
63. }
64. }
65. // 構造響應數據包
66. byte[] response = message.toString().getBytes();
67. DatagramPacket dataPacket = new DatagramPacket(response, response.length, packet.getAddress(), packet.getPort());
68. try {// 發送
69. dataSocket.send(dataPacket);
70. } catch (IOException e) {
71. e.printStackTrace();
72. }
73. }
75. // 下載指定的文件
76. private void download(String fileName) {
77. try {
78. InputStream in = new FileInputStream("upload_download/" + fileName);
79. DatagramPacket dataPacket;
80. byte[] response = new byte[60000];// 每次發送60000字節
81. while (true) {
82. int len = in.read(response, 0, response.length);
83. dataPacket = new DatagramPacket(response, len, packet.getAddress(), packet.getPort());
84. dataSocket.send(dataPacket);// 發送
85. if (in.available() == 0)// 發送完畢
86. break;
87. }
88. in.close();
89. } catch (FileNotFoundException e) {
90. e.printStackTrace();
91. } catch (IOException e) {
92. e.printStackTrace();
93. }
94. }
96. @Override
97. public void run() {
98. // 獲取客戶端傳送過來的數據
99. byte[] data = packet.getData();
100. // 表示客戶端點擊顯示文件按鈕，該請求是要得到所有可以下載的文件
101. if (data[0] == 0)
102. showFiles();
103. else if (data[0] == 1)// 表示客戶端的請求是下載請求
104. download(new String(data, 1, packet.getLength()).trim());
105. else
106. System.out.println("請求錯誤");
107. }
108. }
110. public static void main(String[] args) {
111. new DownloadServer().service();
112. }
113. }

說明：DatagramSocket 類是基於UDP協議，服務器端默認提供下載的文件存放在當前目錄下的upload_download文件夾中。服務器采用線程池能同時為多個用戶服務

2、客戶端代碼

1. package demo.net;
3. import java.awt.BorderLayout;
4. import java.awt.GridLayout;
5. import java.awt.event.ActionEvent;
6. import java.awt.event.ActionListener;
7. import java.io.File;
8. import java.io.FileOutputStream;
9. import java.io.IOException;
10. import java.io.OutputStream;
11. import java.net.DatagramPacket;
12. import java.net.DatagramSocket;
13. import java.net.InetAddress;
15. import javax.swing.*;
17. /**
18. * 下載客戶端
19. */
20. public class DownloadClient extends JFrame {
21. // 顯示可下載的文件
22. private JTextArea textArea = new JTextArea();
24. private JPanel panel = new JPanel();
25. // 下載時保存文件
26. private JFileChooser saveFile = new JFileChooser(".");
28. private JButton showButton = new JButton("顯示文件");
29. private JButton downloadButton = new JButton("下載...");
30. // 下載時填入要下載的文件名，注意文件名必須是textArea顯示的文件名
31. private JTextField downloadFile = new JTextField("");
33. private DatagramSocket dataSocket=null;
35. public DownloadClient() {
36. // frame 的基本設置
37. this.setTitle("下載客戶端");
38. this.setVisible(true);
39. this.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
40. this.setSize(400, 500);
41. this.setLayout(new BorderLayout());
42. this.setResizable(false);
44. // 設置不可編輯
45. textArea.setEditable(false);
47. panel.setLayout(new GridLayout(3, 2, 5, 5));
48. panel.add(new JLabel("點擊按鈕顯示可下載的文件"));
49. panel.add(showButton);
50. panel.add(downloadFile);
51. panel.add(downloadButton);
53. // 組件加入frame中
54. add(new JScrollPane(textArea));
55. add(panel, BorderLayout.SOUTH);
57. // saveFile只能打開目錄
58. saveFile.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
60. // 顯示文件按鈕注冊事件
61. showButton.addActionListener(new ActionListener() {
62. @Override
63. public void actionPerformed(ActionEvent e) {
64. showButton.setEnabled(false);
65. downloadButton.setEnabled(false);
66. showFiles();
67. showButton.setEnabled(true);
68. downloadButton.setEnabled(true);
69. }
70. });
72. // 下載按鈕注冊事件
73. downloadButton.addActionListener(new ActionListener() {
74. @Override
75. public void actionPerformed(ActionEvent e) {
76. showButton.setEnabled(false);
77. downloadButton.setEnabled(false);
78. downloadFile();
79. showButton.setEnabled(true);
80. downloadButton.setEnabled(true);
81. }
82. });
83. }
85. // 顯示文件
86. private void showFiles() {
87. try {
88. if (dataSocket == null)
89. dataSocket = new DatagramSocket();
90. // 創建發送數據包並發送給服務器
91. byte[] request = { 0 };
92. DatagramPacket requestPacket = new DatagramPacket(request, request.length, InetAddress.getLocalHost(), 8289);
93. dataSocket.send(requestPacket);
95. // 接收服務器的數據包，顯���在textArea中
96. byte[] receive = new byte[1024 * 1024];
97. DatagramPacket receivePacket = new DatagramPacket(receive, receive.length);
98. dataSocket.receive(receivePacket);
99. String str = new String(receivePacket.getData(), 0, receivePacket.getLength());
100. textArea.setText(str);
101. } catch (IOException e) {
102. e.printStackTrace();
103. }
104. }
106. // 下載文件
107. private void downloadFile() {
108. // 獲取要下載的文件名
109. String fileName = downloadFile.getText().trim();
110. // 所有可以下載的文件
111. String allFiles = textArea.getText();
112. // 文件名為空
113. if (fileName == null || "".equals(fileName))
114. JOptionPane.showMessageDialog(null, "請選中正確的文件名", "文件名錯誤", JOptionPane.WARNING_MESSAGE);
115. // 文件名是在可以下載的文件中
116. else if (allFiles.contains((fileName + '\n'))) {
117. saveFile.showSaveDialog(null);
118. File f = saveFile.getSelectedFile();// 獲取選中的文件夾
119. if (f.exists()) {
120. // 檢測該文件是否已經存在於目錄中
121. String[] fileNames = f.list();
122. boolean exit = false;
123. for (String name : fileNames)
124. if (name.equals(fileName)) {
125. exit = true;
126. break;
127. }
129. if (exit)// 如果要下載的文件已經存在
130. JOptionPane.showMessageDialog(null, "此文件已經存在", "請選擇另外的文件下載", JOptionPane.WARNING_MESSAGE);
131. else {
132. // 發送的請求
133. byte[] request = (new String(new byte[] { 1 }) + fileName).getBytes();
134. try {
135. if (dataSocket == null)
136. dataSocket = new DatagramSocket();
137. // 創建發送數據包並發送給服務器
138. DatagramPacket requestPacket = new DatagramPacket(request, request.length, InetAddress.getLocalHost(), 8289);
139. dataSocket.send(requestPacket);
141. // 接收服務器的數據包,把文件保存在選中的文件夾中
142. OutputStream out = new FileOutputStream(f.getAbsolutePath() + "/" + fileName, true);
143. byte[] receive = new byte[60000];// 每次接收60000字節
144. DatagramPacket receivePacket;
145. // 不斷接收來自服務器的數據包
146. while (true) {
147. receivePacket = new DatagramPacket(receive, receive.length);
148. dataSocket.receive(receivePacket);
149. out.write(receivePacket.getData(), 0, receivePacket.getLength());// 輸出流把文件內容輸出到文件中
150. out.flush();
151. if (receivePacket.getLength() != receive.length)
152. break;
153. }
154. out.close();
155. } catch (IOException e) {
156. e.printStackTrace();
157. }
158. }
160. } else
161. // 選擇的文件夾不存在
162. JOptionPane.showMessageDialog(null, "請選擇正確的存儲路徑", "存儲路徑錯誤", JOptionPane.WARNING_MESSAGE);
164. } else {// 文件名錯誤
165. JOptionPane.showMessageDialog(null, "請選擇正確的文件名", "文件名錯誤", JOptionPane.WARNING_MESSAGE);
166. }
167. }
169. public static void main(String[] args) {
170. new DownloadClient();
171. }
172. }

說明：客戶端下載時先點擊顯示文件，然後輸入要下載的文件名，最後點擊下載按鈕、選擇保存路徑

3、運行時先開服務器，然後再開客戶端。運行結果：

點擊顯示文件按鈕、輸入01.jpg後：

點擊下載按鈕後、選擇保存路徑

點擊保存按鈕後,文件下載到指定目錄下