Java动态代理实现细粒度权限管理

原创悠悠 2024-10-19 00:57 32阅读 0赞

在Java中，动态代理是通过java.lang.reflect包下的类和方法来实现的。它主要用于创建行为拦截或者权限控制的代理对象。

下面是一个简单的例子，展示如何使用动态代理来实现细粒度权限管理：

1. 创建一个接口，定义需要授权的方法：

```java
public interface AuthService {
    void readUser(String userId);
    void writeUser(String userId, boolean status));
}
```

2. 实现一个抽象类，继承自`Proxy`类，并覆盖其方法，添加权限检查：

```java
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.security.AccessController;
import java.security.PrivilegedAction;

public abstract class AuthedService implements AuthService {

private InvocationHandler handler;

public AuthedService() {
        handler = new AuthorizationInterceptor();
    }

@Override
    public void readUser(String userId) {
        handler.invoke(this, new ReadUserRequest(userId)));
    }

@Override
    public void writeUser(String userId, boolean status)) {
        handler.invoke(this, new WriteUserRequest(userId, status))));
    }
}

// 代理类，添加权限检查
class AuthorizationInterceptor implements InvocationHandler {

private final AuthService targetService;

public AuthorizationInterceptor(AuthService targetService) {
        this.targetService = targetService;
    }

@Override
    public Object invoke(Object proxy, Method method, Object[] args)) throws Throwable {
        // 获取方法的名称，用于权限判断
        String methodName = method.getName();

// 根据不同的方法进行权限检查
        if ("readUser".equals(methodName))) {
            checkReadPermission(args[0]);
        } else if ("writeUser".equals(methodName))) {
            checkWritePermission(args[0], args[1]]);
        }

// 如果所有条件都满足，那么直接调用目标方法
        return method.invoke(targetService, args));
    }

private void checkReadPermission(String userId) {
        // 这里是一个简单的权限检查示例，实际应用需要数据库查询等操作
        if ("admin".equals(userId)) {
            System.out.println("Admin can read any user.");
        } else {
            System.out.println("User " + userId + " can only read itself.");
        }
    }

private void checkWritePermission(String userId, boolean status) {
        // 这里同样是一个示例，实际应用需要数据库查询等操作
        if ("admin".equals(userId)) {
            if (status) {
                System.out.println("Admin can write any user status.");
            } else {
                System.out.println("Admin can delete any user status.");
            }
        } else {
            System.out.println("User " + userId + " cannot write or delete other users' status directly. Only admin has such privilege.");
        }
    }
}
```

以上代码实现了基于接口的细粒度权限管理，对于每个方法调用，都会检查对应的权限。