From 5e23b5966d393f1eb94bca82498e09a67274e321 Mon Sep 17 00:00:00 2001 From: kaiyu <404897439@qq.com> Date: Mon, 15 Mar 2021 08:24:16 +0800 Subject: [PATCH] Merge branch 'dev' of http://blit.7drlb.com:8888/r/moral into dev --- screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java | 264 +++++++++++++++++++++++++++++++++++++++------------- 1 files changed, 197 insertions(+), 67 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 2b73e59..8950238 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 @@ -1,26 +1,31 @@ package com.moral.api.service.impl; -import io.swagger.models.auth.In; - import java.util.ArrayList; +import java.util.Comparator; import java.util.Date; import java.util.HashMap; +import java.util.LinkedHashMap; import java.util.List; import java.util.Map; -import java.util.Set; -import java.util.UUID; +import java.util.stream.Collectors; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; -import com.moral.api.entity.Group; import com.moral.api.entity.Menu; import com.moral.api.entity.User; +import com.moral.api.mapper.GroupMapper; +import com.moral.api.mapper.MenuMapper; +import com.moral.api.mapper.OrganizationMapper; import com.moral.api.mapper.UserMapper; import com.moral.api.service.UserService; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; -import com.moral.redis.RedisUtil; +import com.moral.util.AESUtils; +import com.moral.util.MD5Utils; +import com.moral.util.RegexUtils; +import com.moral.util.TokenUtils; import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Service; /** @@ -37,114 +42,239 @@ @Autowired private UserMapper userMapper; + @Autowired + private GroupMapper groupMapper; + + @Autowired + private MenuMapper menuMapper; + + @Autowired + private OrganizationMapper organizationMapper; + + @Value("${AES.KEY}") + private String AESKey; + @Override - public Map<String, Object> login(String account, String password) { - Map<String, Object> resultMap = new HashMap<>(); + public Map<String, Object> login(Map<String, Object> parameters) { + Map<String, Object> resultMap = new LinkedHashMap<>(); + //������������ + String account = parameters.get("account").toString(); + String password = parameters.get("password").toString(); + //������ + /*account = AESUtils.decrypt(account, AESKey); + password = AESUtils.decrypt(password, AESKey);*/ QueryWrapper<User> queryWrapper = new QueryWrapper<>(); - queryWrapper.eq("account", account).eq("password", password).eq("is_Delete", "0"); + //������������ + queryWrapper.eq("account", account); User user = userMapper.selectOne(queryWrapper); String msg; - String token = null; if (user == null) { - msg = "������������������������"; - } else if (user.getExpireTime() != null && user.getExpireTime().getTime() < System.currentTimeMillis()) { - msg = "���������������������"; + msg = "������������������"; } else { - msg = "������������"; - token = user.getId() + "-" + UUID.randomUUID().toString().replaceAll("-", ""); - Set<Group> groups = userMapper.selectUserGroup(user.getId()); - Set<Menu> menus = userMapper.selectUserMenu(user.getId()); - user.setGroups(groups); - user.setMenus(menus); - RedisUtil.set(token, user, 1296000); + if (!MD5Utils.saltMD5Verify(password, user.getPassword())) {//������������ + msg = "������������"; + } else if ("1".equals(user.getIsDelete())) { + msg = "������������������"; + } else if (user.getExpireTime() != null && user.getExpireTime().getTime() < System.currentTimeMillis()) { + msg = "���������������������"; + } else { + int userId = user.getId(); + //������������������ + Map<String, Object> orgMap = organizationMapper.selectOrganizationById(user.getOrganizationId()); + String locationCode = orgMap.get(orgMap.get("location_level").toString()).toString(); + //������������������ + Map<String, Object> userInfo = new LinkedHashMap<>(); + userInfo.put("userId", userId); + userInfo.put("account", user.getAccount()); + userInfo.put("userName", user.getUserName()); + userInfo.put("organizationId", user.getOrganizationId()); + userInfo.put("locationCode", locationCode); + userInfo.put("expireTime", user.getExpireTime()); + userInfo.put("isAdmin", user.getIsAdmin()); + List<Map<String, Object>> groups = groupMapper.selectUserGroup(userId); + userInfo.put("groups", groups); + userInfo.putAll(getMenus(userId)); + //������token������������redis + String token = TokenUtils.getToken(user.getId().toString(), userInfo); + resultMap.put("token", token); + resultMap.putAll(userInfo); + return resultMap; + } } - resultMap.put("token", token); resultMap.put("msg", msg); return resultMap; } + //������userId������������������������ + private Map<String, Object> getMenus(int userId) { + List<Menu> allMenus = menuMapper.selectUserMenu(userId); + Map<String, Object> resultMap = new LinkedHashMap<>(); + //��������������� + List<Menu> oneMenu = allMenus.stream() + .filter(o -> o.getParentId().equals(0)) + .sorted(Comparator.comparing(Menu::getOrder)) + .collect(Collectors.toList()); + + List<Map<String, Object>> newList = new ArrayList<>(); + //������������������ + oneMenu.forEach(o -> { + Map<String, Object> menuMap = new LinkedHashMap<>(); + menuMap.put("id", o.getId()); + menuMap.put("name", o.getName()); + menuMap.put("url", o.getUrl()); + menuMap.put("icon", o.getIcon()); + menuMap.put("menus", getMenusByRecursion(o, allMenus)); + newList.add(menuMap); + }); + resultMap.put("menus", newList); + return resultMap; + } + + //������������������������������������ + private List<Map<String, Object>> getMenusByRecursion(Menu menu, List<Menu> menus) { + 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())) + .collect(Collectors.toList()); + List<Map<String, Object>> list = new ArrayList<>(); + if (nextLevelMenus.size() > 0) { + //��������������������������������������������������� + nextLevelMenus.forEach(o -> { + Map<String, Object> menuMap = new LinkedHashMap<>(); + menuMap.put("id", o.getId()); + menuMap.put("name", o.getName()); + menuMap.put("url", o.getUrl()); + menuMap.put("icon", o.getIcon()); + //��������������� + menuMap.put("menus", getMenusByRecursion(o, menus)); + list.add(menuMap); + }); + resultList.add(list); + } + return list; + } + @Override - public Map<String, Object> addUser(User user, Integer userId) { + public Map<String, Object> addUser(User user, String token) { Map<String, Object> resultMap = new HashMap<>(); - User currentUser = userMapper.selectById(userId); - if (!currentUser.getIsAdmin()) { - resultMap.put("flag", false); - resultMap.put("msg", "���������������������������"); + 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()); userMapper.selectOne(queryWrapper); - if (userMapper.selectOne(queryWrapper) == null) { - user.setOrganizationId(currentUser.getOrganizationId()); - user.setExpireTime(currentUser.getExpireTime()); - userMapper.insert(user); - resultMap.put("flag", true); - resultMap.put("msg", "������������"); - } else { - resultMap.put("flag", false); - resultMap.put("msg", "���������������������������������"); + if (userMapper.selectOne(queryWrapper) != null) { + resultMap.put("msg", "������������������"); + return resultMap; } + //������������������������������������ + List<String> msgs = checkUserInfo(user); + if (!msgs.isEmpty()) { + resultMap.put("msg", msgs); + return resultMap; + } + //������������ + String password = MD5Utils.saltMD5(user.getPassword()); + user.setPassword(password); + user.setIsAdmin(false); + user.setOrganizationId(Integer.parseInt(currentUserInfo.get("organizationId").toString())); + //��������������������������� + Date userExpireTime = user.getExpireTime(); + //��������������������������� + Date expireTime = (Date) currentUserInfo.get("expireTime"); + if (userExpireTime == null || userExpireTime.getTime() > expireTime.getTime()) { + user.setExpireTime(expireTime); + } + userMapper.insert(user); return resultMap; } @Override - public Map<String, Object> deleteUser(Integer userId, Integer currentUserId) { + public Map<String, Object> deleteUser(int userId, String token) { Map<String, Object> resultMap = new HashMap<>(); - User currentUser = userMapper.selectById(currentUserId); - if (!currentUser.getIsAdmin()) { - resultMap.put("flag", false); - resultMap.put("msg", "���������������������������"); + 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("1"); userMapper.updateById(user); - resultMap.put("flag", true); - resultMap.put("msg", "������������"); } return resultMap; } @Override - public Map<String, Object> updateUser(User user, Integer currentUserId) { + public Map<String, Object> updateUser(User user, String token) { Map<String, Object> resultMap = new HashMap<>(); - User currentUser = userMapper.selectById(currentUserId); - if (!currentUser.getIsAdmin()) { - resultMap.put("flag", false); - resultMap.put("msg", "���������������������������"); + 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()); - if (userMapper.selectOne(queryWrapper) == null) { - userMapper.updateById(user); - resultMap.put("flag", true); - resultMap.put("msg", "������������"); - } else { - resultMap.put("flag", false); - resultMap.put("msg", "������������������������������"); + if (userMapper.selectOne(queryWrapper) != null) { + resultMap.put("msg", "���������������"); + return resultMap; } + //������������������������������������ + List<String> msgs = checkUserInfo(user); + if (!msgs.isEmpty()) { + resultMap.put("msg", msgs); + return resultMap; + } + //������Md5������ + user.setPassword(MD5Utils.saltMD5(user.getPassword())); + userMapper.updateById(user); return resultMap; } @Override - public List<User> getUsersByOrgId(Integer currentUserId) { - User currentUser = userMapper.selectById(currentUserId); - if (!currentUser.getIsAdmin()) { - return null; + public Map<String, Object> getUsers(Map<String, Object> parameters) { + Map<String, Object> resultMap = new HashMap<>(); + Map<String, Object> currentUserInfo = (Map<String, Object>) TokenUtils.getUserInfoByToken(parameters.get("token").toString()); + if (!(boolean) currentUserInfo.get("isAdmin")) { + resultMap.put("msg", "������������"); + return resultMap; } - Map<String, Object> queryMap = new HashMap<>(); - queryMap.put("organizationId", currentUser.getOrganizationId()); - return userMapper.selectByMap(queryMap); + if (!parameters.containsKey("userId")) { + parameters.put("orgId", currentUserInfo.get("organizationId")); + } + List<Map<String, Object>> users = userMapper.selectUsers(parameters); + resultMap.put("users", users); + return resultMap; } - @Override - public User getUserById(Integer userId, Integer currentUserId) { - User currentUser = userMapper.selectById(currentUserId); - if (!currentUser.getIsAdmin()) { - return null; + private List<String> checkUserInfo(User user) { + List<String> msgs = new ArrayList<>(); + //������������ + if (!RegexUtils.checkAccount(user.getAccount())) { + msgs.add("���������������������"); } - return userMapper.selectById(userId); + //������������ + 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