\(^_^)/ Java 使用Socket多线程传输整个文件夹

一.列出莫文件夹中所有的文件和文件夹的工具类，磁盘列表类

 

Java代码  

1. package upload;  
2.   
3. import java.io.File;  
4. import java.util.ArrayList;  
5. import java.util.List;  
6.   
7. public class ListFiles {  
8.     private List<File> fileList = new ArrayList<File>();  
9.   
10.     public void listAllFiles(File file) {  
11.         File[] files = file.listFiles();  
12.         for (int i = 0; i < files.length; i++) {  
13.             if (files[i].isDirectory()) {  
14.                 if (files[i].listFiles() != null && files[i].listFiles().length > 0) {  
15.                     listAllFiles(files[i]);  
16.                 } else {  
17.                     fileList.add(files[i]);  
18.                 }  
19.             } else {  
20.                 fileList.add(files[i]);  
21.             }  
22.         }  
23.     }  
24.   
25.     public List<File> getAllFiles(File file) {  
26.         this.listAllFiles(file);  
27.         return fileList;  
28.     }  
29.   
30.     public static void main(String[] args) {  
31.         System.out.println(new ListFiles().getAllFiles(new File("E:\\temp\\")));  
32.     }  
33. }  

 

Java代码  

1. package upload;  
2.   
3. public enum DiskName {  
4.     C, D, E, F, G, H, I, J, K  
5. }  

 

 

 

 

 

二.服务器端程序

1.文件处理

Java代码  

1. package upload;  
2.   
3. import java.io.DataInputStream;  
4. import java.io.File;  
5. import java.io.FileOutputStream;  
6. import java.io.IOException;  
7. import java.net.Socket;  
8.   
9. public class UploadFileServer implements Runnable {  
10.   
11.     private Socket socket;  
12.     private DiskName diskName;  
13.   
14.     public UploadFileServer(Socket socket, DiskName diskName) {  
15.         this.socket = socket;  
16.         this.diskName = diskName;  
17.     }  
18.   
19.     @Override  
20.     public void run() {  
21.         DataInputStream dis = null;  
22.         FileOutputStream fos = null;  
23.   
24.         try {  
25.             long start = System.currentTimeMillis();  
26.             dis = new DataInputStream(socket.getInputStream());  
27.             String filePath = dis.readUTF();  
28.             String fileName = filePath.substring(filePath.lastIndexOf("\\") + 1);  
29.             String strDir = diskName.name()  
30.                     + filePath.substring(filePath.indexOf(":"), filePath.lastIndexOf("\\"))  
31.                     + File.separator;  
32.   
33.             File d = new File(strDir);  
34.             d.mkdirs();  
35.   
36.             File file = new File(strDir + fileName);  
37.             file.createNewFile();  
38.   
39.             if (file.isFile()) {  
40.                 fos = new FileOutputStream(file);  
41.                 byte[] buffer = new byte[1024];  
42.                 int len = 0;  
43.                 while ((len = dis.read(buffer, 0, buffer.length)) > 0) {  
44.                     fos.write(buffer, 0, len);  
45.                     fos.flush();  
46.                 }  
47.             }  
48.   
49.             long end = System.currentTimeMillis();  
50.             System.out.println("cost: " + ((end - start) / 1000) + " s, size: "  
51.                     + (file.length() / 1024.0) + " KB , file: " + file.getAbsolutePath());  
52.         } catch (IOException e) {  
53.             e.printStackTrace();  
54.         } finally {  
55.             if (fos != null) {  
56.                 try {  
57.                     fos.close();  
58.                 } catch (IOException e) {  
59.                     e.printStackTrace();  
60.                 }  
61.             }  
62.             if (dis != null) {  
63.                 try {  
64.                     dis.close();  
65.                 } catch (IOException e) {  
66.                     e.printStackTrace();  
67.                 }  
68.             }  
69.             if (socket != null) {  
70.                 try {  
71.                     socket.close();  
72.                 } catch (IOException e) {  
73.                     e.printStackTrace();  
74.                 }  
75.             }  
76.         }  
77.   
78.     }  
79.   
80. }  

 

