1.4.5.2

EdgeAPI/internal/db/models/log_dao.go

@@ -0,0 +1,188 @@
+package models
+
+import (
+	"encoding/json"
+	dbutils "github.com/TeaOSLab/EdgeAPI/internal/db/utils"
+	"github.com/TeaOSLab/EdgeAPI/internal/errors"
+	"github.com/TeaOSLab/EdgeAPI/internal/utils"
+	"github.com/TeaOSLab/EdgeCommon/pkg/langs"
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+	"github.com/iwind/TeaGo/types"
+	timeutil "github.com/iwind/TeaGo/utils/time"
+	"regexp"
+	"strings"
+	"time"
+)
+
+type LogDAO dbs.DAO
+
+func NewLogDAO() *LogDAO {
+	return dbs.NewDAO(&LogDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeLogs",
+			Model:  new(Log),
+			PkName: "id",
+		},
+	}).(*LogDAO)
+}
+
+var SharedLogDAO *LogDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedLogDAO = NewLogDAO()
+	})
+}
+
+// CreateLog 创建管理员日志
+func (this *LogDAO) CreateLog(tx *dbs.Tx, adminType string, adminId int64, level string, description string, action string, ip string, langMessageCode langs.MessageCode, langMessageArgs []any) error {
+	var op = NewLogOperator()
+	op.Level = level
+	op.Description = utils.LimitString(description, 1000)
+	op.Action = action
+	op.Ip = ip
+	op.Type = adminType
+
+	switch adminType {
+	case "admin":
+		op.AdminId = adminId
+	case "user":
+		op.UserId = adminId
+	case "provider":
+		op.ProviderId = adminId
+	}
+
+	// i18n
+	op.LangMessageCode = langMessageCode
+	if len(langMessageArgs) > 0 {
+		langMessageArgsJSON, err := json.Marshal(langMessageArgs)
+		if err != nil {
+			return err
+		}
+		op.LangMessageArgs = langMessageArgsJSON
+	}
+
+	op.Day = timeutil.Format("Ymd")
+	op.Type = LogTypeAdmin
+	err := this.Save(tx, op)
+	return err
+}
+
+// CountLogs 计算所有日志数量
+func (this *LogDAO) CountLogs(tx *dbs.Tx, dayFrom string, dayTo string, keyword string, userType string, level string) (int64, error) {
+	dayFrom = this.formatDay(dayFrom)
+	dayTo = this.formatDay(dayTo)
+
+	var query = this.Query(tx)
+
+	if len(dayFrom) > 0 {
+		query.Gte("day", dayFrom)
+	}
+	if len(dayTo) > 0 {
+		query.Lte("day", dayTo)
+	}
+	if len(keyword) > 0 {
+		query.Where("(description LIKE :keyword OR ip LIKE :keyword OR action LIKE :keyword)").
+			Param("keyword", dbutils.QuoteLike(keyword))
+	}
+	if len(level) > 0 {
+		query.Attr("level", level)
+	}
+
+	// 用户类型
+	switch userType {
+	case "admin":
+		query.Where("adminId>0")
+	case "user":
+		query.Where("userId>0")
+	}
+
+	return query.Count()
+}
+
+// ListLogs 列出单页日志
+func (this *LogDAO) ListLogs(tx *dbs.Tx, offset int64, size int64, dayFrom string, dayTo string, keyword string, userType string, level string) (result []*Log, err error) {
+	dayFrom = this.formatDay(dayFrom)
+	dayTo = this.formatDay(dayTo)
+
+	var query = this.Query(tx)
+	if len(dayFrom) > 0 {
+		query.Gte("day", dayFrom)
+	}
+	if len(dayTo) > 0 {
+		query.Lte("day", dayTo)
+	}
+	if len(keyword) > 0 {
+		query.Where("(description LIKE :keyword OR ip LIKE :keyword OR action LIKE :keyword)").
+			Param("keyword", dbutils.QuoteLike(keyword))
+	}
+
+	if len(level) > 0 {
+		query.Attr("level", level)
+	}
+
+	// 用户类型
+	switch userType {
+	case "admin":
+		query.Where("adminId>0")
+	case "user":
+		query.Where("userId>0")
+	}
+
+	_, err = query.
+		Offset(offset).
+		Limit(size).
+		Slice(&result).
+		DescPk().
+		FindAll()
+	return
+}
+
+// DeleteLogPermanently 物理删除日志
+func (this *LogDAO) DeleteLogPermanently(tx *dbs.Tx, logId int64) error {
+	if logId <= 0 {
+		return errors.New("invalid logId")
+	}
+	_, err := this.Delete(tx, logId)
+	return err
+}
+
+// DeleteAllLogsPermanently 物理删除所有日志
+func (this *LogDAO) DeleteAllLogsPermanently(tx *dbs.Tx) error {
+	_, err := this.Query(tx).
+		Delete()
+	return err
+}
+
+// DeleteLogsPermanentlyBeforeDays 物理删除某些天之前的日志
+func (this *LogDAO) DeleteLogsPermanentlyBeforeDays(tx *dbs.Tx, days int) error {
+	if days <= 0 {
+		days = 0
+	}
+	untilDay := timeutil.Format("Ymd", time.Now().AddDate(0, 0, -days))
+	_, err := this.Query(tx).
+		Lte("day", untilDay).
+		Delete()
+	return err
+}
+
+// SumLogsSize 计算当前日志容量大小
+func (this *LogDAO) SumLogsSize() (int64, error) {
+	col, err := this.Instance.FindCol(0, "SELECT DATA_LENGTH FROM information_schema.TABLES WHERE TABLE_SCHEMA=? AND TABLE_NAME=? LIMIT 1", this.Instance.Name(), this.Table)
+	if err != nil {
+		return 0, err
+	}
+	return types.Int64(col), nil
+}
+
+// 格式化日期
+func (this *LogDAO) formatDay(day string) string {
+	if !regexp.MustCompile(`^\d{4}-\d{2}-\d{2}$`).MatchString(day) {
+		return ""
+	}
+	day = strings.ReplaceAll(day, "-", "")
+	return day
+}