Socket文件传输demo

一、Socket本地上传

客户端

package com.demo.client;

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

public class Client
{
    public static void main(String[] args) throws IOException
    {
        //1.连接服务器
        Socket socket = new Socket("127.0.0.1", 9999);
        //2.获取网络输出流
        OutputStream netOut = socket.getOutputStream();					//获取网络输入流  OutputStream  socket.getOutputStream
        //3.创建一个File，获取要上传的"文件名"
        File src = new File("E:\\123.txt");
        //4.发送文件名;
        if (src.exists()) {
            netOut.write(src.getName().getBytes());
        }else{
            System.out.println("要上传的文件不存在！");
            return;
        }
        //5.接收反馈
        InputStream netIn = socket.getInputStream();     				//获取网络输出流 InputStream  socket.getInputStream
        byte[] byteArray = new byte[1024];
        int len = netIn.read(byteArray);
        String msg = new String(byteArray, 0, len);
        //6.如果反馈是:OK，创建文件输入流
        if("ok".equals(msg.toLowerCase().trim())) {						//文件输入流    FileInputStream
            //7.循环读取文件，并且发送到网络;
            FileInputStream fileIn = new FileInputStream(src);
            len = 0;
            while ((len = fileIn.read(byteArray)) != -1) {
                netOut.write(byteArray,0,len);
            }
            fileIn.close();
        }
        //关闭输出流
        socket.shutdownOutput();//向网络发送一个结束符EOF

        System.out.println("【发送完毕】等待反馈...");
        //接收反馈
        len = netIn.read(byteArray);
        String fk = new String(byteArray, 0, len);
        System.out.println("【客户端】收到反馈：" + fk);
        //8.循环结束，关闭资源
        socket.close();
        System.out.println("【客戶端】上传完毕！");
    }
}

服务器端

package com.demo.server;



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

public class Server
{
    public static void main(String[] args) throws IOException {
        //1.监听端口
        ServerSocket server = new ServerSocket(9999);
        //2.等待连接
        System.out.println("【服务器】等待连接...");
        Socket socket = server.accept();
        //3.获取网络输入流，接收文件名
        InputStream netIn = socket.getInputStream();
        byte[] byteArray = new byte[1024];
        int len = netIn.read(byteArray);
        String fileName = new String(byteArray, 0, len);
        System.out.println("【服务器】收到文件名：" + fileName);
        //4.获取网络输出流，发送OK
        OutputStream netOut = socket.getOutputStream();
        netOut.write("ok".getBytes());
        //5.创建一个"文件输出流"，使用接收的"文件名"
        FileOutputStream fileOut = new FileOutputStream("E:\\upload/" + fileName);
        //6.使用"网络输入流"循环从网络读取byte[]数组，向文件输出一个byte[]数组
        System.out.println("【服务器】开始上传...");
        while ((len = netIn.read(byteArray)) != -1) {
            fileOut.write(byteArray,0,len);
        }
        System.out.println("【服务器】处理上传完毕，发送反馈....");
        //给客户端发送【反馈】
        netOut.write("接收完毕".getBytes());
        //7.循环结束，关闭资源
        fileOut.close();
        socket.close();
        server.close();
    }
}

二、点对点多线程聊天

客户端

package com.demo.client;

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

public class Client
{
    public static void main(String[] args) throws IOException
    {
        //1.连接服务器-默认本机
        Socket socket = new Socket("127.0.0.1", 9999);
        //2.开启线程，用于接收服务器端信息
        new ClientThread(socket).start();

        //控制台输入
        OutputStream out = socket.getOutputStream();
        Scanner scanner = new Scanner(System.in);


        //循环从控制台接受信息，并发送给服务器
        while (true)
        {
            System.out.println("请输入信息：");
//            从控制台接受消息
            String msg = scanner.next();
            out.write(msg.getBytes());
        }
    }
}

客户端线程

package com.demo.client;

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

public class ClientThread extends Thread
{
    private Socket socket;

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

    @Override
    public void run()
    {
        //死循环，从网络读取数据
        try
        {
            //输入流
            InputStream in = socket.getInputStream();
            byte[] bytes = new byte[1024];

            int len = 0;
            while (true)
            {
                len = in.read(bytes);
                String msg = new String(bytes, 0, len);
                System.out.println("【服务器】:" + msg);
                //聊天结束 语句
                if (msg.equals("88"))
                {
                    //强制关闭虚拟机  结束线程
                    System.exit(0);

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

服务器端

package com.demo.server;



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

public class Server
{
    public static void main(String[] args) throws IOException {

        //1.启动服务器， 监听端口 9999
        ServerSocket server = new ServerSocket(9999);
        //2.等待客户端连接
        Socket socket = server.accept();
        //3.有客户端连接成功 启动线程
        new ServerThread(socket).start();
        //4.读取信息
        OutputStream out = socket.getOutputStream();
        Scanner scanner = new Scanner(System.in);

        while (true)
        {
            System.out.println("请输入信息:");
            String msg = scanner.next();
            try
            {
                out.write(msg.getBytes());
            } catch (IOException e)
            {
                System.out.println("聊天结束");
            }
        }
    }
}

服务器端线程

package com.demo.server;


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

public class ServerThread extends Thread
{
    private Socket socket;

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

    @Override
    public void run()
    {
        try
        {
            //输入流
            InputStream in = socket.getInputStream();
            byte[] bytes = new byte[1024];
            int len = 0;
            while (true)
            {
                try
                {
                    len = in.read(bytes);
                } catch (IOException e)
                {
                    e.printStackTrace();
                }
                String msg = new String(bytes, 0, len);
                System.out.println("【客户端】:" + msg);
                if (msg.equals("88"))
                {
                    System.exit(0);

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