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

来源：互联网发布：淘宝怎么添加宝贝编辑：程序博客网时间：2024/06/06 01:31

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 [])方法也是一个阻塞方法，这样服务器端和客户端进入了一个“死锁”状态，两个程序都不能结束。

0 0