From bc47afc824f8d7097855f16328f644930570fe58 Mon Sep 17 00:00:00 2001
From: k94314517 <8417338+k94314517@user.noreply.gitee.com>
Date: 星期四, 31 十月 2024 17:26:20 +0800
Subject: [PATCH] 代码初始化

---
 server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/CarUseBookServiceImpl.java |    2 
 server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/ApproveServiceImpl.java    |    3 +
 server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/MemberCloudController.java             |   24 +++++++++++-
 server/visits/dmvisit_admin/src/main/java/com/doumee/api/business/CompanyController.java                |    2 
 server/system_service/src/main/java/com/doumee/dao/business/model/Company.java                          |    3 +
 server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java    |   12 +++++
 server/visits/dmvisit_service/src/main/java/com/doumee/dao/business/model/Member.java                   |    7 +++
 server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/CompanyCloudController.java            |   11 ++---
 8 files changed, 52 insertions(+), 12 deletions(-)

diff --git a/server/system_service/src/main/java/com/doumee/dao/business/model/Company.java b/server/system_service/src/main/java/com/doumee/dao/business/model/Company.java
index ddaf90a..03ae864 100644
--- a/server/system_service/src/main/java/com/doumee/dao/business/model/Company.java
+++ b/server/system_service/src/main/java/com/doumee/dao/business/model/Company.java
@@ -196,4 +196,7 @@
     @ApiModelProperty(value = "erp閮ㄩ棬绾у埆璺緞", example = "1")
     @ExcelColumn(name="erp閮ㄩ棬绾у埆璺緞")
     private String erpCompanyPath;
+    @ApiModelProperty(value = "鍙煡璇㈤殣鎮i厤缃儴闂細0=鍚︼紱1=鏄�", example = "1")
+    @TableField(exist = false)
+    private Integer queryHiddenDanger;
 }
