如何参考这些Java代码实现DH密钥协商？

import java.security.KeyPair;
import java.security.KeyPairGenerator;
import java.security.KeyAgreement;
import javax.crypto.KeyAgreement;
import javax.crypto.interfaces.DHPublicKey;
import javax.crypto.spec.DHParameterSpec;

// 生成DH参数
KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance("DH");
keyPairGenerator.initialize(512); // 可根据需要调整密钥长度
KeyPair keyPair = keyPairGenerator.generateKeyPair();

// 获取公钥和私钥
DHPublicKey publicKey = (DHPublicKey) keyPair.getPublic();
DHPrivateKey privateKey = (DHPrivateKey) keyPair.getPrivate();

byte[] publicKeyBytes = publicKey.getEncoded();
// 将publicKeyBytes传递给对方

// 接收对方的publicKeyBytes
KeyFactory keyFactory = KeyFactory.getInstance("DH");
X509EncodedKeySpec publicKeySpec = new X509EncodedKeySpec(publicKeyBytes);
PublicKey receivedPublicKey = keyFactory.generatePublic(publicKeySpec);

// 生成本地的DH密钥对
KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance("DH");
keyPairGenerator.initialize(receivedPublicKey.getParams());
KeyPair keyPair = keyPairGenerator.generateKeyPair();

// 生成本地的共享密钥
KeyAgreement keyAgreement = KeyAgreement.getInstance("DH");
keyAgreement.init(keyPair.getPrivate());
keyAgreement.doPhase(receivedPublicKey, true);
byte[] sharedSecret = keyAgreement.generateSecret();