TCP之网络文件上传

客户端实现

package internet_coding;

import java.io.*;
import java.net.Socket;
import java.util.Scanner;

/**
 * @author swordsmanye
 */
public class TCP_Client {
    public static void main(String[] args) {
        File file = getFile();
        try {
            // 创建socket对象
            Socket socket = new Socket("10.90.92.109", 12345);
            // 读取数据 将字节流包装成字符流
            BufferedReader br = new BufferedReader(new InputStreamReader(socket.getInputStream()));
            // 写数据
            PrintStream ps = new PrintStream(socket.getOutputStream());
            // 向服务器发送文件名
            ps.println(file.getName());

            // 读取服务器端发送的消息
            String result = br.readLine();
            // 如果存在给予提示，程序直接退出
            if ("存在".equals(result)) {
                System.out.println("您上传的文件已存在，请不要重复上传");
                socket.close();
                return;
            }
            // 如果不存在，定义FIleInputStream对象读取文件，上传到服务器
            FileInputStream fis = new FileInputStream(file);
            byte[] bytes = new byte[1024 * 4];
            int len;
            while ((len = fis.read(bytes)) != -1) {
                ps.write(bytes, 0, len);
            }
            fis.close();
            socket.close();

        } catch (IOException e) {
            e.printStackTrace();
        }

    }

    public static File getFile() {
        // 创建键盘录入对象
        Scanner sc = new Scanner(System.in);
        // 提示信息
        System.out.println("请输入您需要上传的文件路径");
        // 用户可以一直上传
        while (true) {
            String line = sc.nextLine();
            // 包装成File对象，好使用File对象内的方法
            File file = new File(line);
            if (!file.exists()) {
                System.out.println("您输入的文件路径不存在，请重新输入");
            } else if (file.isDirectory()) {
                System.out.println("您输入的文件路径是文件夹，请重新输入");
            } else {
                return file;
            }
        }
    }
}

服务端实现

package internet_coding;

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

/**
 * @author swordsmanye
 */
public class TCP_Server {
    public static void main(String[] args) throws IOException {
        // 创建serverSocket对象
        ServerSocket server = new ServerSocket(12345);
        System.out.println("服务器已启动...");

        // 读取文件名
        while (true) {
            // 接收客户端发来的请求
            final Socket socket = server.accept();
            new Thread() {
                @Override
                public void run() {
                    try {
                        // 创键输入流
                        InputStream is = socket.getInputStream();
                        // 读取数据
                        BufferedReader br = new BufferedReader(new InputStreamReader(socket.getInputStream()));
                        // 写数据
                        PrintStream ps = new PrintStream(socket.getOutputStream());
                        // 读取客户端发送的数据
                        String fileName = br.readLine();
                        // 判断文件是否存在，将结果发回客户端。
                        File dir = new File("update");
                        dir.mkdir();
                        File file = new File(dir, fileName);
                        // 如果服务器已经存在该文件
                        if (file.exists()) {
                            // 将存在发回给客户端
                            ps.println("存在");
                            socket.close();
                            return;
                        } else {
                            ps.println("不存在");
                        }

                        // 定义FileOutputStream对象，从网络读取数据存储到本地
                        FileOutputStream fos = new FileOutputStream(file);
                        byte[] bytes = new byte[1024 * 4];
                        int len = 0;
                        while ((len = is.read()) != -1) {
                            fos.write(bytes, 0, len);
                        }
                        fos.close();
                        socket.close();

                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
            }.start();
        }
    }

}