2.启动类

Java代码  

1. package upload;  
2.   
3. import java.io.IOException;  
4. import java.net.ServerSocket;  
5. import java.net.Socket;  
6.   
7. public class SocketServer {  
8.     public static void main(String[] args) throws IOException {  
9.         ServerSocket server = new ServerSocket(9999);  
10.         while (true) {  
11.             Socket socket = server.accept();  
12.             new Thread(new UploadFileServer(socket, DiskName.D)).start();  
13.         }  
14.     }  
15. }  

 

 

 

 

 

 

三.客户端程序

1.文件处理

Java代码  

1. package upload;  
2.   
3. import java.io.DataOutputStream;  
4. import java.io.File;  
5. import java.io.FileInputStream;  
6. import java.io.IOException;  
7. import java.net.Socket;  
8.   
9. public class UploadFileClient implements Runnable {  
10.     private Socket socket;  
11.     private File file;  
12.   
13.     public UploadFileClient(Socket socket, File file) {  
14.         this.socket = socket;  
15.         this.file = file;  
16.     }  
17.   
18.     @Override  
19.     public void run() {  
20.         DataOutputStream dos = null;  
21.         FileInputStream fis = null;  
22.         try {  
23.             long start = System.currentTimeMillis();  
24.             dos = new DataOutputStream(socket.getOutputStream());  
25.             if (file.isFile()) {  
26.                 fis = new FileInputStream(file);  
27.   
28.                 dos.writeUTF(file.getAbsolutePath());  
29.                 byte[] sendBuffer = new byte[1024];  
30.                 int len = 0;  
31.                 while ((len = fis.read(sendBuffer, 0, sendBuffer.length)) > 0) {  
32.                     dos.write(sendBuffer, 0, len);  
33.                     dos.flush();  
34.                 }  
35.             } else {  
36.                 dos.writeUTF(file.getAbsolutePath() + File.separator);  
37.             }  
38.             long end = System.currentTimeMillis();  
39.             System.out.println("cost: " + ((end - start) / 1000) + " s, size: "  
40.                     + (file.length() / 1024.0) + " KB , file: " + file.getAbsolutePath());  
41.         } catch (IOException e) {  
42.             e.printStackTrace();  
43.         } finally {  
44.             if (dos != null) {  
45.                 try {  
46.                     dos.close();  
47.                 } catch (IOException e) {  
48.                     e.printStackTrace();  
49.                 }  
50.             }  
51.             if (fis != null) {  
52.                 try {  
53.                     fis.close();  
54.                 } catch (IOException e) {  
55.                     e.printStackTrace();  
56.                 }  
57.             }  
58.             if (socket != null) {  
59.                 try {  
60.                     socket.close();  
61.                 } catch (IOException e) {  
62.                     e.printStackTrace();  
63.                 }  
64.             }  
65.         }  
66.   
67.     }  
68.   
69. }  

 

 

2.启动类

Java代码  

1. package upload;  
2.   
3. import java.io.File;  
4. import java.io.IOException;  
5. import java.net.InetSocketAddress;  
6. import java.net.Socket;  
7. import java.util.List;  
8.   
9. public class SocketClient {  
10.     public static void main(String[] args) throws IOException {  
11.         List<File> fileList = new ListFiles().getAllFiles(new File("E:\\java standard\\"));  
12.         for (int i = 0; i < fileList.size(); i++) {  
13.             Socket socket = new Socket();  
14.             socket.connect(new InetSocketAddress("127.0.0.1", 9999));  
15.             new Thread(new UploadFileClient(socket, fileList.get(i))).start();  
16.         }  
17.     }  
18. }  

 

• FileUpload.zip (21.9 KB)
• 下载次数: 1

• FileUpload.7z (13.7 KB)
• 下载次数: 1

转自：http://yanguz123.iteye.com/blog/2019151