指示 Go TCP 服务器 Java TCP 客户端已完成对流的写入

Question

我正在编写一个连接到Golang TCP服务器的Java TCP客户端。

服务器使用以下代码从客户端读取消息：

func (tcpHandler TCPHandler) getClientMsgBytes(connection *net.TCPConn) ([]byte, error) {
    clientMsgBytes, err := ioutil.ReadAll(connection)
    if err != nil {            
        return nil, err
    }       
    return clientMsgBytes, nil
}

我的客户端使用以下代码向服务器发送消息：

try (Socket socket = new Socket("localhost", 9000)) {
    byte[] message = getMessage();
    DataOutputStream outputStream = new DataOutputStream(socket.getOutputStream());
    outputStream.write(commandMessage);     
    // Read message from the sever...          
}

我的问题是，即使在消息的所有字节都已写入流之后，服务器也会继续等待客户端写入其消息。这似乎是因为ioutil.ReadAll函数正在等待io.EOF错误作为停止从流中读取的信号。

我怎样才能告诉 Go 我已经完成了从 Java 写入 TCP 流？我可以更改Java TCP客户端和Golang TCP服务器代码，如果有帮助的话。

注意：服务器是这样编写的，因为我们的Go TCP客户端使用以下代码：

func writeToConnection(connection *net.TCPConn, tcpCommand structs.TCPCommand) error {
    messageBytes, err := json.Marshal(tcpCommand)
    if err != nil {
        err = merry.Wrap(err)
        return err
    }
    _, err = connection.Write(messageBytes)
    if err != nil {
        err = merry.Wrap(err)
        return err
    }
    err = connection.CloseWrite()
    if err != nil {
        err = merry.Wrap(err)
        return err
    }
    return nil
}

Answer 1

调用 shutdownOputput 以匹配 Go 客户端中的代码。