From 311cc48341d2cb59c89dd427a6ffde65733c8066 Mon Sep 17 00:00:00 2001
From: fit2cloud-chenyw <yawen.chen@fit2cloud.com>
Date: Fri, 12 Mar 2021 11:31:55 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=BB=84=E7=BB=87=E6=9C=BA=E6=9E=84?=
 =?UTF-8?q?=E6=8C=89=E5=90=8D=E7=A7=B0=E6=A3=80=E7=B4=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../io/dataease/service/sys/DeptService.java  |  2 +-
 frontend/src/views/system/dept/index.vue      | 58 ++++++++++++++++---
 2 files changed, 51 insertions(+), 9 deletions(-)

diff --git a/backend/src/main/java/io/dataease/service/sys/DeptService.java b/backend/src/main/java/io/dataease/service/sys/DeptService.java
index 101533d193..290d3468af 100644
--- a/backend/src/main/java/io/dataease/service/sys/DeptService.java
+++ b/backend/src/main/java/io/dataease/service/sys/DeptService.java
@@ -162,7 +162,7 @@ public class DeptService {
      * @return
      */
     private List<Long> upTree(List<SimpleTreeNode> allNodes, List<SimpleTreeNode> targetNodes){
-        final Map<Long, SimpleTreeNode> map = targetNodes.stream().collect(Collectors.toMap(SimpleTreeNode::getId, node -> node));
+        final Map<Long, SimpleTreeNode> map = allNodes.stream().collect(Collectors.toMap(SimpleTreeNode::getId, node -> node));
         List<Long> results = targetNodes.parallelStream().flatMap(targetNode -> {
             //向上逐级找爹
             List<Long> ids = new ArrayList<>();
diff --git a/frontend/src/views/system/dept/index.vue b/frontend/src/views/system/dept/index.vue
index 1dcac01e5c..2e9b5d8cc9 100644
--- a/frontend/src/views/system/dept/index.vue
+++ b/frontend/src/views/system/dept/index.vue
@@ -4,7 +4,7 @@
     <complex-table
       ref="table"
       :data="tableData"
-      lazy
+      :lazy="isLazy"
       :load="loadExpandDatas"
       :columns="columns"
       :buttons="buttons"
@@ -12,6 +12,7 @@
       :search-config="searchConfig"
       :pagination-config="paginationConfig"
       :tree-props="{children: 'children', hasChildren: 'hasChildren'}"
+      :default-expand-all="isTableExpand"
       row-key="deptId"
       @search="search"
     >
@@ -22,7 +23,7 @@
       <!-- <el-table-column type="selection" fix /> -->
       <el-table-column label="名称" prop="name" />
       <el-table-column label="下属组织数" prop="subCount" />
-      <el-table-column label="状态" align="center" prop="enabled">
+      <!-- <el-table-column label="状态" align="center" prop="enabled">
         <template slot-scope="scope">
           <el-switch
             v-model="scope.row.enabled"
@@ -32,7 +33,7 @@
             @change="changeEnabled(scope.row, scope.row.enabled,)"
           />
         </template>
-      </el-table-column>
+      </el-table-column> -->
       <el-table-column prop="createTime" label="创建日期">
         <template v-slot:default="scope">
           <span>{{ scope.row.createTime | timestampFormatDate }}</span>
@@ -47,7 +48,7 @@
       :close-on-click-modal="false"
       :title="formType=='add' ? $t('organization.create') : $t('organization.modify')"
       :visible.sync="dialogOrgAddVisible"
-      width="30%"
+      width="500px"
       :destroy-on-close="true"
       @closed="closeFunc"
     >
@@ -74,7 +75,7 @@
         </el-form-item>
 
         <el-form-item label="状态" prop="enabled">
-          <el-radio-group v-model="form.enabled" style="width: 140px">
+          <el-radio-group v-model="form.enabled" style="width: 140px" disabled>
             <el-radio :label="true">启用</el-radio>
             <el-radio :label="false">停用</el-radio>
           </el-radio-group>
@@ -176,15 +177,19 @@ export default {
         field: 'pid',
         operator: 'eq',
         value: 0
-      }
-
+      },
+      defaultForm: { deptId: null, top: true, enabled: true, pid: null },
+      isTableExpand: false,
+      isLazy: true
     }
   },
   activated() {
+    this.form = Object.assign({}, this.defaultForm)
     this.search()
   },
   methods: {
     create() {
+      this.form = Object.assign({}, this.defaultForm)
       this.dialogOrgAddVisible = true
       this.formType = 'add'
     },
@@ -254,8 +259,20 @@ export default {
       }
       return Object.assign({}, condition)
     },
+
+    setTableAttr(isSearch) {
+      if (isSearch) {
+        this.lazy = false
+        this.isTableExpand = true
+      } else {
+        this.lazy = true
+        this.isTableExpand = false
+      }
+    },
     // 加载表格数据
     search(condition) {
+      this.setTableAttr()
+      this.tableData = []
       let param = {}
       if (condition && condition.quick) {
         const con = this.quick_condition(condition)
@@ -273,11 +290,36 @@ export default {
           }
           return obj
         })
+
+        if (condition && condition.quick) {
+          data = this.buildTree(data)
+          this.setTableAttr(true)
+        }
         this.tableData = data
         this.depts = null
       })
     },
 
+    buildTree(arrs) {
+      const idMapping = arrs.reduce((acc, el, i) => {
+        acc[el.deptId] = i
+        return acc
+      }, {})
+      const roots = []
+      arrs.forEach(el => {
+        // 判断根节点
+        if (el.pid === null || el.pid === 0) {
+          roots.push(el)
+          return
+        }
+        // 用映射表找到父元素
+        const parentEl = arrs[idMapping[el.pid]]
+        // 把当前元素添加到父元素的`children`数组中
+        parentEl.children = [...(parentEl.children || []), el]
+      })
+      return roots
+    },
+
     // 加载下一级子节点数据
     loadExpandDatas(row, treeNode, resolve) {
       getDeptTree(row.deptId).then(res => {
@@ -359,7 +401,7 @@ export default {
         id: node.deptId,
         pid: node.pid,
         label: node.name,
-        children: node.children
+        children: node.children || null
       }
     },
     // 改变状态