案例:实现文件上传至服务器。服务器循环接收,使用多线程提高效率,接收后对文件重命名并保存

服务器端:

package com.chunzhi.Test02FileUpload;

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

public class TCPServer {
    public static void main(String[] args) throws IOException {
        // 1.创建一个服务器ServerSocket对象,和系统要指定的端口号
        ServerSocket server = new ServerSocket(8888);

        /*
            让服务器一直处于监听状态(死循环accept方法)
            有一个客户端上传文件,就保存一个文件
         */
        while (true) {
            // 2.使用ServerSocket对象中的方法accept,获取到请求的客户端Socket对象
            Socket socket = server.accept();

            /*
                使用多线程技术,提高程序的效率
                有一个客户端上传文件,就开启一个线程,完成文件的上传
             */
            new Thread(new Runnable() {
                @Override
                public void run() {
                    try {
                        // 3.使用Socket对象中的方法getInputStream,获取到网络字节输入流InputStream对象
                        InputStream is = socket.getInputStream();
                        // 4.判断d:\\Upload文件夹是否存在,不存在则创建
                        File file = new File("D:\\Upload");
                        if (!file.exists()) {
                            file.mkdir(); // 创建文件夹
                        }

                        /*
                            自定义一个文件的命名规则:防止同名的文件被覆盖
                            规则:域名+毫秒值+随机数
                         */
                        String fileName = "chunzhi" + System.currentTimeMillis() + new Random().nextInt();
                        // 5.创建一个本地字节输出流FileOutputStream对象,构造方法中绑定要输出的目的地
                        FileOutputStream fos = new FileOutputStream(file + "\\" + fileName + ".png");
                        // 6.使用网络字节输出流InputStream对象中的方法read,读取客户端上传的文件
                        int len = 0;
                        byte[] bytes = new byte[1024];
                        while ((len = is.read(bytes)) != -1) {
                            // 7
                            fos.write(bytes, 0, len);
                        }
                        // 8.使用Socket对象中的方法OutputStream,获取到网络字节输出流OutputStream对象
                        // 9.使用网络字节输出流OutputStream对象中的方法write,给客户端回写“上传成功”
                        socket.getOutputStream().write("上传成功".getBytes());
                        // 10.释放资源
                        fos.close();
                        socket.close();
                    } catch (IOException e) {
                        System.out.println(e);
                    }
                }
            }).start();
        }
        // 既然想让服务器一直接收数据,那么服务器就不用关闭了
        // server.close();
    }
}

客户端:

package com.chunzhi.Test02FileUpload;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.Socket;

/*
    文件上传案例的客户端:读取本地文件,上传到服务器,读取服务器回写的数据
 */
public class TCPClient {
    public static void main(String[] args) throws IOException {
        // 1.创建一个本地字输入流FileInputStream对象,构造方法中绑定要读取的数据源
        FileInputStream fis = new FileInputStream("C:\\1.png");
        // 2.创建一个客户端Socket对象,构造方法中绑定服务器的IP地址和端口号
        Socket socket = new Socket("192.168.0.106", 8888);
        // 3.使用Socket中的方法getOutputStream,获取网络字节输出流OutputStream对象
        OutputStream os = socket.getOutputStream();
        // 4.使用本地字节输入流FileInputStream对象中的方法read,读取本地文件
        int len = 0;
        byte[] bytes = new byte[1024];
        while ((len = fis.read(bytes)) != -1) {
            // 5.使用网络字节输出流OutputStream对象中的方法write,把读取到的文件上传到服务器
            os.write(bytes, 0, len);
        }

        /*
            发送完数据程序继续运行的原因,因为while循环读取不到结束标记,程序进入阻塞状态
            解决方法:上传完文件,给服务器写一个结束标记
            void shutdownOutput():禁用此套接字的输出流。
         */
        socket.shutdownOutput();

        // 6.使用Socket中的方法getInputStream,获取网络字节输入流InputStream对象
        InputStream is = socket.getInputStream();
        // 7.使用网络字节输入流InputStream对象中的方法read读取服务器回写的数据
        while ((len = is.read(bytes)) != -1) {
            System.out.println( new String(bytes, 0, len));
        }
        // 8.释放资源
        fis.close();
        socket.close();
    }
}

 

posted @ 2020-11-27 00:02  春志  阅读(271)  评论(0)    收藏  举报