超市报表

src/views/admin/system/menu/index.vue

@@ -40,6 +40,7 @@
 				lazy
 				:load="load"
 				:tree-props="{ children: 'children', hasChildren: 'hasChildren' }"
+				@expand-change="handleExpandChange"
 			>
 				<el-table-column :label="$t('sysmenu.name')" fixed prop="name" align="left" show-overflow-tooltip min-width="200" />
 				<el-table-column :label="$t('sysmenu.sortOrder')" prop="sortOrder" align="center" show-overflow-tooltip />
@@ -107,6 +108,8 @@ const tableRef = ref()
 const menuDialogRef = ref()
 const showSearch = ref(true)
 const queryRef = ref()
+// 保存已展开的节点ID
+const expandedRowIds = ref<Set<string>>(new Set())
 const state: BasicTableProps = reactive<BasicTableProps>({
 	createdIsNeed: false,
 	pageList: pageList,
@@ -136,6 +139,10 @@ const expandRowKeys = computed(() => {
 
 // 打开新增菜单弹窗
 const onOpenAddMenu = (type?: string, row?: any) => {
+	// 如果是添加子菜单，记录父节点ID，以便刷新后自动展开
+	if (row && row.id) {
+		expandedRowIds.value.add(row.id)
+	}
 	menuDialogRef.value.openDialog(type, row)
 }
 // 打开编辑菜单弹窗
@@ -149,11 +156,24 @@ const deleteMenuDisabled = (row: any) => {
 }
 
 // 清空搜索条件
-const resetQuery = () => {
+const resetQuery = async () => {
 	queryRef.value.resetFields()
 	state.queryForm.parentId = "-1"
 	state.dataList = []
-	getDataList()
+	await getDataList()
+	// 恢复展开状态
+	await restoreExpandedRows()
+}
+
+// 监听展开/折叠事件
+const handleExpandChange = (row: any, expanded: boolean) => {
+	if (expanded) {
+		// 展开时记录
+		expandedRowIds.value.add(row.id)
+	} else {
+		// 折叠时移除
+		expandedRowIds.value.delete(row.id)
+	}
 }
 
 const load = (row: any, treeNode: unknown, resolve: (date: any[]) => void) => {
@@ -186,15 +206,64 @@ const tableList = computed(() => {
 	return list
 })
 
-const query = () => {
+// 恢复展开状态（支持多级展开）
+const restoreExpandedRows = async () => {
+	await nextTick()
+	if (!tableRef.value || expandedRowIds.value.size === 0) return
+	
+	// 递归加载并展开节点
+	const loadAndExpandRows = async (list: any[]) => {
+		for (const row of list || []) {
+			if (expandedRowIds.value.has(row.id)) {
+				// 如果节点有子节点且未加载，先加载子节点
+				if (row.hasChildren && (!row.children || row.children.length === 0)) {
+					try {
+						const res = await pageList({ parentId: row.id })
+						const childrenList = res.data
+						if (Array.isArray(childrenList) && childrenList.length > 0) {
+							setHasChildren(childrenList)
+							row.children = childrenList
+						}
+					} catch (error) {
+						console.error('加载子节点失败:', error)
+					}
+				}
+				
+				// 展开该节点
+				try {
+					tableRef.value.toggleRowExpansion(row, true)
+					// 等待 DOM 更新，确保子节点渲染完成
+					await nextTick()
+				} catch (error) {
+					console.error('展开节点失败:', error)
+				}
+				
+				// 如果有子节点，递归处理
+				if (row.children && row.children.length > 0) {
+					await loadAndExpandRows(row.children)
+				}
+			}
+		}
+	}
+	
+	await loadAndExpandRows(state.dataList)
+}
+
+const query = async () => {
 	state.dataList = []
 	if (state.queryForm.menuName) {
 		state.queryForm.parentId = ""
+		// 搜索时清空展开记录
+		expandedRowIds.value.clear()
 	} else {
 		state.queryForm.parentId = "-1"
 	}
 
-	getDataList()
+	await getDataList()
+	// 刷新后恢复展开状态
+	if (!state.queryForm.menuName) {
+		await restoreExpandedRows()
+	}
 }
 // 删除操作
 const handleDelete = async (row: any) => {
@@ -206,7 +275,11 @@ const handleDelete = async (row: any) => {
 
 	try {
 		await delObj(row.id)
-		getDataList()
+		// 删除后刷新数据
+		state.dataList = []
+		await getDataList()
+		// 恢复展开状态
+		await restoreExpandedRows()
 		useMessage().success(t("common.delSuccessText"))
 	} catch (err: any) {
 		useMessage().error(err.msg)