java实现角色及菜单权限-CFANZ编程社区

在 Java 中实现角色和菜单权限管理涉及到用户角色、菜单权限和访问控制的设计。以下是一个基本的实现思路和示例代码来展示如何在 Java 应用程序中实现角色和菜单权限管理：

1. 定义数据模型

首先，定义角色（Role）、菜单（Menu）和权限（Permission）的数据模型。

import java.util.Set;

// 角色类
public class Role {
    private Long id;
    private String name;
    private Set<Permission> permissions; // 角色拥有的权限

    // Constructors, getters, and setters
}

// 菜单类
public class Menu {
    private Long id;
    private String name;
    private Set<Permission> permissions; // 菜单的权限

    // Constructors, getters, and setters
}

// 权限类
public class Permission {
    private Long id;
    private String name; // 权限名称
    private Set<Menu> menus; // 权限对应的菜单

    // Constructors, getters, and setters
}

2. 定义用户和角色关联

在用户类中，关联角色。

import java.util.Set;

// 用户类
public class User {
    private Long id;
    private String username;
    private Set<Role> roles; // 用户拥有的角色

    // Constructors, getters, and setters

    // 检查用户是否具有指定权限
    public boolean hasPermission(String permissionName) {
        for (Role role : roles) {
            for (Permission permission : role.getPermissions()) {
                if (permission.getName().equals(permissionName)) {
                    return true;
                }
            }
        }
        return false;
    }
}

3. 创建权限管理服务

创建一个服务来管理权限检查。这个服务可以用来判断用户是否有权访问某个菜单或执行某个操作。

import java.util.HashSet;
import java.util.Set;

public class PermissionService {

    // 模拟数据库中的角色和权限
    private Set<Role> roles = new HashSet<>();
    private Set<Menu> menus = new HashSet<>();

    // 添加角色
    public void addRole(Role role) {
        roles.add(role);
    }

    // 添加菜单
    public void addMenu(Menu menu) {
        menus.add(menu);
    }

    // 检查用户是否有权限访问指定菜单
    public boolean hasAccess(User user, Menu menu) {
        for (Role role : user.getRoles()) {
            if (role.getPermissions().containsAll(menu.getPermissions())) {
                return true;
            }
        }
        return false;
    }
}

4. 示例使用

下面是如何使用上述模型和服务的一个示例：

public class Main {
    public static void main(String[] args) {
        // 创建权限
        Permission readPermission = new Permission(1L, "READ", new HashSet<>());
        Permission writePermission = new Permission(2L, "WRITE", new HashSet<>());

        // 创建菜单
        Menu dashboardMenu = new Menu(1L, "Dashboard", Set.of(readPermission));
        Menu adminMenu = new Menu(2L, "Admin", Set.of(readPermission, writePermission));

        // 创建角色
        Role userRole = new Role(1L, "USER", Set.of(readPermission));
        Role adminRole = new Role(2L, "ADMIN", Set.of(readPermission, writePermission));

        // 创建用户
        User user = new User(1L, "john_doe", Set.of(userRole));
        User admin = new User(2L, "admin", Set.of(adminRole));

        // 创建权限服务
        PermissionService permissionService = new PermissionService();
        permissionService.addRole(userRole);
        permissionService.addRole(adminRole);
        permissionService.addMenu(dashboardMenu);
        permissionService.addMenu(adminMenu);

        // 检查用户访问权限
        System.out.println("User has access to Dashboard: " + permissionService.hasAccess(user, dashboardMenu));
        System.out.println("User has access to Admin: " + permissionService.hasAccess(user, adminMenu));
        System.out.println("Admin has access to Dashboard: " + permissionService.hasAccess(admin, dashboardMenu));
        System.out.println("Admin has access to Admin: " + permissionService.hasAccess(admin, adminMenu));
    }
}