怎么使用Go语言实现WebSocket通信

   2024-10-22 8970
核心提示:在Go语言中实现WebSocket通信,可以使用标准库中提供的net/http和github.com/gorilla/websocket包。以下是一个简单的示例:packa

在Go语言中实现WebSocket通信,可以使用标准库中提供的net/httpgithub.com/gorilla/websocket包。以下是一个简单的示例:

package mainimport ("fmt""net/http""github.com/gorilla/websocket")var upgrader = websocket.Upgrader{CheckOrigin: func(r *http.Request) bool {return true},}func handler(w http.ResponseWriter, r *http.Request) {conn, err := upgrader.Upgrade(w, r, nil)if err != nil {fmt.Println("Error upgrading to WebSocket:", err)return}for {// 读取客户端发送的消息_, message, err := conn.ReadMessage()if err != nil {fmt.Println("Error reading message:", err)break}// 输出客户端发送的消息fmt.Printf("Received message: %s\n", message)// 发送消息给客户端err = conn.WriteMessage(websocket.TextMessage, []byte("Message received"))if err != nil {fmt.Println("Error writing message:", err)break}}}func main() {http.HandleFunc("/", handler)http.ListenAndServe(":8080", nil)}

在上面的代码中,我们首先创建了一个upgrader实例,用于将HTTP连接升级为WebSocket连接。然后定义了一个处理函数handler,当有客户端连接时会调用该函数。在处理函数中,我们通过upgrader.Upgrade方法将HTTP连接升级为WebSocket连接,并通过conn.ReadMessage读取客户端发送的消息,然后通过conn.WriteMessage发送消息给客户端。

最后在main函数中,我们通过http.HandleFunc将处理函数注册到路由上,然后通过http.ListenAndServe启动HTTP服务器,监听在8080端口上。

在客户端可以使用JavaScript来与该WebSocket服务器进行通信,例如:

const ws = new WebSocket("ws://localhost:8080/");ws.onopen = function() {    console.log("Connected to WebSocket server");    ws.send("Hello, server!");};ws.onmessage = function(event) {    console.log("Received message from server:", event.data);};

通过这种方式,可以使用Go语言实现WebSocket通信。

 
举报打赏
 
更多>同类网点查询
推荐图文
推荐网点查询
点击排行

网站首页  |  关于我们  |  联系方式网站留言    |  赣ICP备2021007278号