案例——TCP上传图片

TCP上传图片

  继续做一个TCP的相关案例，在客户端上传一张图片到服务端，服务端收到图片后，给客户端回馈上传成功信息。

实现细节：

• 对于一些文件的读取，若文件偏大，可以始用缓冲区去读取和写入，可以提高效率
• 服务端面向的是很多用户，所以必须要开线程来接收
• 如果同一个用户上传多张图片则有可能文件名重复，所以要对文件名做处理，具体在代码处有注释
• 客户端图片上传完成后需要给服务端发送结束信息，s.shutdownOutput();
  就像是写入了一个终止符一样

客户端

需要在上传结束的时候写入终止符告诉服务端上传结束，可以双方约定一个终止的符号，也可以用shutdownOutput()方法来传递结束信息

import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.Socket;

public class UploadPicClient {

    public static void main(String[] args) throws IOException {
        System.out.println("上传图片客户端启动...");
        
        //1、创建socket
        Socket s = new Socket("127.233.24.3", 10018);
        
        //2、读取源图片
        File picFile = new File("D:\\forio\\download.jpg");
        BufferedInputStream fis = new BufferedInputStream(new FileInputStream(picFile));
        
        //3、目的：socket输出流，发送到服务端
        OutputStream out = s.getOutputStream();
        byte[] buf = new byte[1024];
        
        int len = 0;
        while((len = fis.read(buf)) != -1)
        {
            out.write(buf, 0, len);
        }
        
        //告诉服务端   图片发送完成，不要在等着读了
        s.shutdownOutput();
        
        //读取上传成功字样
        BufferedReader in = new BufferedReader(new InputStreamReader(s.getInputStream()));
        String info = in.readLine();
        System.out.println(info);
        
        //关闭
        fis.close();
        s.close();
    }
}

服务端

服务端需要多线程接收多个客户的上传信息，对于有可能存在的文件名重复，添加(1)(2)(3)...加以区分

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStreamWriter;
import java.net.ServerSocket;
import java.net.Socket;

public class UploadPicServer {

    public static void main(String[] args) throws IOException {
        System.out.println("上传图片服务端启动...");

        // 创建ServerSocket
        ServerSocket ss = new ServerSocket(10018);

        while (true) {
            // 获取客户端
            Socket s = ss.accept();

            // 实现多个客户端并发上传，服务器端必须启动做个线程来完成
            new Thread(new UploadPic(s)).start();

        }
    }
}

class UploadPic implements Runnable {

    private Socket s;

    public UploadPic(Socket s) {
        super();
        this.s = s;
    }

    @Override
    public void run() {

        try {

            String ip = s.getInetAddress().getHostAddress();
            System.out.println(ip + "connected...");

            // 写图片数据到文件的输出流
            File dir = new File("D:\\forio\\photo");
            // 为了避免重名覆盖，通过给重名文件进行编号
            int count = 1;
            File picFile = new File(dir, ip + "(" + count + ")" + ".jpg");
            while (picFile.exists()) {
                count++;
                picFile = new File(dir, ip + "(" + count + ")" + ".jpg");
            }

            FileOutputStream fos = new FileOutputStream(picFile);

            // 读取客户端发来的数据
            InputStream in = s.getInputStream();
            byte[] buf = new byte[1024];
            int len = 0;

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

            // 给客户端一个回馈信息
            BufferedWriter out = new BufferedWriter(new OutputStreamWriter(s.getOutputStream()));
            out.write("上传成功！"); // 写完需要刷新一下，或者关闭流

            // 关闭资源
            out.close();
            fos.close();
            s.close();

        } catch (Exception e) {

        }

    }

}

结果演示