mirror of
https://gitee.com/goploy/goploy.git
synced 2024-12-05 21:48:39 +08:00
68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package model
|
|
|
|
import (
|
|
sq "github.com/Masterminds/squirrel"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
const permissionTable = "`permission`"
|
|
|
|
type Permission struct {
|
|
ID int64 `json:"id"`
|
|
PID int64 `json:"pid"`
|
|
Name string `json:"name"`
|
|
Sort int `json:"sort"`
|
|
Description string `json:"description"`
|
|
InsertTime string `json:"insertTime,omitempty"`
|
|
UpdateTime string `json:"updateTime,omitempty"`
|
|
}
|
|
|
|
type Permissions []Permission
|
|
|
|
func (p Permission) GetList() (Permissions, error) {
|
|
rows, err := sq.
|
|
Select("id, pid, name, sort, description").
|
|
From(permissionTable).
|
|
RunWith(DB).
|
|
Query()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
permissions := Permissions{}
|
|
for rows.Next() {
|
|
var permission Permission
|
|
|
|
if err := rows.Scan(&permission.ID, &permission.PID, &permission.Name, &permission.Sort, &permission.Description); err != nil {
|
|
return nil, err
|
|
}
|
|
permissions = append(permissions, permission)
|
|
}
|
|
|
|
return permissions, nil
|
|
}
|
|
|
|
func (p Permission) GetIDs() (map[int64]struct{}, error) {
|
|
var permissionIDs string
|
|
err := sq.Select("GROUP_CONCAT(id)").
|
|
From(permissionTable).
|
|
RunWith(DB).
|
|
QueryRow().
|
|
Scan(&permissionIDs)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ids := map[int64]struct{}{}
|
|
for _, permissionID := range strings.Split(permissionIDs, ",") {
|
|
v, err := strconv.ParseInt(permissionID, 10, 64)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ids[v] = struct{}{}
|
|
}
|
|
return ids, nil
|
|
}
|