From 511fbc6392d46828284bd72b47966ef217893275 Mon Sep 17 00:00:00 2001
From: jinpengyong <jpy123456>
Date: Wed, 17 Mar 2021 17:37:44 +0800
Subject: [PATCH] 登陆修改
---
screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java | 262 +++++++++++++++++++++++++++++++++++++---------------
1 files changed, 185 insertions(+), 77 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..4a1b6f2 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,33 @@
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.Organization;
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 +44,215 @@
@Autowired
private UserMapper userMapper;
+ @Autowired
+ private MenuMapper menuMapper;
+
@Override
- public Map<String, Object> login(String account, String password) {
- Map<String, Object> resultMap = new HashMap<>();
- QueryWrapper<User> queryWrapper = new QueryWrapper<>();
- queryWrapper.eq("account", account).eq("password", password).eq("is_Delete", "0");
- 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 = "���������������������";
- } 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);
- }
- resultMap.put("token", token);
- resultMap.put("msg", msg);
- return resultMap;
+ public User selectUserInfo(Map<String, Object> parameters) {
+ return userMapper.selectUserInfo(parameters);
}
@Override
- public Map<String, Object> addUser(User user, Integer userId) {
- Map<String, Object> resultMap = new HashMap<>();
- User currentUser = userMapper.selectById(userId);
- if (!currentUser.getIsAdmin()) {
- resultMap.put("flag", false);
- resultMap.put("msg", "���������������������������");
- return resultMap;
+ public Map<String, Object> login(User user) {
+ Map<String, Object> resultMap = new LinkedHashMap<>();
+ //������������������
+ Map<String, Object> userInfo = new LinkedHashMap<>();
+ List<Map<String, Object>> groups = new ArrayList<>();
+ for (Group group : user.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());
+ userInfo.put("organizationName", organization.getName());
+ userInfo.put("locationLevel", organization.getLocationLevel());
+ userInfo.put("groups", groups);
+ userInfo.putAll(getMenus(user.getId()));
+ try {
+ //������token������������redis
+ String token = TokenUtils.getToken(user.getId().toString(), userInfo);
+ resultMap.put("token", token);
+ } catch (Exception e) {
+ log.error("token���������������"+e.getMessage());
+ }
+ resultMap.putAll(userInfo);
+ 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);
+
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