golang实现密钥交换，代码如何写？-Go服务器开发

以下是一个使用Golang实现Diffie-Hellman密钥交换的示例代码：

package main
import (
	"crypto/rand"
	"fmt"
	"math/big"
)
// 生成随机素数
func generatePrime() *big.Int {
	prime, err := rand.Prime(rand.Reader, 128)
	if err != nil {
		panic(err)
	}
	return prime
}
// 计算 g^x mod p
func computePublicKey(g, x, p *big.Int) *big.Int {
	publicKey := new(big.Int).Exp(g, x, p)
	return publicKey
}
// 计算公共密钥 g^xy mod p
func computeSharedSecret(publicKey, x, p *big.Int) *big.Int {
	sharedSecret := new(big.Int).Exp(publicKey, x, p)
	return sharedSecret
}
func main() {
	// 生成素数p和基数g（可以自行选择合适的值）
	p := generatePrime()
	g := big.NewInt(2)
	fmt.Println("Prime (p):", p)
	fmt.Println("Base (g):", g)
	// Alice 和 Bob 分别选择私钥（秘密指数）
	xAlice := big.NewInt(123)
	xBob := big.NewInt(456)
	// Alice 和 Bob 分别计算公钥并进行交换
	yAlice := computePublicKey(g, xAlice, p)
	yBob := computePublicKey(g, xBob, p)
	fmt.Println("Alice's public key (yAlice):", yAlice)
	fmt.Println("Bob's public key (yBob):", yBob)
	// Alice 和 Bob 分别计算共享密钥（用对方的公钥和自己的私钥）
	secretAlice := computeSharedSecret(yBob, xAlice, p)
	secretBob := computeSharedSecret(yAlice, xBob, p)
	fmt.Println("Alice's shared secret:", secretAlice)
	fmt.Println("Bob's shared secret:", secretBob)
}