moral.git - Gitblit

			@@ -1,11 +1,7 @@
			package com.moral.api.service.impl;

			import java.util.ArrayList;
			import java.util.List;
			import java.util.Map;
			import java.util.Objects;

			import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
			import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
			import com.moral.api.entity.UserGroup;
			import com.moral.api.mapper.GroupMapper;
			@@ -20,10 +16,7 @@
			import org.springframework.beans.factory.annotation.Autowired;
			import org.springframework.stereotype.Service;
			import org.springframework.transaction.annotation.Transactional;
			import org.springframework.web.context.request.RequestContextHolder;
			import org.springframework.web.context.request.ServletRequestAttributes;

			import javax.servlet.http.HttpServletRequest;
			import org.springframework.util.ObjectUtils;

			/**
			* <p>
			@@ -45,44 +38,34 @@
			@Autowired
			private GroupMapper groupMapper;

			@Autowired
			private OperationLogUtils operationLogUtils;

			@Override
			@Transactional
			public void allotGroups(Map<String, Object> parameters) {
			HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();
			String token = request.getHeader("token");

			Object o = parameters.get("groupId");
			Integer userId = Integer.parseInt(parameters.get("userId").toString());
			List<Integer> groupIds = (ArrayList) parameters.get("groupIds");
			Map<String, Object> currentUserInfo = (Map<String, Object>) TokenUtils.getUserInfoByToken(token);
			Map<String, Object> currentUserInfo = (Map<String, Object>) TokenUtils.getUserInfo();
			Map<String, Object> orgInfo = (Map<String, Object>) currentUserInfo.get("organization");

			UpdateWrapper<UserGroup> deleteWrapper = new UpdateWrapper<>();
			deleteWrapper.eq("user_id", userId);
			userGroupMapper.delete(deleteWrapper);
			List<String> groups = new ArrayList<>();
			if (groupIds != null && !groupIds.isEmpty()) {
			groupIds.forEach(groupId -> {
			groups.add(groupMapper.selectById(groupId).getGroupName());
			UserGroup userGroup = new UserGroup();
			userGroup.setUserId(userId);
			userGroup.setGroupId(groupId);
			userGroup.setOrganizationId((Integer) orgInfo.get("id"));
			userGroupMapper.insert(userGroup);
			});

			if (!ObjectUtils.isEmpty(o)) {
			int groupId = Integer.parseInt(o.toString());
			String groupName = groupMapper.selectById(groupId).getGroupName();
			UserGroup userGroup = new UserGroup();
			userGroup.setUserId(userId);
			userGroup.setGroupId(groupId);
			userGroup.setOrganizationId((Integer) orgInfo.get("id"));
			//user_group表insert
			userGroupMapper.insert(userGroup);
			//日志
			String account = userMapper.selectById((Integer) parameters.get("userId")).getAccount();
			String content = "给用户：" + account + "分配了组：" + groupName;
			operationLogUtils.insertLog(content, Constants.UPDATE_OPERATE_TYPE);
			}

			//日志
			String account = userMapper.selectById((Integer) parameters.get("userId")).getAccount();
			String content = "给用户：" + account + "分配了组：" + groups.toString();
			OperationLogUtils.insertLog(request, content, Constants.UPDATE_OPERATE_TYPE);
			}

			@Override
			public List<Integer> getGroupIds(Integer userId) {
			QueryWrapper<UserGroup> queryWrapper = new QueryWrapper<>();
			queryWrapper.eq("user_id", userId).eq("is_delete", Constants.NOT_DELETE);
			List<UserGroup> userGroups = userGroupMapper.selectList(queryWrapper);
			List<Integer> groupIds = new ArrayList<>();
			userGroups.forEach(userGroup -> groupIds.add(userGroup.getGroupId()));
			return groupIds;
			}
			}