109 lines
2.8 KiB
Go
109 lines
2.8 KiB
Go
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
|
|
}
|