权限接口

spring5-admin/src/main/java/com/yaozhitech/spring5/controller/ResourceController.java

@@ -14,7 +14,8 @@ import org.springframework.web.bind.annotation.RestController;
  * @since 2019-12-15
  */
 @RestController
-@RequestMapping("/resource-domain")
+@RequestMapping("/resource")
 public class ResourceController {
 
+	
 }

spring5-admin/src/main/java/com/yaozhitech/spring5/controller/RoleController.java

@@ -1,10 +1,20 @@
 package com.yaozhitech.spring5.controller;
 
 
-import org.springframework.web.bind.annotation.RequestMapping;
+import java.util.List;
 
+import org.springframework.util.CollectionUtils;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
+import com.yaozhitech.spring5.model.ResourceDomain;
+import com.yaozhitech.spring5.model.RoleDomain;
+import com.yaozhitech.spring5.service.UserService;
+
+import lombok.AllArgsConstructor;
+
 /**
  * <p>
  * 用户角色 前端控制器
@@ -14,7 +24,34 @@ import org.springframework.web.bind.annotation.RestController;
  * @since 2019-12-15
  */
 @RestController
-@RequestMapping("/role-domain")
+@AllArgsConstructor
+@RequestMapping("/role")
 public class RoleController {
 
+	private final UserService userService;
+	
+	@GetMapping(value="/user/{uid}")
+	public List<RoleDomain> getRolesByUserId(@PathVariable Integer uid) {
+		return userService.getRolesByUserId(uid);
+	}
+	
+	/**
+	 * 是否有权限
+	 * @param url
+	 * @param uid
+	 * @return
+	 */
+	@GetMapping(value="/hasPermission")
+	public Boolean hasPermission(String url, Integer uid) {
+		List<ResourceDomain> resourceDomains = userService.getApiResourcesByUserId(uid);
+		if (!CollectionUtils.isEmpty(resourceDomains)) {
+			for (ResourceDomain resourceDomain : resourceDomains) {
+				if (resourceDomain.getUrl().equalsIgnoreCase(url)) {
+					return Boolean.TRUE;
+				}
+			}
+		}
+		return Boolean.FALSE;
+	}
+
 }

spring5-admin/src/main/java/com/yaozhitech/spring5/controller/UserController.java

@@ -3,7 +3,6 @@ package com.yaozhitech.spring5.controller;
 
 import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.RequestMapping;
-
 import org.springframework.web.bind.annotation.RestController;
 
 import com.yaozhitech.spring5.model.UserDomain;
@@ -30,4 +29,5 @@ public class UserController {
 	public UserDomain getById(@PathVariable Integer id) {
 		return userService.getById(id);
 	}
+	
 }

spring5-admin/src/main/java/com/yaozhitech/spring5/model/ResourceDomain.java

@@ -1,14 +1,15 @@
 package com.yaozhitech.spring5.model;
 
-import com.baomidou.mybatisplus.annotation.TableName;
+import java.io.Serializable;
+import java.time.LocalDateTime;
+
 import com.baomidou.mybatisplus.annotation.IdType;
 import com.baomidou.mybatisplus.annotation.TableId;
-import java.time.LocalDateTime;
-import java.io.Serializable;
+import com.baomidou.mybatisplus.annotation.TableName;
+
 import io.swagger.annotations.ApiModel;
 import io.swagger.annotations.ApiModelProperty;
 import lombok.Data;
-import lombok.EqualsAndHashCode;
 import lombok.experimental.Accessors;
 
 /**
@@ -20,7 +21,6 @@ import lombok.experimental.Accessors;
  * @since 2019-12-15
  */
 @Data
-@EqualsAndHashCode(callSuper = false)
 @Accessors(chain = true)
 @TableName("sys_resource")
 @ApiModel(value="ResourceDomain对象", description="权限")
@@ -61,5 +61,35 @@ public class ResourceDomain implements Serializable {
 
     private Integer isactive;
 
+	@Override
+	public boolean equals(Object obj) {
+		if (this == obj)
+			return true;
+		if (obj == null)
+			return false;
+		if (getClass() != obj.getClass())
+			return false;
+		ResourceDomain other = (ResourceDomain) obj;
+		if (id == null) {
+			if (other.id != null)
+				return false;
+		} else if (!id.equals(other.id))
+			return false;
+		if (resourceUuid == null) {
+			if (other.resourceUuid != null)
+				return false;
+		} else if (!resourceUuid.equals(other.resourceUuid))
+			return false;
+		return true;
+	}
+
+	@Override
+	public int hashCode() {
+		final int prime = 31;
+		int result = 1;
+		result = prime * result + ((id == null) ? 0 : id.hashCode());
+		result = prime * result + ((resourceUuid == null) ? 0 : resourceUuid.hashCode());
+		return result;
+	}
 
 }

