Check for error (#7711)

* Custom error
This commit is contained in:
Amir Omidi
2019-10-14 16:26:51 -07:00
committed by GitHub
parent eb465fde1a
commit 5da0e16e44
3 changed files with 12 additions and 1 deletions

View File

@@ -48,6 +48,7 @@ import { CollapseAllAction } from 'vs/base/browser/ui/tree/treeDefaults';
import { ITreeItem, ITreeView } from 'sql/workbench/common/views';
import { IOEShimService } from 'sql/workbench/parts/objectExplorer/browser/objectExplorerViewTreeShim';
import { NodeContextKey } from 'sql/workbench/parts/dataExplorer/browser/nodeContext';
import { UserCancelledConnectionError } from 'sql/base/common/errors';
export class CustomTreeViewPanel extends ViewletPanel {
@@ -700,6 +701,9 @@ class TreeDataSource implements IAsyncDataSource<ITreeItem, ITreeItem> {
// So in order to enable this we need to tell the tree to refresh this node so it will ask us for the data again
setTimeout(() => {
this.treeView.collapse(node);
if (e instanceof UserCancelledConnectionError) {
return;
}
this.treeView.refresh([node]);
});
return [];