diff --git a/server/visits/dmvisit_admin/src/main/java/com/doumee/api/business/CompanyController.java b/server/visits/dmvisit_admin/src/main/java/com/doumee/api/business/CompanyController.java
index b65d924..beec7ca 100644
--- a/server/visits/dmvisit_admin/src/main/java/com/doumee/api/business/CompanyController.java
+++ b/server/visits/dmvisit_admin/src/main/java/com/doumee/api/business/CompanyController.java
@@ -101,7 +101,7 @@
     @ApiOperation("缁勭粐鏍戞煡璇�")
     @GetMapping("/tree")
     @ApiImplicitParams({
-            @ApiImplicitParam(name = "type", dataType = "Integer", value = "绫诲瀷 0鍔冲姟鍏徃 1鍐呴儴缁勭粐", required = true),
+            @ApiImplicitParam(name = "type", dataType = "Integer", value = "绫诲瀷 0鍔冲姟鍏徃 1鍐呴儴缁勭粐", required = true)
     })
     @RequiresPermissions("business:company:query")
     public ApiResponse<List<Company>> tree (@RequestParam Integer type) {
diff --git a/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/CompanyCloudController.java b/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/CompanyCloudController.java
index ec0beda..b5a5677 100644
--- a/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/CompanyCloudController.java
+++ b/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/CompanyCloudController.java
@@ -1,5 +1,6 @@
 package com.doumee.cloud.admin;
 
+import com.baomidou.mybatisplus.annotation.TableField;
 import com.doumee.api.BaseController;
 import com.doumee.config.DataSyncConfig;
 import com.doumee.config.annotation.CloudRequiredPermission;
@@ -16,10 +17,7 @@
 import com.doumee.service.business.ext.ERPSyncService;
 import com.doumee.service.business.impl.hksync.fhk.HkSyncOrgUserFromHKServiceImpl;
 import com.doumee.service.business.impl.hksync.fhk.HkSyncOrgUserFromSelfServiceImpl;
-import io.swagger.annotations.Api;
-import io.swagger.annotations.ApiImplicitParam;
-import io.swagger.annotations.ApiImplicitParams;
-import io.swagger.annotations.ApiOperation;
+import io.swagger.annotations.*;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
@@ -115,10 +113,11 @@
     @ApiOperation("缁勭粐鏍戞煡璇�")
     @GetMapping("/tree")
     @ApiImplicitParams({
-            @ApiImplicitParam(name = "type", dataType = "Integer", value = "绫诲瀷 0鍔冲姟鍏徃 1鍐呴儴缁勭粐", required = true),
+            @ApiImplicitParam(name = "type", dataType = "Integer", value = "绫诲瀷 0鍔冲姟鍏徃 1鍐呴儴缁勭粐 2鎸囧畾缁勭粐", required = true),
     })
     @CloudRequiredPermission("business:company:query")
-    public ApiResponse<List<Company>> tree (@RequestParam(required=false) Integer type,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){
+    public ApiResponse<List<Company>> tree (@RequestParam(required=false) Integer type,
+                                            @RequestHeader(Constants.HEADER_USER_TOKEN) String token){
         return ApiResponse.success(companyService.companyTree(type));
     }
 
diff --git a/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/MemberCloudController.java b/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/MemberCloudController.java
index 8d65a14..3a44e45 100644
--- a/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/MemberCloudController.java
+++ b/server/visits/dmvisit_admin/src/main/java/com/doumee/cloud/admin/MemberCloudController.java
@@ -24,7 +24,9 @@
 import com.doumee.service.business.ext.ERPSyncService;
 import com.doumee.service.business.impl.hksync.fhk.HkSyncOrgUserFromHKServiceImpl;
 import com.doumee.service.business.impl.hksync.fhk.HkSyncOrgUserFromSelfServiceImpl;
+import com.github.xiaoymin.knife4j.core.util.CollectionUtils;
 import io.swagger.annotations.*;
+import org.apache.commons.lang3.StringUtils;
 import org.apache.shiro.authz.annotation.RequiresPermissions;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.validation.annotation.Validated;
@@ -266,8 +268,26 @@
     @ApiOperation("瀵煎嚭Excel")
     @PostMapping("/exportExcel")
     @CloudRequiredPermission("business:member:exportExcel")
-    public void exportExcel (@RequestBody PageWrap<Member> pageWrap, HttpServletResponse response,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){
-        ExcelExporter.build(Member.class).export(memberService.findPage(pageWrap).getRecords(), "浜哄憳淇℃伅琛�", response);
+    public void exportExcel (@RequestBody PageWrap<MemberQuery> pageWrap, HttpServletResponse response,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){
+        List<Member> memberList =  new ArrayList<>();
+        if(pageWrap.getModel().getType().equals(Constants.memberType.visitor)){
+             memberList =  memberService.findVisitPage(pageWrap).getRecords();
+        }else if(pageWrap.getModel().getType().equals(Constants.memberType.internal)){
+            pageWrap.getModel().setLoginUserInfo(this.getLoginUser(token));
+            memberList =  memberService.findMemberInfoPage(pageWrap).getRecords();
+        }
+        if(CollectionUtils.isNotEmpty(memberList)){
+            for (Member member:memberList) {
+                if(StringUtils.isNotBlank(member.getFaceImg())){
+                    member.setHavaFaceImg(Constants.ONE);
+                }else{
+                    member.setHavaFaceImg(Constants.ZERO);
+                }
+
+            }
+        }
+
+        ExcelExporter.build(Member.class).export(memberList, "浜哄憳淇℃伅琛�", response);
     }
 
     @ApiOperation("鏍规嵁ID鏌ヨ")
diff --git a/server/visits/dmvisit_service/src/main/java/com/doumee/dao/business/model/Member.java b/server/visits/dmvisit_service/src/main/java/com/doumee/dao/business/model/Member.java
index 98ad7d6..8d72649 100644
--- a/server/visits/dmvisit_service/src/main/java/com/doumee/dao/business/model/Member.java
+++ b/server/visits/dmvisit_service/src/main/java/com/doumee/dao/business/model/Member.java
@@ -182,7 +182,6 @@
     private String visitCompanyName;
 
     @ApiModelProperty(value = "缁勭粐鍚嶇О")
-    @ExcelColumn(name="鎵�灞為儴闂�",index = 4,width = 12)
     @TableField(exist = false)
     private String companyName;
     @ApiModelProperty(value = "鏈�鍚庢搷浣滀汉")
@@ -203,6 +202,7 @@
     @TableField(exist = false)
     private Integer headId;
     @ApiModelProperty(value = "缁勭粐璺緞")
+    @ExcelColumn(name="鎵�灞為儴闂�",index = 4,width = 12)
     @TableField(exist = false)
     private String companyNamePath;
     @ApiModelProperty(value = "缁勭粐娴峰悍缂栫爜")
@@ -280,5 +280,10 @@
     @TableField(exist = false)
     private Integer querySpecial;
 
+    @ApiModelProperty(value = "鏄惁鏈変汉鑴�" )
+    @ExcelColumn(name="鏄惁鏈変汉鑴�",index = 15,width = 12,valueMapping = "1=鏈�;0=鏃�;")
+    @TableField(exist = false)
+    private Integer havaFaceImg;
+
 
 }
diff --git a/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/ApproveServiceImpl.java b/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/ApproveServiceImpl.java
index b0ed16f..91ce4b9 100644
--- a/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/ApproveServiceImpl.java
+++ b/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/ApproveServiceImpl.java
@@ -791,6 +791,9 @@
      * @param objLevel  鏌ヨ绾у埆
      */
     public Company getAuditCompanyHead(Integer companyId, Integer objLevel){
+        if(Objects.isNull(objLevel)||Objects.isNull(companyId)){
+            throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"瀹℃壒娴侀厤缃紓甯革紝璇疯仈绯荤鐞嗗憳");
+        }
         Company auditCompany =  companyMapper.selectById(companyId);
         Integer queryLevel = Constants.ONE;
         while(queryLevel<objLevel){
diff --git a/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/CarUseBookServiceImpl.java b/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/CarUseBookServiceImpl.java
index f62e20f..3751492 100644
--- a/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/CarUseBookServiceImpl.java
+++ b/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/CarUseBookServiceImpl.java
@@ -118,7 +118,7 @@
         CarDriver carDriver = carDriverMapper.selectOne(new QueryWrapper<CarDriver>().lambda()
                 .eq(CarDriver::getCarId,cars.getId())
                 .eq(CarDriver::getIsdeleted,Constants.ZERO)
-                .eq(CarDriver::getStatus,Constants.ONE)
+//                .eq(CarDriver::getStatus,Constants.ZERO)
                 .last( " limit 1 ")
         );
         if(Objects.nonNull(carDriver)&&Objects.nonNull(carDriver.getMemberId())){
diff --git a/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java b/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java
index e7782a0..fe28eb8 100644
--- a/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java
+++ b/server/visits/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java
@@ -449,6 +449,9 @@
             queryWrapper.ge(Company::getFsDate, Utils.Date.getStart(company.getFsDate()));
             queryWrapper.le(Company::getFsDate, Utils.Date.getEnd(company.getFsDate()));
         }
+        if (company.getQueryHiddenDanger() != null && Constants.equalsInteger(company.getQueryHiddenDanger(),Constants.ONE)) {
+            queryWrapper.apply(" t.id in ( select hdp.COMPANY_ID from hidden_danger_param hdp where hdp.ISDELETED = 0 and hdp.STATUS = 0 and hdp.TYPE = 0  ) ");
+        }
         queryWrapper.orderByDesc(Company::getCreateDate);
         queryWrapper.leftJoin(Company.class,Company::getId,Company::getParentId);
         queryWrapper.select(" t.* ")
@@ -678,7 +681,14 @@
         queryWrapper.leftJoin(Member.class,Member::getId,Company::getHeadId);
         queryWrapper.leftJoin(SystemUser.class,SystemUser::getId,Company::getEditor);
         queryWrapper.eq(Company::getIsdeleted,Constants.ZERO);
-        queryWrapper.eq(type!=null,Company::getType,type);
+        if(Objects.nonNull(type) && ( Constants.equalsInteger(type,Constants.ONE) || Constants.equalsInteger(type,Constants.ZERO))){
+            queryWrapper.eq(type!=null,Company::getType,type);
+        }else{
+            String companySpecialId = systemDictDataBiz.queryByCode(Constants.SYSTEM,Constants.SPECIAL_COMPANY_ID).getCode();
+            if(StringUtils.isNotBlank(companySpecialId)){
+                queryWrapper.apply(" (find_in_set('"+companySpecialId+"',REPLACE(t.company_path,'/',','))  or t.parent_id  is null  )  ");
+            }
+        }
         queryWrapper.orderByAsc(true,"t.sortnum");
         List<Company> companyList = companyJoinMapper.selectJoinList(Company.class,queryWrapper);
         //閰嶇疆缁勫憳浜哄憳绫诲瀷鏉ユ簮鏂瑰紡 0鑷缓 1ERP

--
Gitblit v1.9.3