TCP实现文件上传功能

服务端

package com.WL;

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

public class TcpServerDemo2 {
    public static void main(String[] args) throws IOException {
        ServerSocket serverSocket = new ServerSocket(9000);

        Socket socket = serverSocket.accept();//阻塞式监听

        //获取输入流
        InputStream is = socket.getInputStream();

        //文件输出
        FileOutputStream fs = new FileOutputStream(new File("s.jpg"));
		//创建一个字节数组，大小根据实际调整，用于读取输入流的内容
        byte[] buffer = new byte[1024];

        int len;
        while (( len= is.read(buffer)) != -1){
            fs.write(buffer,0,len);
        }
		//socket这个管道可以实现客户端和服务端数据传输
        OutputStream os = socket.getOutputStream();
        os.write("接收完毕".getBytes());

        fs.close();
        is.close();
        socket.close();
        serverSocket.close();
    }
}

客户端

package com.WL;

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

public class TcpClientDemo2 {
    public static void main(String[] args) throws IOException {
        //创建一个socket连接
        Socket socket = new Socket(InetAddress.getByName("127.0.0.1"), 9000);
        //创建一个输出流
        OutputStream os = socket.getOutputStream();
        //读取文件
        FileInputStream fileInputStream = new FileInputStream(new File("d.jpg"));
        //将读取的内容输出
        byte[] buffer = new byte[1024];
        int len;
        while ((len=fileInputStream.read(buffer))!=-1){
            os.write(buffer,0,len);
        }
        //告知服务端数据传输完成
        socket.shutdownOutput();
		
		//接收服务端的输出
        InputStream is = socket.getInputStream();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        byte[] buffer1 = new byte[1024];
        int len1;
        while ((len1=is.read(buffer1)) !=-1){
            baos.write(buffer1,0,len1);
        }
        System.out.println(baos.toString());

        fileInputStream.close();
        os.close();
        socket.close();

    }
}