Java系列 - socket 缓冲流传输文件

• 看的老韩的视频，记录一下socket缓冲流传输。
• 封装的方法注意 read(buff) 传参
• 设置结束标记时，注意

package com.network.ioexercise.networkmain;

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

/**
 * 1. 编写服务端和客户端，服务器再9999端口监听
 * 2. 客户端连接服务端，发送一张图片 D:\\pic.png
 * 3. 服务端收到图片保存再src下，返回 “收到图片”在退出
 * 4.客户端收到服务端的消息后退出
 */
public class TCPFileUploadServer {
    public static void main(String[] args) throws IOException {

        System.out.println("服务器已创建");
        // 1. 服务端在本机监听 9999 端口
        ServerSocket serverSocket = new ServerSocket(7888);

        // 2. 等待连接
        Socket socket = serverSocket.accept();

        System.out.println("客户端已连接");

        // 3. 读取客户端发送的数据
        BufferedInputStream bis = new BufferedInputStream(socket.getInputStream());

        byte[] bytes = StreamUtilTest.streamToByteArray(bis);
        System.out.println(bytes.length);

        // 4. 得到 byteArray,写入指定路径
        String destFilePath = "src\\copy.png";
        BufferedOutputStream saveStream = new BufferedOutputStream(new FileOutputStream(destFilePath));
        saveStream.write(bytes);
        saveStream.close();
        System.out.println("服务端收到图片，并保存完毕");

        // 4. 给客户端回复一个收到了的消息
        BufferedWriter bufferedWriter = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));

        bufferedWriter.write("服务端: 已经收到图片了\r\n");
        bufferedWriter.flush();

        bufferedWriter.write("服务端: 我还想多说一句");
        bufferedWriter.flush();

        // 关闭，关闭的时候会自动刷新
        bufferedWriter.close();
        bis.close();
        socket.close();
        serverSocket.close();

    }
}

package com.network.ioexercise.networkmain;

import java.io.*;
import java.net.InetAddress;
import java.net.Socket;

public class TCPFileUploadClient {
    public static void main(String[] args) throws IOException {
        // 1. 客户端连接
        Socket socket = new Socket(InetAddress.getLocalHost(),7888);

        // 2. 获取磁盘文件的输入流
        String filePath = "D:\\pic.png";
        BufferedInputStream fileStream = new BufferedInputStream(new FileInputStream(filePath));
        byte[] byteArray = StreamUtilTest.streamToByteArray(fileStream);

        // 3.通过socket获取输出流，将bytes数据发送给服务端
        BufferedOutputStream bos = new BufferedOutputStream(socket.getOutputStream());
        bos.write(byteArray); // 将文件对应的字节数组数据写入通道
        fileStream.close();
        bos.flush();
        socket.shutdownOutput();// 设置写入数据的结束标记

        System.out.println("客户端: 图片发送完成");

        InputStream bis = socket.getInputStream();
        String s = StreamUtilTest.streamToString(bis);

        System.out.println(s);

        bis.close();
        bos.close();
        socket.close();
    }
}

package com.network.ioexercise.networkmain;

import java.io.*;

public class StreamUtilTest {

    /**
     * stream转化为字节数组
     *
     * @param is
     * @return
     * @throws IOException
     */
    public static byte[] streamToByteArray(InputStream is) throws IOException {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();// 创建输出流
        byte[] buffer = new byte[1024]; // 字节数组
        int len = 0;
        // read()方法是有参数的！
        while ((len = is.read(buffer)) != -1) {
            bos.write(buffer,0,len);
        }
        byte[] array = bos.toByteArray();//输出流转换为数组
        bos.close();
        return array;
    }

    /**
     * stream 转化为 字符串
     * @param is
     * @return
     * @throws IOException
     */
    public static String streamToString(InputStream is) throws IOException {
        BufferedReader stream = new BufferedReader(new InputStreamReader(is));
        StringBuffer stringBuffer = new StringBuffer();

        String line;
        while ((line = stream.readLine()) != null) {
            stringBuffer.append(line + "\r\n");
        }
        return stringBuffer.toString();
    }
}

源码

---

------ 如果文章对你有用，感谢 >>>点赞 | 收藏 <<<