package com.moral.api.service.impl;
|
|
import java.util.ArrayList;
|
import java.util.Comparator;
|
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.TokenUtils;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
|
/**
|
* <p>
|
* 用户表 服务实现类
|
* </p>
|
*
|
* @author moral
|
* @since 2021-03-09
|
*/
|
@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);
|
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, 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;
|
}
|
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", "添加失败，账户名已存在");
|
}
|
return resultMap;
|
}
|
|
@Override
|
public Map<String, Object> deleteUser(Integer userId, Integer currentUserId) {
|
Map<String, Object> resultMap = new HashMap<>();
|
User currentUser = userMapper.selectById(currentUserId);
|
if (!currentUser.getIsAdmin()) {
|
resultMap.put("flag", false);
|
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) {
|
Map<String, Object> resultMap = new HashMap<>();
|
User currentUser = userMapper.selectById(currentUserId);
|
if (!currentUser.getIsAdmin()) {
|
resultMap.put("flag", false);
|
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", "修改失败，账户已存在");
|
}
|
return resultMap;
|
}
|
|
@Override
|
public List<User> getUsersByOrgId(Integer currentUserId) {
|
User currentUser = userMapper.selectById(currentUserId);
|
if (!currentUser.getIsAdmin()) {
|
return null;
|
}
|
Map<String, Object> queryMap = new HashMap<>();
|
queryMap.put("organizationId", currentUser.getOrganizationId());
|
return userMapper.selectByMap(queryMap);
|
}
|
|
@Override
|
public User getUserById(Integer userId, Integer currentUserId) {
|
User currentUser = userMapper.selectById(currentUserId);
|
if (!currentUser.getIsAdmin()) {
|
return null;
|
}
|
return userMapper.selectById(userId);
|
}
|
}
|
