1Panel/core/app/service/logs.go

113 lines
3.1 KiB
Go

package service
import (
"github.com/1Panel-dev/1Panel/core/buserr"
"github.com/1Panel-dev/1Panel/core/utils/common"
geo2 "github.com/1Panel-dev/1Panel/core/utils/geo"
"github.com/gin-gonic/gin"
"github.com/1Panel-dev/1Panel/core/app/dto"
"github.com/1Panel-dev/1Panel/core/app/model"
"github.com/1Panel-dev/1Panel/core/app/repo"
"github.com/1Panel-dev/1Panel/core/global"
"github.com/1Panel-dev/1Panel/core/utils/cmd"
"github.com/jinzhu/copier"
)
type LogService struct{}
const logs = "https://resource.fit2cloud.com/installation-log.sh"
type ILogService interface {
CreateLoginLog(operation model.LoginLog) error
PageLoginLog(ctx *gin.Context, search dto.SearchLgLogWithPage) (int64, interface{}, error)
CreateOperationLog(operation *model.OperationLog) error
PageOperationLog(search dto.SearchOpLogWithPage) (int64, interface{}, error)
CleanLogs(logtype string) error
}
func NewILogService() ILogService {
return &LogService{}
}
func (u *LogService) CreateLoginLog(operation model.LoginLog) error {
return logRepo.CreateLoginLog(&operation)
}
func (u *LogService) PageLoginLog(ctx *gin.Context, req dto.SearchLgLogWithPage) (int64, interface{}, error) {
options := []global.DBOption{
repo.WithOrderBy("created_at desc"),
}
if len(req.IP) != 0 {
options = append(options, logRepo.WithByIP(req.IP))
}
if len(req.Status) != 0 {
options = append(options, repo.WithByStatus(req.Status))
}
total, ops, err := logRepo.PageLoginLog(
req.Page,
req.PageSize,
options...,
)
var dtoOps []dto.LoginLog
geoDB, _ := geo2.NewGeo()
for _, op := range ops {
var item dto.LoginLog
if err := copier.Copy(&item, &op); err != nil {
return 0, nil, buserr.WithErr("ErrTransform", err)
}
if geoDB != nil {
item.Address, _ = geo2.GetIPLocation(geoDB, item.IP, common.GetLang(ctx))
}
dtoOps = append(dtoOps, item)
}
return total, dtoOps, err
}
func (u *LogService) CreateOperationLog(operation *model.OperationLog) error {
return logRepo.CreateOperationLog(operation)
}
func (u *LogService) PageOperationLog(req dto.SearchOpLogWithPage) (int64, interface{}, error) {
options := []global.DBOption{
repo.WithOrderBy("created_at desc"),
logRepo.WithByLikeOperation(req.Operation),
}
if len(req.Source) != 0 {
options = append(options, logRepo.WithBySource(req.Source))
}
if len(req.Status) != 0 {
options = append(options, repo.WithByStatus(req.Status))
}
if len(req.Node) != 0 {
options = append(options, repo.WithByNode(req.Node))
}
total, ops, err := logRepo.PageOperationLog(
req.Page,
req.PageSize,
options...,
)
var dtoOps []dto.OperationLog
for _, op := range ops {
var item dto.OperationLog
if err := copier.Copy(&item, &op); err != nil {
return 0, nil, buserr.WithErr("ErrTransform", err)
}
dtoOps = append(dtoOps, item)
}
return total, dtoOps, err
}
func (u *LogService) CleanLogs(logtype string) error {
if logtype == "operation" {
return logRepo.CleanOperation()
}
return logRepo.CleanLogin()
}
func writeLogs(version string) {
_, _ = cmd.RunDefaultWithStdoutBashCf("curl -sfL %s | sh -s 1p upgrade %s", logs, version)
}