daohualiuxiang
/
vscode
mirror of https://github.com/microsoft/vscode.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improve pause connection (#147318)

pull/147914/head
Alex Dima 2022-04-22 12:35:33 +03:00
parent 2e2654c97f
commit 01ab50b929
No known key found for this signature in database
GPG Key ID: 39563C1504FDD0C9
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
extensions/vscode-test-resolver/src/extension.ts
View File

@ -24,7 +24,13 @@ let outputChannel: vscode.OutputChannel;
export function activate(context: vscode.ExtensionContext) { export function activate(context: vscode.ExtensionContext) {
let connectionPaused = false;
let connectionPausedEvent = new vscode.EventEmitter<boolean>();
function doResolve(_authority: string, progress: vscode.Progress<{ message?: string; increment?: number }>): Promise<vscode.ResolvedAuthority> { function doResolve(_authority: string, progress: vscode.Progress<{ message?: string; increment?: number }>): Promise<vscode.ResolvedAuthority> {
if (connectionPaused) {
throw vscode.RemoteAuthorityResolverError.TemporarilyNotAvailable('Not available right now');
}
const connectionToken = String(crypto.randomInt(0xffffffffff)); const connectionToken = String(crypto.randomInt(0xffffffffff));
// eslint-disable-next-line no-async-promise-executor // eslint-disable-next-line no-async-promise-executor
@ -151,8 +157,8 @@ export function activate(context: vscode.ExtensionContext) {
let remoteReady = true, localReady = true; let remoteReady = true, localReady = true;
const remoteSocket = net.createConnection({ port: serverAddr.port }); const remoteSocket = net.createConnection({ port: serverAddr.port });
let isDisconnected = connectionPaused; let isDisconnected = false;
connectionPausedEvent.event(_ => { const handleConnectionPause = () => {
let newIsDisconnected = connectionPaused; let newIsDisconnected = connectionPaused;
if (isDisconnected !== newIsDisconnected) { if (isDisconnected !== newIsDisconnected) {
outputChannel.appendLine(`Connection state: ${newIsDisconnected ? 'open' : 'paused'}`); outputChannel.appendLine(`Connection state: ${newIsDisconnected ? 'open' : 'paused'}`);
@ -175,7 +181,10 @@ export function activate(context: vscode.ExtensionContext) {
} }
} }
} }
}); };
connectionPausedEvent.event(_ => handleConnectionPause());
handleConnectionPause();
proxySocket.on('data', (data) => { proxySocket.on('data', (data) => {
remoteReady = remoteSocket.write(data); remoteReady = remoteSocket.write(data);
@ -251,9 +260,6 @@ export function activate(context: vscode.ExtensionContext) {
}); });
} }
let connectionPaused = false;
let connectionPausedEvent = new vscode.EventEmitter<boolean>();
const authorityResolverDisposable = vscode.workspace.registerRemoteAuthorityResolver('test', { const authorityResolverDisposable = vscode.workspace.registerRemoteAuthorityResolver('test', {
async getCanonicalURI(uri: vscode.Uri): Promise<vscode.Uri> { async getCanonicalURI(uri: vscode.Uri): Promise<vscode.Uri> {
return vscode.Uri.file(uri.path); return vscode.Uri.file(uri.path);