spring5-admin/src/main/java/com/yaozhitech/spring5/service/ResourceService.java

@@ -13,4 +13,5 @@ import com.yaozhitech.spring5.model.ResourceDomain;
  */
 public interface ResourceService extends IService<ResourceDomain> {
 
+	ResourceDomain getByUuid(String uuid);
 }

spring5-admin/src/main/java/com/yaozhitech/spring5/service/RoleService.java

@@ -1,6 +1,9 @@
 package com.yaozhitech.spring5.service;
 
+import java.util.List;
+
 import com.baomidou.mybatisplus.extension.service.IService;
+import com.yaozhitech.spring5.model.ResourceDomain;
 import com.yaozhitech.spring5.model.RoleDomain;
 
 /**
@@ -13,4 +16,13 @@ import com.yaozhitech.spring5.model.RoleDomain;
  */
 public interface RoleService extends IService<RoleDomain> {
 
+	public RoleDomain getByUuid(String uuid);
+	
+	/**
+	 * 
+	 * @param roleUuid
+	 * @param resourceType	资源类型
+	 * @return
+	 */
+	public List<ResourceDomain> getResourcesByRoleUuid(String roleUuid, Integer resourceType);
 }

spring5-admin/src/main/java/com/yaozhitech/spring5/service/UserService.java

@@ -1,6 +1,10 @@
 package com.yaozhitech.spring5.service;
 
+import java.util.List;
+
 import com.baomidou.mybatisplus.extension.service.IService;
+import com.yaozhitech.spring5.model.ResourceDomain;
+import com.yaozhitech.spring5.model.RoleDomain;
 import com.yaozhitech.spring5.model.UserDomain;
 
 /**
@@ -13,4 +17,26 @@ import com.yaozhitech.spring5.model.UserDomain;
  */
 public interface UserService extends IService<UserDomain> {
 
+	/**
+	 * 用户的角色
+	 * @param userId
+	 * @return
+	 */
+	List<RoleDomain> getRolesByUserId(Integer userId);
+	
+	List<ResourceDomain> getResourcesByUserId(Integer userId);
+	
+	/**
+	 * 用户的API级权限
+	 * @param userId
+	 * @return
+	 */
+	List<ResourceDomain> getApiResourcesByUserId(Integer userId);
+	
+	/**
+	 * 用户的菜单及权限
+	 * @param userId
+	 * @return
+	 */
+	List<ResourceDomain> getMenuResourcesByUserId(Integer userId);
 }

spring5-admin/src/main/java/com/yaozhitech/spring5/service/impl/ResourceServiceImpl.java

@@ -2,6 +2,7 @@ package com.yaozhitech.spring5.service.impl;
 
 import org.springframework.stereotype.Service;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.yaozhitech.spring5.mapper.ResourceMapper;
 import com.yaozhitech.spring5.model.ResourceDomain;
@@ -18,4 +19,9 @@ import com.yaozhitech.spring5.service.ResourceService;
 @Service
 public class ResourceServiceImpl extends ServiceImpl<ResourceMapper, ResourceDomain> implements ResourceService {
 
+	@Override
+	public ResourceDomain getByUuid(String uuid) {
+		return baseMapper.selectOne(new QueryWrapper<ResourceDomain>().eq("resource_uuid", uuid));
+	}
+
 }

spring5-admin/src/main/java/com/yaozhitech/spring5/service/impl/RoleServiceImpl.java

@@ -1,10 +1,21 @@
 package com.yaozhitech.spring5.service.impl;
 
+import java.util.ArrayList;
+import java.util.List;
+import java.util.stream.Collectors;
+
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.util.CollectionUtils;
 
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.yaozhitech.spring5.mapper.RoleMapper;
+import com.yaozhitech.spring5.mapper.RoleRelResourceMapper;
+import com.yaozhitech.spring5.model.ResourceDomain;
 import com.yaozhitech.spring5.model.RoleDomain;
