管理端全部功能跑通

EdgeAPI/internal/db/models/httpdns_runtime_log_dao.go

@@ -0,0 +1,108 @@
+package models
+
+import (
+	"strconv"
+	"time"
+
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/iwind/TeaGo/Tea"
+	"github.com/iwind/TeaGo/dbs"
+)
+
+type HTTPDNSRuntimeLogDAO dbs.DAO
+
+func NewHTTPDNSRuntimeLogDAO() *HTTPDNSRuntimeLogDAO {
+	return dbs.NewDAO(&HTTPDNSRuntimeLogDAO{
+		DAOObject: dbs.DAOObject{
+			DB:     Tea.Env,
+			Table:  "edgeHTTPDNSRuntimeLogs",
+			Model:  new(HTTPDNSRuntimeLog),
+			PkName: "id",
+		},
+	}).(*HTTPDNSRuntimeLogDAO)
+}
+
+var SharedHTTPDNSRuntimeLogDAO *HTTPDNSRuntimeLogDAO
+
+func init() {
+	dbs.OnReady(func() {
+		SharedHTTPDNSRuntimeLogDAO = NewHTTPDNSRuntimeLogDAO()
+	})
+}
+
+func (this *HTTPDNSRuntimeLogDAO) CreateLog(tx *dbs.Tx, log *HTTPDNSRuntimeLog) error {
+	lastLog, err := this.Query(tx).
+		Result("id", "clusterId", "nodeId", "level", "type", "module", "description", "createdAt").
+		DescPk().
+		Find()
+	if err != nil {
+		return err
+	}
+	if lastLog != nil {
+		nodeLog := lastLog.(*HTTPDNSRuntimeLog)
+		if nodeLog.ClusterId == log.ClusterId &&
+			nodeLog.NodeId == log.NodeId &&
+			nodeLog.Level == log.Level &&
+			nodeLog.Type == log.Type &&
+			nodeLog.Module == log.Module &&
+			nodeLog.Description == log.Description &&
+			time.Now().Unix()-int64(nodeLog.CreatedAt) < 1800 {
+			
+			count := log.Count
+			if count <= 0 {
+				count = 1
+			}
+
+			return this.Query(tx).
+				Pk(nodeLog.Id).
+				Set("count", dbs.SQL("count+"+strconv.FormatInt(count, 10))).
+				UpdateQuickly()
+		}
+	}
+
+	var op = NewHTTPDNSRuntimeLogOperator()
+	op.ClusterId = log.ClusterId
+	op.NodeId = log.NodeId
+	op.Level = log.Level
+	op.Type = log.Type
+	op.Module = log.Module
+	op.Description = log.Description
+	op.Count = log.Count
+	op.RequestId = log.RequestId
+	op.CreatedAt = log.CreatedAt
+	op.Day = log.Day
+	return this.Save(tx, op)
+}
+
+func (this *HTTPDNSRuntimeLogDAO) BuildListQuery(tx *dbs.Tx, day string, clusterId int64, nodeId int64, level string, keyword string) *dbs.Query {
+	query := this.Query(tx).DescPk()
+	if len(day) > 0 {
+		query = query.Attr("day", day)
+	}
+	if clusterId > 0 {
+		query = query.Attr("clusterId", clusterId)
+	}
+	if nodeId > 0 {
+		query = query.Attr("nodeId", nodeId)
+	}
+	if len(level) > 0 {
+		query = query.Attr("level", level)
+	}
+	if len(keyword) > 0 {
+		query = query.Where("(type LIKE :kw OR module LIKE :kw OR description LIKE :kw OR requestId LIKE :kw)").Param("kw", "%"+keyword+"%")
+	}
+	return query
+}
+
+func (this *HTTPDNSRuntimeLogDAO) CountLogs(tx *dbs.Tx, day string, clusterId int64, nodeId int64, level string, keyword string) (int64, error) {
+	return this.BuildListQuery(tx, day, clusterId, nodeId, level, keyword).Count()
+}
+
+func (this *HTTPDNSRuntimeLogDAO) ListLogs(tx *dbs.Tx, day string, clusterId int64, nodeId int64, level string, keyword string, offset int64, size int64) (result []*HTTPDNSRuntimeLog, err error) {
+	_, err = this.BuildListQuery(tx, day, clusterId, nodeId, level, keyword).
+		Offset(offset).
+		Limit(size).
+		Slice(&result).
+		FindAll()
+	return
+}