既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,涵盖了95%以上Go语言开发知识点,真正体系化!
由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新
要构建一个WebSocket服务器,首先需要导入github.com/gorilla/websocket
包,并注册一个处理WebSocket请求的处理程序。下面是一个简单的示例,演示了如何使用Golang构建一个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")
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 {
![img](https://img-blog.csdnimg.cn/img_convert/6fe2f84d0e880c38a31b632b175d8af5.png)
![img](https://img-blog.csdnimg.cn/img_convert/028deab7f8e172b1541326a69d45a647.png)
![img](https://img-blog.csdnimg.cn/img_convert/a8d11f097aacb186d1b03fc655f77430.png)
**既有适合小白学习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,涵盖了95%以上Go语言开发知识点,真正体系化!**
**由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新**
**[如果你需要这些资料,可以戳这里获取](https://bbs.csdn.net/topics/618658159)**
习的零基础资料,也有适合3年以上经验的小伙伴深入学习提升的进阶课程,涵盖了95%以上Go语言开发知识点,真正体系化!**
**由于文件比较多,这里只是将部分目录截图出来,全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频,并且后续会持续更新**
**[如果你需要这些资料,可以戳这里获取](https://bbs.csdn.net/topics/618658159)**