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

1、服務器端代碼

1. package demo.net;  
2.   
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;  
13.   
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);  
25.   
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.     }  
39.   
40.     // 內部類，為每個客戶提供服務   
41.     private class WorkThread implements Runnable {  
42.         private DatagramPacket packet;  
43.         private DatagramSocket dataSocket;  
44.   
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.         }  
53.   
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.         }  
74.   
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.         }  
95.   
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.     }  
109.   
110.     public static void main(String[] args) {  
111.         new DownloadServer().service();  
112.     }  
113. }   

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

2、客戶端代碼

1. package demo.net;  
2.   
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;  
14.   
15. import javax.swing.*;  
16.   
17. /** 
18.  * 下載客戶端 
19.  */  
20. public class DownloadClient extends JFrame {  
21.     // 顯示可下載的文件   
22.     private JTextArea textArea = new JTextArea();  
23.   
24.     private JPanel panel = new JPanel();  
25.     // 下載時保存文件   
26.     private JFileChooser saveFile = new JFileChooser(".");  
27.   
28.     private JButton showButton = new JButton("顯示文件");  
29.     private JButton downloadButton = new JButton("下載...");  
30.     // 下載時填入要下載的文件名，注意文件名必須是textArea顯示的文件名   
31.     private JTextField downloadFile = new JTextField("");  
32.   
33.     private DatagramSocket dataSocket=null;  
34.   
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);  
43.   
44.         // 設置不可編輯   
45.         textArea.setEditable(false);  
46.   
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);  
52.   
53.         // 組件加入frame中   
54.         add(new JScrollPane(textArea));  
55.         add(panel, BorderLayout.SOUTH);  
56.   
57.         // saveFile只能打開目錄   
58.         saveFile.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);  
59.   
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.         });  
71.   
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.     }  
84.   
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);  
94.   
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.     }  
105.   
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.                     }  
128.   
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);  
140.   
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.                 }  
159.   
160.             } else  
161.                 // 選擇的文件夾不存在   
162.                 JOptionPane.showMessageDialog(null, "請選擇正確的存儲路徑", "存儲路徑錯誤", JOptionPane.WARNING_MESSAGE);  
163.   
164.         } else {// 文件名錯誤   
165.             JOptionPane.showMessageDialog(null, "請選擇正確的文件名", "文件名錯誤", JOptionPane.WARNING_MESSAGE);  
166.         }  
167.     }  
168.   
169.     public static void main(String[] args) {  
170.         new DownloadClient();  
171.     }  
172. }  

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

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

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

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

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