Golang如何使用代理模式实现权限控制

package main

import "fmt"

// FileManager 定义文件操作接口
type FileManager interface {
    ReadFile(filename string) string
    WriteFile(filename, content string) bool
}

// RealFileManager 真实的文件管理器
type RealFileManager struct{}

func (r *RealFileManager) ReadFile(filename string) string {
    return fmt.Sprintf("读取文件内容: %s", filename)
}

func (r *RealFileManager) WriteFile(filename, content string) bool {
    fmt.Printf("写入文件: %s, 内容: %s\n", filename, content)
    return true
}

// SecureFileManager 代理：带权限控制的文件管理器
type SecureFileManager struct {
    realManager *RealFileManager
    userRole    string // 用户角色：guest、user、admin
}

func NewSecureFileManager(role string) *SecureFileManager {
    return &SecureFileManager{
        realManager: &RealFileManager{},
        userRole:    role,
    }
}

func (s *SecureFileManager) ReadFile(filename string) string {
    if s.userRole == "guest" || s.userRole == "user" || s.userRole == "admin" {
        fmt.Printf("[%s] 正在尝试读取文件: %s\n", s.userRole, filename)
        return s.realManager.ReadFile(filename)
    }
    fmt.Printf("拒绝读取：用户权限不足 [%s]\n", s.userRole)
    return ""
}

func (s *SecureFileManager) WriteFile(filename, content string) bool {
    if s.userRole == "admin" {
        fmt.Printf("[%s] 正在写入文件: %s\n", s.userRole, filename)
        return s.realManager.WriteFile(filename, content)
    }
    fmt.Printf("拒绝写入：仅管理员可修改文件 [%s]\n", s.userRole)
    return false
}

// 示例使用
func main() {
    // 普通用户只能读，不能写
    userProxy := NewSecureFileManager("user")
    <strong>fmt.Println(userProxy.ReadFile("config.txt"))</strong>
    userProxy.WriteFile("config.txt", "new data")

    fmt.Println("---")

    // 管理员拥有全部权限
    adminProxy := NewSecureFileManager("admin")
    <strong>fmt.Println(adminProxy.ReadFile("secret.txt"))</strong>
    adminProxy.WriteFile("secret.txt", "top secret")
}