- 增加管理员 角色分配，

admin-web/src/models/admin/adminList.js

@@ -1,11 +1,15 @@
 import { message } from 'antd';
+import { buildTreeNode, findCheckedKeys } from '../../utils/tree.utils';
 import {
   addAdmin,
   updateAdmin,
   updateAdminStatus,
   deleteAdmin,
   queryAdmin,
+  queryAdminRoleList,
+  adminRoleAssign,
 } from '../../services/admin';
+import { arrayToStringParams } from '../../utils/request.qs';
 
 export default {
   namespace: 'adminList',
@@ -15,6 +19,10 @@ export default {
     count: 0,
     pageNo: 0,
     pageSize: 10,
+
+    roleList: [],
+    roleCheckedKeys: [],
+    roleAssignLoading: false,
   },
 
   effects: {
@@ -78,6 +86,40 @@ export default {
         },
       });
     },
+    *queryRoleList({ payload }, { call, put }) {
+      yield put({
+        type: 'changeRoleAssignLoading',
+        payload: true,
+      });
+
+      const response = yield call(queryAdminRoleList, payload);
+      const roleList = response.data;
+      const roleTreeData = buildTreeNode(roleList, 'name', 'id');
+      const roleCheckedKeys = findCheckedKeys(roleList);
+
+      yield put({
+        type: 'querySuccess',
+        payload: {
+          roleList: roleTreeData,
+          roleCheckedKeys,
+        },
+      });
+
+      yield put({
+        type: 'changeRoleAssignLoading',
+        payload: false,
+      });
+    },
+    *roleAssign({ payload }, { call }) {
+      const params = {
+        id: payload.id,
+        roleIds: arrayToStringParams(payload.roleIds),
+      };
+      const response = yield call(adminRoleAssign, params);
+      if (response.code === 0) {
+        message.info('操作成功!');
+      }
+    },
   },
 
   reducers: {
@@ -87,5 +129,17 @@ export default {
         ...payload,
       };
     },
+    changeRoleCheckedKeys(state, { payload }) {
+      return {
+        ...state,
+        roleCheckedKeys: payload,
+      };
+    },
+    changeRoleAssignLoading(state, { payload }) {
+      return {
+        ...state,
+        roleAssignLoading: payload,
+      };
+    },
   },
 };

admin-web/src/models/admin/roleList.js

@@ -1,5 +1,6 @@
 import { message } from 'antd';
 import { arrayToStringParams } from '../../utils/request.qs';
+import { buildTreeNode, findAllNodes, findCheckedKeys } from '../../utils/tree.utils';
 import {
   addRole,
   updateRole,
@@ -9,71 +10,6 @@ import {
   roleAssignResource,
 } from '../../services/admin';
 
-function buildTreeNode(nodes, titleKey, nodeKey) {
-  return nodes.map(item => {
-    const res = {};
-    if (item.children) {
-      res.children = buildTreeNode(item.children, titleKey, nodeKey);
-    }
-    res.title = `${item.id}-${item[titleKey]}`;
-    res.key = item[nodeKey];
-    return res;
-  });
-}
-
-function findNodes(id, nodes) {
-  const res = [];
-  for (let i = 0; i < nodes.length; i += 1) {
-    const node = nodes[i];
-    if (node.key === id) {
-      res.push(node.key);
-      break;
-    } else {
-      const childNodes = findNodes(id, node.children);
-      if (childNodes.length) {
-        res.push(node.key);
-        for (let j = 0; j < childNodes.length; j += 1) {
-          res.push(childNodes[j]);
-        }
-        break;
-      }
-    }
-  }
-  return res;
-}
-
-function findAllNodes(resourceIds, nodes) {
-  const findNodesArray = [];
-  for (let i = 0; i < resourceIds.length; i += 1) {
-    const findNodesData = findNodes(resourceIds[i], nodes);
-    if (findNodesData) {
-      for (let j = 0; j < findNodesData.length; j += 1) {
-        const jD = findNodesData[j];
-        if (findNodesArray.indexOf(jD) === -1) {
-          findNodesArray.push(jD);
-        }
-      }
-    }
-  }
-  return findNodesArray;
-}
-
-function findCheckedKeys(nodes) {
-  let res = [];
-  for (let i = 0; i < nodes.length; i += 1) {
-    const node = nodes[i];
-    if (node.children) {
-      const findChildrenNodes = findCheckedKeys(node.children);
-      if (findChildrenNodes) {
-        res = res.concat(findChildrenNodes);
-      }
-    } else if (node.assigned === true) {
-      res.push(node.id);
-    }
-  }
-  return res;
-}
-
 export default {
   namespace: 'roleList',