七叶笔记 » golang编程 » 在 Go 中使用 Websockets 和 Socket.IO – 教程

在 Go 中使用 Websockets 和 Socket.IO – 教程

分类: golang编程 | 浏览: 755

Websockets 是我觉得有趣的东西，因为它们为我们提供了应用程序之间通信的替代选项，而不是标准的 RESTful API 解决方案。使用 Sockets，我们可以做很酷的事情，例如在数千到数十万不同的客户端之间进行实时通信，而不必每分钟产生数十万次 RESTful API 调用访问我们的服务器的费用。

现实生活中的例子

透视事物并展示 websockets 的重要性。想象一下，我们有一个聊天应用程序，它从单个服务器获取所有最新消息，并将所有新消息推送到同一台服务器。

REST API 方法

  1. 为了实现实时聊天，您必须轮询每秒提供新消息的 REST API。
  2. 这相当于 每个客户端 每分钟大约 60 个 REST API 调用。
  3. 如果我们构建了一个成功的服务并且我们开始看到越来越多的流量访问我们的应用程序，我们的服务器将开始被每分钟处理数百万个 REST API 调用所淹没。

套接字示例

如果我们考虑使用 websockets 而不是 REST API 调用的场景：

  1. 每个客户端将与服务器保持一个单独的连接。
  2. 对于 1,000 个客户端，我们只需维护 1,000 个套接字连接。
  3. 如果有人发布新消息，只有到那时我们的服务器才会push向我们的 1,000 个客户端发送更新。

通过这种方法，我们极大地减少了访问我们服务器的网络流量。我们已经节省了我们需要运行的服务器应用程序实例数量的成本，而且我们基本上可以毫不费力地处理数千个客户端。

实现一个 Golang 服务器

为了在 Go 中实现 websockets，我们有许多不同的选择。我来自前端背景，前端中最流行的套接字通信库之一是socket-io，因此我们将使用 Golang 等效项，以便轻松地将它们集成在一起。

安装 go-socket.io

我们可以使用如下go get命令安装软件包： 

 go get github.com/googollee/go-socket.io

然后我们可以将它包含在我们的 go 程序中，如下所示： 

 import "github.com/googollee/go-socket.io"

简单服务器

让我们看一下readme.md我们正在使用的库中提供的示例代码。

main.go 

 package main

import (
    "log"
    "net/http"

    socketio "github.com/googollee/go-socket.io"
)

func main() {

    server, err := socketio.NewServer(nil)
    if err != nil {
        log.Fatal(err)
    }

    server.On("connection", func(so socketio.Socket) {

        log.Println("on connection")

        so.Join("chat")

        so.On("chat message", func(msg string) {
            log.Println("emit:", so.Emit("chat message", msg))
            so.BroadcastTo("chat", "chat message", msg)
        })

        so.On("disconnection", func() {
            log.Println("on disconnect")
        })
    })

    server.On("error", func(so socketio.Socket, err error) {
        log.Println("error:", err)
    })

    http.Handle("/socket.io/", server)

    fs := http.FileServer(http.Dir("static"))
    http.Handle("/", fs)

    log.Println("Serving at localhost:5000...")
    log.Fatal(http.ListenAndServe(":5000", nil))
}

运行它

所以在上面的代码示例中，我们在我们的main()函数中做所有事情。我们首先socketio通过调用定义一个新的服务器实例， socketio.NewServer(nil)然后再定义我们希望我们的套接字服务器在连接时和出现错误时具有的行为。

在server.On(‘connection’,…)我们第一次登录，有已经连接成功做出之前就加入了chat使用房间 so.Join(“chat”)。

之后，我们指定当我们”chat message”通过一个连接的套接字接收到事件时我们想要发生的事情 。每当我们的服务器收到这种类型的事件时，我们就会调用 so.BroadcastTo(“chat”, “chat message”, msg)它广播发送到当前连接的每个套接字的任何消息。这意味着一个客户端将看到另一个客户端要发送的任何消息。

最后，我们定义应该发生什么”disconnection”，在这种情况下，我们只记录我们的一个客户端断开连接的事实。

前端客户端

好的，我们已经设法充实了我们的后端基于 Go 的 WebSocket 服务器，但是现在是时候启动并运行一个简单的前端应用程序，以便我们可以测试我们所做的工作。

我们将首先index.html在我们的项目目录中创建一个简单的。

索引.html 

 <!DOCTYPE html>
<html lang="en">
  <head>
    <meta charset="UTF-8" />
    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
    <meta http-equiv="X-UA-Compatible" content="ie=edge" />
    <title>Go WebSocket Tutorial</title>
  </head>
  <body>
    <h2>Hello World</h2>

    <script src="#34;></script>
    <script>
      const socket = io("#34;);
    </script>
  </body>
</html>

当你然后通过调用运行你的 websocket 服务器时：

$ 去运行 main.go 

 2018/06/10 07:54:06 Serving at localhost:5000...
2018/06/10 07:54:15 on connection
2018/06/10 07:54:16 on connection

这应该然后开始运行。然后，您应该能够在浏览器中导航到此 URL，并在服务器的日志输出中看到正在建立的新连接。

您现在已经成功构建了一个直接连接到新创建的后端 websocket 服务器的前端！

如果您发现本教程有用或需要任何进一步的帮助，请随时在下面的评论部分告诉我。

相关文章