角色分配模块初始选中优化

src/views/demo/system/role/ModulesModal.vue

@@ -95,6 +95,8 @@
       unref(asyncBtnExpandTreeRef)?.expandAll(true);
     });
   }
+  let moduleIds = ref([]);
+  let elementIds = ref([]);
   async function getMenusForRol(id) {
     var menusForRol = await loadForRole({
       roleId: id,
@@ -107,9 +109,11 @@
     unref(asyncExpandTreeRef)?.setCheckedKeys(menusForRol.moduleIds);
     checkData.moduleIds = menusForRol.moduleIds;
     checkData.elementIds = menusForRol.elementIds;
+    unref(asyncBtnExpandTreeRef)?.setCheckedKeys(checkData.elementIds);
+    moduleIds.value = menusForRol.moduleIds;
+    elementIds.value = menusForRol.elementIds;
   }
-  let moduleIds = ref([]);
-  let elementIds = ref([]);
+
   function handleSelect(checkedKeys, e: { checked: boolean; checkedNodes; node; event }) {
     const list = e.checkedNodes;
     moduleIds.value = [];
@@ -143,6 +147,9 @@
     try {
       if (step.value == 1) {
         step.value = 2;
+        setTimeout(() => {
+          unref(asyncBtnExpandTreeRef)?.setCheckedKeys(checkData.elementIds);
+        }, 10);
       } else {
         var query = {
           roleId: rowId.value,