mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
1004 lines
27 KiB
Go
1004 lines
27 KiB
Go
/*
|
|
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
|
|
Copyright (C) ITsysCOM GmbH
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>
|
|
*/
|
|
|
|
package engine
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/cgrates/birpc/context"
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
type TpReader struct {
|
|
tpid string
|
|
timezone string
|
|
dm *DataManager
|
|
lr LoadReader
|
|
resProfiles map[utils.TenantID]*utils.TPResourceProfile
|
|
sqProfiles map[utils.TenantID]*utils.TPStatProfile
|
|
trProfiles map[utils.TenantID]*utils.TPTrendsProfile
|
|
rgProfiles map[utils.TenantID]*utils.TPRankingProfile
|
|
thProfiles map[utils.TenantID]*utils.TPThresholdProfile
|
|
filters map[utils.TenantID]*utils.TPFilterProfile
|
|
routeProfiles map[utils.TenantID]*utils.TPRouteProfile
|
|
attributeProfiles map[utils.TenantID]*utils.TPAttributeProfile
|
|
chargerProfiles map[utils.TenantID]*utils.TPChargerProfile
|
|
rateProfiles map[utils.TenantID]*utils.TPRateProfile
|
|
actionProfiles map[utils.TenantID]*utils.TPActionProfile
|
|
accounts map[utils.TenantID]*utils.TPAccount
|
|
cacheConns []string
|
|
//schedulerConns []string
|
|
}
|
|
|
|
func NewTpReader(db *DBConnManager, lr LoadReader, tpid, timezone string,
|
|
cacheConns, schedulerConns []string) (*TpReader, error) {
|
|
tpr := &TpReader{
|
|
tpid: tpid,
|
|
timezone: timezone,
|
|
dm: NewDataManager(db, config.CgrConfig(), connMgr), // ToDo: add CGRConfig as parameter to the NewTpReader
|
|
lr: lr,
|
|
cacheConns: cacheConns,
|
|
//schedulerConns: schedulerConns,
|
|
}
|
|
tpr.Init()
|
|
|
|
return tpr, nil
|
|
}
|
|
|
|
func (tpr *TpReader) Init() {
|
|
tpr.resProfiles = make(map[utils.TenantID]*utils.TPResourceProfile)
|
|
tpr.sqProfiles = make(map[utils.TenantID]*utils.TPStatProfile)
|
|
tpr.rgProfiles = make(map[utils.TenantID]*utils.TPRankingProfile)
|
|
tpr.thProfiles = make(map[utils.TenantID]*utils.TPThresholdProfile)
|
|
tpr.trProfiles = make(map[utils.TenantID]*utils.TPTrendsProfile)
|
|
tpr.routeProfiles = make(map[utils.TenantID]*utils.TPRouteProfile)
|
|
tpr.attributeProfiles = make(map[utils.TenantID]*utils.TPAttributeProfile)
|
|
tpr.chargerProfiles = make(map[utils.TenantID]*utils.TPChargerProfile)
|
|
tpr.rateProfiles = make(map[utils.TenantID]*utils.TPRateProfile)
|
|
tpr.actionProfiles = make(map[utils.TenantID]*utils.TPActionProfile)
|
|
tpr.accounts = make(map[utils.TenantID]*utils.TPAccount)
|
|
tpr.filters = make(map[utils.TenantID]*utils.TPFilterProfile)
|
|
}
|
|
|
|
func (tpr *TpReader) LoadResourceProfilesFiltered(tag string) (err error) {
|
|
rls, err := tpr.lr.GetTPResources(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapRsPfls := make(map[utils.TenantID]*utils.TPResourceProfile)
|
|
for _, rl := range rls {
|
|
if err = verifyInlineFilterS(rl.FilterIDs); err != nil {
|
|
return
|
|
}
|
|
mapRsPfls[utils.TenantID{Tenant: rl.Tenant, ID: rl.ID}] = rl
|
|
}
|
|
tpr.resProfiles = mapRsPfls
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadResourceProfiles() error {
|
|
return tpr.LoadResourceProfilesFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadStatsFiltered(tag string) (err error) {
|
|
tps, err := tpr.lr.GetTPStats(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapSTs := make(map[utils.TenantID]*utils.TPStatProfile)
|
|
for _, st := range tps {
|
|
if err = verifyInlineFilterS(st.FilterIDs); err != nil {
|
|
return
|
|
}
|
|
mapSTs[utils.TenantID{Tenant: st.Tenant, ID: st.ID}] = st
|
|
}
|
|
tpr.sqProfiles = mapSTs
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadStats() error {
|
|
return tpr.LoadStatsFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadRankingsFiltered(tag string) error {
|
|
tps, err := tpr.lr.GetTPRankings(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapRgs := make(map[utils.TenantID]*utils.TPRankingProfile)
|
|
for _, rg := range tps {
|
|
mapRgs[utils.TenantID{Tenant: rg.Tenant, ID: rg.ID}] = rg
|
|
}
|
|
tpr.rgProfiles = mapRgs
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadRankings() error {
|
|
return tpr.LoadRankingsFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadTrendsFiltered(tag string) error {
|
|
tps, err := tpr.lr.GetTPTrends(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapSrs := make(map[utils.TenantID]*utils.TPTrendsProfile)
|
|
for _, sr := range tps {
|
|
mapSrs[utils.TenantID{Tenant: sr.Tenant, ID: sr.ID}] = sr
|
|
}
|
|
tpr.trProfiles = mapSrs
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadTrends() error {
|
|
return tpr.LoadTrendsFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadThresholdsFiltered(tag string) (err error) {
|
|
tps, err := tpr.lr.GetTPThresholds(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapTHs := make(map[utils.TenantID]*utils.TPThresholdProfile)
|
|
for _, th := range tps {
|
|
if err = verifyInlineFilterS(th.FilterIDs); err != nil {
|
|
return
|
|
}
|
|
mapTHs[utils.TenantID{Tenant: th.Tenant, ID: th.ID}] = th
|
|
}
|
|
tpr.thProfiles = mapTHs
|
|
return
|
|
}
|
|
|
|
func (tpr *TpReader) LoadThresholds() error {
|
|
return tpr.LoadThresholdsFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadFiltersFiltered(tag string) error {
|
|
tps, err := tpr.lr.GetTPFilters(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapTHs := make(map[utils.TenantID]*utils.TPFilterProfile)
|
|
for _, th := range tps {
|
|
mapTHs[utils.TenantID{Tenant: th.Tenant, ID: th.ID}] = th
|
|
}
|
|
tpr.filters = mapTHs
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadFilters() error {
|
|
return tpr.LoadFiltersFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadRouteProfilesFiltered(tag string) (err error) {
|
|
rls, err := tpr.lr.GetTPRoutes(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapRsPfls := make(map[utils.TenantID]*utils.TPRouteProfile)
|
|
for _, rl := range rls {
|
|
if err = verifyInlineFilterS(rl.FilterIDs); err != nil {
|
|
return
|
|
}
|
|
mapRsPfls[utils.TenantID{Tenant: rl.Tenant, ID: rl.ID}] = rl
|
|
}
|
|
tpr.routeProfiles = mapRsPfls
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadRouteProfiles() error {
|
|
return tpr.LoadRouteProfilesFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadAttributeProfilesFiltered(tag string) (err error) {
|
|
attrs, err := tpr.lr.GetTPAttributes(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapAttrPfls := make(map[utils.TenantID]*utils.TPAttributeProfile)
|
|
for _, attr := range attrs {
|
|
if err = verifyInlineFilterS(attr.FilterIDs); err != nil {
|
|
return
|
|
}
|
|
for _, at := range attr.Attributes {
|
|
if at.Path == utils.EmptyString { // we do not suppot empty Path in Attributes
|
|
err = fmt.Errorf("empty path in AttributeProfile <%s>", utils.ConcatenatedKey(attr.Tenant, attr.ID))
|
|
return
|
|
}
|
|
}
|
|
mapAttrPfls[utils.TenantID{Tenant: attr.Tenant, ID: attr.ID}] = attr
|
|
}
|
|
tpr.attributeProfiles = mapAttrPfls
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadAttributeProfiles() error {
|
|
return tpr.LoadAttributeProfilesFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadChargerProfilesFiltered(tag string) (err error) {
|
|
rls, err := tpr.lr.GetTPChargers(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapChargerProfile := make(map[utils.TenantID]*utils.TPChargerProfile)
|
|
for _, rl := range rls {
|
|
if err = verifyInlineFilterS(rl.FilterIDs); err != nil {
|
|
return
|
|
}
|
|
mapChargerProfile[utils.TenantID{Tenant: rl.Tenant, ID: rl.ID}] = rl
|
|
}
|
|
tpr.chargerProfiles = mapChargerProfile
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadChargerProfiles() error {
|
|
return tpr.LoadChargerProfilesFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadRateProfiles() error {
|
|
return tpr.LoadRateProfilesFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadRateProfilesFiltered(tag string) (err error) {
|
|
rls, err := tpr.lr.GetTPRateProfiles(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapRateProfiles := make(map[utils.TenantID]*utils.TPRateProfile)
|
|
for _, rl := range rls {
|
|
if err = verifyInlineFilterS(rl.FilterIDs); err != nil {
|
|
return
|
|
}
|
|
mapRateProfiles[utils.TenantID{Tenant: rl.Tenant, ID: rl.ID}] = rl
|
|
}
|
|
tpr.rateProfiles = mapRateProfiles
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadActionProfiles() error {
|
|
return tpr.LoadActionProfilesFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadActionProfilesFiltered(tag string) (err error) {
|
|
aps, err := tpr.lr.GetTPActionProfiles(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapActionProfiles := make(map[utils.TenantID]*utils.TPActionProfile)
|
|
for _, ap := range aps {
|
|
if err = verifyInlineFilterS(ap.FilterIDs); err != nil {
|
|
return
|
|
}
|
|
mapActionProfiles[utils.TenantID{Tenant: ap.Tenant, ID: ap.ID}] = ap
|
|
}
|
|
tpr.actionProfiles = mapActionProfiles
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadAccounts() error {
|
|
return tpr.LoadAccountsFiltered("")
|
|
}
|
|
|
|
func (tpr *TpReader) LoadAccountsFiltered(tag string) (err error) {
|
|
aps, err := tpr.lr.GetTPAccounts(tpr.tpid, "", tag)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mapAccounts := make(map[utils.TenantID]*utils.TPAccount)
|
|
for _, ap := range aps {
|
|
if err = verifyInlineFilterS(ap.FilterIDs); err != nil {
|
|
return
|
|
}
|
|
mapAccounts[utils.TenantID{Tenant: ap.Tenant, ID: ap.ID}] = ap
|
|
}
|
|
tpr.accounts = mapAccounts
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) LoadAll() (err error) {
|
|
if err = tpr.LoadFilters(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadResourceProfiles(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadStats(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadRankings(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadTrends(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadThresholds(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadRouteProfiles(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadAttributeProfiles(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadChargerProfiles(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadRateProfiles(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadActionProfiles(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
if err = tpr.LoadAccounts(); err != nil && err.Error() != utils.NotFoundCaps {
|
|
return
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (tpr *TpReader) WriteToDatabase(verbose, disableReverse bool) (err error) {
|
|
for _, db := range tpr.dm.DataDB() {
|
|
if db == nil {
|
|
return errors.New("no database connection")
|
|
}
|
|
}
|
|
//generate a loadID
|
|
loadID := time.Now().UnixNano()
|
|
loadIDs := make(map[string]int64)
|
|
|
|
if verbose {
|
|
log.Print("Filters:")
|
|
}
|
|
for _, tpTH := range tpr.filters {
|
|
var th *Filter
|
|
if th, err = APItoFilter(tpTH, tpr.timezone); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetFilter(context.TODO(), th, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", th.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.filters) != 0 {
|
|
loadIDs[utils.CacheFilters] = loadID
|
|
}
|
|
if verbose {
|
|
log.Print("ResourceProfiles:")
|
|
}
|
|
for _, tpRsp := range tpr.resProfiles {
|
|
var rsp *utils.ResourceProfile
|
|
if rsp, err = APItoResource(tpRsp, tpr.timezone); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetResourceProfile(context.TODO(), rsp, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", rsp.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.resProfiles) != 0 {
|
|
loadIDs[utils.CacheResourceProfiles] = loadID
|
|
loadIDs[utils.CacheResources] = loadID
|
|
}
|
|
if verbose {
|
|
log.Print("StatQueueProfiles:")
|
|
}
|
|
for _, tpST := range tpr.sqProfiles {
|
|
var st *StatQueueProfile
|
|
if st, err = APItoStats(tpST, tpr.timezone); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetStatQueueProfile(context.TODO(), st, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", st.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.sqProfiles) != 0 {
|
|
loadIDs[utils.CacheStatQueues] = loadID
|
|
loadIDs[utils.CacheStatQueueProfiles] = loadID
|
|
}
|
|
if verbose {
|
|
log.Print("ThresholdProfiles:")
|
|
}
|
|
for _, tpTH := range tpr.thProfiles {
|
|
var th *ThresholdProfile
|
|
if th, err = APItoThresholdProfile(tpTH, tpr.timezone); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetThresholdProfile(context.TODO(), th, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", th.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.thProfiles) != 0 {
|
|
loadIDs[utils.CacheThresholdProfiles] = loadID
|
|
loadIDs[utils.CacheThresholds] = loadID
|
|
}
|
|
if verbose {
|
|
log.Print("TrendProfiles:")
|
|
}
|
|
for _, tpTR := range tpr.trProfiles {
|
|
var tr *utils.TrendProfile
|
|
if tr, err = APItoTrends(tpTR); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetTrendProfile(context.TODO(), tr); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", tr.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.trProfiles) != 0 {
|
|
loadIDs[utils.CacheTrendProfiles] = loadID
|
|
loadIDs[utils.CacheTrends] = loadID
|
|
}
|
|
if verbose {
|
|
log.Print("RankingProfiles:")
|
|
}
|
|
for _, tpRN := range tpr.rgProfiles {
|
|
var rn *utils.RankingProfile
|
|
if rn, err = APItoRanking(tpRN); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetRankingProfile(context.TODO(), rn); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", rn.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.rgProfiles) != 0 {
|
|
loadIDs[utils.CacheRankingProfiles] = loadID
|
|
loadIDs[utils.CacheRankings] = loadID
|
|
}
|
|
if verbose {
|
|
log.Print("RouteProfiles:")
|
|
}
|
|
for _, tpTH := range tpr.routeProfiles {
|
|
var th *utils.RouteProfile
|
|
if th, err = APItoRouteProfile(tpTH, tpr.timezone); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetRouteProfile(context.TODO(), th, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", th.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.routeProfiles) != 0 {
|
|
loadIDs[utils.CacheRouteProfiles] = loadID
|
|
}
|
|
if verbose {
|
|
log.Print("AttributeProfiles:")
|
|
}
|
|
for _, tpTH := range tpr.attributeProfiles {
|
|
var th *utils.AttributeProfile
|
|
if th, err = APItoAttributeProfile(tpTH, tpr.timezone); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetAttributeProfile(context.TODO(), th, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", th.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.attributeProfiles) != 0 {
|
|
loadIDs[utils.CacheAttributeProfiles] = loadID
|
|
}
|
|
if verbose {
|
|
log.Print("ChargerProfiles:")
|
|
}
|
|
for _, tpTH := range tpr.chargerProfiles {
|
|
th := APItoChargerProfile(tpTH, tpr.timezone)
|
|
if err = tpr.dm.SetChargerProfile(context.TODO(), th, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", th.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.chargerProfiles) != 0 {
|
|
loadIDs[utils.CacheChargerProfiles] = loadID
|
|
}
|
|
if verbose {
|
|
log.Print("RateProfiles:")
|
|
}
|
|
for _, tpTH := range tpr.rateProfiles {
|
|
var th *utils.RateProfile
|
|
if th, err = APItoRateProfile(tpTH, tpr.timezone); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetRateProfile(context.Background(), th, false, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", th.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.rateProfiles) != 0 {
|
|
loadIDs[utils.CacheRateProfiles] = loadID
|
|
}
|
|
|
|
if verbose {
|
|
log.Print("ActionProfiles:")
|
|
}
|
|
for _, tpAP := range tpr.actionProfiles {
|
|
var ap *utils.ActionProfile
|
|
if ap, err = APItoActionProfile(tpAP, tpr.timezone); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetActionProfile(context.TODO(), ap, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", ap.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.actionProfiles) != 0 {
|
|
loadIDs[utils.CacheActionProfiles] = loadID
|
|
}
|
|
|
|
if verbose {
|
|
log.Print("Accounts:")
|
|
}
|
|
for _, tpAP := range tpr.accounts {
|
|
var ap *utils.Account
|
|
if ap, err = APItoAccount(tpAP, tpr.timezone); err != nil {
|
|
return
|
|
}
|
|
if err = tpr.dm.SetAccount(context.TODO(), ap, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", ap.TenantID())
|
|
}
|
|
}
|
|
if len(tpr.accounts) != 0 {
|
|
loadIDs[utils.CacheAccounts] = loadID
|
|
}
|
|
|
|
return tpr.dm.SetLoadIDs(context.TODO(), loadIDs)
|
|
}
|
|
|
|
func (tpr *TpReader) ShowStatistics() {
|
|
// resource profiles
|
|
log.Print("ResourceProfiles: ", len(tpr.resProfiles))
|
|
// stats
|
|
log.Print("Stats: ", len(tpr.sqProfiles))
|
|
// thresholds
|
|
log.Print("Thresholds: ", len(tpr.thProfiles))
|
|
// filters
|
|
log.Print("Filters: ", len(tpr.filters))
|
|
// Route profiles
|
|
log.Print("RouteProfiles: ", len(tpr.routeProfiles))
|
|
// Attribute profiles
|
|
log.Print("AttributeProfiles: ", len(tpr.attributeProfiles))
|
|
// Charger profiles
|
|
log.Print("ChargerProfiles: ", len(tpr.chargerProfiles))
|
|
// Rate profiles
|
|
log.Print("RateProfiles: ", len(tpr.rateProfiles))
|
|
// Action profiles
|
|
log.Print("ActionProfiles: ", len(tpr.actionProfiles))
|
|
// Trend profiles
|
|
log.Print("TrendProfiles: ", len(tpr.trProfiles))
|
|
// Ranking profiles
|
|
log.Print("RankingProfiles: ", len(tpr.rgProfiles))
|
|
}
|
|
|
|
// GetLoadedIds returns the identities loaded for a specific category, useful for cache reloads
|
|
func (tpr *TpReader) GetLoadedIds(categ string) ([]string, error) {
|
|
switch categ {
|
|
case utils.ResourceProfilesPrefix:
|
|
keys := make([]string, len(tpr.resProfiles))
|
|
i := 0
|
|
for k := range tpr.resProfiles {
|
|
keys[i] = k.TenantID()
|
|
i++
|
|
}
|
|
return keys, nil
|
|
|
|
case utils.StatQueueProfilePrefix:
|
|
keys := make([]string, len(tpr.sqProfiles))
|
|
i := 0
|
|
for k := range tpr.sqProfiles {
|
|
keys[i] = k.TenantID()
|
|
i++
|
|
}
|
|
return keys, nil
|
|
case utils.ThresholdProfilePrefix:
|
|
keys := make([]string, len(tpr.thProfiles))
|
|
i := 0
|
|
for k := range tpr.thProfiles {
|
|
keys[i] = k.TenantID()
|
|
i++
|
|
}
|
|
return keys, nil
|
|
case utils.FilterPrefix:
|
|
keys := make([]string, len(tpr.filters))
|
|
i := 0
|
|
for k := range tpr.filters {
|
|
keys[i] = k.TenantID()
|
|
i++
|
|
}
|
|
return keys, nil
|
|
case utils.RouteProfilePrefix:
|
|
keys := make([]string, len(tpr.routeProfiles))
|
|
i := 0
|
|
for k := range tpr.routeProfiles {
|
|
keys[i] = k.TenantID()
|
|
i++
|
|
}
|
|
return keys, nil
|
|
case utils.AttributeProfilePrefix:
|
|
keys := make([]string, len(tpr.attributeProfiles))
|
|
i := 0
|
|
for k := range tpr.attributeProfiles {
|
|
keys[i] = k.TenantID()
|
|
i++
|
|
}
|
|
return keys, nil
|
|
case utils.ChargerProfilePrefix:
|
|
keys := make([]string, len(tpr.chargerProfiles))
|
|
i := 0
|
|
for k := range tpr.chargerProfiles {
|
|
keys[i] = k.TenantID()
|
|
i++
|
|
}
|
|
return keys, nil
|
|
case utils.RateProfilePrefix:
|
|
keys := make([]string, len(tpr.rateProfiles))
|
|
i := 0
|
|
for k := range tpr.rateProfiles {
|
|
keys[i] = k.TenantID()
|
|
i++
|
|
}
|
|
return keys, nil
|
|
case utils.ActionProfilePrefix:
|
|
keys := make([]string, len(tpr.actionProfiles))
|
|
i := 0
|
|
for k := range tpr.actionProfiles {
|
|
keys[i] = k.TenantID()
|
|
i++
|
|
}
|
|
return keys, nil
|
|
}
|
|
return nil, errors.New("Unsupported load category")
|
|
}
|
|
|
|
func (tpr *TpReader) RemoveFromDatabase(verbose, disableReverse bool) (err error) {
|
|
loadID := time.Now().UnixNano()
|
|
loadIDs := make(map[string]int64)
|
|
if verbose {
|
|
log.Print("ResourceProfiles:")
|
|
}
|
|
for _, tpRsp := range tpr.resProfiles {
|
|
if err = tpr.dm.RemoveResourceProfile(context.TODO(), tpRsp.Tenant, tpRsp.ID, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", utils.ConcatenatedKey(tpRsp.Tenant, tpRsp.ID))
|
|
}
|
|
}
|
|
if verbose {
|
|
log.Print("StatQueueProfiles:")
|
|
}
|
|
for _, tpST := range tpr.sqProfiles {
|
|
if err = tpr.dm.RemoveStatQueueProfile(context.TODO(), tpST.Tenant, tpST.ID, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", utils.ConcatenatedKey(tpST.Tenant, tpST.ID))
|
|
}
|
|
}
|
|
if verbose {
|
|
log.Print("ThresholdProfiles:")
|
|
}
|
|
for _, tpTH := range tpr.thProfiles {
|
|
if err = tpr.dm.RemoveThresholdProfile(context.TODO(), tpTH.Tenant, tpTH.ID, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", utils.ConcatenatedKey(tpTH.Tenant, tpTH.ID))
|
|
}
|
|
}
|
|
if verbose {
|
|
log.Print("RouteProfiles:")
|
|
}
|
|
for _, tpSpl := range tpr.routeProfiles {
|
|
if err = tpr.dm.RemoveRouteProfile(context.TODO(), tpSpl.Tenant, tpSpl.ID, true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", utils.ConcatenatedKey(tpSpl.Tenant, tpSpl.ID))
|
|
}
|
|
}
|
|
|
|
if verbose {
|
|
log.Print("AttributeProfiles:")
|
|
}
|
|
for _, tpAttr := range tpr.attributeProfiles {
|
|
if err = tpr.dm.RemoveAttributeProfile(context.TODO(), tpAttr.Tenant, tpAttr.ID,
|
|
true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", utils.ConcatenatedKey(tpAttr.Tenant, tpAttr.ID))
|
|
}
|
|
}
|
|
|
|
if verbose {
|
|
log.Print("ChargerProfiles:")
|
|
}
|
|
for _, tpChr := range tpr.chargerProfiles {
|
|
if err = tpr.dm.RemoveChargerProfile(context.TODO(), tpChr.Tenant, tpChr.ID,
|
|
true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", utils.ConcatenatedKey(tpChr.Tenant, tpChr.ID))
|
|
}
|
|
}
|
|
|
|
if verbose {
|
|
log.Print("RateProfiles:")
|
|
}
|
|
for _, tpRp := range tpr.rateProfiles {
|
|
if err = tpr.dm.RemoveRateProfile(context.TODO(), tpRp.Tenant, tpRp.ID,
|
|
true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", utils.ConcatenatedKey(tpRp.Tenant, tpRp.ID))
|
|
}
|
|
}
|
|
|
|
if verbose {
|
|
log.Print("ActionProfiles:")
|
|
}
|
|
for _, tpAp := range tpr.actionProfiles {
|
|
if err = tpr.dm.RemoveActionProfile(context.TODO(), tpAp.Tenant, tpAp.ID,
|
|
true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", utils.ConcatenatedKey(tpAp.Tenant, tpAp.ID))
|
|
}
|
|
}
|
|
|
|
if verbose {
|
|
log.Print("Accounts:")
|
|
}
|
|
for _, tpAp := range tpr.accounts {
|
|
if err = tpr.dm.RemoveAccount(context.TODO(), tpAp.Tenant, tpAp.ID,
|
|
true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", utils.ConcatenatedKey(tpAp.Tenant, tpAp.ID))
|
|
}
|
|
}
|
|
|
|
//We remove the filters at the end because of indexes
|
|
if verbose {
|
|
log.Print("Filters:")
|
|
}
|
|
for _, tpFltr := range tpr.filters {
|
|
if err = tpr.dm.RemoveFilter(context.TODO(), tpFltr.Tenant, tpFltr.ID,
|
|
true); err != nil {
|
|
return
|
|
}
|
|
if verbose {
|
|
log.Print("\t", utils.ConcatenatedKey(tpFltr.Tenant, tpFltr.ID))
|
|
}
|
|
}
|
|
if len(tpr.filters) != 0 {
|
|
loadIDs[utils.CacheFilters] = loadID
|
|
}
|
|
if len(tpr.resProfiles) != 0 {
|
|
loadIDs[utils.CacheResourceProfiles] = loadID
|
|
loadIDs[utils.CacheResources] = loadID
|
|
}
|
|
if len(tpr.sqProfiles) != 0 {
|
|
loadIDs[utils.CacheStatQueueProfiles] = loadID
|
|
loadIDs[utils.CacheStatQueues] = loadID
|
|
}
|
|
if len(tpr.thProfiles) != 0 {
|
|
loadIDs[utils.CacheThresholdProfiles] = loadID
|
|
loadIDs[utils.CacheThresholds] = loadID
|
|
}
|
|
if len(tpr.routeProfiles) != 0 {
|
|
loadIDs[utils.CacheRouteProfiles] = loadID
|
|
}
|
|
if len(tpr.attributeProfiles) != 0 {
|
|
loadIDs[utils.CacheAttributeProfiles] = loadID
|
|
}
|
|
if len(tpr.chargerProfiles) != 0 {
|
|
loadIDs[utils.CacheChargerProfiles] = loadID
|
|
}
|
|
if len(tpr.rateProfiles) != 0 {
|
|
loadIDs[utils.CacheRateProfiles] = loadID
|
|
}
|
|
if len(tpr.actionProfiles) != 0 {
|
|
loadIDs[utils.CacheActionProfiles] = loadID
|
|
}
|
|
if len(tpr.accounts) != 0 {
|
|
loadIDs[utils.CacheAccounts] = loadID
|
|
}
|
|
return tpr.dm.SetLoadIDs(context.TODO(), loadIDs)
|
|
}
|
|
|
|
func (tpr *TpReader) ReloadCache(ctx *context.Context, caching string, verbose bool, opts map[string]any, tenant string) (err error) {
|
|
if len(tpr.cacheConns) == 0 {
|
|
log.Print("Disabled automatic reload")
|
|
return
|
|
}
|
|
// take IDs for each type
|
|
rspIDs, _ := tpr.GetLoadedIds(utils.ResourceProfilesPrefix)
|
|
stqpIDs, _ := tpr.GetLoadedIds(utils.StatQueueProfilePrefix)
|
|
trspfIDs, _ := tpr.GetLoadedIds(utils.ThresholdProfilePrefix)
|
|
flrIDs, _ := tpr.GetLoadedIds(utils.FilterPrefix)
|
|
routeIDs, _ := tpr.GetLoadedIds(utils.RouteProfilePrefix)
|
|
apfIDs, _ := tpr.GetLoadedIds(utils.AttributeProfilePrefix)
|
|
chargerIDs, _ := tpr.GetLoadedIds(utils.ChargerProfilePrefix)
|
|
ratePrfIDs, _ := tpr.GetLoadedIds(utils.RateProfilePrefix)
|
|
actionPrfIDs, _ := tpr.GetLoadedIds(utils.ActionProfilePrefix)
|
|
accountPrfIDs, _ := tpr.GetLoadedIds(utils.AccountPrefix)
|
|
|
|
//compose Reload Cache argument
|
|
cacheArgs := map[string][]string{
|
|
utils.CacheResourceProfiles: rspIDs,
|
|
utils.CacheResources: rspIDs,
|
|
utils.CacheStatQueues: stqpIDs,
|
|
utils.CacheStatQueueProfiles: stqpIDs,
|
|
utils.CacheThresholds: trspfIDs,
|
|
utils.CacheThresholdProfiles: trspfIDs,
|
|
utils.CacheFilters: flrIDs,
|
|
utils.CacheRouteProfiles: routeIDs,
|
|
utils.CacheAttributeProfiles: apfIDs,
|
|
utils.CacheChargerProfiles: chargerIDs,
|
|
utils.CacheRateProfiles: ratePrfIDs,
|
|
utils.CacheActionProfiles: actionPrfIDs,
|
|
}
|
|
|
|
// verify if we need to clear indexes
|
|
var cacheIDs []string
|
|
if len(apfIDs) != 0 {
|
|
cacheIDs = append(cacheIDs, utils.CacheAttributeFilterIndexes)
|
|
}
|
|
if len(routeIDs) != 0 {
|
|
cacheIDs = append(cacheIDs, utils.CacheRouteFilterIndexes)
|
|
}
|
|
if len(trspfIDs) != 0 {
|
|
cacheIDs = append(cacheIDs, utils.CacheThresholdFilterIndexes)
|
|
}
|
|
if len(stqpIDs) != 0 {
|
|
cacheIDs = append(cacheIDs, utils.CacheStatFilterIndexes)
|
|
}
|
|
if len(rspIDs) != 0 {
|
|
cacheIDs = append(cacheIDs, utils.CacheResourceFilterIndexes)
|
|
}
|
|
if len(chargerIDs) != 0 {
|
|
cacheIDs = append(cacheIDs, utils.CacheChargerFilterIndexes)
|
|
}
|
|
if len(ratePrfIDs) != 0 {
|
|
cacheIDs = append(cacheIDs, utils.CacheRateProfilesFilterIndexes)
|
|
cacheIDs = append(cacheIDs, utils.CacheRateFilterIndexes)
|
|
}
|
|
if len(actionPrfIDs) != 0 {
|
|
cacheIDs = append(cacheIDs, utils.CacheActionProfilesFilterIndexes)
|
|
}
|
|
if len(accountPrfIDs) != 0 {
|
|
cacheIDs = append(cacheIDs, utils.CacheAccountsFilterIndexes)
|
|
}
|
|
if len(flrIDs) != 0 {
|
|
cacheIDs = append(cacheIDs, utils.CacheReverseFilterIndexes)
|
|
}
|
|
|
|
if err = CallCache(connMgr, ctx, tpr.cacheConns, caching, cacheArgs, cacheIDs, opts, verbose, tenant); err != nil {
|
|
return
|
|
}
|
|
//get loadIDs for all types
|
|
var loadIDs map[string]int64
|
|
if loadIDs, err = tpr.dm.GetItemLoadIDs(ctx, utils.EmptyString, false); err != nil {
|
|
return
|
|
}
|
|
cacheLoadIDs := populateCacheLoadIDs(loadIDs, cacheArgs)
|
|
for key, val := range cacheLoadIDs {
|
|
if err = Cache.Set(ctx, utils.CacheLoadIDs, key, val, nil,
|
|
cacheCommit(utils.NonTransactional), utils.NonTransactional); err != nil {
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
// CallCache call the cache reload after data load
|
|
func CallCache(connMgr *ConnManager, ctx *context.Context, cacheConns []string, caching string, args map[string][]string, cacheIDs []string, opts map[string]any, verbose bool, tenant string) (err error) {
|
|
var method, reply string
|
|
var cacheArgs any = utils.NewAttrReloadCacheWithOptsFromMap(args, tenant, opts)
|
|
switch caching {
|
|
case utils.MetaNone:
|
|
return
|
|
case utils.MetaReload:
|
|
method = utils.CacheSv1ReloadCache
|
|
case utils.MetaLoad:
|
|
method = utils.CacheSv1LoadCache
|
|
case utils.MetaRemove:
|
|
method = utils.CacheSv1RemoveItems
|
|
case utils.MetaClear:
|
|
method = utils.CacheSv1Clear
|
|
cacheArgs = &utils.AttrCacheIDsWithAPIOpts{APIOpts: opts, Tenant: tenant}
|
|
}
|
|
if verbose {
|
|
log.Print("Reloading cache")
|
|
}
|
|
|
|
if err = connMgr.Call(ctx, cacheConns, method, cacheArgs, &reply); err != nil {
|
|
return
|
|
}
|
|
|
|
if len(cacheIDs) != 0 {
|
|
if verbose {
|
|
log.Print("Clearing indexes")
|
|
}
|
|
if err = connMgr.Call(ctx, cacheConns, utils.CacheSv1Clear, &utils.AttrCacheIDsWithAPIOpts{
|
|
APIOpts: opts,
|
|
CacheIDs: cacheIDs,
|
|
Tenant: tenant,
|
|
}, &reply); err != nil {
|
|
if verbose {
|
|
log.Printf("WARNING: Got error on cache clear: %s\n", err.Error())
|
|
}
|
|
}
|
|
}
|
|
return
|
|
}
|
|
|
|
func (tpr *TpReader) ReloadScheduler(verbose bool) (err error) { // ToDoNext: add reload to new actions
|
|
// var reply string
|
|
// aps, _ := tpr.GetLoadedIds(utils.ActionPlanPrefix)
|
|
// // in case we have action plans reload the scheduler
|
|
// if len(aps) == 0 {
|
|
// return
|
|
// }
|
|
// if verbose {
|
|
// log.Print("Reloading scheduler")
|
|
// }
|
|
// if err = connMgr.Call(tpr.schedulerConns, nil, utils.SchedulerSv1Reload,
|
|
// new(utils.CGREvent), &reply); err != nil {
|
|
// log.Printf("WARNING: Got error on scheduler reload: %s\n", err.Error())
|
|
// }
|
|
return
|
|
}
|