2024年最全Golang中WebSocket和WSS的支持_golang wss(1)，Golang开发基础机构

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以添加戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

	err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, client!"))
	if err != nil {
		log.Println("Failed to write message:", err)
		break
	}
}

}

func main() {
http.HandleFunc(“/websocket”, WebSocketHandler)
log.Println(“WebSocket server is running on :8080”)
http.ListenAndServe(“:8080”, nil)
}

在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并定义了一个`upgrader`对象。然后，我们实现了一个`WebSocketHandler`函数，用于处理WebSocket请求。在该函数中，我们使用`upgrader.Upgrade`方法将HTTP连接升级为WebSocket连接，并通过`conn.ReadMessage`方法读取客户端发送的消息，并使用`conn.WriteMessage`方法向客户端发送消息。最后，我们使用`http.HandleFunc`函数将`WebSocketHandler`函数注册为处理WebSocket请求的处理程序，并使用`http.ListenAndServe`函数启动WebSocket服务器。


#### 构建WebSocket客户端


要构建一个WebSocket客户端，我们可以使用`github.com/gorilla/websocket`包提供的`Dial`函数来建立与WebSocket服务器的连接。下面是一个简单的示例，演示了如何使用Golang构建一个WebSocket客户端：

package main

import (
“fmt”
“log”
“net/url”
“github.com/gorilla/websocket”
)

func main() {
u := url.URL{Scheme: “ws”, Host: “localhost:8080”, Path: “/websocket”}
conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
if err != nil {
log.Println(“Failed to connect to WebSocket server:”, err)
return
}

defer conn.Close()

// 向服务器发送消息
err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, server!"))
if err != nil {
	log.Println("Failed to write message:", err)
	return
}

// 读取服务器发送的消息
\_, message, err := conn.ReadMessage()
if err != nil {
	log.Println("Failed to read message:", err)
	return
}

fmt.Println("Received message:", string(message))

}

在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并使用`websocket.DefaultDialer.Dial`方法建立与WebSocket服务器的连接。然后，我们使用`conn.WriteMessage`方法向服务器发送消息，并使用`conn.ReadMessage`方法读取服务器发送的消息。最后，我们将接收到的消息打印出来。


### Golang中的WSS


#### 生成自签名证书


要在Golang中使用WSS，首先需要生成一个自签名证书。可以使用OpenSSL工具来生成自签名证书。下面是一个简单的示例，演示了如何生成自签名证书：

$ openssl req -x509 -nodes -days 365 -newkey rsa:2048 -keyout key.pem -out cert.pem

在上面的命令中，我们使用`openssl req`命令生成一个自签名证书，并将私钥保存到`key.pem`文件中，将公钥保存到`cert.pem`文件中。在实际使用中，你可能需要根据自己的需求来生成证书。


#### 使用WSS的WebSocket服务器和客户端


要在Golang中使用WSS，我们只需要在HTTP服务器和客户端中使用TLS配置即可。下面是一个简单的示例，演示了如何在Golang中构建一个使用WSS的WebSocket服务器和客户端：

package main

import (
“fmt”
“log”
“net/http”
“github.com/gorilla/websocket”
)

var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}

func WebSocketHandler(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Println(“Failed to upgrade to WebSocket:”, err)
return
}

defer conn.Close()

for {
	\_, message, err := conn.ReadMessage()
	if err != nil {
		log.Println("Failed to read message:", err)
		break
	}

	fmt.Println("Received message:", string(message))

	err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, client!"))
	if err != nil {
		log.Println("Failed to write message:", err)
		break
	}
}

}

func main() {
http.HandleFunc(“/websocket”, WebSocketHandler)
log.Println(“WebSocket server is running on :8080”)
err := http.ListenAndServeTLS(“:8080”, “cert.pem”, “key.pem”, nil)
if err != nil {
log.Fatal(“Failed to start WebSocket server:”, err)
}
}

在上面的代码中，我们首先导入`github.com/gorilla/websocket`包，并定义了一个`upgrader`对象。然后，我们实现了一个`WebSocketHandler`函数，用于处理WebSocket请求。最后，我们使用`http.ListenAndServeTLS`函数启动一个使用WSS的WebSocket服务器，并提供证书文件`cert.pem`和私钥文件`key.pem`。


对于WebSocket客户端，只需要在建立连接时使用`DialTLS`方法，并提供证书的URL即可。下面是一个简单的示例：

package main

import (
“fmt”
“log”
“net/url”
“github.com/gorilla/websocket”
)

func main() {
u := url.URL{Scheme: “wss”, Host: “localhost:8080”, Path: “/websocket”}
conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
if err != nil {
log.Println(“Failed to connect to WebSocket server:”, err)
return
}

defer conn.Close()

err = conn.WriteMessage(websocket.TextMessage, []byte("Hello, server!"))
if err != nil {
	log.Println("Failed to write message:", err)
	return
}

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

如果你需要这些资料，可以戳这里获取

习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！**

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

如果你需要这些资料，可以戳这里获取