Java--网络编程--TCP实现聊天与文件上传

Java–网络编程–TCP实现聊天与文件上传

TCP聊天

服务端

package com.zy.net;

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

/**
 *description: Java--网络编程--TCP服务端
 *@program: 基础语法
 *@author: zy
 *@create: 2023-03-07 21:23
 */
public class StudyTcpServer {

    public static void main(String[] args) {
        initTcpServer();
    }

    /**
     * @Description 服务端初始化
     * @author zy
     * []
     * void
     * @date 2023-3-7 21:26
     */
    public static void initTcpServer(){
        ServerSocket serverSocket = null;
        Socket accept = null;
        InputStream is = null;
        ByteArrayOutputStream baos = null;
        try {
            // 1.创建一个地址
            serverSocket = new ServerSocket(9999);
            // 2.等待客户端连接
            accept = serverSocket.accept();
            // 3.读取客户端消息
            is = accept.getInputStream();

            // 不建议的写法：当长度超出1024则乱码
            /*byte[] buffer = new byte[1024];
            int len;
            while ((len = is.read(buffer)) != -1){
                String msg = new String(buffer, 0, len);
                System.out.println(msg);
            }*/

            // 管道流接收消息：无论如何都不会乱码
            baos = new ByteArrayOutputStream();
            byte[] buffer = new byte[1024];
            int len;
            while ((len = is.read(buffer)) != -1){
                baos.write(buffer,0,len);
            }
            System.out.println(baos.toString());

        } catch (IOException e) {
            e.printStackTrace();
        }finally {
            try {
                // 4.关闭资源：先开后关
                if (baos != null){
                    baos.close();
                }
                if (is != null){
                    is.close();
                }
                if (accept != null){
                    accept.close();
                }
                if (serverSocket != null){
                    serverSocket.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

客户端

package com.zy.net;

import java.io.IOException;
import java.io.OutputStream;
import java.net.InetAddress;
import java.net.Socket;
import java.net.UnknownHostException;

/**
 *description: Java--网络编程--TCP客户端
 *@program: 基础语法
 *@author: zy
 *@create: 2023-03-07 21:23
 */
public class StudyTcpClient {

    public static void main(String[] args) {
        initTcpClient();
    }

    /**
     * @Description 客户端初始化
     * @author zy
     * []
     * void
     * @date 2023-3-7 21:26
     */
    public static void initTcpClient(){
        Socket socket = null;
        OutputStream os = null;
        try {
            // 1.要知道服务器的地址，端口号
            InetAddress serverIP = InetAddress.getByName("127.0.0.1");
            int port = 9999;
            // 2.创建一个socket连接
            socket = new Socket(serverIP,port);
            // 3.发送消息 IO流
            os = socket.getOutputStream();
            os.write("你好，这里是客户端".getBytes());

        } catch (UnknownHostException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }finally {
            try {
                // 4.关闭资源：先开后关
                if(os != null){
                    os.close();
                }
                if(socket != null){
                    socket.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }
}

TCP文件上传

服务端

package com.zy.net;

import javax.imageio.stream.FileImageOutputStream;
import java.io.File;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;

/**
 *description: Java--网络编程--TCP文件上传服务端
 *@program: 基础语法
 *@author: zy
 *@create: 2023-03-07 22:11
 */
public class StudyTcpFileServer {

    public static void main(String[] args) {
        try {
            initTcpFileServer();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void initTcpFileServer() throws Exception {
        // 创建服务端
        ServerSocket serverSocket = new ServerSocket(8888);
        // 监听客户端:阻塞式监听，会一直等待客户端连接
        Socket accept = serverSocket.accept();
        // 获取输入流
        InputStream is = accept.getInputStream();
        // 管道流输出文件
        byte[] buffer = new byte[1024];
        int len;
        FileImageOutputStream fios = new FileImageOutputStream(new File("JavaProjectDemo/statics/netimg/receiveTwo.jpg"));
        while((len = is.read(buffer)) != -1){
            fios.write(buffer,0,len);
        }

        // 通知客户端已接收完毕
        OutputStream os = accept.getOutputStream();
        os.write("已接收完毕！".getBytes());

        // 关闭资源：先开后关
        os.close();
        fios.close();
        is.close();
        accept.close();
        serverSocket.close();
    }
}

客户端

package com.zy.net;

import javax.imageio.stream.FileImageInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.InetAddress;
import java.net.Socket;

/**
 *description: Java--网络编程--TCP文件上传客户端
 *@program: 基础语法
 *@author: zy
 *@create: 2023-03-07 22:11
 */
public class StudyTcpFileClient {

    public static void main(String[] args) {
        try {
            initTcpFileClient();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void initTcpFileClient() throws Exception {
        // 得到服务端的地址
        InetAddress serverIP = InetAddress.getByName("127.0.0.1");
        int port = 8888;
        // 创建客户端连接
        Socket socket = new Socket(serverIP,port);
        // 获取输出流
        OutputStream os = socket.getOutputStream();
        // 读取文件为文件流
        FileImageInputStream fiis = new FileImageInputStream(new File("JavaProjectDemo/statics/netimg/one.jpg"));
        // 写出文件
        byte[] buffer = new byte[1024];
        int len;
        while((len = fiis.read(buffer)) != -1){
            os.write(buffer,0,len);
        }

        // 通知服务端，传输完毕
        socket.shutdownOutput();

        // 确定服务端接收完毕，才能断开连接
        InputStream is = socket.getInputStream();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        byte[] buffer2 = new byte[1024];
        int len2;
        while ((len2 = is.read(buffer2)) != -1){
            baos.write(buffer2,0,len2);
        }
        System.out.println(baos.toString());

        // 关闭资源：先开后关
        baos.close();
        is.close();
        fiis.close();
        os.close();
        socket.close();
    }
}