package/go_process_manager
1
0
Fork 0
mirror of https://github.com/lzh-1625/go_process_manager.git synced 2025-10-05 16:06:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

1

Browse Source
This commit is contained in:
lzh
2025-06-19 11:48:41 +08:00
parent af941de7f2
commit 8a175d33aa
2 changed files with 29 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
internal/app/repository/permission.go
View File

@@ -1,12 +1,12 @@
package repository
import (
"reflect"
"errors"
"github.com/lzh-1625/go_process_manager/internal/app/constants"
"github.com/lzh-1625/go_process_manager/internal/app/model"
"github.com/lzh-1625/go_process_manager/internal/app/repository/query"
"github.com/lzh-1625/go_process_manager/log"
"gorm.io/gorm"
)
@@ -26,21 +26,22 @@ func (p *permissionRepository) GetPermssionList(account string) []model.Permissi
}
func (p *permissionRepository) EditPermssion(data model.Permission) error {
if db.Model(&model.Permission{}).Where(&model.Permission{Account: data.Account, Pid: data.Pid}).First(nil).Error == gorm.ErrRecordNotFound {
db.Omit("name").Create(&model.Permission{
per := query.Permission
if _, err := per.Where(per.Account.Eq(data.Account)).Where(per.Pid.Eq(data.Pid)).First(); errors.Is(err, gorm.ErrRecordNotFound) {
per.Create(&model.Permission{
Account: data.Account,
Pid: data.Pid,
})
}
return db.Model(&model.Permission{}).Where(&model.Permission{Account: data.Account, Pid: data.Pid}).Updates(map[string]interface{}{
"owned": data.Owned,
"start": data.Start,
"stop": data.Stop,
"terminal": data.Terminal,
"log": data.Log,
"write": data.Write,
}).Error
_, err := per.Where(per.Account.Eq(data.Account)).Where(per.Pid.Eq(data.Pid)).Updates(map[string]any{
per.Owned.ColumnName().String(): data.Owned,
per.Start.ColumnName().String(): data.Start,
per.Stop.ColumnName().String(): data.Stop,
per.Terminal.ColumnName().String(): data.Terminal,
per.Log.ColumnName().String(): data.Log,
per.Write.ColumnName().String(): data.Write,
})
return err
}
func (p *permissionRepository) GetPermission(user string, pid int) (result model.Permission) {
@@ -49,8 +50,19 @@ func (p *permissionRepository) GetPermission(user string, pid int) (result model
}
func (p *permissionRepository) GetProcessNameByPermission(user string, op constants.OprPermission) (result []string) {
query := model.PermissionPo{Account: user, Owned: true}
reflect.ValueOf(&query).Elem().FieldByName(string(op)).SetBool(true)
db.Model(&model.Permission{}).Select("name").Joins("right join process p on p.uuid = pid").Where(query).Find(&result)
tx := query.Permission.Select(query.Process.Name).RightJoin(query.Permission, query.Process.Uuid.EqCol(query.Permission.Pid)).Where(query.Permission.Account.Eq(user)).Where(query.Permission.Owned.Is(true))
switch op {
case constants.OPERATION_LOG:
tx = tx.Where(query.Permission.Log.Is(true))
case constants.OPERATION_START:
tx = tx.Where(query.Permission.Start.Is(true))
case constants.OPERATION_STOP:
tx = tx.Where(query.Permission.Stop.Is(true))
case constants.OPERATION_TERMINAL:
tx = tx.Where(query.Permission.Terminal.Is(true))
case constants.OPERATION_TERMINAL_WRITE:
tx = tx.Where(query.Permission.Write.Is(true))
}
tx.Scan(&result)
return
}

2
internal/app/repository/process.go
View File

@@ -23,7 +23,7 @@ func (p *processRepository) GetAllProcessConfig() []model.Process {
func (p *processRepository) GetProcessConfigByUser(username string) []model.Process {
result := []model.Process{}
err := query.Permission.LeftJoin(query.Process, query.Process.Uuid.EqCol(query.Permission.Pid)).
err := query.Process.LeftJoin(query.Permission, query.Process.Uuid.EqCol(query.Permission.Pid)).
Where(query.Permission.Owned.Is(true)).
Where(query.Permission.Account.Eq(username)).
Scan(&result)