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

Added stats csv read methods and test

Browse Source
This commit is contained in:
Edwardro22
2017-07-24 02:27:03 +03:00
parent c6254f1ebb
commit fbacb250b6
12 changed files with 247 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
engine/tp_reader.go
View File

@@ -53,6 +53,7 @@ type TpReader struct {
users map[string]*UserProfile
aliases map[string]*Alias
resLimits map[string]*utils.TPResourceLimit
stats map[string]*utils.TPStats
revDests,
revAliases,
acntActionPlans map[string][]string
@@ -124,6 +125,7 @@ func (tpr *TpReader) Init() {
tpr.aliases = make(map[string]*Alias)
tpr.derivedChargers = make(map[string]*utils.DerivedChargers)
tpr.resLimits = make(map[string]*utils.TPResourceLimit)
tpr.stats = make(map[string]*utils.TPStats)
tpr.revDests = make(map[string][]string)
tpr.revAliases = make(map[string][]string)
tpr.acntActionPlans = make(map[string][]string)
@@ -1603,6 +1605,23 @@ func (tpr *TpReader) LoadResourceLimits() error {
return tpr.LoadResourceLimitsFiltered("")
}
func (tpr *TpReader) LoadStatsFiltered(tag string) error {
tps, err := tpr.lr.GetTPStats(tpr.tpid, tag)
if err != nil {
return err
}
mapSTs := make(map[string]*utils.TPStats)
for _, st := range tps {
mapSTs[st.ID] = st
}
tpr.stats = mapSTs
return nil
}
func (tpr *TpReader) LoadStats() error {
return tpr.LoadStatsFiltered("")
}
func (tpr *TpReader) LoadAll() (err error) {
if err = tpr.LoadDestinations(); err != nil && err.Error() != utils.NotFoundCaps {
return
@@ -1655,6 +1674,9 @@ func (tpr *TpReader) LoadAll() (err error) {
if err = tpr.LoadResourceLimits(); err != nil && err.Error() != utils.NotFoundCaps {
return
}
if err = tpr.LoadStats(); err != nil && err.Error() != utils.NotFoundCaps {
return
}
return nil
}