mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-04 16:00:24 +08:00
104 lines
3.2 KiB
Go
104 lines
3.2 KiB
Go
package stats
|
|
|
|
import (
|
|
"github.com/TeaOSLab/EdgeAPI/internal/errors"
|
|
"github.com/TeaOSLab/EdgeAPI/internal/remotelogs"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
"github.com/iwind/TeaGo/Tea"
|
|
"github.com/iwind/TeaGo/dbs"
|
|
"github.com/iwind/TeaGo/maps"
|
|
timeutil "github.com/iwind/TeaGo/utils/time"
|
|
"time"
|
|
)
|
|
|
|
type ServerDomainHourlyStatDAO dbs.DAO
|
|
|
|
func init() {
|
|
dbs.OnReadyDone(func() {
|
|
// 清理数据任务
|
|
var ticker = time.NewTicker(24 * time.Hour)
|
|
go func() {
|
|
for range ticker.C {
|
|
err := SharedServerDomainHourlyStatDAO.Clean(nil, 60) // 只保留60天
|
|
if err != nil {
|
|
remotelogs.Error("ServerDomainHourlyStatDAO", "clean expired data failed: "+err.Error())
|
|
}
|
|
}
|
|
}()
|
|
})
|
|
}
|
|
|
|
func NewServerDomainHourlyStatDAO() *ServerDomainHourlyStatDAO {
|
|
return dbs.NewDAO(&ServerDomainHourlyStatDAO{
|
|
DAOObject: dbs.DAOObject{
|
|
DB: Tea.Env,
|
|
Table: "edgeServerDomainHourlyStats",
|
|
Model: new(ServerDomainHourlyStat),
|
|
PkName: "id",
|
|
},
|
|
}).(*ServerDomainHourlyStatDAO)
|
|
}
|
|
|
|
var SharedServerDomainHourlyStatDAO *ServerDomainHourlyStatDAO
|
|
|
|
func init() {
|
|
dbs.OnReady(func() {
|
|
SharedServerDomainHourlyStatDAO = NewServerDomainHourlyStatDAO()
|
|
})
|
|
}
|
|
|
|
// IncreaseHourlyStat 增加统计数据
|
|
func (this *ServerDomainHourlyStatDAO) IncreaseHourlyStat(tx *dbs.Tx, clusterId int64, nodeId int64, serverId int64, domain string, hour string, bytes int64, cachedBytes int64, countRequests int64, countCachedRequests int64) error {
|
|
if len(hour) != 10 {
|
|
return errors.New("invalid hour '" + hour + "'")
|
|
}
|
|
err := this.Query(tx).
|
|
Param("bytes", bytes).
|
|
Param("cachedBytes", cachedBytes).
|
|
Param("countRequests", countRequests).
|
|
Param("countCachedRequests", countCachedRequests).
|
|
InsertOrUpdateQuickly(maps.Map{
|
|
"clusterId": clusterId,
|
|
"nodeId": nodeId,
|
|
"serverId": serverId,
|
|
"hour": hour,
|
|
"domain": domain,
|
|
"bytes": bytes,
|
|
"cachedBytes": cachedBytes,
|
|
"countRequests": countRequests,
|
|
"countCachedRequests": countCachedRequests,
|
|
}, maps.Map{
|
|
"bytes": dbs.SQL("bytes+:bytes"),
|
|
"cachedBytes": dbs.SQL("cachedBytes+:cachedBytes"),
|
|
"countRequests": dbs.SQL("countRequests+:countRequests"),
|
|
"countCachedRequests": dbs.SQL("countCachedRequests+:countCachedRequests"),
|
|
})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// FindTopDomainStatsWithClusterId 取得一定时间内的节点排行数据
|
|
func (this *ServerDomainHourlyStatDAO) FindTopDomainStatsWithClusterId(tx *dbs.Tx, clusterId int64, hourFrom string, hourTo string) (result []*ServerDomainHourlyStat, err error) {
|
|
// TODO 节点如果已经被删除，则忽略
|
|
_, err = this.Query(tx).
|
|
Attr("clusterId", clusterId).
|
|
Between("hour", hourFrom, hourTo).
|
|
Result("domain, MIN(serverId) AS serverId, SUM(bytes) AS bytes, SUM(cachedBytes) AS cachedBytes, SUM(countRequests) AS countRequests, SUM(countCachedRequests) AS countCachedRequests").
|
|
Group("domain").
|
|
Desc("countRequests").
|
|
Slice(&result).
|
|
FindAll()
|
|
return
|
|
}
|
|
|
|
// Clean 清理历史数据
|
|
func (this *ServerDomainHourlyStatDAO) Clean(tx *dbs.Tx, days int) error {
|
|
var hour = timeutil.Format("Ymd00", time.Now().AddDate(0, 0, -days))
|
|
_, err := this.Query(tx).
|
|
Lt("hour", hour).
|
|
Delete()
|
|
return err
|
|
}
|