moral.git - Gitblit

			@@ -1,26 +1,29 @@
			package com.moral.api.service.impl;

			import io.swagger.models.auth.In;

			import java.util.ArrayList;
			import java.util.Date;
			import java.util.Comparator;
			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.TokenUtils;

			import org.springframework.beans.factory.annotation.Autowired;
			import org.springframework.beans.factory.annotation.Value;
			import org.springframework.stereotype.Service;

			/**
			@@ -37,32 +40,120 @@
			@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);
			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) {
			Map<String, Object> resultMap = new HashMap<>();