Java网络编程Demo（二）TCP文件上传

Demo_03.java

package TCP;

import java.io.*;
import java.net.*;

public class Demo_03 {
    public static void main(String[] args) throws Exception {
        ServerSocket serverSocket = new ServerSocket(10001);

        while (true) {
            Socket socket = serverSocket.accept();
            new Thread(new ServerThread(socket)).start();
        }
    }
}

// TCP服务端
class ServerThread implements Runnable {
    private Socket socket;

    public ServerThread(Socket socket) {
        this.socket = socket;
    }

    public void run() {
        String name = socket.getInetAddress().getHostName();    // 获取客户端的主机名
        int count = 1;                                          // 上传文件个数

        try {
            InputStream in = socket.getInputStream();
            File parentFile = new File("D:\\Test\\");   // 创建上传文件目录的File对象

            // 如果不存在, 就创建这个目录
            if (!parentFile.exists()) {
                parentFile.mkdir();
            }

            File file = new File(parentFile, name + "(" + count + ").jpg");

            while (file.exists()) {
                file = new File(parentFile, name + "(" + (count++) + ").jpg");
            }

            FileOutputStream fos = new FileOutputStream(file);
            byte[] buf = new byte[1024];
            int len = 0;

            while ((len = in.read(buf)) != -1) {
                fos.write(buf, 0, len);
            }

            OutputStream out = socket.getOutputStream();
            out.write("上传文件成功!".getBytes());    // 上传文件成功后向客户端写出"上传文件成功!"

            fos.close();
            socket.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

Demo_04.java

package TCP;

import java.io.*;
import java.net.*;

// TCP客户端
public class Demo_04 {
    public static void main(String[] args) throws Exception {
        Socket socket = new Socket("192.168.1.100", 10001);     // 创建客户端Socket
        socket.setSoTimeout(10000);     // 设置超时

        OutputStream out = socket.getOutputStream();
        FileInputStream fis = new FileInputStream("D:\\Test.jpg");  // 要上传的文件
        byte[] buf = new byte[1024];
        int len;

        while ((len = fis.read(buf)) != -1) {
            out.write(buf, 0, len);
        }

        socket.shutdownOutput();    // 关闭客户端输出流
        InputStream in = socket.getInputStream();
        byte[]bufMsg = new byte[1024];
        int num = in.read(bufMsg);  // 接收服务端的信息
        System.out.println(new String(bufMsg, 0, num));

        fis.close();
        socket.close();
    }
}

运行结果

• Demo_03.java
  依次开启3个客户端，分别上传文件到服务器端。为了验证图片是否上传成功，进入D:\Test目录下，在该目录下可以看见三张以name+count编号命名的图片，说明文件上传成功。
  
• Demo_04.java
  客户端上传文件成功后，会读取服务器端发送的“上传文件成功!”信息。
  

总结

shutdownOutput()方法非常重要，因为服务器端程序（Demo_03.java）在while循环中读取客户端（Demo_04.java）发送的数据，当读取到-1时才会结束循环。如果在客户端不调用shutdownOutput()方法关闭客户端的输出流，服务器端就不会读到-1，而会一直执行while循环，同时客户端读取服务器端数据的read(byte [])方法也是一个阻塞方法，这样服务器端和客户端进入了一个“死锁”状态，两个程序都不能结束。