Java中使用Socket实现客户端与服务器的数据收发技术咨询

我帮你把这个Socket客户端的收发方法补全并优化一下，先看完整的实现代码，再聊聊可以改进的地方：

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.Socket;
import java.nio.charset.StandardCharsets;

public static String sendAndReceiveDataToNode(String address, int port, String data) throws ConnectionFailedException {
    // 使用try-with-resources自动管理Socket和流资源，无需手动关闭
    try (Socket me = new Socket(address, port);
         DataOutputStream outputStream = new DataOutputStream(me.getOutputStream());
         DataInputStream inputStream = new DataInputStream(me.getInputStream())) {
        
        // 设置读取超时，避免无限等待（可选但推荐）
        me.setSoTimeout(5000); // 5秒超时
        
        // 发送数据：先写入数据长度（帮助服务器确定读取边界），再写入实际内容
        byte[] dataBytes = data.getBytes(StandardCharsets.UTF_8);
        outputStream.writeInt(dataBytes.length);
        outputStream.write(dataBytes);
        outputStream.flush(); // 确保数据立即发送
        
        // 读取响应：先读取长度，再读取对应字节数的内容
        int responseLength = inputStream.readInt();
        byte[] responseBytes = new byte[responseLength];
        inputStream.readFully(responseBytes); // 确保读取完整的响应数据
        String responseData = new String(responseBytes, StandardCharsets.UTF_8);
        
        return responseData;
        
    } catch (IOException e) {
        // 将IO异常包装成自定义异常抛出，便于上层业务处理
        throw new ConnectionFailedException("连接或数据收发失败: " + e.getMessage(), e);
    }
}

// 自定义连接失败异常类
class ConnectionFailedException extends Exception {
    public ConnectionFailedException(String message, Throwable cause) {
        super(message, cause);
    }
}

接下来聊聊几个关键的优化点：

• 自动资源管理：用try-with-resources替代手动try-finally关闭Socket和流，Java会在代码块结束后自动关闭所有实现了AutoCloseable接口的资源，彻底避免因遗漏关闭导致的资源泄漏。
• 明确的数据边界：发送数据时先写长度、再写内容，读取时先读长度、再读对应字节数，这是Socket通信的常用最佳实践，能避免服务器/客户端因不知道数据何时结束而出现读取阻塞或内容截断的问题。
• 超时控制：通过setSoTimeout设置读取超时，防止客户端因服务器无响应而一直挂起；如果需要连接超时，可以用Socket重载构造方法：new Socket(address, port, null, 5000)（最后一个参数为连接超时毫秒数）。
• 编码一致性：明确指定StandardCharsets.UTF_8而非依赖系统默认字符集，避免不同环境下的编码不一致导致乱码。
• 异常封装：将底层的IOException包装成自定义的ConnectionFailedException，上层调用方可以更清晰地处理业务层面的连接失败场景，同时保留原始异常栈便于排查问题。

如果你的业务有特定的收发格式要求（比如不需要先发送长度），可以根据实际情况调整读写逻辑，但上述优化点大多是通用的，能让你的Socket客户端更健壮、更易于维护。

内容的提问来源于stack exchange，提问作者Kumar Gaurav