From b64946fb4c4b32592633ff1dc0b95a77cb440246 Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Thu, 18 Mar 2021 17:15:45 +0800
Subject: [PATCH] 用户增删改查修改

---
 screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java |  253 ++++++++++++++++++++++++++++++++------------------
 1 files changed, 162 insertions(+), 91 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java
index e30cd14..fb42e30 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java
@@ -10,18 +10,21 @@
 import java.util.stream.Collectors;
 
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
 import com.moral.api.entity.Group;
 import com.moral.api.entity.Menu;
 import com.moral.api.entity.Organization;
 import com.moral.api.entity.User;
 
-import com.moral.api.mapper.GroupMapper;
+import com.moral.api.entity.UserGroup;
 import com.moral.api.mapper.MenuMapper;
-import com.moral.api.mapper.OrganizationMapper;
+import com.moral.api.mapper.UserGroupMapper;
 import com.moral.api.mapper.UserMapper;
+import com.moral.api.pojo.UserBo;
 import com.moral.api.service.UserService;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.moral.constant.Constants;
+import com.moral.constant.ResponseCodeEnum;
 import com.moral.util.AESUtils;
 import com.moral.util.MD5Utils;
 import com.moral.util.RegexUtils;
@@ -30,6 +33,7 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
 
 /**
  * <p>
@@ -48,43 +52,81 @@
     @Autowired
     private MenuMapper menuMapper;
 
+    @Autowired
+    private UserGroupMapper userGroupMapper;
+
+    @Value("${AES.KEY}")
+    private String AESKey;
+
     @Override
-    public User selectUserInfo(Map<String, Object> parameters) {
+    public UserBo selectUserInfo(Map<String, Object> parameters) {
         return userMapper.selectUserInfo(parameters);
     }
 
     @Override
-    public Map<String, Object> login(User user) {
-        Map<String, Object> resultMap = new LinkedHashMap<>();
+    public Map<String, Object> login(Map<String, Object> parameters) {
+        UserBo userBo = selectUserInfo(parameters);
+        Map<String, Object> result = new HashMap<>();
+        //������������
+        if (userBo == null) {
+            result.put("code", ResponseCodeEnum.ACCOUNT_NOT_EXIST.getCode());
+            result.put("msg", ResponseCodeEnum.ACCOUNT_NOT_EXIST.getMsg());
+            return result;
+        }
+        //������������
+        String password = parameters.get("password").toString();
+        //������������
+        //password = AESUtils.decrypt(password, AESKey);
+        if (!MD5Utils.saltMD5Verify(password, userBo.getPassword())) {
+            result.put("code", ResponseCodeEnum.PASSWORD_INVALID.getCode());
+            result.put("msg", ResponseCodeEnum.PASSWORD_INVALID.getMsg());
+            return result;
+        }
+        //������������������
+        if (Constants.DELETE.equals(userBo.getIsDelete())) {
+            result.put("code", ResponseCodeEnum.ACCOUNT_IS_DELETE.getCode());
+            result.put("msg", ResponseCodeEnum.ACCOUNT_IS_DELETE.getMsg());
+            return result;
+        }
+        //������������������
+        if (userBo.getExpireTime() != null && userBo.getExpireTime().getTime() < System.currentTimeMillis()) {
+            result.put("code", ResponseCodeEnum.ACCOUNT_IS_EXPIRE.getCode());
+            result.put("msg", ResponseCodeEnum.ACCOUNT_IS_EXPIRE.getMsg());
+            return result;
+        }
         //������������������
         Map<String, Object> userInfo = new LinkedHashMap<>();
         List<Map<String, Object>> groups = new ArrayList<>();
-        for (Group group : user.getGroups()) {
+        for (Group group : userBo.getGroups()) {
             HashMap<String, Object> groupMap = new HashMap<>();
             groupMap.put("groupId", group.getId());
             groupMap.put("groupName", group.getGroupName());
             groups.add(groupMap);
         }
-        Organization organization = user.getOrganization();
-        userInfo.put("userId", user.getId());
-        userInfo.put("account", user.getAccount());
-        userInfo.put("userName", user.getUserName());
-        userInfo.put("expireTime", user.getExpireTime());
-        userInfo.put("isAdmin", user.getIsAdmin());
-        userInfo.put("organizationId", user.getOrganizationId());
+        Organization organization = userBo.getOrganization();
+        userInfo.put("userId", userBo.getId());
+        userInfo.put("account", userBo.getAccount());
+        userInfo.put("userName", userBo.getUserName());
+        userInfo.put("expireTime", userBo.getExpireTime());
+        userInfo.put("isAdmin", userBo.getIsAdmin());
+        userInfo.put("organizationId", userBo.getOrganizationId());
         userInfo.put("organizationName", organization.getName());
         userInfo.put("locationLevel", organization.getLocationLevel());
         userInfo.put("groups", groups);
-        userInfo.putAll(getMenus(user.getId()));
+        userInfo.putAll(getMenus(userBo.getId()));
         try {
             //������token������������redis
-            String token = TokenUtils.getToken(user.getId().toString(), userInfo);
-            resultMap.put("token", token);
+            String token = TokenUtils.getToken(userBo.getId().toString(), userInfo);
+            Map<String, Object> data = new HashMap<>();
+            data.put("token", token);
+            data.put("user", userInfo);
+            result.put("data", data);
         } catch (Exception e) {
-            log.error("token���������������"+e.getMessage());
+            log.error("token���������������" + e.getMessage());
+            result.put("code", ResponseCodeEnum.TOKEN_CREATE_ERROR.getCode());
+            result.put("msg", ResponseCodeEnum.TOKEN_CREATE_ERROR.getMsg());
         }
-        resultMap.putAll(userInfo);
-        return resultMap;
+        return result;
     }
 
     //������userId������������������������
@@ -117,7 +159,6 @@
         List<List<Map<String, Object>>> resultList = new ArrayList();
         Menu newMenu = new Menu();
         newMenu.setParentId(menu.getId());
-
         //������������������������������
         List<Menu> nextLevelMenus = menus.stream()
                 .filter(o -> o.getParentId().equals(menu.getId()))
@@ -141,77 +182,133 @@
     }
 
     @Override
+    @Transactional
     public Map<String, Object> addUser(User user, String token) {
-        Map<String, Object> resultMap = new HashMap<>();
+        Map<String, Object> result = new HashMap<>();
         Map<String, Object> currentUserInfo = (Map<String, Object>) TokenUtils.getUserInfoByToken(token);
 
         QueryWrapper<User> queryWrapper = new QueryWrapper<>();
-        queryWrapper.eq("account", user.getAccount());
-        if (userMapper.selectOne(queryWrapper) != null) {
-            resultMap.put("msg", "������������������");
-            return resultMap;
+        queryWrapper.eq("account", user.getAccount()).eq("is_delete", Constants.NOT_DELETE);
+        //������������������������
+        if (user.selectOne(queryWrapper) != null) {
+            result.put("code", ResponseCodeEnum.ACCOUNT_EXIST.getCode());
+            result.put("msg", ResponseCodeEnum.ACCOUNT_EXIST.getMsg());
+            return result;
         }
-        //������������������������������������
-        List<String> msgs = checkUserInfo(user);
-        if (!msgs.isEmpty()) {
-            resultMap.put("msg", msgs);
-            return resultMap;
+        //������������������
+        if (!RegexUtils.checkAccount(user.getAccount())) {
+            result.put("code", ResponseCodeEnum.ACCOUNT_INVALID.getCode());
+            result.put("msg", ResponseCodeEnum.ACCOUNT_INVALID.getMsg());
+            return result;
+        }
+        //������������������
+        if (user.getEmail() != null && !RegexUtils.checkEmail(user.getEmail())) {
+            result.put("code", ResponseCodeEnum.EMAIL_INVALID.getCode());
+            result.put("msg", ResponseCodeEnum.EMAIL_INVALID.getMsg());
+            return result;
+        }
+        //������������������
+        String password = user.getPassword();
+        //������������
+        //password = AESUtils.decrypt(password, AESKey);
+        if (!RegexUtils.checkPassword(password)) {
+            result.put("code", ResponseCodeEnum.PASSWORD_INVALID.getCode());
+            result.put("msg", ResponseCodeEnum.PASSWORD_INVALID.getMsg());
+            return result;
+        }
+        //���������������������
+        if (user.getMobile() != null && !RegexUtils.checkMobile(user.getMobile())) {
+            result.put("code", ResponseCodeEnum.MOBILE_INVALID.getCode());
+            result.put("msg", ResponseCodeEnum.MOBILE_INVALID.getMsg());
+            return result;
         }
         //������������
-        String password = MD5Utils.saltMD5(user.getPassword());
-        user.setPassword(password);
-        user.setIsAdmin(false);
+        user.setPassword(MD5Utils.saltMD5(password));
         user.setOrganizationId(Integer.parseInt(currentUserInfo.get("organizationId").toString()));
+        user.setIsAdmin(false);
         //���������������������������
         Date userExpireTime = user.getExpireTime();
         //���������������������������
-        Date expireTime = (Date) currentUserInfo.get("expireTime");
-        if (userExpireTime == null || userExpireTime.getTime() > expireTime.getTime()) {
-            user.setExpireTime(expireTime);
+        Date currentUserExpireTime = (Date) currentUserInfo.get("expireTime");
+        if (userExpireTime == null || userExpireTime.getTime() > currentUserExpireTime.getTime()) {
+            user.setExpireTime(currentUserExpireTime);
         }
         userMapper.insert(user);
-        return resultMap;
+        return result;
     }
 
     @Override
-    public Map<String, Object> deleteUser(int userId, String token) {
-        Map<String, Object> resultMap = new HashMap<>();
-        Map<String, Object> currentUserInfo = (Map<String, Object>) TokenUtils.getUserInfoByToken(token);
-        if (!(boolean) currentUserInfo.get("isAdmin")) {
-            resultMap.put("msg", "������������");
-        } else {
-            User user = new User();
-            user.setId(userId);
-            user.setIsDelete(Constants.DELETE);
-            userMapper.updateById(user);
-        }
-        return resultMap;
+    @Transactional
+    public void deleteUser(User user) {
+        //������������������
+        user.setIsDelete(Constants.DELETE);
+        //������������������������������
+        UpdateWrapper<UserGroup> updateWrapper = new UpdateWrapper<>();
+        updateWrapper.set("is_delete", Constants.DELETE).eq("user_id", user.getId());
+        userGroupMapper.update(null, updateWrapper);
     }
 
     @Override
+    @Transactional
     public Map<String, Object> updateUser(User user, String token) {
-        Map<String, Object> resultMap = new HashMap<>();
+        Map<String, Object> result = new HashMap<>();
         Map<String, Object> currentUserInfo = (Map<String, Object>) TokenUtils.getUserInfoByToken(token);
-        if (!(boolean) currentUserInfo.get("isAdmin")) {
-            resultMap.put("msg", "������������");
-            return resultMap;
-        }
+
         QueryWrapper<User> queryWrapper = new QueryWrapper<>();
-        queryWrapper.eq("account", user.getAccount());
+        queryWrapper.eq("id", user.getId()).eq("is_delete", Constants.NOT_DELETE);
+        if (userMapper.selectOne(queryWrapper) == null) {
+            result.put("code", ResponseCodeEnum.ACCOUNT_IS_DELETE.getCode());
+            result.put("msg", ResponseCodeEnum.ACCOUNT_IS_DELETE.getMsg());
+            return result;
+        }
+        queryWrapper.clear();
+        queryWrapper.eq("account", user.getAccount()).eq("is_delete", Constants.NOT_DELETE);
+        //������������������������
         if (userMapper.selectOne(queryWrapper) != null) {
-            resultMap.put("msg", "���������������");
-            return resultMap;
+            result.put("code", ResponseCodeEnum.ACCOUNT_EXIST.getCode());
+            result.put("msg", ResponseCodeEnum.ACCOUNT_EXIST.getMsg());
+            return result;
         }
-        //������������������������������������
-        List<String> msgs = checkUserInfo(user);
-        if (!msgs.isEmpty()) {
-            resultMap.put("msg", msgs);
-            return resultMap;
+        //������������������
+        if (!RegexUtils.checkAccount(user.getAccount())) {
+            result.put("code", ResponseCodeEnum.ACCOUNT_INVALID.getCode());
+            result.put("msg", ResponseCodeEnum.ACCOUNT_INVALID.getMsg());
+            return result;
         }
-        //������Md5������
-        user.setPassword(MD5Utils.saltMD5(user.getPassword()));
+        //������������������
+        if (user.getEmail() != null && !RegexUtils.checkEmail(user.getEmail())) {
+            result.put("code", ResponseCodeEnum.EMAIL_INVALID.getCode());
+            result.put("msg", ResponseCodeEnum.EMAIL_INVALID.getMsg());
+            return result;
+        }
+        //������������������
+        String password = user.getPassword();
+        //������������
+        //password = AESUtils.decrypt(password, AESKey);
+        if (!RegexUtils.checkPassword(password)) {
+            result.put("code", ResponseCodeEnum.PASSWORD_INVALID.getCode());
+            result.put("msg", ResponseCodeEnum.PASSWORD_INVALID.getMsg());
+            return result;
+        }
+        //���������������������
+        if (user.getMobile() != null && !RegexUtils.checkMobile(user.getMobile())) {
+            result.put("code", ResponseCodeEnum.MOBILE_INVALID.getCode());
+            result.put("msg", ResponseCodeEnum.MOBILE_INVALID.getMsg());
+            return result;
+        }
+        //������������
+        user.setPassword(MD5Utils.saltMD5(password));
+        user.setOrganizationId(Integer.parseInt(currentUserInfo.get("organizationId").toString()));
+        user.setIsAdmin(false);
+        //���������������������������
+        Date userExpireTime = user.getExpireTime();
+        //���������������������������
+        Date currentUserExpireTime = (Date) currentUserInfo.get("expireTime");
+        if (userExpireTime == null || userExpireTime.getTime() > currentUserExpireTime.getTime()) {
+            user.setExpireTime(currentUserExpireTime);
+        }
         userMapper.updateById(user);
-        return resultMap;
+        return result;
     }
 
     @Override
@@ -229,31 +326,5 @@
         resultMap.put("users", users);
         return resultMap;
     }
-
-    private List<String> checkUserInfo(User user) {
-        List<String> msgs = new ArrayList<>();
-        //������������
-        if (!RegexUtils.checkAccount(user.getAccount())) {
-            msgs.add("���������������������");
-        }
-        //������������
-        if (!RegexUtils.checkPassword(user.getPassword())) {
-            msgs.add("���������������������");
-        }
-        //������������
-        if (user.getEmail() != null) {
-            if (!RegexUtils.checkEmail(user.getEmail())) {
-                msgs.add("���������������������");
-            }
-        }
-        //���������������
-        if (user.getMobile() != null) {
-            if (!RegexUtils.checkMobile(user.getMobile())) {
-                msgs.add("������������������������");
-            }
-        }
-        return msgs;
-    }
-
 
 }

--
Gitblit v1.8.0