当前位置：首页 > 文章列表 > Golang > Go教程 > 在Go中使用WebSocket实现实时通信

在Go中使用WebSocket实现实时通信

2023-06-19 17:27:50 0浏览收藏

推广推荐

支持 PC / 移动端，安全直达

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《在Go中使用WebSocket实现实时通信》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

WebSockets是一种在Web浏览器和服务器之间提供双向通信的技术。它允许实时通信，使得在应用程序中进行交互变得更加容易和快捷。在Go语言中，使用WebSocket从服务器向浏览器发送数据或从浏览器接收数据都变得非常简单。本文将详细介绍如何在Go中使用WebSocket实现实时通信。

什么是WebSocket

WebSocket是一种在Web浏览器和服务器之间提供双向通信的技术。与HTTP请求/响应模式不同，WebSocket提供了一种持久连接，允许服务器和客户端之间实时通信，从而提高了应用程序的性能和用户体验。

WebSocket协议建立在TCP协议之上，并通过HTTP / 1.1协议中的Upgrade头部将其升级为WebSocket协议。在WebSockets之前，HTTP协议是Web应用程序中使用的主要协议。然而，HTTP是一种无状态协议，需要浏览器不断地发送请求来进行交互，这可能会浪费大量的带宽和增加延迟。

使用WebSocket，服务器可以向客户端发送消息，而无需等待客户端发起请求。这极大地提高了实时通信的效率和速度。

在Go中使用WebSocket

在Go语言中，使用WebSocket实现实时通信非常简单。Go内置了一个标准库net/http和golang.org/x/net/websocket，可以方便地实现WebSocket。

首先，编写一个处理WebSocket请求的处理程序。

func wsHandler(ws *websocket.Conn) {
  for {
    var msg string
    err := websocket.Message.Receive(ws, &msg)
    if err != nil {
      fmt.Println(err)
      return
    }
    fmt.Println("Received message:", msg)

    // 处理消息并发送回客户端
    err = websocket.Message.Send(ws, "Received: "+msg)
    if err != nil {
      fmt.Println(err)
      return
    }
  }
}

上面的代码创建了一个处理WebSocket请求的处理程序。处理程序接受一个指向websocket.Conn的指针并在无限循环中等待客户端发送消息。一旦收到消息，它将打印消息，然后将相同的消息发送回客户端。

在处理程序中使用websocket.Message.Receive()函数来接收消息，并使用websocket.Message.Send()函数来发送消息回客户端。

接下来，编写一个处理普通HTTP请求的处理程序。

func httpHandler(w http.ResponseWriter, r *http.Request) {
  // 将HTTP请求升级为WebSocket请求
  wsHandler := websocket.Handler(wsHandler)
  wsHandler.ServeHTTP(w, r)
}

上面的代码创建了一个处理普通HTTP请求的处理程序。它使用websocket.Handler()函数将HTTP请求升级为WebSocket请求。该处理程序使用之前编写的WebSocket处理程序来处理升级后的WebSocket请求。

最后，启动WebSocket服务器来处理来自客户端的请求。

func main() {
  http.HandleFunc("/", httpHandler)
  fmt.Println("Starting server...")
  err := http.ListenAndServe(":8080", nil)
  if err != nil {
    fmt.Println(err)
    return
  }
}

上面的代码在main()函数中启动WebSocket服务器，将处理程序注册到HTTP服务器中，并开始侦听端口8080。

在网页中使用WebSocket

在客户端，可以使用JavaScript的WebSocket对象来创建WebSocket连接。

const socket = new WebSocket("ws://localhost:8080/");
socket.addEventListener("open", (event) => {
  console.log("WebSocket connected");
  socket.send("Hello, WebSocket!");
});
socket.addEventListener("message", (event) => {
  console.log("Received message:", event.data);
  socket.close();
});
socket.addEventListener("close", (event) => {
  console.log("WebSocket disconnected");
});

上面的代码创建了一个WebSocket对象，并连接到服务器的ws://localhost:8080/地址。一旦连接启动，将发送一条消息Hello, WebSocket!。当客户端收到消息时，它将打印消息并关闭连接。如果连接断开，将打印WebSocket disconnected消息。