DESKTOP-72TV0V4\caoxiaozhu
ecb885ee5e
- main.go: 添加 Swagger 文档、初始化默认管理员 - 认证模块: 完善用户角色管理 - 新增工具模块: tool_handler, tool_repo, tool_service, tool model - 更新 go.mod 依赖 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
57 lines
1.4 KiB
Go
57 lines
1.4 KiB
Go
package model
|
|
|
|
import (
|
|
"encoding/json"
|
|
"time"
|
|
)
|
|
|
|
// PermissionLevel 权限级别
|
|
type PermissionLevel int
|
|
|
|
const (
|
|
PermissionRead PermissionLevel = iota + 1
|
|
PermissionWrite
|
|
PermissionExecute
|
|
PermissionAdmin
|
|
)
|
|
|
|
// Role 角色
|
|
type Role struct {
|
|
ID string `json:"id" gorm:"primaryKey"`
|
|
Name string `json:"name" gorm:"uniqueIndex"`
|
|
Permissions string `json:"permissions" gorm:"type:text"` // 存储 JSON 格式的权限数组
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// User 用户
|
|
type User struct {
|
|
ID string `json:"id" gorm:"primaryKey"`
|
|
Username string `json:"username" gorm:"uniqueIndex;size:50;not null"`
|
|
Password string `json:"-" gorm:"not null"`
|
|
Email string `json:"email" gorm:"index"`
|
|
RoleID string `json:"role_id" gorm:"size:50;not null"`
|
|
Role *Role `json:"role,omitempty" gorm:"foreignKey:RoleID"`
|
|
IsActive bool `json:"is_active" gorm:"default:true"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
// HasPermission 检查是否有权限
|
|
func (u *User) HasPermission(level PermissionLevel) bool {
|
|
if u.Role == nil {
|
|
return false
|
|
}
|
|
// 解析 JSON 格式的权限
|
|
var perms []int
|
|
if err := json.Unmarshal([]byte(u.Role.Permissions), &perms); err != nil {
|
|
return false
|
|
}
|
|
for _, p := range perms {
|
|
if PermissionLevel(p) >= level {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|