在 Go（Golang）中实现 RBAC（Role-Based Access Control，基于角色的访问控制） 权限管理，可以通过定义角色、权限和用户的结构体，结合合适的逻辑来控制用户的权限。下面是一个简单的 RBAC 权限管理系统的实现示例。
角色是系统中的一种身份，它定义了用户可以执行的操作。每个角色可以包含一个或多个权限。
Life is like a journey
package main
import "fmt"
// Role 结构体，表示一个角色
type Role struct {
Name string
Permissions []string
}
权限是指用户可以在系统中执行的操作，比如查看数据、修改数据、删除数据等。可以把权限表示为字符串，或使用常量进行定义。
// 权限常量定义
const (
PermissionRead = "read"
PermissionWrite = "write"
PermissionDelete = "delete"
PermissionAdmin = "admin"
)
每个用户可以拥有多个角色。用户的角色决定了他们所拥有的权限。
// User 结构体，表示一个用户
type User struct {
Name string
Roles []Role
}
每个角色都有一组与之关联的权限。这里我们创建角色，并赋予它们相应的权限。
// 创建角色并赋予权限
func createRoles() []Role {
return []Role{
{
Name: "admin",
Permissions: []string{PermissionRead, PermissionWrite, PermissionDelete, PermissionAdmin},
},
{
Name: "manager",
Permissions: []string{PermissionRead, PermissionWrite},
},
{
Name: "employee",
Permissions: []string{PermissionRead},
},
}
}
用户可以拥有一个或多个角色。通过将用户分配给特定角色，用户将拥有该角色所关联的权限。
// 创建用户并为其分配角色
func createUser(name string, roles []Role) User {
return User{
Name: name,
Roles: roles,
}
}
验证用户是否具有某个权限。我们需要遍历用户的所有角色，并检查角色是否具有该权限。
// 检查用户是否具有某个权限
func hasPermission(user User, permission string) bool {
for _, role := range user.Roles {
for _, perm := range role.Permissions {
if perm == permission {
return true
}
}
}
return false
}
最后，我们可以通过创建角色、用户并进行权限检查来测试我们的 RBAC 系统。
func main() {
// 创建角色
roles := createRoles()
// 创建用户并分配角色
user1 := createUser("Alice", []Role{roles[0]}) // admin 角色
user2 := createUser("Bob", []Role{roles[1]}) // manager 角色
user3 := createUser("Charlie", []Role{roles[2]}) // employee 角色
// 权限检查
fmt.Println("Does Alice have read permission?", hasPermission(user1, PermissionRead)) // true
fmt.Println("Does Alice have delete permission?", hasPermission(user1, PermissionDelete)) // true
fmt.Println("Does Bob have write permission?", hasPermission(user2, PermissionWrite)) // true
fmt.Println("Does Bob have delete permission?", hasPermission(user2, PermissionDelete)) // false
fmt.Println("Does Charlie have read permission?", hasPermission(user3, PermissionRead)) // true
fmt.Println("Does Charlie have write permission?", hasPermission(user3, PermissionWrite)) // false
}
Does Alice have read permission? true
Does Alice have delete permission? true
Does Bob have write permission? true
Does Bob have delete permission? false
Does Charlie have read permission? true
Does Charlie have write permission? false
这种设计将权限控制从用户身上转移到角色上，从而简化了权限管理。RBAC（基于角色的访问控制）允许灵活地管理复杂的权限系统，并在多个用户和角色之间复用权限逻辑。
manager继承employee的权限，或让admin继承manager的权限。这个简单的RBAC实现提供了一个基础的权限管理系统，可以根据实际需求扩展和完善。