aop+泛型+反射 解决 若依框架分页问题 ( 实现从注解给实体类某个属性赋值)

因为项目需要，我们的数据都是与部门关联的，但是因为采用的若依框架中的分页插件在controller层使用 starePage()方法，会对第一条sql语句进行查询,我们项目需要在真正的查询表数据之前，先查询当前选择的部门，及部门下的id，所以需要将该方法放到controller层去执行，为了代码简洁，所以自定义了aop注解,上代码

首先是 泛型+反射 实现对实体类某属性赋值

 private <T>T setDeptIds(T data)throws Exception{

        /**
         * 得到类对象
         */
        Class entity = (Class) data.getClass();

        //获取指定属性
        Field field = entity.getDeclaredField("deptId");
        
        Field fields = entity.getDeclaredField("deptIds");


        field.setAccessible(true);

        fields.setAccessible(true);
        
        //获取指定属性的值
        Long deptId = (Long)(field.get(data));

        //获取部门下所有子部门
        List<Long> deptIds = ListUtils.newArrayList();

        if(deptId!=null && deptId > 0L){
            deptIds = sysDeptService.selectNormalChildrenDeptIdsByDeptId(deptId);
        }
        
        //给指定属性赋值
        fields.set(data,deptIds);

        return data;
    }

 private <T>T setDeptIds(T data)throws Exception{

        /**
         * 得到类对象
         */
        Class entity = (Class) data.getClass();

        //获取指定属性
        Field field = entity.getDeclaredField("deptId");
        
        Field fields = entity.getDeclaredField("deptIds");


        field.setAccessible(true);

        fields.setAccessible(true);
        
        //获取指定属性的值
        Long deptId = (Long)(field.get(data));

        //获取部门下所有子部门
        List<Long> deptIds = ListUtils.newArrayList();

        if(deptId!=null && deptId > 0L){
            deptIds = sysDeptService.selectNormalChildrenDeptIdsByDeptId(deptId);
        }
        
        //给指定属性赋值
        fields.set(data,deptIds);

        return data;
    }

 

然后是aop 注解 获取当前方法的参数(此处百度很久，还删掉了一些不太理解的代码才实现了)

   /**
     * 方法执行前执行
     * @param joinPoint
     */
    @Before("checkParam()")
    public <T>T doBefore(JoinPoint joinPoint)throws Exception {
        Object[] params = joinPoint.getArgs();
        if(params.length == 0){
            return null;
        }
        //获取方法，此处可将signature强转为MethodSignature
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        Method method = signature.getMethod();
        //参数注解，1维是参数，2维是注解
        Annotation[][] annotations = method.getParameterAnnotations();
        T result = null;
        for (int i = 0; i < annotations.length; i++) {
            Object param = params[i];
            Annotation[] paramAnn = annotations[i];

            result =(T)setDeptIds(param);

            //参数为空，直接下一个参数
            if(param == null || paramAnn.length == 0){
                continue;
            }


//            for (Annotation annotation : paramAnn) {
//                //这里判断当前注解是否为Test.class
//                if(annotation.annotationType().equals(DeptDataScope.class)){
//                    //校验该参数，验证一次退出该注解
//                    //TODO 校验参数
//                    break;
//                }
//            }
        }
        return result ;

    }

接下来就是使用注解了

   /**
     * 查询备件调整单管理列表
     */
    @PreAuthorize("@ss.hasPermi('adjustDocuments:documents:list')")
    @GetMapping("/list")
    @DeptDataScope()
    public TableDataInfo list(SparePartsAdjustDocuments sparePartsAdjustDocuments) {
        startPage();
        List<SparePartsAdjustDocuments> list = sparePartsAdjustDocumentsService.selectSparePartsAdjustDocumentsList(sparePartsAdjustDocuments);
        return getDataTable(list);
    }

 

ok，至此大功告成！！！ 忙了一天，虽然还有许多不足之处，但是基本需求满足了，后面慢慢优化吧。

这是aop实现类的完整代码

@Component
@Aspect
public class DeptDataScopeAop {

    @Resource
    private ISysDeptService sysDeptService;

    /**
     * 定义有一个切入点
     */
    @Pointcut("@annotation(com.*.*.aop.DeptDataScope)")
    public void checkParam() {
    }

    /**
     * 方法执行前执行
     * @param joinPoint
     */
    @Before("checkParam()")
    public <T>T doBefore(JoinPoint joinPoint)throws Exception {
        Object[] params = joinPoint.getArgs();
        if(params.length == 0){
            return null;
        }
        //获取方法，此处可将signature强转为MethodSignature
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        Method method = signature.getMethod();
        //参数注解，1维是参数，2维是注解
        Annotation[][] annotations = method.getParameterAnnotations();
        T result = null;
        for (int i = 0; i < annotations.length; i++) {
            Object param = params[i];
            Annotation[] paramAnn = annotations[i];

            result =(T)setDeptIds(param);

            //参数为空，直接下一个参数
            if(param == null || paramAnn.length == 0){
                continue;
            }


//            for (Annotation annotation : paramAnn) {
//                //这里判断当前注解是否为Test.class
//                if(annotation.annotationType().equals(DeptDataScope.class)){
//                    //校验该参数，验证一次退出该注解
//                    //TODO 校验参数
//                    break;
//                }
//            }
        }
        return result ;

    }

    @Around(value = "(execution(public * com.nongshiyun.modules.*.*(..)))", argNames = "joinPoint")
    public <T>T doAround(ProceedingJoinPoint  joinPoint) throws Exception  {

        Object[] params = joinPoint.getArgs();
        if(params.length == 0){
            return null;
        }
        //获取方法，此处可将signature强转为MethodSignature
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        Method method = signature.getMethod();
        //参数注解，1维是参数，2维是注解
        Annotation[][] annotations = method.getParameterAnnotations();
        T result = null;
        for (int i = 0; i < annotations.length; i++) {
            Object param = params[i];
            Annotation[] paramAnn = annotations[i];
            //参数为空，直接下一个参数
            if(param == null || paramAnn.length == 0){
                continue;
            }
            for (Annotation annotation : paramAnn) {
                //这里判断当前注解是否为Test.class
                if(annotation.annotationType().equals(DeptDataScope.class)){
                    //校验该参数，验证一次退出该注解
                    //TODO 校验参数
                    result =(T)setDeptIds(param);
                    break;
                }
            }
        }
        return result ;
    }

    /**
     * 在切入点return内容之后切入内容（可以用来对处理返回值做一些加工处理）
     * TODO 留着做以后处理返回值用
     * @param joinPoint
     */
    @AfterReturning(value = "checkParam()", returning = "result")
    public void doAfterReturning(JoinPoint joinPoint, Object result) {

    }


    private <T>T setDeptIds(T data)throws Exception{

        /**
         * 得到类对象
         */
        Class entity = (Class) data.getClass();

        //获取指定属性
        Field field = entity.getDeclaredField("deptId");

        Field fields = entity.getDeclaredField("deptIds");


        field.setAccessible(true);

        fields.setAccessible(true);

        //获取指定属性的值
        Long deptId = (Long)(field.get(data));

        //获取部门下所有子部门
        List<Long> deptIds = ListUtils.newArrayList();

        if(deptId!=null && deptId > 0L){
            deptIds = sysDeptService.selectNormalChildrenDeptIdsByDeptId(deptId);
        }

        //给指定属性赋值
        fields.set(data,deptIds);

        return data;
    }

}