mirror of
https://github.com/TeaOSLab/EdgeAPI.git
synced 2025-11-04 16:00:24 +08:00
102 lines
2.4 KiB
Go
102 lines
2.4 KiB
Go
package models
|
|
|
|
import (
|
|
"encoding/json"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
"github.com/iwind/TeaGo/Tea"
|
|
"github.com/iwind/TeaGo/dbs"
|
|
"github.com/iwind/TeaGo/maps"
|
|
"strconv"
|
|
)
|
|
|
|
type MetricStatDAO dbs.DAO
|
|
|
|
func NewMetricStatDAO() *MetricStatDAO {
|
|
return dbs.NewDAO(&MetricStatDAO{
|
|
DAOObject: dbs.DAOObject{
|
|
DB: Tea.Env,
|
|
Table: "edgeMetricStats",
|
|
Model: new(MetricStat),
|
|
PkName: "id",
|
|
},
|
|
}).(*MetricStatDAO)
|
|
}
|
|
|
|
var SharedMetricStatDAO *MetricStatDAO
|
|
|
|
func init() {
|
|
dbs.OnReady(func() {
|
|
SharedMetricStatDAO = NewMetricStatDAO()
|
|
})
|
|
}
|
|
|
|
// CreateStat 创建统计数据
|
|
func (this *MetricStatDAO) CreateStat(tx *dbs.Tx, hash string, clusterId int64, nodeId int64, serverId int64, itemId int64, keys []string, value float64, time string, version int32) error {
|
|
hash += "@" + strconv.FormatInt(nodeId, 10)
|
|
var keysString string
|
|
if len(keys) > 0 {
|
|
keysJSON, err := json.Marshal(keys)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
keysString = string(keysJSON)
|
|
} else {
|
|
keysString = "[]"
|
|
}
|
|
return this.Query(tx).
|
|
Param("value", value).
|
|
InsertOrUpdateQuickly(maps.Map{
|
|
"hash": hash,
|
|
"clusterId": clusterId,
|
|
"nodeId": nodeId,
|
|
"serverId": serverId,
|
|
"itemId": itemId,
|
|
"value": value,
|
|
"time": time,
|
|
"version": version,
|
|
"keys": keysString,
|
|
}, maps.Map{
|
|
"value": value,
|
|
})
|
|
}
|
|
|
|
// DeleteOldItemStats 删除以前版本的统计数据
|
|
func (this *MetricStatDAO) DeleteOldItemStats(tx *dbs.Tx, itemId int64, version int32) error {
|
|
_, err := this.Query(tx).
|
|
Attr("itemId", itemId).
|
|
Where("version<:version").
|
|
Delete()
|
|
return err
|
|
}
|
|
|
|
// DeleteItemStats 删除某个指标相关的统计数据
|
|
func (this *MetricStatDAO) DeleteItemStats(tx *dbs.Tx, itemId int64) error {
|
|
_, err := this.Query(tx).
|
|
Attr("itemId", itemId).
|
|
Delete()
|
|
return err
|
|
}
|
|
|
|
// CountItemStats 计算统计数据数量
|
|
func (this *MetricStatDAO) CountItemStats(tx *dbs.Tx, itemId int64, version int32) (int64, error) {
|
|
return this.Query(tx).
|
|
Attr("itemId", itemId).
|
|
Attr("version", version).
|
|
Count()
|
|
}
|
|
|
|
// ListItemStats 列出单页统计数据
|
|
func (this *MetricStatDAO) ListItemStats(tx *dbs.Tx, itemId int64, version int32, offset int64, size int64) (result []*MetricStat, err error) {
|
|
_, err = this.Query(tx).
|
|
Attr("itemId", itemId).
|
|
Attr("version", version).
|
|
Offset(offset).
|
|
Limit(size).
|
|
Desc("time").
|
|
Desc("serverId").
|
|
Desc("value").
|
|
Slice(&result).
|
|
FindAll()
|
|
return
|
|
}
|