2022-11-15 09:20:57 +00:00
|
|
|
package repo
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/1Panel-dev/1Panel/backend/app/model"
|
|
|
|
"github.com/1Panel-dev/1Panel/backend/global"
|
2023-02-07 10:48:32 +00:00
|
|
|
"gorm.io/gorm"
|
2022-11-15 09:20:57 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type LogRepo struct{}
|
|
|
|
|
|
|
|
type ILogRepo interface {
|
2022-11-15 10:44:50 +00:00
|
|
|
CleanLogin() error
|
2022-11-15 09:20:57 +00:00
|
|
|
CreateLoginLog(user *model.LoginLog) error
|
|
|
|
PageLoginLog(limit, offset int, opts ...DBOption) (int64, []model.LoginLog, error)
|
|
|
|
|
2023-02-07 10:48:32 +00:00
|
|
|
WithByIP(ip string) DBOption
|
|
|
|
WithByStatus(status string) DBOption
|
|
|
|
WithByGroup(group string) DBOption
|
|
|
|
WithByLikeOperation(operation string) DBOption
|
2022-11-15 10:44:50 +00:00
|
|
|
CleanOperation() error
|
2022-11-15 09:20:57 +00:00
|
|
|
CreateOperationLog(user *model.OperationLog) error
|
|
|
|
PageOperationLog(limit, offset int, opts ...DBOption) (int64, []model.OperationLog, error)
|
|
|
|
}
|
|
|
|
|
|
|
|
func NewILogRepo() ILogRepo {
|
|
|
|
return &LogRepo{}
|
|
|
|
}
|
|
|
|
|
2022-11-15 10:44:50 +00:00
|
|
|
func (u *LogRepo) CleanLogin() error {
|
|
|
|
return global.DB.Exec("delete from login_logs;").Error
|
|
|
|
}
|
|
|
|
|
2022-11-15 09:20:57 +00:00
|
|
|
func (u *LogRepo) CreateLoginLog(log *model.LoginLog) error {
|
|
|
|
return global.DB.Create(log).Error
|
|
|
|
}
|
|
|
|
|
|
|
|
func (u *LogRepo) PageLoginLog(page, size int, opts ...DBOption) (int64, []model.LoginLog, error) {
|
|
|
|
var ops []model.LoginLog
|
|
|
|
db := global.DB.Model(&model.LoginLog{})
|
|
|
|
for _, opt := range opts {
|
|
|
|
db = opt(db)
|
|
|
|
}
|
|
|
|
count := int64(0)
|
|
|
|
db = db.Count(&count)
|
|
|
|
err := db.Limit(size).Offset(size * (page - 1)).Find(&ops).Error
|
|
|
|
return count, ops, err
|
|
|
|
}
|
|
|
|
|
2022-11-15 10:44:50 +00:00
|
|
|
func (u *LogRepo) CleanOperation() error {
|
|
|
|
return global.DB.Exec("delete from operation_logs").Error
|
|
|
|
}
|
|
|
|
|
2022-11-15 09:20:57 +00:00
|
|
|
func (u *LogRepo) CreateOperationLog(log *model.OperationLog) error {
|
|
|
|
return global.DB.Create(log).Error
|
|
|
|
}
|
|
|
|
|
|
|
|
func (u *LogRepo) PageOperationLog(page, size int, opts ...DBOption) (int64, []model.OperationLog, error) {
|
|
|
|
var ops []model.OperationLog
|
|
|
|
db := global.DB.Model(&model.OperationLog{})
|
|
|
|
for _, opt := range opts {
|
|
|
|
db = opt(db)
|
|
|
|
}
|
|
|
|
count := int64(0)
|
|
|
|
db = db.Count(&count)
|
|
|
|
err := db.Limit(size).Offset(size * (page - 1)).Find(&ops).Error
|
|
|
|
return count, ops, err
|
|
|
|
}
|
2023-02-07 10:48:32 +00:00
|
|
|
|
|
|
|
func (c *LogRepo) WithByStatus(status string) DBOption {
|
|
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
|
|
if len(status) == 0 {
|
|
|
|
return g
|
|
|
|
}
|
|
|
|
return g.Where("status = ?", status)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func (c *LogRepo) WithByGroup(group string) DBOption {
|
|
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
|
|
if len(group) == 0 {
|
|
|
|
return g
|
|
|
|
}
|
|
|
|
return g.Where("source = ?", group)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func (c *LogRepo) WithByIP(ip string) DBOption {
|
|
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
|
|
if len(ip) == 0 {
|
|
|
|
return g
|
|
|
|
}
|
|
|
|
return g.Where("ip LIKE ?", "%"+ip+"%")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
func (c *LogRepo) WithByLikeOperation(operation string) DBOption {
|
|
|
|
return func(g *gorm.DB) *gorm.DB {
|
|
|
|
if len(operation) == 0 {
|
|
|
|
return g
|
|
|
|
}
|
|
|
|
infoStr := "%" + operation + "%"
|
|
|
|
return g.Where("detail_zh LIKE ? OR detail_en LIKE ?", infoStr, infoStr)
|
|
|
|
}
|
|
|
|
}
|