mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-29 05:56:38 +08:00
ryan
106 lines
3.3 KiB
Go
106 lines
3.3 KiB
Go
package model
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/rain-kl/openflare/pkg/utils"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type NodeRequestReport struct {
|
|
ID uint `json:"id" gorm:"primaryKey"`
|
|
NodeID string `json:"node_id" gorm:"index;size:64;not null"`
|
|
WindowStartedAt time.Time `json:"window_started_at" gorm:"index"`
|
|
WindowEndedAt time.Time `json:"window_ended_at" gorm:"index"`
|
|
RequestCount int64 `json:"request_count"`
|
|
ErrorCount int64 `json:"error_count"`
|
|
UniqueVisitorCount int64 `json:"unique_visitor_count"`
|
|
StatusCodesJSON string `json:"status_codes_json" gorm:"type:text"`
|
|
TopDomainsJSON string `json:"top_domains_json" gorm:"type:text"`
|
|
SourceCountriesJSON string `json:"source_countries_json" gorm:"type:text"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
}
|
|
|
|
func (report *NodeRequestReport) GetID() uint {
|
|
return report.ID
|
|
}
|
|
|
|
func (report *NodeRequestReport) GetTime() time.Time {
|
|
return report.WindowEndedAt
|
|
}
|
|
|
|
func (report *NodeRequestReport) BeforeCreate(tx *gorm.DB) error {
|
|
return assignObservabilityID(&report.ID)
|
|
}
|
|
|
|
func (report *NodeRequestReport) Insert() error {
|
|
return DB.Create(report).Error
|
|
}
|
|
|
|
func ListNodeRequestReports(nodeID string, since time.Time, limit int) (reports []*NodeRequestReport, err error) {
|
|
rows, err := queryAcrossShards("node_request_reports", func(tx *gorm.DB) ([]*NodeRequestReport, error) {
|
|
var shardRows []*NodeRequestReport
|
|
query := tx.Order("window_ended_at desc, id desc")
|
|
if nodeID != "" {
|
|
query = query.Where("node_id = ?", nodeID)
|
|
}
|
|
if !since.IsZero() {
|
|
query = query.Where("window_ended_at >= ?", since)
|
|
}
|
|
if err := query.Find(&shardRows).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return shardRows, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return utils.SortAndLimitRecords(rows, limit), nil
|
|
}
|
|
|
|
func ListRequestReportsSince(since time.Time) (reports []*NodeRequestReport, err error) {
|
|
rows, err := queryAcrossShards("node_request_reports", func(tx *gorm.DB) ([]*NodeRequestReport, error) {
|
|
var shardRows []*NodeRequestReport
|
|
query := tx.Order("window_ended_at desc")
|
|
if !since.IsZero() {
|
|
query = query.Where("window_ended_at >= ?", since)
|
|
}
|
|
if err := query.Find(&shardRows).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return shardRows, nil
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return utils.SortAndLimitRecords(rows, 0), nil
|
|
}
|
|
|
|
func NodeRequestReportExists(db *gorm.DB, nodeID string, windowStartedAt time.Time, windowEndedAt time.Time) (bool, error) {
|
|
db = normalizeShardedDB(db)
|
|
for _, table := range observabilityShardTables("node_request_reports") {
|
|
var count int64
|
|
if err := db.Table(table).
|
|
Where("node_id = ? AND window_started_at = ? AND window_ended_at = ?", nodeID, windowStartedAt, windowEndedAt).
|
|
Limit(1).
|
|
Count(&count).Error; err != nil {
|
|
return false, err
|
|
}
|
|
if count > 0 {
|
|
return true, nil
|
|
}
|
|
}
|
|
return false, nil
|
|
}
|
|
|
|
func DeleteNodeRequestReportsBefore(db *gorm.DB, before time.Time) (int64, error) {
|
|
return deleteAcrossShards(db, "node_request_reports", &NodeRequestReport{}, func(tx *gorm.DB) *gorm.DB {
|
|
return tx.Where("window_ended_at < ?", before)
|
|
})
|
|
}
|
|
|
|
func DeleteAllNodeRequestReports(db *gorm.DB) (int64, error) {
|
|
return deleteAcrossShards(db, "node_request_reports", &NodeRequestReport{}, nil)
|
|
}
|