Java TCP文件上传小程序

本文最后更新于：2022年12月19日 晚上

一个基于TCP的简单的文件上传程序

客户端

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import java.io.*;
import java.net.*;
import java.util.Scanner;

public class LocalClient {
    public static void main(String[] args) throws IOException {
        //文件路径和文件
        String filePath;
        Scanner sc = new Scanner(System.in);
        System.out.println("输入要上传的文件地址：");
        filePath = sc.nextLine();
        //创建本地字节输入流，读取文件
        FileInputStream fis = new FileInputStream(filePath);
        //服务器地址及端口
        Socket socket = new Socket("127.0.0.1", 7222);
        //创建网络字节输出流，向服务器传递文件
        OutputStream os = socket.getOutputStream();
        byte[] bytes = new byte[1024];
        int len = 0;
        while((len = fis.read(bytes)) != -1) {
            os.write(bytes, 0, len);
        }
        //shutdown结束上传
        socket.shutdownOutput();
        //创建网络字节输入流，读取服务器传递的信息
        InputStream is = socket.getInputStream();
        while((len = is.read(bytes)) != -1) {
            System.out.println(new String(bytes, 0, len));
        }
        //关闭
        fis.close();
        socket.close();
    }
}

服务端

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import java.io.*;
import java.net.*;
import java.util.Random;
public class Server {
    public static void main(String[] args) throws IOException {
        //创建服务器端口
        ServerSocket server = new ServerSocket(7222);
        //使用while使服务器保持监听状态
        while (true) {
            //接受客户端请求
            Socket socket = server.accept();
            //使用多线程上传
            new Thread(new Runnable() {
                @Override
                public void run() {
                    try {
                        //创建网络字节输入流，收取客户端上传的文件
                        InputStream is = socket.getInputStream();
                        //判断文件夹是否存在，不存在则创建文件夹
                        File file = new File("C:\\UpdatePath");
                        if (!file.exists()) {
                            file.mkdirs();
                        }
                        //创建上传文件的命名规则
                        //int nums = 0;
                        String fileName = "File" + System.currentTimeMillis() + new Random().nextInt(99) + ".zip";
                        //nums++;
                        //创建本地字节输出流，向硬盘中保存文件
                        FileOutputStream fos = new FileOutputStream(file + "\\" + fileName);
                        byte[] bytes = new byte[1024];
                        int len = 0;
                        while ((len = is.read(bytes)) != -1) {
                            fos.write(bytes, 0, len);
                        }
                        //创建网络字节输出流，向客户端发送已经上传完成的信息
                        OutputStream os = socket.getOutputStream();
                        os.write("Update complete! =w= ~".getBytes());
                        //关闭
                        fos.close();
                        socket.close();
                    } catch (IOException e) {
                        System.out.println(e);
                    }
                }
            }).start();
        }
        //server.close();
    }
}

以后有空可以再完善一下，识别上传文件的文件类型和文件名