Examples with DataPermission com.ruoyi.common.mybatis.annotation.DataPermission used on opensource projects

Example 1 with DataPermission

use of com.ruoyi.common.mybatis.annotation.DataPermission in project RuoYi-Cloud-Plus by JavaLionLi.

the class PlusDataPermissionHandler method findAnnotation.

private DataColumn[] findAnnotation(String mappedStatementId) {
    StringBuilder sb = new StringBuilder(mappedStatementId);
    int index = sb.lastIndexOf(".");
    String clazzName = sb.substring(0, index);
    String methodName = sb.substring(index + 1, sb.length());
    Class<?> clazz = ClassUtil.loadClass(clazzName);
    List<Method> methods = Arrays.stream(ClassUtil.getDeclaredMethods(clazz)).filter(method -> method.getName().equals(methodName)).collect(Collectors.toList());
    DataPermission dataPermission;
    // 获取方法注解
    for (Method method : methods) {
        dataPermission = dataPermissionCacheMap.get(mappedStatementId);
        if (ObjectUtil.isNotNull(dataPermission)) {
            return dataPermission.value();
        }
        if (AnnotationUtil.hasAnnotation(method, DataPermission.class)) {
            dataPermission = AnnotationUtil.getAnnotation(method, DataPermission.class);
            dataPermissionCacheMap.put(mappedStatementId, dataPermission);
            return dataPermission.value();
        }
    }
    dataPermission = dataPermissionCacheMap.get(clazz.getName());
    if (ObjectUtil.isNotNull(dataPermission)) {
        return dataPermission.value();
    }
    // 获取类注解
    if (AnnotationUtil.hasAnnotation(clazz, DataPermission.class)) {
        dataPermission = AnnotationUtil.getAnnotation(clazz, DataPermission.class);
        dataPermissionCacheMap.put(clazz.getName(), dataPermission);
        return dataPermission.value();
    }
    return null;
}