跳转到主要内容

WebSocketRoute

每当使用 page.routeWebSocket()browserContext.routeWebSocket() 设置 WebSocket 路由时,WebSocketRoute 对象允许像实际服务器那样处理 WebSocket。

模拟

默认情况下,路由的 WebSocket 不会连接到服务器。通过这种方式,您可以模拟整个 WebSocket 通信。这是一个响应 "request" 并返回 "response" 的示例。

await page.routeWebSocket('wss://example.com/ws', ws => {
ws.onMessage(message => {
if (message === 'request')
ws.send('response');
});
});

由于我们没有在 WebSocket 路由处理程序中调用 webSocketRoute.connectToServer(),Playwright 假定 WebSocket 将被模拟,并自动在页面内打开 WebSocket。

这是另一个处理 JSON 消息的示例

await page.routeWebSocket('wss://example.com/ws', ws => {
ws.onMessage(message => {
const json = JSON.parse(message);
if (json.request === 'question')
ws.send(JSON.stringify({ response: 'answer' }));
});
});

拦截

或者,您可能希望连接到实际服务器,但在此期间拦截消息并修改或阻止它们。调用 webSocketRoute.connectToServer() 会返回一个服务器端 WebSocketRoute 实例,您可以向其发送消息或处理传入消息。

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

await page.routeWebSocket('/ws', ws => {
const server = ws.connectToServer();
ws.onMessage(message => {
if (message === 'request')
server.send('request2');
else
server.send(message);
});
});

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

然而,如果您在原始路由上调用 webSocketRoute.onMessage(),从页面到服务器的消息将不再转发,而应由 处理程序 处理。

类似地,在服务器端 WebSocket 上调用 webSocketRoute.onMessage()停止转发从服务器到页面的消息,并且 处理程序 应该处理它们。

以下示例阻止了双向的一些消息。由于它在两个方向都调用了 webSocketRoute.onMessage(),因此根本没有自动转发。

await page.routeWebSocket('/ws', ws => {
const server = ws.connectToServer();
ws.onMessage(message => {
if (message !== 'blocked-from-the-page')
server.send(message);
});
server.onMessage(message => {
if (message !== 'blocked-from-the-server')
ws.send(message);
});
});

方法

关闭

新增于: v1.48 webSocketRoute.close

关闭 WebSocket 连接的一端。

用法

await webSocketRoute.close();
await webSocketRoute.close(options);

参数

返回


connectToServer

新增于: v1.48 webSocketRoute.connectToServer

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

连接到服务器后

有关更多详细信息,请参见顶部的示例。

用法

webSocketRoute.connectToServer();

返回


onClose

新增于: v1.48 webSocketRoute.onClose

允许处理 WebSocket.close

默认情况下,关闭连接的一端(无论是页面还是服务器)都会关闭另一端。但是,当设置了 webSocketRoute.onClose() 处理程序时,默认的关闭转发将被禁用,处理程序应该负责处理它。

用法

webSocketRoute.onClose(handler);

参数


onMessage

新增于: v1.48 webSocketRoute.onMessage

此方法允许处理由 WebSocket 发送的消息,无论是从页面还是从服务器。

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

一旦调用此方法,消息不会自动转发到服务器或页面——您应该通过调用 webSocketRoute.send() 手动完成。有关更多详细信息,请参见顶部的示例。

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

用法

webSocketRoute.onMessage(handler);

参数


send

新增于: v1.48 webSocketRoute.send

向 WebSocket 发送消息。当在原始 WebSocket 上调用时,将消息发送到页面。当在 webSocketRoute.connectToServer() 的结果上调用时,将消息发送到服务器。有关更多详细信息,请参见顶部的示例。

用法

webSocketRoute.send(message);

参数


url

新增于: v1.48 webSocketRoute.url

在页面中创建的 WebSocket 的 URL。

用法

webSocketRoute.url();

返回