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 | 260 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 260 insertions(+), 0 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 24a141e..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,9 +1,31 @@ package com.moral.api.service.impl; +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.stream.Collectors; + +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +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.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; /** @@ -17,4 +39,242 @@ @Service public class UserServiceImpl extends ServiceImpl<UserMapper, User> implements UserService { + @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(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); + User user = userMapper.selectOne(queryWrapper); + String msg; + if (user == null) { + msg = "������������������"; + } else { + 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("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, 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", "������������"); + return resultMap; + } + QueryWrapper<User> queryWrapper = new QueryWrapper<>(); + queryWrapper.eq("account", user.getAccount()); + userMapper.selectOne(queryWrapper); + 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(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("1"); + userMapper.updateById(user); + } + return resultMap; + } + + @Override + public Map<String, Object> updateUser(User user, 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", "������������"); + return resultMap; + } + QueryWrapper<User> queryWrapper = new QueryWrapper<>(); + queryWrapper.eq("account", user.getAccount()); + 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 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; + } + if (!parameters.containsKey("userId")) { + parameters.put("orgId", currentUserInfo.get("organizationId")); + } + List<Map<String, Object>> users = userMapper.selectUsers(parameters); + 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