当前位置：首页 > article >正文

C# WinForms 控制权限到按钮级别

article 2024/10/9 14:33:51

在 C# WinForms 应用程序中，实现细粒度的权限控制，特别是到按钮级别，是确保应用程序安全性的重要手段。以下是如何在 WinForms 应用程序中实现按钮级别的权限控制的详细步骤和示例代码。

1. 设计权限数据模型

首先，需要设计一个权限数据模型，通常包括用户、角色、权限和按钮权限等表。这些表可以通过数据库来实现，也可以使用文件或其他存储方式。

示例数据库设计

sql

CREATE TABLE Users (
    UserID INT PRIMARY KEY,
    Username NVARCHAR(50),
    Password NVARCHAR(50)
);

CREATE TABLE Roles (
    RoleID INT PRIMARY KEY,
    RoleName NVARCHAR(50)
);

CREATE TABLE UserRoles (
    UserID INT,
    RoleID INT,
    FOREIGN KEY (UserID) REFERENCES Users(UserID),
    FOREIGN KEY (RoleID) REFERENCES Roles(RoleID)
);

CREATE TABLE Permissions (
    PermissionID INT PRIMARY KEY,
    PermissionName NVARCHAR(50)
);

CREATE TABLE RolePermissions (
    RoleID INT,
    PermissionID INT,
    FOREIGN KEY (RoleID) REFERENCES Roles(RoleID),
    FOREIGN KEY (PermissionID) REFERENCES Permissions(PermissionID)
);

CREATE TABLE Buttons (
    ButtonID INT PRIMARY KEY,
    ButtonName NVARCHAR(50),
    PermissionID INT,
    FOREIGN KEY (PermissionID) REFERENCES Permissions(PermissionID)
);

2. 获取用户权限

在应用程序启动时或用户登录后，需要获取当前用户的权限列表。这通常涉及到查询数据库并获取与当前用户关联的角色和权限。

示例代码

csharp

public List<string> GetUserPermissions(int userID)
{
    List<string> permissions = new List<string>();
    // 假设有一个数据库访问类 DBAccess
    using (var db = new DBAccess())
    {
        var userRoles = db.Query("SELECT RoleID FROM UserRoles WHERE UserID = @UserID", new { UserID = userID });
        foreach (var role in userRoles)
        {
            var rolePermissions = db.Query("SELECT PermissionID FROM RolePermissions WHERE RoleID = @RoleID", new { RoleID = role.RoleID });
            foreach (var permission in rolePermissions)
            {
                var button = db.QueryFirstOrDefault("SELECT ButtonName FROM Buttons WHERE PermissionID = @PermissionID", new { PermissionID = permission.PermissionID });
                if (button != null)
                {
                    permissions.Add(button.ButtonName);
                }
            }
        }
    }
    return permissions;
}

3. 控制按钮权限

在 WinForms 应用程序中，根据获取到的用户权限列表，控制按钮的可见性和可用性。

示例代码

csharp

public void SetButtonPermissions(Form form, List<string> permissions)
{
    foreach (Control control in form.Controls)
    {
        if (control is Button)
        {
            Button button = control as Button;
            if (!permissions.Contains(button.Name))
            {
                button.Visible = false; // 或者设置为不可用 button.Enabled = false;
            }
        }
        else if (control.HasChildren)
        {
            SetButtonPermissions(control as Form, permissions);
        }
    }
}

4. 在窗体加载时应用权限

在窗体的 Load 事件中调用 SetButtonPermissions 方法，以确保在窗体显示之前应用权限。

示例代码

csharp

private void MainForm_Load(object sender, EventArgs e)
{
    int userID = GetLoggedInUserID(); // 获取当前登录用户的ID
    List<string> permissions = GetUserPermissions(userID);
    SetButtonPermissions(this, permissions);
}

5. 处理权限变更

在用户的角色或权限变更时，需要更新 UI 中的权限控制。这可能涉及到重新加载窗体或动态更新按钮的可见性和可用性。

示例代码

csharp

private void UpdatePermissionsButton_Click(object sender, EventArgs e)
{
    int userID = GetLoggedInUserID(); // 获取当前登录用户的ID
    List<string> permissions = GetUserPermissions(userID);
    SetButtonPermissions(this, permissions);
}