package com.whyc.service; import com.whyc.mapper.UserRoleMapper; import com.whyc.pojo.Role; import com.whyc.pojo.User; import com.whyc.pojo.UserRole; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import javax.annotation.Resource; import java.util.HashMap; import java.util.LinkedList; import java.util.List; import java.util.Map; /** * 用户对应角色 */ @Service public class UserRoleService { @Resource private UserRoleMapper mapper; public List getUserWithNoRole() { return mapper.getUserWithNoRole(); } public List getUserWithRole() { List userRoles = mapper.getUserWithRole(); return userRoles; } public Map> getUserWithRoleMap() { HashMap> map = new HashMap<>(); List userRoleList = new LinkedList<>(); List userRoles= mapper.getUserWithRoleMap(); for (UserRole temp:userRoles){ String[] userIds = temp.getUserIds().split(","); String[] userNames = temp.getUserNames().split(","); for (int i = 0; i < userIds.length; i++) { UserRole userRole = new UserRole(); userRole.setUser(new User(Integer.parseInt(userIds[i]),userNames[i])); userRole.setRoleId(temp.getRoleId()); userRole.setRoleName(temp.getRoleName()); userRoleList.add(userRole); } map.put(temp.getRoleName(),userRoleList); } return map; } public boolean bindingUserWithRole(int userId,int roleId) { return mapper.insert(new UserRole(userId,roleId))>0; } public boolean bindingUserWithRoleBatch(List userRoles) { return mapper.insertBatchSomeColumn(userRoles)==userRoles.size(); } }