Golang如何使用代理模式进行权限控制_Golang代理模式权限控制实践详解

type DocumentEditor interface {
    Edit(content string) error
}

type RealDocumentEditor struct{}

func (r *RealDocumentEditor) Edit(content string) error {
    fmt.Println("文档已更新为:", content)
    return nil
}

type ProtectedDocumentEditor struct {
    realEditor *RealDocumentEditor
    username   string
}

func NewProtectedEditor(username string) *ProtectedDocumentEditor {
    return &ProtectedDocumentEditor{
        realEditor: &RealDocumentEditor{},
        username:   username,
    }
}

func (p *ProtectedDocumentEditor) Edit(content string) error {
    if p.username != "admin" {
        return fmt.Errorf("用户 %s 没有编辑权限", p.username)
    }
    return p.realEditor.Edit(content)
}

func main() {
    editor1 := NewProtectedEditor("alice")
    err := editor1.Edit("机密信息")
    if err != nil {
        fmt.Println("编辑失败:", err)
    }

    editor2 := NewProtectedEditor("admin")
    err = editor2.Edit("重要更新")
    if err != nil {
        fmt.Println("编辑失败:", err)
    } else {
        fmt.Println("编辑成功")
    }
}

编辑失败: 用户 alice 没有编辑权限
文档已更新为: 重要更新
编辑成功

func hasEditPermission(username string) bool {
    // 可对接数据库或配置中心
    return username == "admin"
}