Java实现文件上传功能

Hello World, Hello NetWorks

Posted by Resulte on March 9, 2020

Java实现文件上传功能

文件上传功能是互联网中非常常见的应用,例如修改用户头像时需要上传照片,把文件上传到云端网盘等等。

其网络传输协议可以使用TCP协议,在Java中可以通过SocketServerSocket实现。其本质就是把本地客户端的文件复制到服务器端,对文件复制还不熟悉的小伙伴可以再回顾一下我的这篇博客:用Java实现文件复制

下面我将以上传图片文件photo.jpg为例,总结文件上传的几个步骤。

客户端

目的:读取本地文件,上传到服务器,读取服务器回写的数据

实现步骤:

  1. 创建一个本地字节输入流FileInputStream对象,构造方法中绑定要读取的数据源
  2. 创建一个客户端Socket对象,构造方法中绑定服务器的IP地址和端口号
  3. 使用Socket中的方法getOutputStream,获取网络字节输出流OutputStream对象
  4. 使用本地字节输入流FileInputStream对象中的方法read,读取本地文件
  5. 使用网络字节输出流OutputStream对象中的方法write,把读取到的文件上传到服务器
  6. 使用Socket中的方法getInputStream,获取网络字节输入流InputStream对象
  7. 使用网络字节输入流InputStream对象中的方法read读取服务回写的数据
  8. 释放资源(FileInputStream,Socket)

注:本例中我使用的IP地址为本地地址127.0.0.1,端口号为8000,这些可以根据需要修改即可。

代码实现如下:

public class FileUploadClient {
    public static void main(String[] args) throws IOException {
        //1.创建一个本地字节输入流FileInputStream对象,构造方法中绑定要读取的数据源
        FileInputStream fis = new FileInputStream("C:\\photo.jpg");
        //2.创建一个客户端Socket对象,构造方法中绑定服务器的IP地址和端口号
        Socket socket = new Socket("127.0.0.1", 8000);
        //3.使用Socket中的方法getOutputStream,获取网络字节输出流OutputStream对象
        OutputStream os = socket.getOutputStream();
        //4.使用本地字节输入流FileInputStream对象中的方法read,读取本地文件
        byte[] bytes = new byte[1024];
        int len = 0;
        while((len = fis.read(bytes)) != -1){
            //5.使用网络字节输出流OutputStream对象中的方法write,把读取到的文件上传到服务器
            os.write(bytes, 0, len);
        }
        //关闭此套接字的输出流
        socket.shutdownOutput();
        //6.使用Socket中的方法getInputStream,获取网络字节输入流InputStream对象
        InputStream is = socket.getInputStream();
        //7.使用网络字节输入流InputStream对象中的方法read读取服务回写的数据
        while((len = is.read(bytes)) != -1){
            System.out.println(new String(bytes, 0, len));
        }
        //8.释放资源(FileInputStream,Socket)
        fis.close();
        socket.close();
    }
}

服务器端

目的:读取客户端上传的文件,保存到服务器的硬盘,给客户端回写”上传成功”

注:本例中文件上传的目的地:服务器的硬盘 d:\upload文件夹下

实现步骤

  1. 创建一个服务器ServerSocket对象,和系统要指定的端口号
  2. 使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
  3. 使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
  4. 判断d:\\upload文件夹是否存在,不存在则创建
  5. 创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
  6. 使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
  7. 使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上
  8. 使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
  9. 使用网络字节输出流OutputStream对象中的方法write,给客户端回写”上传成功”
  10. 释放资源(FileOutputStream,Socket)

代码实现如下:

public class FileUploadServer {
    public static void main(String[] args) throws IOException {
        //1.创建一个服务器ServerSocket对象,和系统要指定的端口号
        ServerSocket server = new ServerSocket(8000);
        //让服务器一直处于监听状态
        while (true){
            //2.使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
            Socket socket = server.accept();
            //使用多线程技术,提高程序的效率
            new Thread(new Runnable() {
                @Override
                public void run() {
                    try{
                        //3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
                        InputStream is = socket.getInputStream();
						//4.判断d:\\upload文件夹是否存在,不存在则创建
                        File file = new File("D:\\upload");
                        if(!file.exists()){
                            file.mkdir();
                        }
                        //为文件随机创建一个文件名,确保不重复
                        String fileName = System.currentTimeMillis() + new Random().nextInt(9000) + ".txt";
                        //5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
                        FileOutputStream fos = new FileOutputStream(file + File.separator + fileName);
						//6.使用网络字节输入流InputStream对象中的方法read,读取客户端上传的文件
                        byte[] bytes = new byte[1024];
                        int len = 0;
                        while((len = is.read(bytes)) != -1){
                            //7.使用本地字节输出流FileOutputStream对象中的方法write,把读取到的文件保存到服务器的硬盘上
                            fos.write(bytes, 0, len);
                        }
                        //8.使用Socket对象中的方法getOutputStream,获取到网络字节输出流OutputStream对象
                        OutputStream os = socket.getOutputStream();
                        //9.使用网络字节输出流OutputStream对象中的方法write,给客户端回写"上传成功"
                        os.write("upload success".getBytes());
						//10.释放资源(FileOutputStream,Socket)
                        fos.close();
                        socket.close();
                    }catch (IOException e){
                        e.printStackTrace();
                    }
                }
            }).start();
        }
    }
}