81 lines
2.4 KiB
Go
81 lines
2.4 KiB
Go
package repositories
|
|
|
|
import (
|
|
"log"
|
|
|
|
"github.com/niangaodev/art-code/config"
|
|
"github.com/niangaodev/art-code/models"
|
|
)
|
|
|
|
// GetPermissions 获取所有权限
|
|
func GetPermissions() ([]models.Permission, error) {
|
|
query := "SELECT id, name, resource, action, created_at, updated_at FROM permissions"
|
|
rows, err := config.DB.Query(query)
|
|
if err != nil {
|
|
log.Printf("Error getting permissions: %v", err)
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var permissions []models.Permission
|
|
for rows.Next() {
|
|
var permission models.Permission
|
|
if err := rows.Scan(&permission.ID, &permission.Name, &permission.Resource, &permission.Action, &permission.CreatedAt, &permission.UpdatedAt); err != nil {
|
|
log.Printf("Error scanning permission: %v", err)
|
|
continue
|
|
}
|
|
permissions = append(permissions, permission)
|
|
}
|
|
|
|
return permissions, nil
|
|
}
|
|
|
|
// GetPermissionsByRoleID 获取指定角色的权限
|
|
func GetPermissionsByRoleID(roleID uint) ([]models.Permission, error) {
|
|
query := `
|
|
SELECT p.id, p.name, p.resource, p.action, p.created_at, p.updated_at
|
|
FROM permissions p
|
|
JOIN role_permissions rp ON p.id = rp.permission_id
|
|
WHERE rp.role_id = ?
|
|
`
|
|
rows, err := config.DB.Query(query, roleID)
|
|
if err != nil {
|
|
log.Printf("Error getting permissions by role id: %v", err)
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
var permissions []models.Permission
|
|
for rows.Next() {
|
|
var permission models.Permission
|
|
if err := rows.Scan(&permission.ID, &permission.Name, &permission.Resource, &permission.Action, &permission.CreatedAt, &permission.UpdatedAt); err != nil {
|
|
log.Printf("Error scanning permission: %v", err)
|
|
continue
|
|
}
|
|
permissions = append(permissions, permission)
|
|
}
|
|
|
|
return permissions, nil
|
|
}
|
|
|
|
// BuildPermissionResponse 构建权限响应
|
|
func BuildPermissionResponse(permission *models.Permission) *models.PermissionResponse {
|
|
return &models.PermissionResponse{
|
|
ID: permission.ID,
|
|
Name: permission.Name,
|
|
Resource: permission.Resource,
|
|
Action: permission.Action,
|
|
CreatedAt: permission.CreatedAt.Format("2006-01-02 15:04:05"),
|
|
UpdatedAt: permission.UpdatedAt.Format("2006-01-02 15:04:05"),
|
|
}
|
|
}
|
|
|
|
// BuildPermissionsResponse 构建权限列表响应
|
|
func BuildPermissionsResponse(permissions []models.Permission) []models.PermissionResponse {
|
|
var responses []models.PermissionResponse
|
|
for _, permission := range permissions {
|
|
responses = append(responses, *BuildPermissionResponse(&permission))
|
|
}
|
|
return responses
|
|
}
|