java自定义注解实现前后台参数校验 java自定义注解实现aop

转载

mob6454cc7416d1 2024-04-16 22:21:17

文章标签 java自定义注解实现前后台参数校验 java 前端数据库 ViewUI 文章分类 Java 后端开发

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

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

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

java自定义注解实现前后台参数校验 java自定义注解实现aop_前端

这里的参数我们并没有在使用,先随意定义个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里想要限制的方法上即可

java自定义注解实现前后台参数校验 java自定义注解实现aop_java_02