TCP文件上传实现

package edu.wtbu;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
//服务器端
public class Demo05 {
    public static void main(String[] args) throws IOException {
        //1.创建服务
        ServerSocket serverSocket = new ServerSocket(9000);
        //2.监听客户端的连接
        Socket socket = serverSocket.accept();//阻塞式监听，会一直等待客户端连接
        //3.获取输入流
        InputStream is = socket.getInputStream();

        //4.文件输出
        FileOutputStream fos = new FileOutputStream(new File("b.jpg"));
        byte[] bytes = new byte[1024];
        int len;
        while ((len=is.read())!=-1){
            fos.write(bytes,0,len);
        }


        //通知客户端已经接收完毕了
        OutputStream os = socket.getOutputStream();
        os.write("我接收完毕了".getBytes());
        //5.关闭资源

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

}

package edu.wtbu;

import java.io.*;
import java.net.InetAddress;
import java.net.Socket;
import java.net.UnknownHostException;
//客户端
public class Demo04 {
    public static void main(String[] args) throws IOException {
       //1.创建一个Socket连接
        Socket socket = new Socket(InetAddress.getByName("127.0.0.1"), 9000);
        //2.创建一个输出流
        OutputStream os = socket.getOutputStream();

        //3.读取文件
        FileInputStream fis = new FileInputStream(new File("a.jpg"));
        //4.写出文件
        byte[] bytes = new byte[1024];
        int len;
        while ((len=fis.read(bytes))!=-1){
            os.write(bytes,0,len);
        }
        //通知服务器，我已经结束了
        socket.shutdownOutput();//我已经传输完了
        //确定服务器接收完毕，才断开连接
        InputStream is = socket.getInputStream();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        byte[] bytes1 = new byte[1024];
        int len2;
        while ((len2=is.read(bytes1))!=-1){
            baos.write(bytes1,0,len2);
        }
        System.out.println(baos.toString());

        //5.关闭资源
        baos.close();
        is.close();
        fis.close();
        os.close();
        socket.close();
    }
}