如何在套接字java上正确地进行非对称加密和解密

import java.io.*;
import java.net.*;
import java.security.*;
import javax.crypto.*;

public class SocketEncryptionExample {
    public static void main(String[] args) {
        try {
            // 生成密钥对
            KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance("RSA");
            keyPairGenerator.initialize(2048);
            KeyPair keyPair = keyPairGenerator.generateKeyPair();
            PublicKey publicKey = keyPair.getPublic();
            PrivateKey privateKey = keyPair.getPrivate();

            // 启动服务器
            Thread serverThread = new Thread(() -> {
                try {
                    ServerSocket serverSocket = new ServerSocket(8888);
                    Socket socket = serverSocket.accept();

                    // 获取输入输出流
                    InputStream inputStream = socket.getInputStream();
                    OutputStream outputStream = socket.getOutputStream();

                    // 加密数据
                    Cipher encryptCipher = Cipher.getInstance("RSA");
                    encryptCipher.init(Cipher.ENCRYPT_MODE, publicKey);
                    CipherOutputStream cipherOutputStream = new CipherOutputStream(outputStream, encryptCipher);

                    // 发送加密数据
                    String message = "Hello, World!";
                    cipherOutputStream.write(message.getBytes());
                    cipherOutputStream.flush();
                    cipherOutputStream.close();

                    // 关闭连接
                    socket.close();
                    serverSocket.close();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            });
            serverThread.start();

            // 启动客户端
            Thread clientThread = new Thread(() -> {
                try {
                    Socket socket = new Socket("localhost", 8888);

                    // 获取输入输出流
                    InputStream inputStream = socket.getInputStream();
                    OutputStream outputStream = socket.getOutputStream();

                    // 解密数据
                    Cipher decryptCipher = Cipher.getInstance("RSA");
                    decryptCipher.init(Cipher.DECRYPT_MODE, privateKey);
                    CipherInputStream cipherInputStream = new CipherInputStream(inputStream, decryptCipher);

                    // 接收解密数据
                    ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
                    byte[] buffer = new byte[1024];
                    int bytesRead;
                    while ((bytesRead = cipherInputStream.read(buffer)) != -1) {
                        byteArrayOutputStream.write(buffer, 0, bytesRead);
                    }
                    String decryptedMessage = new String(byteArrayOutputStream.toByteArray());
                    System.out.println("Decrypted Message: " + decryptedMessage);

                    // 关闭连接
                    socket.close();
                } catch (Exception e) {
                    e.printStackTrace();
                }
            });
            clientThread.start();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}