跳至主要内容

WebSocketRoute

每当通过Page.RouteWebSocketAsync()BrowserContext.RouteWebSocketAsync()设置WebSocket路由时,WebSocketRoute对象允许像真实服务器一样处理WebSocket。

模拟

默认情况下,路由的WebSocket不会连接到服务器。这样,您可以模拟通过WebSocket的整个通信。以下是一个示例,它用"response"来响应"request"

await page.RouteWebSocketAsync("wss://example.com/ws", ws => {
ws.OnMessage(frame => {
if (frame.Text == "request")
ws.Send("response");
});
});

由于我们没有在WebSocket路由处理程序中调用WebSocketRoute.ConnectToServer,Playwright会假设WebSocket将被模拟,并自动在页面内打开WebSocket。

以下是另一个处理JSON消息的示例:

await page.RouteWebSocketAsync("wss://example.com/ws", ws => {
ws.OnMessage(frame => {
using var jsonDoc = JsonDocument.Parse(frame.Text);
JsonElement root = jsonDoc.RootElement;
if (root.TryGetProperty("request", out JsonElement requestElement) && requestElement.GetString() == "question")
{
var response = new Dictionary<string, string> { ["response"] = "answer" };
string jsonResponse = JsonSerializer.Serialize(response);
ws.Send(jsonResponse);
}
});
});

拦截

或者,您可能希望连接到实际的服务器,但在通信过程中拦截消息并修改或阻止它们。调用WebSocketRoute.ConnectToServer会返回一个服务器端的WebSocketRoute实例,您可以向其发送消息或处理传入消息。

以下是一个修改页面发送到服务器的部分消息的示例。从服务器发送到页面的消息保持不变,依赖于默认转发。

await page.RouteWebSocketAsync("/ws", ws => {
var server = ws.ConnectToServer();
ws.OnMessage(frame => {
if (frame.Text == "request")
server.Send("request2");
else
server.Send(frame.Text);
});
});

连接到服务器后,默认情况下所有消息都会在页面和服务器之间转发

但是,如果您在原始路由上调用WebSocketRoute.OnMessage(),从页面发送到服务器的消息将不再被转发,而应该由handler处理。

类似地,在服务器端WebSocket上调用WebSocketRoute.OnMessage()将会停止转发消息从服务器到页面,此时handler应该负责处理这些消息。

以下示例在两个方向上拦截了一些消息。由于它在两个方向上都调用了WebSocketRoute.OnMessage(),因此完全不会自动转发。

await page.RouteWebSocketAsync("/ws", ws => {
var server = ws.ConnectToServer();
ws.OnMessage(frame => {
if (frame.Text != "blocked-from-the-page")
server.Send(frame.Text);
});
server.OnMessage(frame => {
if (frame.Text != "blocked-from-the-server")
ws.Send(frame.Text);
});
});

方法

CloseAsync

Added in: v1.48 webSocketRoute.CloseAsync

关闭WebSocket连接的一端。

用法

await WebSocketRoute.CloseAsync(options);

参数

返回


连接至服务器

Added in: v1.48 webSocketRoute.ConnectToServer

默认情况下,路由式WebSocket不会连接到服务器,因此您可以模拟整个WebSocket通信。此方法会连接到实际的WebSocket服务器,并返回服务器端的WebSocketRoute实例,从而具备从服务器发送和接收消息的能力。

连接到服务器后:

更多详情请参见顶部的示例。

用法

WebSocketRoute.ConnectToServer

返回


关闭时

Added in: v1.48 webSocketRoute.OnClose

允许处理 WebSocket.close

默认情况下,关闭连接的任何一端(无论是在页面还是服务器端)都会导致另一端也被关闭。但是,当设置了WebSocketRoute.OnClose()处理程序时,默认的关闭转发功能会被禁用,此时应由处理程序来管理关闭操作。

用法

WebSocketRoute.OnClose(handler);

参数

  • handler Action#

    处理WebSocket关闭的函数。接收一个可选的close code和一个可选的close reason


OnMessage

Added in: v1.48 webSocketRoute.OnMessage

该方法允许处理通过WebSocket发送的消息,无论是来自页面还是服务器。

当在原始的WebSocket路由上调用时,此方法处理从页面发送的消息。您可以通过使用WebSocketRoute.Send()响应这些消息、将它们转发到由WebSocketRoute.ConnectToServer返回的服务器端连接,或者执行其他操作来处理这些消息。

一旦调用此方法,消息将不会自动转发到服务器或页面 - 您应通过调用WebSocketRoute.Send()手动完成此操作。更多详情请参见顶部的示例。

再次调用此方法将用新的处理程序覆盖原有处理程序。

用法

WebSocketRoute.OnMessage(handler);

参数


发送

Added in: v1.48 webSocketRoute.Send

向WebSocket发送消息。当在原始WebSocket上调用时,消息会发送到页面。当在WebSocketRoute.ConnectToServer的结果上调用时,消息会发送到服务器。更多详情请参见顶部的示例。

用法

WebSocketRoute.Send(message);

参数


网址

Added in: v1.48 webSocketRoute.Url

页面中创建的WebSocket的URL。

用法

WebSocketRoute.Url

返回