Filip/playwright
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

use addEventListener

Browse source
This commit is contained in:
Simon Knott 2024-10-04 15:52:14 +02:00
parent aebd5c0565
commit a483cab690
No known key found for this signature in database
GPG key ID: 8CEDC00028084AEC
1 changed files with 26 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
packages/playwright-core/src/server/fetch.ts
View file

@ -25,7 +25,7 @@ import zlib from 'zlib';
import type { HTTPCredentials } from '../../types/types';
import { TimeoutSettings } from '../common/timeoutSettings';
import { getUserAgent } from '../utils/userAgent';
import { assert, constructURLBasedOnBaseURL, createGuid, monotonicTime } from '../utils';
import { assert, constructURLBasedOnBaseURL, createGuid, eventsHelper, monotonicTime, type RegisteredListener } from '../utils';
import { HttpsProxyAgent, SocksProxyAgent } from '../utilsBundle';
import { BrowserContext, verifyClientCertificates } from './browserContext';
import { CookieStore, domainMatches, parseRawCookie } from './cookieStore';
@ -312,7 +312,7 @@ export abstract class APIRequestContext extends SdkObject {
let securityDetails: har.SecurityDetails | undefined;
const cleanup: (() => void)[] = [];
const listeners: RegisteredListener[] = [];
const request = requestConstructor(url, requestOptions as any, async response => {
const responseAt = monotonicTime();
@ -481,13 +481,13 @@ export abstract class APIRequestContext extends SdkObject {
});
request.on('error', reject);
const disposeListener = () => {
listeners.push(
eventsHelper.addEventListener(this, APIRequestContext.Events.Dispose, () => {
reject(new Error('Request context disposed.'));
request.destroy();
};
this.on(APIRequestContext.Events.Dispose, disposeListener);
cleanup.push(() => this.off(APIRequestContext.Events.Dispose, disposeListener));
request.on('close', () => cleanup.forEach(c => c()));
})
);
request.on('close', () => eventsHelper.removeEventListeners(listeners));
request.on('socket', socket => {
// happy eyeballs don't emit lookup and connect events, so we use our custom ones
@ -496,13 +496,10 @@ export abstract class APIRequestContext extends SdkObject {
tcpConnectionAt = happyEyeBallsTimings.tcpConnectionAt;
// non-happy-eyeballs sockets
const onLookup = () => { dnsLookupAt = monotonicTime(); };
socket.once('lookup', onLookup);
cleanup.push(() => socket.removeListener('lookup', onLookup));
const onConnect = () => { tcpConnectionAt = monotonicTime(); };
socket.once('connect', onConnect);
cleanup.push(() => socket.removeListener('connect', onConnect));
const onSecureConnect = () => {
listeners.push(
eventsHelper.addEventListener(socket, 'lookup', () => { dnsLookupAt = monotonicTime(); }),
eventsHelper.addEventListener(socket, 'connect', () => { tcpConnectionAt = monotonicTime(); }),
eventsHelper.addEventListener(socket, 'secureConnect', () => {
tlsHandshakeAt = monotonicTime();
if (socket instanceof TLSSocket) {
@ -515,9 +512,8 @@ export abstract class APIRequestContext extends SdkObject {
issuer: peerCertificate.issuer.CN
};
}
};
socket.once('secureConnect', onSecureConnect);
cleanup.push(() => socket.removeListener('secureConnect', onSecureConnect));
}),
);
serverIPAddress = socket.remoteAddress;
serverPort = socket.remotePort;