uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-24 08:38:45 +05:00
Code Issues Packages Projects Releases Wiki Activity

StatsV1.GetQueueIDs API, add integration tests for StatS

Browse Source
This commit is contained in:
DanB
2017-09-25 14:28:19 +02:00
parent c3110af3db
commit e0898fe7bc
5 changed files with 69 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
engine/tp_reader.go
View File

@@ -34,6 +34,7 @@ type TpReader struct {
tpid string
timezone string
dataStorage DataDB
dm *DataManager
lr LoadReader
actions map[string][]*Action
actionPlans map[string]*ActionPlan
@@ -69,6 +70,7 @@ func NewTpReader(db DataDB, lr LoadReader, tpid, timezone string) *TpReader {
tpid: tpid,
timezone: timezone,
dataStorage: db,
dm: NewDataManager(db),
lr: lr,
}
tpr.Init()
@@ -2010,8 +2012,16 @@ func (tpr *TpReader) WriteToDatabase(flush, verbose, disable_reverse bool) (err
log.Print("StatQueues:")
}
for _, sqTntID := range tpr.statQueues {
if err = tpr.dataStorage.SetStoredStatQueue(&StoredStatQueue{Tenant: sqTntID.Tenant, ID: sqTntID.ID,
SQMetrics: make(map[string][]byte)}); err != nil {
sq := &StatQueue{Tenant: sqTntID.Tenant, ID: sqTntID.ID,
SQMetrics: make(map[string]StatMetric)}
for _, metricID := range tpr.sqProfiles[sqTntID.Tenant][sqTntID.ID].Metrics {
if metric, err := NewStatMetric(metricID); err != nil {
return err
} else {
sq.SQMetrics[metricID] = metric
}
}
if err = tpr.dm.SetStatQueue(sq); err != nil {
return
}
if verbose {