SpringAOP的自定义注解实践

Posted 2021-01-09 水滴

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了SpringAOP的自定义注解实践相关的知识，希望对你有一定的参考价值。

springaop属于spring的重要属性,在java中有相当广泛的用途,大家一般都接触过aop实现事务的管理,在xml里配好声明式事务,然后直接在service上直接加上相应注解即可,

今天我们来实现下SpringAOP的自定义注解,用来在前置通知中做下权限校验,有利于我们代码的解藕,提高复用性,增加代码B格;

话不多说,上代码,首先定义一个自定义注解

这里的参数我们并没有在使用,先随意定义个type,以后想使用时可以再赋值

再来看切面

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;

import org.apache.commons.lang.StringUtils;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.AfterReturning;
import org.aspectj.lang.annotation.AfterThrowing;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import com.jiubao.common.core.cache.redis.JedisUtil;
import com.jiubao.common.core.exception.MyException;
import com.jiubao.livenet.cache.JiuBaoCache;
import com.jiubao.livenet.constant.ServiceConstants;
import com.jiubao.livenet.enums.Limitsenum;
import com.jiubao.livenet.service.CustomerService;

/**
* 注解切面
* @author Administrator
*
*/
@Aspect
@Component
public class AuthorityAspect {

private static final Logger logger = LoggerFactory.getLogger(AuthorityAspect.class);

@Resource
private CustomerService customerService;

//Controller层切点
@Pointcut("execution (* com.jiubao.livenet.controller.*.*(..))")
public void controllerAspect() {}

/**
* 前置通知
* 只拦截带有Authority注解的controller方法
* @param joinPoint 切点
* @return
*/

//@org.springframework.stereotype.Controller *) 表示拦截controller方法 @annotation(Authority) 表示只拦截带此注解的方法

@Before("within(@org.springframework.stereotype.Controller *) && @annotation(Authority)")
public void doBefore(JoinPoint joinPoint) throws MyException {
　　logger.info("==========执行商户权限controller前置通知===============");
　 Object[] args = joinPoint.getArgs();
　　HttpServletRequest request=null;
　　for (int i = 0; i < args.length; i++) {
　　　　Object object = args[i];
　　　　if(object instanceof HttpServletRequest) {
　　　　request=(HttpServletRequest) args[i];
　　　　break;
　　　}
}
　　Map<String, Object> paramMap = new HashMap<String, Object>();
　　if(request ==null) {
　　　　return;
　　}

　　String url = request.getRequestURI().toString(); //请求的url,注意,只有方法参数里有HttpServletRequest request才能获取到request

　　/**
　　* 权限控制
　 */
JedisUtil jedis = JiuBaoCache.getJedisUtil();
String userId = jedis.getUserId(request);
paramMap.put("uid", userId); //登录人UID
boolean isAccess=false;
List<Map<String, String>> userAuthlist= customerService.getUserAuth(paramMap); //查询登录人的商户角色列表
for (int i = 0; i < userAuthlist.size(); i++) {
if(userAuthlist.get(i).get("role")!=null) {
String[] limitsArray = Limitsenum.limitsArray(Integer.valueOf(userAuthlist.get(i).get("role"))); //使用枚举得到此角色对应的URL数组
for (int a = 0; a < limitsArray.length; a++) {
if(url.contains(limitsArray[a])) { //判断是否有,如果有则跳出
isAccess=true;
break;
}
}
if(isAccess) {
break;
}
}
}

　　if (!isAccess){
　　 throw new MyException(-1,ServiceConstants.NO_ACCESS); //无权限访问,如果没有权限直接扔出个自定义异常,在那里会用response返回给前端提示
}
}

　　//全部拦截,这样的before就是全局拦截,会拦截所有方法
　　@Before("controllerAspect()")
　　public void doBefore2(JoinPoint joinPoint) {
　　　　System.out.println("==========执行全部拦截controller前置通知===============");
　　}

　　@AfterReturning(returning="rvt",pointcut="controllerAspect()") //打印所有方法的返回值
　　public Object AfterExec(JoinPoint joinPoint,Object rvt){ //pointcut是对应的注解类 rvt就是方法运行完之后要返回的值
　　　　logger.info("返回值为：" + rvt);
　　　　return rvt;
　　}

}

再看看调用过程

直接加在controller里想要限制的方法上即可