update code

yanzhu-modules/yanzhu-system/src/main/java/com/yanzhu/system/controller/SysDeptController.java

@@ -1,5 +1,6 @@
 package com.yanzhu.system.controller;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.concurrent.TimeUnit;
 
@@ -7,6 +8,7 @@ import com.yanzhu.common.core.constant.CacheConstants;
 import com.yanzhu.common.core.constant.Constants;
 import com.yanzhu.common.core.enums.IsDelEnums;
 import com.yanzhu.common.redis.service.RedisService;
+import com.yanzhu.system.domain.vo.TreeSelect;
 import org.apache.commons.lang3.ArrayUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.validation.annotation.Validated;
@@ -159,4 +161,14 @@ public class SysDeptController extends BaseController
         redisService.setCacheObject(key,depts, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES);
         return success(depts);
     }
+    @RequiresPermissions("system:dept:list")
+    @GetMapping("/deptTree")
+    public AjaxResult userDeptTree(){
+        SysDept comp= SecurityUtils.getLoginUser().getSysUser().getComp();
+        if(comp==null){
+            return  success(deptService.buildDeptTreeSelect(new ArrayList<>()));
+        }
+        List<TreeSelect> treeList = deptService.buildDeptTreeSelect(deptService.selectDeptListForComp(comp));
+        return success(treeList);
+    }
 }

yanzhu-modules/yanzhu-system/src/main/java/com/yanzhu/system/mapper/SysDeptMapper.java

@@ -19,6 +19,11 @@ public interface SysDeptMapper
      */
     public List<SysDept> selectDeptList(SysDept dept);
 
+
+    /**
+     * 根据deptId查询此ID及下面的子项，用于构建树
+     */
+    public List<SysDept> selectDeptListForComp(SysDept dept);
     /**
      * 根据角色ID查询部门树信息
      * 

yanzhu-modules/yanzhu-system/src/main/java/com/yanzhu/system/service/ISysDeptService.java

@@ -121,4 +121,8 @@ public interface ISysDeptService
      * @return 结果
      */
     public int deleteDeptById(Long deptId);
+    /**
+     * 根据deptId查询此ID及下面的子项，用于构建树
+     */
+    List<SysDept> selectDeptListForComp(SysDept comp);
 }

yanzhu-modules/yanzhu-system/src/main/java/com/yanzhu/system/service/impl/SysDeptServiceImpl.java

@@ -293,6 +293,13 @@ public class SysDeptServiceImpl implements ISysDeptService
     {
         return deptMapper.deleteDeptById(deptId);
     }
