From 4ff16885c1eb9c3550be47d2248784e92fb302b4 Mon Sep 17 00:00:00 2001
From: Aasim Khan <aasimkhan30@gmail.com>
Date: Thu, 30 Mar 2023 15:17:45 -0700
Subject: [PATCH] Cleaning update children logic in async server tree (#22550)

---
 .../objectExplorer/browser/asyncServerTree.ts       | 13 ++-----------
 1 file changed, 2 insertions(+), 11 deletions(-)

diff --git a/src/sql/workbench/services/objectExplorer/browser/asyncServerTree.ts b/src/sql/workbench/services/objectExplorer/browser/asyncServerTree.ts
index c458a9552e..bdca7cc7f4 100644
--- a/src/sql/workbench/services/objectExplorer/browser/asyncServerTree.ts
+++ b/src/sql/workbench/services/objectExplorer/browser/asyncServerTree.ts
@@ -125,18 +125,9 @@ export class AsyncServerTree extends WorkbenchAsyncDataTree<ConnectionProfileGro
 	}
 
 	public override async updateChildren(element?: ServerTreeElement, recursive?: boolean, rerender?: boolean, options?: IAsyncDataTreeUpdateChildrenOptions<ServerTreeElement>): Promise<void> {
-		const viewState = this.getViewState();
-		const expandedElementIds = viewState?.expanded;
+		const expandedChildren = this.getExpandedState(element);
 		await super.updateChildren(element, recursive, rerender, options);
-		if (expandedElementIds) {
-			for (let i = 0; i <= expandedElementIds.length; i++) {
-				const id = expandedElementIds[i];
-				const node = this.getDataNodeById(id);
-				if (node) {
-					await this.expand(node.element);
-				}
-			}
-		}
+		await this.expandElements(expandedChildren);
 	}
 
 	public async expandElements(elements: ServerTreeElement[]): Promise<void> {