java实现socket客户端与服务端实现数据相互发送

1. socket服务端实现代码

1.1 socket服务端

package com.exchange.api.utils;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

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


/**
 * @Author wangchang
 * @Description socket服务端
 * @Date 2023/7/11 10:55
 */
@Component
@Slf4j
public class SocketServer {

    @Value("${socket.port}")
    private Integer port;
    private ServerSocket serverSocket;

    public void startServer(String data) {
        try {
            System.out.println("启动服务端");
            //创建ServerSocket的同时要申请服务端口，该端口不能与系统其他程序
            serverSocket = new ServerSocket(port);
            System.out.println("服务器已启动端口号为：9527，等待连接中...");

            while (true) {
                //该方法是一个阻塞方法，调用会程序会卡住，直到一个客户端与服务端建立连接
                Socket socket = serverSocket.accept();
                /*
                 * 启用一个线程处理与该客户端的交互
                 */
                ServerThread ch = new ServerThread(socket, data);
                Thread thread = new Thread(ch);
                thread.start();
            }

        } catch (IOException e) {
            log.error(e.getMessage());
        }
    }

}

1.2 服务端接收的数据在单独的线程上处理

package com.exchange.api.utils;


import lombok.extern.slf4j.Slf4j;

import java.io.*;
import java.net.Socket;
import java.nio.charset.StandardCharsets;

/**
 * @Author wangchang
 * @Description 将处理客户端交互的操作在单独的线程上进行，
 * @Date 2023/7/11 11:23
 */
@Slf4j
public class ServerThread implements Runnable {

    private Socket socket;

    private String data;

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

    @Override
    public void run() {
        PrintWriter pw = null;
        /*
         * 通过Socket获取输入流可以读取来自远端计算机发送过来的消息
         */
        try {
            InputStream is = socket.getInputStream();
            InputStreamReader isr = new InputStreamReader(is, StandardCharsets.UTF_8);
            BufferedReader br = new BufferedReader(isr);

            //输出消息给客户端
            OutputStream ops = socket.getOutputStream();
            OutputStreamWriter osw = new OutputStreamWriter(ops, StandardCharsets.UTF_8);
            BufferedWriter bw = new BufferedWriter(osw);
//            pw = new PrintWriter(bw, true);


            String message;
            /*
             * 这里循环读取客户端传递过来的消息可能会出现异常：
             *   java.net.SocketException: Connection reset
             * 这个错误是由于远端异常断开导致的（没有进行TCP挥手断开操作），
             * 该异常无法通过逻辑来避免
             * */
            while ((message = br.readLine()) != null) {
                System.out.println("收到客户端消息：：" + message);
                //响应客户端
//                pw.println("我收你的到消息！");
                bw.write(data + "\n");
                bw.flush();
            }
        } catch (IOException e) {
            log.error(e.getMessage());
        } finally {
            try {
                socket.close();
            } catch (IOException e) {
                log.error(e.getMessage());
            }
        }

    }
}

2. socket客户端实现代码

2.1 socket客户端

package com.exchange.api.utils;

import cn.hutool.core.util.ObjectUtil;
import com.exchange.api.entity.ResponseResult;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import java.io.*;
import java.net.Socket;
import java.nio.charset.StandardCharsets;

/**
 * @Author wangchang
 * @Description socket客户端
 * @Date 2023/7/11 10:54
 */
@Component
@Slf4j
public class SocketClient {

    @Value("${socket.ip}")
    private String ip;
    @Value("${socket.port}")
    private Integer port;

    private Socket socket;


    public void startClient() {
        try {

            System.out.println("连接服务端");
            // 实例化Socket就是与远端计算机建立连接的过程。
            socket = new Socket(ip, port);
            System.out.println("已连接服务端");


            // 启动用来读取服务端消息的线程
            ClientThread sh = new ClientThread(socket);
            Thread thread = new Thread(sh);
            // 守护线程
            thread.setDaemon(true);
            thread.start();

            OutputStream ops = socket.getOutputStream();
            OutputStreamWriter opsw = new OutputStreamWriter(ops, StandardCharsets.UTF_8);
            BufferedWriter bw = new BufferedWriter(opsw);

            while (true) {
                // 3秒执行一次
                Thread.sleep(3000);
                ResponseResult result = sh.getResult();
                String dataID = "";
                if (ObjectUtil.isNotNull(result)) {
                    dataID = result.getDataID();
                }

                if ("aa".equals(dataID)) {
                    bw.write(dataID + "\n");
                    bw.flush();
                } else {
                    bw.write("请传数据！" + "\n");
                    bw.flush();
                }
            }

        } catch (IOException e) {
            log.error(e.getMessage());
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        } finally {
            try {
                //关闭socket时会与服务端进行挥手操作
                socket.close();
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }

    }


}

2.1 客户端数据处理也在单独线程执行

package com.exchange.api.utils;

import com.exchange.api.action.HandleDataAction;
import com.exchange.api.entity.ResponseResult;
import lombok.extern.slf4j.Slf4j;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.Socket;
import java.nio.charset.StandardCharsets;

/**
 * @Author wangchang
 * @Description 该线程负责读取服务端发送过来的消息
 * @Date 2023/7/11 11:17
 */
@Slf4j
public class ClientThread implements Runnable {

    private Socket socket;

    // 响应结果
    private ResponseResult result;

    public ResponseResult getResult() {
        return result;
    }

    public void setResult(ResponseResult result) {
        this.result = result;
    }

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


    @Override
    public void run() {
        try {
            InputStream is = socket.getInputStream();
            InputStreamReader isr = new InputStreamReader(is, StandardCharsets.UTF_8);
            BufferedReader br = new BufferedReader(isr);

            String message;
            while ((message = br.readLine()) != null) {
                // 处理接收的数据
                HandleDataAction action = new HandleDataAction();
                action.handleData(message);
            }
        } catch (IOException e) {
            log.error(e.getMessage());
        }
    }

}