85 lines
2.5 KiB
Go
85 lines
2.5 KiB
Go
package isql
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/eryajf-world/go-ldap-admin/model"
|
|
"github.com/eryajf-world/go-ldap-admin/public/common"
|
|
"github.com/eryajf-world/go-ldap-admin/public/tools"
|
|
"github.com/eryajf-world/go-ldap-admin/svc/request"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type OperationLogService struct{}
|
|
|
|
//var Logs []model.OperationLog //全局变量多个线程需要加锁，所以每个线程自己维护一个
|
|
//处理OperationLogChan将日志记录到数据库
|
|
func (s OperationLogService) SaveOperationLogChannel(olc <-chan *model.OperationLog) {
|
|
// 只会在线程开启的时候执行一次
|
|
Logs := make([]model.OperationLog, 0)
|
|
|
|
// 一直执行--收到olc就会执行
|
|
for log := range olc {
|
|
Logs = append(Logs, *log)
|
|
// 每10条记录到数据库
|
|
if len(Logs) > 5 {
|
|
common.DB.Create(&Logs)
|
|
Logs = make([]model.OperationLog, 0)
|
|
}
|
|
}
|
|
}
|
|
|
|
// List 获取数据列表
|
|
func (s OperationLogService) List(req *request.OperationLogListReq) ([]*model.OperationLog, error) {
|
|
var list []*model.OperationLog
|
|
db := common.DB.Model(&model.OperationLog{}).Order("start_time DESC")
|
|
|
|
username := strings.TrimSpace(req.Username)
|
|
if username != "" {
|
|
db = db.Where("username LIKE ?", fmt.Sprintf("%%%s%%", username))
|
|
}
|
|
ip := strings.TrimSpace(req.Ip)
|
|
if ip != "" {
|
|
db = db.Where("ip LIKE ?", fmt.Sprintf("%%%s%%", ip))
|
|
}
|
|
path := strings.TrimSpace(req.Path)
|
|
if path != "" {
|
|
db = db.Where("path LIKE ?", fmt.Sprintf("%%%s%%", path))
|
|
}
|
|
status := req.Status
|
|
if status != 0 {
|
|
db = db.Where("status = ?", status)
|
|
}
|
|
|
|
pageReq := tools.NewPageOption(req.PageNum, req.PageSize)
|
|
err := db.Offset(pageReq.PageNum).Limit(pageReq.PageSize).Find(&list).Error
|
|
|
|
return list, err
|
|
}
|
|
|
|
// Count 获取数据总数
|
|
func (s OperationLogService) Count() (count int64, err error) {
|
|
err = common.DB.Model(&model.OperationLog{}).Count(&count).Error
|
|
return count, err
|
|
}
|
|
|
|
// 获取单个用户
|
|
func (s OperationLogService) Find(filter map[string]interface{}, data *model.OperationLog) error {
|
|
return common.DB.Where(filter).First(&data).Error
|
|
}
|
|
|
|
// Exist 判断资源是否存在
|
|
func (s OperationLogService) Exist(filter map[string]interface{}) bool {
|
|
var dataObj model.OperationLog
|
|
err := common.DB.Debug().Order("created_at DESC").Where(filter).First(&dataObj).Error
|
|
return !errors.Is(err, gorm.ErrRecordNotFound)
|
|
}
|
|
|
|
// Delete 删除资源
|
|
func (s OperationLogService) Delete(operationLogIds []uint) error {
|
|
return common.DB.Where("id IN (?)", operationLogIds).Unscoped().Delete(&model.OperationLog{}).Error
|
|
}
|