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_admin/src/main/java/com/doumee/cloud/admin/MemberCloudController.java |   24 ++++++++++++++++++++++--
 1 files changed, 22 insertions(+), 2 deletions(-)

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鏌ヨ")

--
Gitblit v1.9.3