Java实现TCP通信的文件上传 之 代码优化开启多线程

1. 添加了文件名不重复,这样可以上传多个文件
2. 添加while循环,可以一直accept 客户端的socket
3. 开启多线程,可以多用户访问

服务器端: TCPServer.java

package com.zhanshen.demo_file_upload;


import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.Random;

/**
 * 读取客户端上传的文件,把文件保存到本地  然后回写保存成功等提示消息
 */
public class TCPServer {
    public static void main(String[] args) throws IOException {
        //1. 创建ServerSocket,指定端口号8888
        ServerSocket serverSocket = new ServerSocket(8888);

        new Thread(new Runnable() {
            @Override
            public void run() {
                try {


                    while (true) {
                        //2. 使用ServerSocket的accept来获得客户端请求的socket对象
                        Socket socket = serverSocket.accept();



                        //3. 获取socket的输入流
                        InputStream inputStream = socket.getInputStream();

                        //4. 判断是否有要保存文件的文件夹,若没有就创建一个
                        File file = new File("/home/zhanshen/IdeaProjects/Net/src/com/zhanshen/demo_file_upload/upload");
                        if(!file.exists()) {   //如果这个文件夹不存在
                            file.mkdirs();   //若没有,就创建他
                        }


                        String fileName = "file" + System.currentTimeMillis() + new Random().nextInt(999999)+".png";

                        //5. 创建本地输出流,写入到本地硬盘的位置
                        FileOutputStream fileOutputStream = new FileOutputStream(file+"/"+fileName);

                        //6. 使用InputStream 来读取客户端传过来的数据
                        int len = 0;
                        byte[] bytes = new byte[1024];
                        while( (len = inputStream.read(bytes) ) != -1) {

                            //7. 写入到硬盘
                            fileOutputStream.write(bytes, 0, len);
                        }

                        //8. 获取socket的OutputStream来给客户端回显数据
                        socket.getOutputStream().write("上传成功!".getBytes());

                        //9. 释放资源
                        fileOutputStream.close();
                        socket.close();
                        //serverSocket.close();
                    }

                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }).start();


    }
}

客户端: TCPClient.java

package com.zhanshen.demo_file_upload;


import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;

/**
 * 客户端读取本地文件,上传到服务器,读取服务器回写的数据
 */
public class TCPClient {

    public static void main(String[] args) throws IOException {

        //1. 创建一个本地字节输入流fileInputStream, 构造方法中绑定要上传的文件
        FileInputStream fileInputStream = new FileInputStream("/home/zhanshen/IdeaProjects/Net/src/com/zhanshen/demo_file_upload/xv.png");

        //2. 创建客户端Socket对象,绑定要上传到服务器的IP地址和端口号
        Socket socket = new Socket("127.0.0.1", 8888);

        //3. 获取网络字节输出流OutputStream
        OutputStream outputStream = socket.getOutputStream();

        //4. 使用FIleInputStream 读取本地文件
        /**
         * 这一次我们的读取是循环读取,因为一个文件会很大,所以我们循环读取每次读取1024
         */
        int len = 0;
        byte[] bytes = new byte[1024];
        while( (len = fileInputStream.read(bytes) ) != -1) {
            //5. 上传到服务器
            outputStream.write(bytes, 0, len);
        }

        socket.shutdownOutput();  //当上传文件结束高速服务器结束了,防止阻塞

        //6. 获取InputStream
        InputStream inputStream = socket.getInputStream();

        //7. 读取服务器返回的数据
        while ( (len = inputStream.read(bytes) ) != -1) {
            System.out.println(new String(bytes, 0, len));
        }

        //8. 释放资源
        fileInputStream.close();
        socket.close();
    }
}