优化学员列表的部门

src/compenents/tree-department/index.tsx

@@ -10,19 +10,18 @@ interface Option {
 }
 
 interface PropInterface {
-  type: string;
   text: string;
-  refresh: boolean;
   showNum: boolean;
   selected: any;
+  depUserCount?: KeyNumberObject;
+  userCount?: number;
   onUpdate: (keys: any, title: any) => void;
 }
 
 export const TreeDepartment = (props: PropInterface) => {
   const [treeData, setTreeData] = useState<any>([]);
-  const [loading, setLoading] = useState<boolean>(true);
   const [selectKey, setSelectKey] = useState<number[]>([]);
-  const [userTotal, setUserTotal] = useState(0);
+  // 本地缓存
   const localDepartments = useSelector(
     (state: any) => state.systemConfig.value.departments
   );
@@ -34,44 +33,25 @@ export const TreeDepartment = (props: PropInterface) => {
   }, [props.selected]);
 
   useEffect(() => {
-    setLoading(true);
-    if (props.showNum) {
-      department.departmentList().then((res: any) => {
-        const departments: DepartmentsBoxModel = res.data.departments;
-        const departCount: DepIdsModel = res.data.dep_user_count;
-        setUserTotal(res.data.user_total);
-        if (JSON.stringify(departments) !== "{}") {
-          const new_arr: any[] = checkNewArr(departments, 0, departCount);
-          setTreeData(new_arr);
-        } else {
-          const new_arr: Option[] = [
-            {
-              key: "",
-              title: "全部",
-              children: [],
-            },
-          ];
-          setTreeData(new_arr);
-        }
-        setLoading(false);
-      });
-    } else {
     if (JSON.stringify(localDepartments) !== "{}") {
-        const new_arr: any[] = checkArr(localDepartments, 0);
+      let data: any[] = [];
-        setTreeData(new_arr);
+      if (props.depUserCount) {
+        data = checkNewArr(localDepartments, 0, props.depUserCount);
       } else {
-        const new_arr: Option[] = [
+        data = checkArr(localDepartments, 0);
+      }
+      setTreeData(data);
+    } else {
+      const data: Option[] = [
         {
           key: "",
           title: "全部",
           children: [],
         },
       ];
-        setTreeData(new_arr);
+      setTreeData(data);
     }
-      setLoading(false);
+  }, [localDepartments, props.depUserCount]);
-    }
-  }, [props.refresh, localDepartments]);
 
   const checkNewArr = (
     departments: DepartmentsBoxModel,
@@ -173,7 +153,7 @@ export const TreeDepartment = (props: PropInterface) => {
         onClick={() => onSelect([], "")}
       >
         全部{props.text}
-        {props.showNum && userTotal ? "(" + userTotal + ")" : ""}
+        {props.showNum && props.userCount ? "(" + props.userCount + ")" : ""}
       </div>
       {treeData.length > 0 && (
         <Tree

src/pages/course/index.tsx

@@ -139,9 +139,7 @@ const CoursePage = () => {
         <div className="float-left">
           <TreeDepartment
             selected={dep_ids}
-            refresh={refresh}
             showNum={false}
-            type="no-course"
             text={"部门"}
             onUpdate={(keys: any, title: any) => {
               resetLocalSearchParams({

src/pages/member/index.tsx

@@ -70,7 +70,9 @@ const MemberPage = () => {
   const [loading, setLoading] = useState(false);
   const [list, setList] = useState<DataType[]>([]);
   const [total, setTotal] = useState(0);
+  const [pureTotal, setPureTotal] = useState(0);
   const [refresh, setRefresh] = useState(false);
+  const [depUserCount, setDepUserCount] = useState<KeyNumberObject>();
 
   const [dep_ids, setDepIds] = useState<number[]>([]);
   const [selLabel, setLabel] = useState<string>(
@@ -252,6 +254,8 @@ const MemberPage = () => {
         setDepartments(res.data.departments);
         setUserDepIds(res.data.user_dep_ids);
         setTotal(res.data.total);
+        setPureTotal(res.data.pure_total);
+        setDepUserCount(res.data.dep_user_count);
         setLoading(false);
       });
   };
@@ -333,9 +337,9 @@ const MemberPage = () => {
         <div className="left-box">
           <TreeDepartment
             selected={dep_ids}
-            refresh={refresh}
             showNum={true}
-            type=""
+            userCount={pureTotal}
+            depUserCount={depUserCount}
             text={"部门"}
             onUpdate={(keys: any, title: any) => {
               resetLocalSearchParams({

src/playedu.d.ts

@@ -1,4 +1,8 @@
 declare global {
+  interface KeyNumberObject {
+    [key: number]: number;
+  }
+
   interface FileItem {
     id: string; //上传文件的唯一id
     file: File; //上传的文件资源