TCP Socket 实现文件上传

文件上传过程一个单向Socket通信过程。客户端通过文件输入流读取文件,然后从Socket获取输出流写入数据。服务端从Socket中获得输入流,然后写入文件输出流,写入数据完成则上传完成。

服务端UploadServer:

public class UplaodServer {
    public static void main(String []args){
        try(
            // 创建一个ServerSocket监听8080端口的请求
            // ServerSocket 实现了 AutoCloseable接口, 可以自动释放资源
        	ServerSocket server = new ServerSocket(8080);
            // accept()会阻塞当前线程, 等待客户端请求 
            Socket socket = server.accept();
            // 获取输入流
            BufferedInputStream in = new BufferedInputStream(socket.getInputStream());
            // 由文件输出流创建缓冲输出流, 写入本地upload.jpg 文件
            FileOutputStream outputStream = new FileOutputStream("./upload.jpg")
        ){
            // 缓冲区 
            byte [] buffer = new byte[1024];
            int len = in.read(buffer);
            while(len != -1){
                // 写入数据到文件
                outputStream.write(buffer, 0, len);
                // 再次从Socket中读取数据
                len = in.read(buffer);
            }
            System.out.println("文件已上传");
        }catch (IOException e) {
            e.printStackTrace();
        }
    }
}

客户端UploadClient:

public class UploadClient {
    public static void main(String []args){
        try(
        	// Socket 实现了 AutoCloseable接口 可自动关闭资源
            Socket socket = new Socket("127.0.0.1", 8080);
             // 又Socket获得输出流, 并创建缓冲输出流
            BufferedOutputStream out = new BufferedOutputStream(socket.getOutputStream());
            FileInputStream fileInput = new FileInputStream("F:\file\oxx6j5.jpg");
            // 由文件输入流创建缓冲输入流
            BufferedInputStream in = new BufferedInputStream(fileInputStream);
        ){
             // 准备缓冲区
            byte[] buffer = new byte[1024];
            int len = in.read(buffer);
            while (len != -1) {
                // 写入Socket
                out.write(buffer, 0, len);
                // 再次读取文件
                len = in.read(buffer);
            }
            System.out.println("上传成功!");
        }catch (ConnectException e) {
            System.out.println("服务器未启动");
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

注意

server.accept()会阻塞线程,所以最好把server.accept()语句放入子线程中。

posted @ 2020-12-11 00:46  weiQLog  阅读(271)  评论(0编辑  收藏  举报