67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
|
package entity
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"go.mongodb.org/mongo-driver/v2/bson"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Permissions 權限映射表
|
||
|
|
type Permissions map[string]int
|
||
|
|
|
||
|
|
// Role 角色實體
|
||
|
|
type Role struct {
|
||
|
|
ID bson.ObjectID `bson:"_id,omitempty" json:"id"`
|
||
|
|
ClientID string `bson:"client_id" json:"client_id"`
|
||
|
|
UID string `bson:"uid" json:"uid"`
|
||
|
|
Name string `bson:"name" json:"name"`
|
||
|
|
Status int `bson:"status" json:"status"`
|
||
|
|
Permissions Permissions `bson:"permissions" json:"permissions"`
|
||
|
|
CreateTime time.Time `bson:"create_time" json:"create_time"`
|
||
|
|
UpdateTime time.Time `bson:"update_time" json:"update_time"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// CollectionName 返回集合名稱
|
||
|
|
func (r *Role) CollectionName() string {
|
||
|
|
return "roles"
|
||
|
|
}
|
||
|
|
|
||
|
|
// // Validate 驗證角色數據
|
||
|
|
//
|
||
|
|
// func (r *Role) Validate() error {
|
||
|
|
// if r.ClientID == "" {
|
||
|
|
// return mongo.WriteError{Code: 400, Message: "client_id is required"}
|
||
|
|
// }
|
||
|
|
// if r.Name == "" {
|
||
|
|
// return mongo.WriteError{Code: 400, Message: "role name is required"}
|
||
|
|
// }
|
||
|
|
// return nil
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
// // HasPermission 檢查是否有指定權限
|
||
|
|
//
|
||
|
|
// func (r *Role) HasPermission(key string) bool {
|
||
|
|
// if !r.IsActive() {
|
||
|
|
// return false
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
// permission, exists := r.Permissions[key]
|
||
|
|
// return exists && permission == 1 // 1 表示有權限
|
||
|
|
// }
|
||
|
|
//
|
||
|
|
|
||
|
|
// AddPermission 添加權限
|
||
|
|
func (r *Role) AddPermission(key string) {
|
||
|
|
if r.Permissions == nil {
|
||
|
|
r.Permissions = make(Permissions)
|
||
|
|
}
|
||
|
|
r.Permissions[key] = 1
|
||
|
|
}
|
||
|
|
|
||
|
|
// RemovePermission 移除權限
|
||
|
|
func (r *Role) RemovePermission(key string) {
|
||
|
|
if r.Permissions != nil {
|
||
|
|
delete(r.Permissions, key)
|
||
|
|
}
|
||
|
|
}
|