随着云计算和 Serverless 的兴起，越来越多的开发者转向了基于 Serverless 架构的应用程序开发。然而，这种服务架构在 WebSocket 实现方面存在一些挑战性。
在本文中，我们将深入探讨在 Serverless 架构下实现 WebSocket 的方法。我们将从 WebSocket 的基本原理开始，逐步引入 Serverless 架构的概念，并提供一个完整的示例，以帮助您更好地理解和实践 WebSocket 在 Serverless 环境下的应用。
WebSocket 基础知识
WebSocket 是一个在 Web 浏览器和服务器之间建立持久性连接的技术。它可以像 HTTP 的请求和响应一样通过一个 URL 发送或接收消息。与标准 HTTP 请求和响应不同，WebSocket 的连接在建立之后可以保持打开状态，也就是说，服务器和客户端之间可以实时地传输数据。
WebSocket 提供了双向通信的能力，使得客户端可以实时地接收服务器上的消息，而不用自己反复查询服务器状态。这样，WebSocket 技术为实时应用程序带来了许多有趣的应用场景——例如在线聊天，在线博弈等。
下面是 WebSocket 的运作原理：
- 客户端通过一个 WebSocket URL 发起连接
- 服务器接收这个连接，并向客户端发送一个 HTTP 响应来确认连接，这个响应中包含一些自定义的标头，以表示这是一个 WebSocket 连接
- 一旦 WebSocket 连接建立起来，服务器和客户端都可以通过它来传输消息，直到其中一方关闭连接
Serverless 架构
Serverless 架构是一种完全由云服务提供商管理和维护底层架构的服务模型。也就是说，开发者只需专注于编写应用程序，而不用担心服务器的维护、配置、扩展等问题。云服务提供商会根据程序的实际用量向开发者提供资源，并收取相应的费用。
在 Serverless 架构中，应用程序由一系列函数组成，每个函数都由一个事件触发器激活。每个函数都是一段短小的代码，它们分别执行自己的任务并产生输出，然后结束运行。整个应用程序是在云服务提供商的服务器上执行的。
Serverless 架构的优点：
- 开发者只需专注于应用程序本身，不需要关心底层服务器的复杂性和维护成本。
- 应用程序的弹性和可扩展性可以轻松地进行调整，以适应预期和非预期量的流量。
- 应用程序开发的速度和效率都比传统的应用程序开发领域更高。
WebSocket 在 Serverless 环境下的挑战
虽然 WebSocket 在实际应用中的优势很明显，但是在 Serverless 环境下实现 WebSocket 却面临一些挑战：
- WebSocket 需要保持持久化连接，这对于类型为“按量付费”的 Serverless 服务来说是一个问题，因为它们是动态调整资源的，这可能导致 WebSocket 连接中断。
- WebSocket 需要确保消息可以在客户端和服务器之间实时传输，这在 Serverless 服务中需要大量的网络带宽和计算资源。
对于这些挑战，我们可以通过以下步骤来实现一个 WebSocket 连接：
- 在客户端启动 WebSocket 并连接到服务器。
- 当服务端接收到 WebSocket 连接请求后，使用 Serverless 中的一个无服务器功能去处理请求，并在处理完成后建立 WebSocket 连接。
- 如果在处理中需要获得某些状态信息，则可以使用服务器端的 WebSocket Key 获取信息。
- 一旦建立起 WebSocket 连接，服务器就可以向客户端发送消息，并且客户端也可以向服务器发起消息。
Serverless 环境下 WebSocket 实现的示例
为了更好地说明 Serverless 环境下的 WebSocket 实现方法，我们将编写一个简单而完整的示例程序。
这个示例程序基于 Node.js 和 AWS 服务，那么你需要具备基本的 Node.js 和 AWS 知识。
Step 1：创建 Lambda 函数
首先，我们需要创建一个 AWS Lambda 函数来处理来自 WebSocket 的请求。将以下代码保存为 serverless-websocket.js 并上传到 AWS Lambda 中：
-- -------------------- ---- -------
--------------- - ------- -------- --------- -- -
-- -- --------- ---
----- --- - ------------------------------
-- ---- --- ----------- -
-- - --------- -----
---------------------- --------
-------------- - ----------- ---- ----- ---------- ------ ---
- ---- -- ---- --- -------------- -
-- - --------- -----
---------------------- --------
-------------- - ----------- ---- ----- ---------- ------ ---
- ---- -- ---- --- ----------- -
-- -- --------- -----
----------------------------------------
-------------- - ----------- ---- ----- --------- ---
-
--这段代码使用了 AWS Lambda 的编程模型，将 event 作为输入参数，响应作为输出参数，这也是 AWS Lambda 中使用 WebSocket 的真正用法。这个函数可以接受三个不同的事件 - $connect、$disconnect 和 $default。当服务端收到 WebSocket 连接时，它会发出 $connect 事件。断开连接时，服务器会发出 $disconnect 事件，而 $default 事件会处理 WebSocket 客户端发送的请求。
Step 2：配置 WebSocket API
AWS ApiGateway 是一个帮助开发者构建和发布 RESTful API 或 WebSocket API 的云服务。它可以与 AWS Lambda 和其他 AWS 服务集成，提供了一组功能强大的工具，帮助开发者轻松构建可扩展的 Serverless API。
接下来，我们需要在 AWS ApiGateway 上创建一个 WebSocket API 并将其映射到 AWS Lambda 函数：
- 登录 AWS 并访问 ApiGateway 控制台
- 在左侧面板中，选择 WebSocket API，然后单击“创建 WebSocket API”
- 输入 API 名称并单击“创建 WebSocket API”按钮
接下来，我们需要为 WebSocket API 创建一个 Lambda 集成，这样它才能够与 AWS Lambda 集成：
- 在 WebSocket API 中，选择“集成请求”，然后单击“设定新的 WebSocket 集成”按钮
- 选择 Lambda 作为集成类型
- 选择您之前上传到 AWS Lambda 的函数
- 单击“保存”
现在你已完成 WebSocket API 的配置，接下来需要添加资源路由，详情如下：
- 在资源创建之前，我们需要在“WebSocket API”页面上单击“创建资源”
- 在“WebSocket API”页面上，单击资源名称以选择它，然后在该资源的上下文菜单中选择“设定 WebSocket Action”
- 将
$connect、$disconnect和$default添加到“WebSocket Action”。
现在 WebSocket API 和 AWS Lambda 集成完成了。接下来，我们将学习如何在客户端实现 WebSocket。
Step 3：客户端实现
最后，我们需要在客户端中实现 WebSocket，以便与服务器建立连接和发送消息。将以下代码保存为 client.js 并在浏览器中运行：
-- -------------------- ---- ------- ----- ------ - --- ------------------------------------------ ------------- - ----- -- - ---------------------- ---------- ------------------------- - ---------------- - ----- -- - -------------------------------------- - -------------- - ----- -- - ---------------------- ---------- -
在客户端中，我们使用 WebSocket 对象建立了一个 WebSocket 连接。当连接成功建立时，onopen 回调函数会自动执行。在此回调函数当中，我们可以向 WebSocket 服务器发送消息。当服务器向客户端发送消息时，onmessage 回调函数会自动执行。
最后，要关闭WebSocket连接，我们可以使用 socket.close() 函数。
总结
本文介绍了如何在 Serverless 架构下实现 WebSocket，包括如何创建 Lambda 函数、配置 WebSocket API、向 Lambda 构建 WebSocket 集成，以及如何在客户端中实现 WebSocket。
虽然 Serverless 架构中实现 WebSocket 存在一些挑战，在上述示例中，我们可以看到它的优点。借助于 Serverless 和 AWS Lambda，WebSocket 可以用于令人信服的实时 Web 应用程序的开发。
在 Serverless 架构应用中，WebSocket 相关的实现流程需要进行适当的调整，对于开发者而言也需要付出比较大的努力去交流实现相关项目、技术方案等。
来源：JavaScript中文网 ，转载请注明来源 https://www.javascriptcn.com/post/64df45a6f6b2d6eab3a79cc4