From 87a395a6024278594a0a80d25a24ba4ca5993513 Mon Sep 17 00:00:00 2001
From: zhanghua <314079846@qq.com>
Date: 星期六, 18 三月 2023 17:42:10 +0800
Subject: [PATCH] 统计修改

---
 ycl-common/src/main/java/com/ycl/controller/user/UmsAdminController.java |   43 +++++++++++++++++++++++++++++++++----------
 1 files changed, 33 insertions(+), 10 deletions(-)

diff --git a/ycl-common/src/main/java/com/ycl/controller/user/UmsAdminController.java b/ycl-common/src/main/java/com/ycl/controller/user/UmsAdminController.java
index f3d12e7..fa325a3 100644
--- a/ycl-common/src/main/java/com/ycl/controller/user/UmsAdminController.java
+++ b/ycl-common/src/main/java/com/ycl/controller/user/UmsAdminController.java
@@ -4,15 +4,15 @@
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.ycl.annotation.LogSave;
 import com.ycl.api.CommonResult;
+import com.ycl.bo.AdminUserDetails;
 import com.ycl.controller.BaseController;
-import com.ycl.dto.UmsAdminLoginParam;
-import com.ycl.dto.UmsAdminParam;
-import com.ycl.dto.UpdateAdminPasswordParam;
+import com.ycl.dto.*;
 import com.ycl.entity.user.UmsAdmin;
 import com.ycl.entity.user.UmsRole;
 import com.ycl.service.redis.RedisService;
 import com.ycl.service.user.UmsAdminService;
-import com.ycl.service.user.UmsRoleService;
+import com.ycl.service.user.UmsMenuService;
+import com.ycl.utils.EasyExcelUtils;
 import com.ycl.utils.redis.RedisKey;
 import com.ycl.vo.user.UserVO;
 import io.swagger.annotations.Api;
@@ -21,12 +21,16 @@
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.security.core.context.SecurityContextHolder;
 import org.springframework.stereotype.Controller;
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
 
 import javax.annotation.Resource;
 import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
 import java.security.Principal;
 import java.util.HashMap;
 import java.util.List;
@@ -47,7 +51,7 @@
     @Autowired
     private UmsAdminService adminService;
     @Autowired
-    private UmsRoleService roleService;
+    UmsMenuService umsMenuService;
     @Resource
     private RedisService redisService;
 
@@ -101,11 +105,10 @@
         }
         String username = principal.getName();
         UmsAdmin umsAdmin = adminService.getAdminByUsername(username);
+        AdminUserDetails user = (AdminUserDetails) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
         Map<String, Object> data = new HashMap<>();
-        data.put("userId", umsAdmin.getId());
-        data.put("username", umsAdmin.getUsername());
-        data.put("menus", roleService.getMenuList(umsAdmin.getId()));
-        data.put("icon", umsAdmin.getIcon());
+        data.put("user", umsAdmin);
+        data.put("menus", umsMenuService.ListTreeMyself(user.getUserId()));
         List<UmsRole> roleList = adminService.getRoleList(umsAdmin.getId());
         if (CollUtil.isNotEmpty(roleList)) {
             List<String> roles = roleList.stream().map(UmsRole::getName).collect(Collectors.toList());
@@ -130,6 +133,24 @@
         IPage<UmsAdmin> page = adminService.pageUser(pageUserVO);
         return CommonResult.success(page);
     }
+
+
+    @ApiOperation("鐢ㄦ埛瀵煎嚭")
+    @PostMapping("/export")
+    public void exportUser(HttpServletResponse response) {
+        List<UmsUserDto> umsUserDtos = adminService.userExp();
+        String sheetName = "鐢ㄦ埛瀵煎嚭";
+        EasyExcelUtils.export(response, sheetName, UmsUserDto.class, umsUserDtos);
+    }
+
+    @ApiOperation("鐢ㄦ埛瀵煎叆")
+    @PostMapping("/import")
+    public CommonResult importUser(MultipartFile file) throws IOException {
+        EasyExcelUtils.importDepartFile(file);
+        return CommonResult.success(true, "瀵煎叆鎴愬姛");
+    }
+
+
 
     @ApiOperation("鑾峰彇鎸囧畾鐢ㄦ埛淇℃伅")
     @RequestMapping(value = "/{id}", method = RequestMethod.GET)
@@ -160,11 +181,13 @@
         if (status > 0) {
             return CommonResult.success(status);
         } else if (status == -1) {
-            return CommonResult.failed("鎻愪氦鍙傛暟涓嶅悎娉�");
+            return CommonResult.failed("瀵嗙爜闀垮害8鍒�16浣嶄笖瀵嗙爜涓殑瀛楃蹇呴』鍖呭惈瀛楁瘝锛堝ぇ鍐欐垨鑰呭皬鍐欙級鍜屽繀椤诲寘鍚暟瀛楋紝涓嶈兘鍖呭惈绌烘牸");
         } else if (status == -2) {
             return CommonResult.failed("鎵句笉鍒拌鐢ㄦ埛");
         } else if (status == -3) {
             return CommonResult.failed("鏃у瘑鐮侀敊璇�");
+        }else if (status == -4) {
+            return CommonResult.failed("瀵嗙爜涓嶈兘鍜岀櫥褰曞悕瀹屽叏涓�鑷�");
         } else {
             return CommonResult.failed();
         }

--
Gitblit v1.8.0