网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。
一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!
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
}
\_, message, err := conn.ReadMessage()
if err != nil {
log.Println("Failed to read message:", err)
return
}
fmt.Println("Received message:", string(message))
}
![img](https://img-blog.csdnimg.cn/img_convert/7899f2059830f7cc38b35cbd2df4401b.png)
![img](https://i-blog.csdnimg.cn/blog_migrate/bd3e1a38ea86ac9ddad0af158494def9.png)
**网上学习资料一大堆,但如果学到的知识不成体系,遇到问题时只是浅尝辄止,不再深入研究,那么很难做到真正的技术提升。**
**[需要这份系统化的资料的朋友,可以添加戳这里获取](https://bbs.csdn.net/topics/618658159)**
**一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!**
入研究,那么很难做到真正的技术提升。**
**[需要这份系统化的资料的朋友,可以添加戳这里获取](https://bbs.csdn.net/topics/618658159)**
**一个人可以走的很快,但一群人才能走的更远!不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人,都欢迎加入我们的的圈子(技术交流、学习资源、职场吐槽、大厂内推、面试辅导),让我们一起学习成长!**