From 1ec0374d08be45c6c6854705211dbd6c067dfab7 Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Thu, 18 Mar 2021 17:23:20 +0800
Subject: [PATCH] Merge branch 'dev' of http://blit.7drlb.com:8888/r/moral into dev

---
 screen-api/src/main/java/com/moral/api/controller/UserController.java |   47 ++++++++++++++++++++++++++---------------------
 1 files changed, 26 insertions(+), 21 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/controller/UserController.java b/screen-api/src/main/java/com/moral/api/controller/UserController.java
index 12b071a..34e7464 100644
--- a/screen-api/src/main/java/com/moral/api/controller/UserController.java
+++ b/screen-api/src/main/java/com/moral/api/controller/UserController.java
@@ -15,6 +15,7 @@
 import org.springframework.web.bind.annotation.RequestMethod;
 import org.springframework.web.bind.annotation.RestController;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.moral.api.entity.User;
 import com.moral.api.service.UserService;
 import com.moral.constant.ResponseCodeEnum;
@@ -32,8 +33,8 @@
 
     @ApiOperation(value = "������������", notes = "������������")
     @ApiImplicitParams({
-            @ApiImplicitParam(name = "account", value = "������,������6-10", required = false, paramType = "query", dataType = "String"),
-            @ApiImplicitParam(name = "password", value = "������,������6-20", required = false, paramType = "query", dataType = "String"),
+            @ApiImplicitParam(name = "account", value = "������,������6-10", required = true, paramType = "query", dataType = "String"),
+            @ApiImplicitParam(name = "password", value = "������,������6-20", required = true, paramType = "query", dataType = "String"),
             @ApiImplicitParam(name = "userName", value = "������������", required = false, paramType = "query", dataType = "String"),
             @ApiImplicitParam(name = "email", value = "������,������123456@qq.com", required = false, paramType = "query", dataType = "String"),
             @ApiImplicitParam(name = "mobile", value = "���������,1������11���������", required = false, paramType = "query", dataType = "String"),
@@ -42,16 +43,14 @@
     })
     @RequestMapping(value = "addUser", method = RequestMethod.POST)
     public ResultMessage addUser(User user, HttpServletRequest request) {
-        Map<String, Object> parameters = WebUtils.getParametersStartingWith(request, null);
-        if (!(parameters.containsKey("account") && parameters.containsKey("password"))) {
-            return ResultMessage.fail(ResponseCodeEnum.PARAMETERS_IS_MISSING.getCode(), ResponseCodeEnum.PARAMETERS_IS_MISSING.getMsg());
+        if (user.getAccount().isEmpty() || user.getPassword().isEmpty()) {
+            return ResultMessage.fail(ResponseCodeEnum.PARAMETERS_IS_MISSING.getCode(),
+                    ResponseCodeEnum.PARAMETERS_IS_MISSING.getMsg());
         }
         String token = request.getHeader("token");
-        Map<String, Object> map = userService.addUser(user, token);
-
-
-        if (map.containsKey("msg")) {
-            return ResultMessage.fail(map.get("msg").toString());
+        Map<String, Object> result = userService.addUser(user, token);
+        if (!result.isEmpty()) {
+            return ResultMessage.fail((int) result.get("code"), result.get("msg").toString());
         }
         return ResultMessage.ok();
     }
@@ -62,22 +61,24 @@
             @ApiImplicitParam(name = "token", value = "token", required = true, paramType = "header", dataType = "String")
     })
     @RequestMapping(value = "deleteUser", method = RequestMethod.POST)
-    public ResultMessage deleteUser(String userId, HttpServletRequest request) {
+    public ResultMessage deleteUser(String userId) {
         if (userId == null) {
-            return ResultMessage.fail("������������������");
+            return ResultMessage.fail(ResponseCodeEnum.PARAMETERS_IS_MISSING.getCode(), ResponseCodeEnum.PARAMETERS_IS_MISSING.getMsg());
         }
-        String token = request.getHeader("token");
-        Map<String, Object> map = userService.deleteUser(Integer.parseInt(userId), token);
-        if (map.containsKey("msg")) {
-            return ResultMessage.fail(map.get("msg").toString());
+        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
+        queryWrapper.eq("id", userId);
+        User user = userService.getOne(queryWrapper);
+        if (user == null) {
+            return ResultMessage.fail(ResponseCodeEnum.ACCOUNT_NOT_EXIST.getCode(), ResponseCodeEnum.ACCOUNT_NOT_EXIST.getMsg());
         }
+        userService.deleteUser(user);
         return ResultMessage.ok();
     }
 
     @ApiOperation(value = "������������������", notes = "������������������")
     @ApiImplicitParams({
-            @ApiImplicitParam(name = "account", value = "������,������6-10", required = false, paramType = "query", dataType = "String"),
-            @ApiImplicitParam(name = "password", value = "������,������6-20", required = false, paramType = "query", dataType = "String"),
+            @ApiImplicitParam(name = "account", value = "������,������6-10", required = true, paramType = "query", dataType = "String"),
+            @ApiImplicitParam(name = "password", value = "������,������6-20", required = true, paramType = "query", dataType = "String"),
             @ApiImplicitParam(name = "userName", value = "������������", required = false, paramType = "query", dataType = "String"),
             @ApiImplicitParam(name = "email", value = "������,������123456@qq.com", required = false, paramType = "query", dataType = "String"),
             @ApiImplicitParam(name = "mobile", value = "���������,1������11���������", required = false, paramType = "query", dataType = "String"),
@@ -86,10 +87,14 @@
     })
     @RequestMapping(value = "updateUser", method = RequestMethod.POST)
     public ResultMessage updateUser(User user, HttpServletRequest request) {
+        if (user.getAccount().isEmpty() || user.getPassword().isEmpty()) {
+            return ResultMessage.fail(ResponseCodeEnum.PARAMETERS_IS_MISSING.getCode(),
+                    ResponseCodeEnum.PARAMETERS_IS_MISSING.getMsg());
+        }
         String token = request.getHeader("token");
-        Map<String, Object> map = userService.updateUser(user, token);
-        if (map.containsKey("msg")) {
-            return ResultMessage.fail(map.get("msg").toString());
+        Map<String, Object> result = userService.addUser(user, token);
+        if (!result.isEmpty()) {
+            return ResultMessage.fail((int) result.get("code"), result.get("msg").toString());
         }
         return ResultMessage.ok();
     }

--
Gitblit v1.8.0