+import com.yaozhitech.spring5.model.RoleRelResourceDomain;
+import com.yaozhitech.spring5.service.ResourceService;
 import com.yaozhitech.spring5.service.RoleService;
 
 /**
@@ -17,5 +28,31 @@ import com.yaozhitech.spring5.service.RoleService;
  */
 @Service
 public class RoleServiceImpl extends ServiceImpl<RoleMapper, RoleDomain> implements RoleService {
+	
+	@Autowired
+	private RoleRelResourceMapper roleRelResourceMapper;
+	
+	@Autowired
+	private ResourceService resourceService;
+	
+	@Override
+	public RoleDomain getByUuid(String uuid) {
+		return baseMapper.selectOne(new QueryWrapper<RoleDomain>().eq("role_uuid", uuid));
+	}
+
+	@Override
+	public List<ResourceDomain> getResourcesByRoleUuid(String roleUuid, Integer resourceType) {
+		List<RoleRelResourceDomain> list = roleRelResourceMapper.selectList(
+				new QueryWrapper<RoleRelResourceDomain>()
+					.eq("role_uuid", roleUuid));
+		List<ResourceDomain> resultList = new ArrayList<ResourceDomain>();
+		if (!CollectionUtils.isEmpty(list)) {
+			resultList = list.stream().map(roleRelResource -> {
+				return resourceService.getByUuid(roleRelResource.getResourceUuid());
+			}).collect(Collectors.toList());
+//			.stream().filter(resource -> resource.getType() == resourceType).collect(Collectors.toList());
+		}
+		return resultList;
+	}
 
 }

spring5-admin/src/main/java/com/yaozhitech/spring5/service/impl/UserServiceImpl.java

@@ -1,10 +1,23 @@
 package com.yaozhitech.spring5.service.impl;
 
+import java.util.ArrayList;
+import java.util.List;
+import java.util.stream.Collectors;
+
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.util.CollectionUtils;
 
+import com.baomidou.mybatisplus.core.conditions.Wrapper;
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.yaozhitech.spring5.mapper.UserMapper;
+import com.yaozhitech.spring5.mapper.UserRelRoleMapper;
+import com.yaozhitech.spring5.model.ResourceDomain;
+import com.yaozhitech.spring5.model.RoleDomain;
 import com.yaozhitech.spring5.model.UserDomain;
+import com.yaozhitech.spring5.model.UserRelRoleDomain;
+import com.yaozhitech.spring5.service.RoleService;
 import com.yaozhitech.spring5.service.UserService;
 
 /**
@@ -18,4 +31,54 @@ import com.yaozhitech.spring5.service.UserService;
 @Service
 public class UserServiceImpl extends ServiceImpl<UserMapper, UserDomain> implements UserService {
 
+	@Autowired
+	private UserRelRoleMapper userRelRoleMapper;
+	
+	@Autowired
+	private RoleService roleService;
+	
+	@Override
+	public List<RoleDomain> getRolesByUserId(Integer userId) {
+		UserDomain userDomain = baseMapper.selectById(userId);
+		
+		Wrapper<UserRelRoleDomain> queryWrapper = new QueryWrapper<UserRelRoleDomain>().eq("user_uuid", userDomain.getUserUuid());
+		List<UserRelRoleDomain> list = userRelRoleMapper.selectList(queryWrapper);
+		List<RoleDomain> result = new ArrayList<RoleDomain>();
+		if (!CollectionUtils.isEmpty(list)) {
+			result = list.stream().map(userRelRole -> {
+				return roleService.getByUuid(userRelRole.getRoleUuid());
+			}).collect(Collectors.toList());
+		}
+		return result;
+	}
+
+	@Override
+	public List<ResourceDomain> getApiResourcesByUserId(Integer userId) {
+		return getResourcesByUserId(userId, 1);
+	}
+
+	@Override
+	public List<ResourceDomain> getMenuResourcesByUserId(Integer userId) {
+		return getResourcesByUserId(userId, 0);
+	}
+
+	@Override
+	public List<ResourceDomain> getResourcesByUserId(Integer userId) {
+		// TODO Auto-generated method stub
+		return null;
+	}
+	
+	private List<ResourceDomain> getResourcesByUserId(Integer userId, Integer type) {
+		UserDomain userDomain = baseMapper.selectById(userId);
+		
+		Wrapper<UserRelRoleDomain> queryWrapper = new QueryWrapper<UserRelRoleDomain>().eq("user_uuid", userDomain.getUserUuid());
+		List<UserRelRoleDomain> list = userRelRoleMapper.selectList(queryWrapper);
+		
+		List<ResourceDomain> result = new ArrayList<ResourceDomain>();
+		if (!CollectionUtils.isEmpty(list)) {
+			list.forEach(userRelRole -> result.addAll(roleService.getResourcesByRoleUuid(userRelRole.getRoleUuid(), type)));
+		}
+		return result.stream().distinct().collect(Collectors.toList());
+	}
+	
 }