基于Socket实现文件上传(多线程版本)

一、编写文件上传的客户端

public class FileUploadClient {

    public static void main(String[] args) throws IOException {
        /*1.创建客户端*/
        Socket socket = new Socket("127.0.0.1", 8888);

        /*2.获取Socket的输出流*/
        OutputStream os = socket.getOutputStream();

        /*3.创建文件输入流*/
        FileInputStream fis = new FileInputStream("D:/Document/Picture/one.jpg");

        /*4.循环读写数据*/
        int len;
        byte[] buff = new byte[1024];
        while ((len = fis.read(buff)) != -1) {
            os.write(buff, 0, len);
        }
        /*文件发送完毕了, 给Socket输出流写一个结束标记*/
        socket.shutdownOutput();
        System.out.println("客户端上传完成!");

        /*5.得到Socket的输入流*/
        InputStream is = socket.getInputStream();

        /*6.读取数据*/
        byte[] buff2 = new byte[1024];
        int len2 = is.read(buff2);
        System.out.println("客户端收到：" + new String(buff2, 0, len2));

        /*7.关闭资源*/
        is.close();
        fis.close();
        os.close();
        socket.close();
    }

}

二、编写线程执行的类

public class FileUploadTask implements Runnable {
    
    private Socket socket;

    public FileUploadTask(Socket socket) {
        this.socket = socket;
    }

    @Override
    public void run() {
        
        try {
            /*3.获取输入流*/
            InputStream is = socket.getInputStream();

            /*文件名的组成：时间的毫秒值 + 随机数 + 上传者的IP*/
            String ip = socket.getInetAddress().getHostAddress();
            String fileName = System.currentTimeMillis() + "_" + new Random().nextInt(88888888) + "_" + ip + ".jpg";

            /*4.创建文件输出流*/
            FileOutputStream fos = new FileOutputStream("D:/Document/Images/upload/" + fileName);

            /*5.循环读写数据*/
            int len;
            byte[] buff = new byte[1024];
            while ((len = is.read(buff)) != -1) {
                fos.write(buff, 0, len);
            }
            System.out.println("服务端上传完成！");

            /*6.得到输出流*/
            OutputStream out = socket.getOutputStream();

            /*7.发送数据给客户端*/
            out.write("上传完成啦！".getBytes());

            /*8.关闭资源*/
            out.close();
            fos.close();
            is.close();
            socket.close();
        } catch (Exception e) {
            System.out.println("文件上传失败！");
        }
        
    }
    
}

三、编写文件上传的服务端

public class FileUploadServer {

    public static void main(String[] args) throws IOException {
        System.out.println("文件上传服务端启动啦！");

        /*1.创建服务端*/
        ServerSocket server = new ServerSocket(8888);

        while (true) {
            /*2.同意客户端的请求*/
            Socket socket = server.accept();

            /*创建一个新的线程执行上传的任务*/
            FileUploadTask task = new FileUploadTask(socket);
            new Thread(task).start();
        }
    }

}