+    /**
+     * 根据deptId查询此ID及下面的子项，用于构建树
+     */
+    @Override
+    public List<SysDept> selectDeptListForComp(SysDept comp) {
+        return deptMapper.selectDeptListForComp(comp);
+    }
 
     /**
      * 递归列表

yanzhu-modules/yanzhu-system/src/main/resources/mapper/system/SysDeptMapper.xml

@@ -56,7 +56,14 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 		${params.dataScope}
 		order by d.parent_id, d.order_num
     </select>
-    
+	<select id="selectDeptListForComp" parameterType="SysDept" resultMap="SysDeptResult">
+		<include refid="selectDeptVo"/>
+		where d.del_flag = '0'
+		<if test="deptId != null and deptId != 0">
+			AND(find_in_set( #{deptId}, ancestors) or dept_id= #{deptId})
+		</if>
+		order by d.parent_id, d.order_num
+	</select>
     <select id="selectDeptListByRoleId" resultType="Long">
 		select d.dept_id
 		from sys_dept d

yanzhu-ui-vue3/src/api/system/dept.js

@@ -58,3 +58,10 @@ export function projectDeptList() {
     method: 'get'
   })
 }
+
+export function deptTree(){
+  return request({
+    url: '/system/dept/deptTree',
+    method: 'get'
+  })
+}

yanzhu-ui-vue3/src/views/manage/proProjectInfo/index.vue

@@ -2,21 +2,12 @@
   <div class="app-container">
     <el-form :model="queryParams" ref="queryRef" :inline="true" v-show="showSearch" label-width="68px">
       <el-form-item label="项目名称" prop="projectName">
-        <el-input
+        <el-input v-model="queryParams.projectName" placeholder="请输入项目名称" clearable @keyup.enter="handleQuery" />
-          v-model="queryParams.projectName"
-          placeholder="请输入项目名称"
-          clearable
-          @keyup.enter="handleQuery"
-        />
       </el-form-item>
       <el-form-item label="项目类型" prop="projectType">
         <el-select v-model="queryParams.projectType" placeholder="请选择项目类型" clearable>
-          <el-option
+          <el-option v-for="dict in pro_project_type" :key="dict.value" :label="dict.label"
-            v-for="dict in pro_project_type"
+            :value="dict.value"></el-option>
-            :key="dict.value"
-            :label="dict.label"
-            :value="dict.value"
-          ></el-option>
         </el-select>
       </el-form-item>
       <el-form-item>
@@ -27,42 +18,20 @@
 
     <el-row :gutter="10" class="mb8">
       <el-col :span="1.5">
-        <el-button
+        <el-button type="primary" plain icon="Plus" @click="handleAdd"
-          type="primary"
+          v-hasPermi="['manage:proProjectInfo:add']">新增</el-button>
-          plain
-          icon="Plus"
-          @click="handleAdd"
-          v-hasPermi="['manage:proProjectInfo:add']"
-        >新增</el-button>
       </el-col>
       <el-col :span="1.5">
-        <el-button
+        <el-button type="success" plain icon="Edit" :disabled="single" @click="handleUpdate"
-          type="success"
+          v-hasPermi="['manage:proProjectInfo:edit']">修改</el-button>
-          plain
-          icon="Edit"
-          :disabled="single"
-          @click="handleUpdate"
-          v-hasPermi="['manage:proProjectInfo:edit']"
-        >修改</el-button>
       </el-col>
       <el-col :span="1.5">
-        <el-button
+        <el-button type="danger" plain icon="Delete" :disabled="multiple" @click="handleDelete"
-          type="danger"
+          v-hasPermi="['manage:proProjectInfo:remove']">删除</el-button>
-          plain
-          icon="Delete"
-          :disabled="multiple"
-          @click="handleDelete"
-          v-hasPermi="['manage:proProjectInfo:remove']"
-        >删除</el-button>
       </el-col>
       <el-col :span="1.5">
-        <el-button
+        <el-button type="warning" plain icon="Download" @click="handleExport"
-          type="warning"
+          v-hasPermi="['manage:proProjectInfo:export']">导出</el-button>
-          plain
-          icon="Download"
-          @click="handleExport"
-          v-hasPermi="['manage:proProjectInfo:export']"
-        >导出</el-button>
       </el-col>
       <right-toolbar v-model:showSearch="showSearch" @queryTable="getList"></right-toolbar>
     </el-row>
@@ -133,54 +102,36 @@
       </el-table-column>
       <el-table-column label="操作" align="center" fixed="right" class-name="small-padding fixed-width">
         <template #default="scope">
-          <el-button link type="primary" icon="Edit" @click="handleUpdate(scope.row)" v-hasPermi="['manage:proProjectInfo:edit']">修改</el-button>
+          <el-button link type="primary" icon="Edit" @click="handleUpdate(scope.row)"
-          <el-button link type="primary" icon="Delete" @click="handleDelete(scope.row)" v-hasPermi="['manage:proProjectInfo:remove']">删除</el-button>
+            v-hasPermi="['manage:proProjectInfo:edit']">修改</el-button>
+          <el-button link type="primary" icon="Delete" @click="handleDelete(scope.row)"
+            v-hasPermi="['manage:proProjectInfo:remove']">删除</el-button>
         </template>
       </el-table-column>
     </el-table>
 
-    <pagination
+    <pagination v-show="total > 0" :total="total" v-model:page="queryParams.pageNum"
-      v-show="total>0"
+      v-model:limit="queryParams.pageSize" @pagination="getList" />
-      :total="total"
-      v-model:page="queryParams.pageNum"
-      v-model:limit="queryParams.pageSize"
-      @pagination="getList"
-    />
 
     <!-- 添加或修改项目信息对话框 -->
     <el-dialog :title="title" v-model="open" width="1080px" append-to-body>
       <el-form ref="proProjectInfoRef" :model="form" :rules="rules" label-width="100px">
-        <el-tabs
+        <el-tabs v-model="activeName" type="card" class="demo-tabs">
-          v-model="activeName"
-          type="card"
-          class="demo-tabs"
-        >
           <el-tab-pane label="项目基本信息" name="base">
             <el-row :gutter="20">
               <el-col :span="16">
-                <el-form-item label="所属部门" prop="deptId">
+                <el-form-item label="归属单位" prop="deptId">
-                  <el-tree-select
+                  <el-select v-model="form.deptId" placeholder="请选择公司" clearable>
-                    v-model="form.deptId"
+                    <el-option v-for="dict in deptOptions" :key="dict.deptId" :label="dict.deptName"
-                    :data="deptOptions"
+                      :value="dict.deptId"></el-option>
-                    :props="{ value: 'id', label: 'label', children: 'children' }"
+                  </el-select>
-                    value-key="id"
-                    placeholder="请选择所属部门"
-                    check-strictly
-                    style="width:100%"
-                  />
                 </el-form-item>
               </el-col>
               <el-col :span="8">
                 <el-form-item label="显示层级" prop="disDeptId">
-                  <el-tree-select
+                  <el-tree-select v-model="form.disDeptId" :data="deptOptions"
-                    v-model="form.disDeptId"
+                    :props="{ value: 'id', label: 'label', children: 'children' }" value-key="id" placeholder="请选择显示层级"
-                    :data="deptOptions"
+                    check-strictly style="width:100%" />
-                    :props="{ value: 'id', label: 'label', children: 'children' }"
-                    value-key="id"
-                    placeholder="请选择显示层级"
-                    check-strictly
-                    style="width:100%"
-                  />
                 </el-form-item>
               </el-col>
             </el-row>
@@ -200,12 +151,8 @@
               <el-col :span="8">
                 <el-form-item label="项目类型" prop="projectType">
                   <el-select v-model="form.projectType" placeholder="请选择项目类型" style="width:100%">
-                    <el-option
+                    <el-option v-for="dict in pro_project_type" :key="dict.value" :label="dict.label"
-                      v-for="dict in pro_project_type"
+                      :value="dict.value"></el-option>
-                      :key="dict.value"
-                      :label="dict.label"
-                      :value="dict.value"
-                    ></el-option>
                   </el-select>
                 </el-form-item>
               </el-col>
@@ -223,13 +170,11 @@
             <el-row :gutter="20">
               <el-col :span="24">
                 <el-form-item label="项目地址" prop="projectAddress">
-                  <el-input
+                  <el-input v-model="addressInfos" placeholder="请选择项目地址信息" @focus="getMapInfo">
-                    v-model="addressInfos"
-                    placeholder="请选择项目地址信息"
-                    @focus="getMapInfo"
-                  >
                     <template #prefix>
-                      <el-icon class="el-input__icon"><Location /></el-icon>
+                      <el-icon class="el-input__icon">
+                        <Location />
+                      </el-icon>
                     </template>
                   </el-input>
                 </el-form-item>
@@ -269,69 +214,43 @@
               <el-col :span="8">
                 <el-form-item label="项目进度" prop="projectSchedule">
                   <el-select v-model="form.projectSchedule" placeholder="请选择项目进度" style="width:100%">
-                    <el-option
+                    <el-option v-for="dict in pro_project_schedule" :key="dict.value" :label="dict.label"
-                      v-for="dict in pro_project_schedule"
+                      :value="dict.value"></el-option>
-                      :key="dict.value"
-                      :label="dict.label"
-                      :value="dict.value"
-                    ></el-option>
                   </el-select>
                 </el-form-item>
               </el-col>
               <el-col :span="8">
                 <el-form-item label="计划开工时间" prop="scheduledStartTime">
-                  <el-date-picker
+                  <el-date-picker v-model="form.scheduledStartTime" type="date" placeholder="请选择计划开工时间"
-                    v-model="form.scheduledStartTime"
+                    style="width:100%" />
-                    type="date"
-                    placeholder="请选择计划开工时间"
-                    style="width:100%"
-                  />
                 </el-form-item>
               </el-col>
             </el-row>
             <el-row :gutter="20">
               <el-col :span="8">
                 <el-form-item label="计划完工时间" prop="plannedCompletionTime">
-                  <el-date-picker
+                  <el-date-picker v-model="form.plannedCompletionTime" type="date" placeholder="请选择计划完工时间"
-                    v-model="form.plannedCompletionTime"
+                    style="width:100%" />
-                    type="date"
-                    placeholder="请选择计划完工时间"
-                    style="width:100%"
-                  />
                 </el-form-item>
               </el-col>
               <el-col :span="8">
                 <el-form-item label="实际开工时间" prop="actualOperatingTime">
-                  <el-date-picker
+                  <el-date-picker v-model="form.actualOperatingTime" type="date" placeholder="请选择实际开工时间"
-                    v-model="form.actualOperatingTime"
+                    style="width:100%" />
-                    type="date"
-                    placeholder="请选择实际开工时间"
-                    style="width:100%"
-                  />
                 </el-form-item>
               </el-col>
               <el-col :span="8">
                 <el-form-item label="实际完工时间" prop="actualCompletionTime">
-                  <el-date-picker
+                  <el-date-picker v-model="form.actualCompletionTime" type="date" placeholder="请选择实际完工时间"
-                    v-model="form.actualCompletionTime"
+                    style="width:100%" />
-                    type="date"
-                    placeholder="请选择实际完工时间"
-                    style="width:100%"
-                  />
                 </el-form-item>
               </el-col>
             </el-row>
             <el-row :gutter="20">
               <el-col :span="24">
                 <el-form-item label="项目概述" prop="projectSummarize">
-                  <el-input
+                  <el-input v-model="form.projectSummarize" type="textarea" :rows="3" placeholder="请输入项目概述"
-                    v-model="form.projectSummarize"
+                    maxlength="1024" show-word-limit />
-                    type="textarea"
-                    :rows="3"
-                    placeholder="请输入项目概述"
-                    maxlength="1024"
-                    show-word-limit
-                  />
                 </el-form-item>
               </el-col>
             </el-row>
@@ -345,18 +264,15 @@
                 <el-button type="danger" icon="Delete" @click="handleDeleteDepts">删除</el-button>
               </el-col>
             </el-row>
-            <el-table :data="projectDeptsList" :row-class-name="rowProjectDeptsIndex" @selection-change="handleDeptsSelectionChange" ref="projectDeptsRef">
+            <el-table :data="projectDeptsList" :row-class-name="rowProjectDeptsIndex"
+              @selection-change="handleDeptsSelectionChange" ref="projectDeptsRef">
               <el-table-column type="selection" width="50" align="center" />
               <el-table-column label="序号" align="center" prop="index" width="50" />
               <el-table-column label="单位类型" prop="deptType" align="center">
                 <template #default="scope">
                   <el-select v-model="scope.row.deptType" placeholder="请选择单位类型">
-                    <el-option
+                    <el-option v-for="dict in pro_dept_type" :key="dict.value" :label="dict.label"
-                      v-for="dict in pro_dept_type"
+                      :value="dict.value"></el-option>
-                      :key="dict.value"
-                      :label="dict.label"
-                      :value="dict.value"
-                    ></el-option>
                   </el-select>
                 </template>
               </el-table-column>
@@ -423,10 +339,10 @@
 
 <script setup name="ProProjectInfo">
 import { deptTreeSelect } from "@/api/system/user";
-import { projectDeptList } from "@/api/system/dept";
+import { projectDeptList, deptTree, listDept } from "@/api/system/dept";
 import { listProProjectInfo, getProProjectInfo, delProProjectInfo, addProProjectInfo, updateProProjectInfo } from "@/api/manage/proProjectInfo";
 import BaiduMap from "@/components/BaiduMap/Map.vue";
-
+import useUserStore from '@/store/modules/user'
 const { proxy } = getCurrentInstance();
 const { pro_project_type, pro_project_schedule, pro_dept_type, sys_is_del } = proxy.useDict('pro_project_type', 'pro_project_schedule', 'pro_dept_type', 'sys_is_del');
 
@@ -446,7 +362,7 @@ const projectDeptsList = ref([]);
 const checkedProjectDepts = ref([]);
 
 const deptOptions = ref([]);
-
+const userStore = useUserStore()
 const data = reactive({
   form: {
     setting: {}
@@ -679,7 +595,7 @@ function handleExport() {
 
 /** 查询部门下拉树结构 */
 function getDeptTree() {
-  deptTreeSelect().then(response => {
+  listDept({ deptType: 2 }).then(response => {
     deptOptions.value = response.data;
   });
 };