/* 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 */ package engine import ( "fmt" "slices" "sort" "strings" "time" "github.com/cgrates/cgrates/config" "github.com/cgrates/cgrates/utils" ) // GetTpIds implements LoadReader interface func (iDB *InternalDB) GetTpIds(colName string) (ids []string, err error) { tpIDs := make(utils.StringSet) if colName == utils.EmptyString { // if colName is empty we need to parse all partitions for _, conNm := range utils.CacheStorDBPartitions { // iterate through all columns for _, key := range iDB.db.GetItemIDs(conNm, utils.EmptyString) { tpIDs.Add(strings.Split(key, utils.InInFieldSep)[0]) } } } else { for _, key := range iDB.db.GetItemIDs(utils.CacheStorDBPartitions[colName], utils.EmptyString) { tpIDs.Add(strings.Split(key, utils.InInFieldSep)[0]) } } return tpIDs.AsSlice(), nil } func (iDB *InternalDB) GetTpTableIds(tpid, table string, distinct utils.TPDistinctIds, filters map[string]string, paginator *utils.PaginatorWithSearch) (ids []string, err error) { fullIDs := iDB.db.GetItemIDs(utils.CacheStorDBPartitions[table], tpid) idSet := make(utils.StringSet) for _, fullID := range fullIDs { switch table { // in case of account action and rating profile // the retutned value may be only the loadID case utils.TBLTPAccountActions, utils.TBLTPRatingProfiles: if len(distinct) == 1 { // special case when to return only the loadID sliceID := strings.Split(fullID[len(tpid)+1:], utils.ConcatenatedKeySep) idSet.Add(sliceID[0]) continue } // in the rest of the cases we return all ID every time default: } idSet.Add(fullID[len(tpid)+1:]) } ids = idSet.AsSlice() return } func (iDB *InternalDB) GetTPTimings(tpid, id string) (timings []*utils.ApierTPTiming, err error) { key := tpid if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPTimings, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPTimings, id) if !ok || x == nil { return nil, utils.ErrNotFound } timings = append(timings, x.(*utils.ApierTPTiming)) } if len(timings) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPDestinations(tpid, id string) (dsts []*utils.TPDestination, err error) { key := tpid if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPDestinations, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPDestinations, id) if !ok || x == nil { return nil, utils.ErrNotFound } dsts = append(dsts, x.(*utils.TPDestination)) } if len(dsts) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPRates(tpid, id string) (rates []*utils.TPRateRALs, err error) { key := tpid if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPRates, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPRates, id) if !ok || x == nil { return nil, utils.ErrNotFound } result := x.(*utils.TPRateRALs) for _, rs := range result.RateSlots { rs.SetDurations() } rates = append(rates, result) } if len(rates) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPDestinationRates(tpid, id string, paginator *utils.Paginator) (dRates []*utils.TPDestinationRate, err error) { key := tpid if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPDestinationRates, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPDestinationRates, id) if !ok || x == nil { return nil, utils.ErrNotFound } dRates = append(dRates, x.(*utils.TPDestinationRate)) } if len(dRates) == 0 { return nil, utils.ErrNotFound } if paginator != nil { var limit, offset int if paginator.Limit != nil && *paginator.Limit > 0 { limit = *paginator.Limit } if paginator.Offset != nil && *paginator.Offset > 0 { offset = *paginator.Offset } if limit == 0 && offset == 0 { return dRates, nil } if offset > len(dRates) { return } if offset != 0 { limit = limit + offset } if limit == 0 { limit = len(dRates[offset:]) } else if limit > len(dRates) { limit = len(dRates) } dRates = dRates[offset:limit] } return } func (iDB *InternalDB) GetTPRatingPlans(tpid, id string, paginator *utils.Paginator) (rPlans []*utils.TPRatingPlan, err error) { key := tpid if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPRatingPlans, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPRatingPlans, id) if !ok || x == nil { return nil, utils.ErrNotFound } rPlans = append(rPlans, x.(*utils.TPRatingPlan)) } if len(rPlans) == 0 { return nil, utils.ErrNotFound } if paginator != nil { var limit, offset int if paginator.Limit != nil && *paginator.Limit > 0 { limit = *paginator.Limit } if paginator.Offset != nil && *paginator.Offset > 0 { offset = *paginator.Offset } if limit == 0 && offset == 0 { return rPlans, nil } if offset > len(rPlans) { return } if offset != 0 { limit = limit + offset } if limit == 0 { limit = len(rPlans[offset:]) } else if limit > len(rPlans) { limit = len(rPlans) } rPlans = rPlans[offset:limit] } return } func (iDB *InternalDB) GetTPRatingProfiles(filter *utils.TPRatingProfile) (rProfiles []*utils.TPRatingProfile, err error) { key := filter.TPid if filter.LoadId != utils.EmptyString { key += utils.ConcatenatedKeySep + filter.LoadId } if filter.Tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + filter.Tenant } if filter.Category != utils.EmptyString { key += utils.ConcatenatedKeySep + filter.Category } if filter.Subject != utils.EmptyString { key += utils.ConcatenatedKeySep + filter.Subject } ids := iDB.db.GetItemIDs(utils.CacheTBLTPRatingProfiles, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPRatingProfiles, id) if !ok || x == nil { return nil, utils.ErrNotFound } rProfiles = append(rProfiles, x.(*utils.TPRatingProfile)) } if len(rProfiles) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPSharedGroups(tpid, id string) (sGroups []*utils.TPSharedGroups, err error) { key := tpid if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPSharedGroups, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPSharedGroups, id) if !ok || x == nil { return nil, utils.ErrNotFound } sGroups = append(sGroups, x.(*utils.TPSharedGroups)) } if len(sGroups) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPActions(tpid, id string) (actions []*utils.TPActions, err error) { key := tpid if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPActions, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPActions, id) if !ok || x == nil { return nil, utils.ErrNotFound } actions = append(actions, x.(*utils.TPActions)) } if len(actions) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPActionPlans(tpid, id string) (aPlans []*utils.TPActionPlan, err error) { key := tpid if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPActionPlans, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPActionPlans, id) if !ok || x == nil { return nil, utils.ErrNotFound } aPlans = append(aPlans, x.(*utils.TPActionPlan)) } if len(aPlans) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPActionTriggers(tpid, id string) (aTriggers []*utils.TPActionTriggers, err error) { key := tpid if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPActionTriggers, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPActionTriggers, id) if !ok || x == nil { return nil, utils.ErrNotFound } aTriggers = append(aTriggers, x.(*utils.TPActionTriggers)) } if len(aTriggers) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPAccountActions(filter *utils.TPAccountActions) (accounts []*utils.TPAccountActions, err error) { key := filter.TPid if filter.LoadId != utils.EmptyString { key += utils.ConcatenatedKeySep + filter.LoadId } if filter.Tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + filter.Tenant } if filter.Account != utils.EmptyString { key += utils.ConcatenatedKeySep + filter.Account } ids := iDB.db.GetItemIDs(utils.CacheTBLTPAccountActions, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPAccountActions, id) if !ok || x == nil { return nil, utils.ErrNotFound } accounts = append(accounts, x.(*utils.TPAccountActions)) } if len(accounts) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPResources(tpid, tenant, id string) (resources []*utils.TPResourceProfile, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPResources, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPResources, id) if !ok || x == nil { return nil, utils.ErrNotFound } resources = append(resources, x.(*utils.TPResourceProfile)) } if len(resources) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPIPs(tpid, tenant, id string) ([]*utils.TPIPProfile, error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPIPs, key) ips := make([]*utils.TPIPProfile, 0, len(ids)) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPIPs, id) if !ok || x == nil { return nil, utils.ErrNotFound } ips = append(ips, x.(*utils.TPIPProfile)) } if len(ips) == 0 { return nil, utils.ErrNotFound } return ips, nil } func (iDB *InternalDB) GetTPStats(tpid, tenant, id string) (stats []*utils.TPStatProfile, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPStats, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPStats, id) if !ok || x == nil { return nil, utils.ErrNotFound } stats = append(stats, x.(*utils.TPStatProfile)) } if len(stats) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPTrends(tpid string, tenant string, id string) (trends []*utils.TPTrendsProfile, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPTrends, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPTrends, id) if !ok || x == nil { return nil, utils.ErrNotFound } trends = append(trends, x.(*utils.TPTrendsProfile)) } if len(trends) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPRankings(tpid string, tenant string, id string) (rankings []*utils.TPRankingProfile, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPRankings, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPRankings, id) if !ok || x == nil { return nil, utils.ErrNotFound } rankings = append(rankings, x.(*utils.TPRankingProfile)) } if len(rankings) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPThresholds(tpid, tenant, id string) (ths []*utils.TPThresholdProfile, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPThresholds, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPThresholds, id) if !ok || x == nil { return nil, utils.ErrNotFound } ths = append(ths, x.(*utils.TPThresholdProfile)) } if len(ths) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPFilters(tpid, tenant, id string) (fltrs []*utils.TPFilterProfile, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPFilters, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPFilters, id) if !ok || x == nil { return nil, utils.ErrNotFound } fltrs = append(fltrs, x.(*utils.TPFilterProfile)) } if len(fltrs) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPRoutes(tpid, tenant, id string) (supps []*utils.TPRouteProfile, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPRoutes, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPRoutes, id) if !ok || x == nil { return nil, utils.ErrNotFound } supps = append(supps, x.(*utils.TPRouteProfile)) } if len(supps) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPAttributes(tpid, tenant, id string) (attrs []*utils.TPAttributeProfile, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPAttributes, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPAttributes, id) if !ok || x == nil { return nil, utils.ErrNotFound } attrs = append(attrs, x.(*utils.TPAttributeProfile)) } if len(attrs) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPChargers(tpid, tenant, id string) (cpps []*utils.TPChargerProfile, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPChargers, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPChargers, id) if !ok || x == nil { return nil, utils.ErrNotFound } cpps = append(cpps, x.(*utils.TPChargerProfile)) } if len(cpps) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPDispatcherProfiles(tpid, tenant, id string) (dpps []*utils.TPDispatcherProfile, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPDispatchers, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPDispatchers, id) if !ok || x == nil { return nil, utils.ErrNotFound } dpps = append(dpps, x.(*utils.TPDispatcherProfile)) } if len(dpps) == 0 { return nil, utils.ErrNotFound } return } func (iDB *InternalDB) GetTPDispatcherHosts(tpid, tenant, id string) (dpps []*utils.TPDispatcherHost, err error) { key := tpid if tenant != utils.EmptyString { key += utils.ConcatenatedKeySep + tenant } if id != utils.EmptyString { key += utils.ConcatenatedKeySep + id } ids := iDB.db.GetItemIDs(utils.CacheTBLTPDispatcherHosts, key) for _, id := range ids { x, ok := iDB.db.Get(utils.CacheTBLTPDispatcherHosts, id) if !ok || x == nil { return nil, utils.ErrNotFound } dpps = append(dpps, x.(*utils.TPDispatcherHost)) } if len(dpps) == 0 { return nil, utils.ErrNotFound } return } // implement LoadWriter interface func (iDB *InternalDB) RemTpData(table, tpid string, args map[string]string) (err error) { if table == utils.EmptyString { return iDB.Flush(utils.EmptyString) } key := tpid if args != nil { if table == utils.TBLTPAccountActions { key += utils.ConcatenatedKeySep + args["loadid"] + utils.ConcatenatedKeySep + args["tenant"] + utils.ConcatenatedKeySep + args["account"] } else if table == utils.TBLTPRatingProfiles { key += utils.ConcatenatedKeySep + args["loadid"] + utils.ConcatenatedKeySep + args["tenant"] + utils.ConcatenatedKeySep + args["category"] + utils.ConcatenatedKeySep + args["subject"] } else if tag, has := args["tag"]; has { key += utils.ConcatenatedKeySep + tag } else if id, has := args["id"]; has { key += utils.ConcatenatedKeySep + args["tenant"] + utils.ConcatenatedKeySep + id } } ids := iDB.db.GetItemIDs(utils.CacheStorDBPartitions[table], key) for _, id := range ids { iDB.db.Remove(utils.CacheStorDBPartitions[table], id, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPTimings(timings []*utils.ApierTPTiming) (err error) { if len(timings) == 0 { return nil } for _, timing := range timings { iDB.db.Set(utils.CacheTBLTPTimings, utils.ConcatenatedKey(timing.TPid, timing.ID), timing, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPDestinations(dests []*utils.TPDestination) (err error) { if len(dests) == 0 { return nil } for _, destination := range dests { iDB.db.Set(utils.CacheTBLTPDestinations, utils.ConcatenatedKey(destination.TPid, destination.ID), destination, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPRates(rates []*utils.TPRateRALs) (err error) { if len(rates) == 0 { return nil } for _, rate := range rates { iDB.db.Set(utils.CacheTBLTPRates, utils.ConcatenatedKey(rate.TPid, rate.ID), rate, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPDestinationRates(dRates []*utils.TPDestinationRate) (err error) { if len(dRates) == 0 { return nil } for _, dRate := range dRates { iDB.db.Set(utils.CacheTBLTPDestinationRates, utils.ConcatenatedKey(dRate.TPid, dRate.ID), dRate, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPRatingPlans(ratingPlans []*utils.TPRatingPlan) (err error) { if len(ratingPlans) == 0 { return nil } for _, rPlan := range ratingPlans { iDB.db.Set(utils.CacheTBLTPRatingPlans, utils.ConcatenatedKey(rPlan.TPid, rPlan.ID), rPlan, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPRatingProfiles(ratingProfiles []*utils.TPRatingProfile) (err error) { if len(ratingProfiles) == 0 { return nil } for _, rProfile := range ratingProfiles { iDB.db.Set(utils.CacheTBLTPRatingProfiles, utils.ConcatenatedKey(rProfile.TPid, rProfile.LoadId, rProfile.Tenant, rProfile.Category, rProfile.Subject), rProfile, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPSharedGroups(groups []*utils.TPSharedGroups) (err error) { if len(groups) == 0 { return nil } for _, group := range groups { iDB.db.Set(utils.CacheTBLTPSharedGroups, utils.ConcatenatedKey(group.TPid, group.ID), group, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPActions(acts []*utils.TPActions) (err error) { if len(acts) == 0 { return nil } for _, action := range acts { iDB.db.Set(utils.CacheTBLTPActions, utils.ConcatenatedKey(action.TPid, action.ID), action, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPActionPlans(aPlans []*utils.TPActionPlan) (err error) { if len(aPlans) == 0 { return nil } for _, aPlan := range aPlans { iDB.db.Set(utils.CacheTBLTPActionPlans, utils.ConcatenatedKey(aPlan.TPid, aPlan.ID), aPlan, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPActionTriggers(aTriggers []*utils.TPActionTriggers) (err error) { if len(aTriggers) == 0 { return nil } for _, aTrigger := range aTriggers { iDB.db.Set(utils.CacheTBLTPActionTriggers, utils.ConcatenatedKey(aTrigger.TPid, aTrigger.ID), aTrigger, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPAccountActions(accActions []*utils.TPAccountActions) (err error) { if len(accActions) == 0 { return nil } for _, accAction := range accActions { iDB.db.Set(utils.CacheTBLTPAccountActions, utils.ConcatenatedKey(accAction.TPid, accAction.LoadId, accAction.Tenant, accAction.Account), accAction, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPResources(resources []*utils.TPResourceProfile) (err error) { if len(resources) == 0 { return nil } for _, resource := range resources { iDB.db.Set(utils.CacheTBLTPResources, utils.ConcatenatedKey(resource.TPid, resource.Tenant, resource.ID), resource, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPIPs(ips []*utils.TPIPProfile) (err error) { if len(ips) == 0 { return nil } for _, ip := range ips { iDB.db.Set(utils.CacheTBLTPIPs, utils.ConcatenatedKey(ip.TPid, ip.Tenant, ip.ID), ip, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPStats(stats []*utils.TPStatProfile) (err error) { if len(stats) == 0 { return nil } for _, stat := range stats { iDB.db.Set(utils.CacheTBLTPStats, utils.ConcatenatedKey(stat.TPid, stat.Tenant, stat.ID), stat, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPRankings(rankings []*utils.TPRankingProfile) (err error) { if len(rankings) == 0 { return nil } for _, ranking := range rankings { iDB.db.Set(utils.CacheTBLTPRankings, utils.ConcatenatedKey(ranking.TPid, ranking.Tenant, ranking.ID), ranking, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPTrends(trends []*utils.TPTrendsProfile) (err error) { if len(trends) == 0 { return nil } for _, trend := range trends { iDB.db.Set(utils.CacheTBLTPTrends, utils.ConcatenatedKey(trend.TPid, trend.Tenant, trend.ID), trend, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPThresholds(thresholds []*utils.TPThresholdProfile) (err error) { if len(thresholds) == 0 { return nil } for _, threshold := range thresholds { iDB.db.Set(utils.CacheTBLTPThresholds, utils.ConcatenatedKey(threshold.TPid, threshold.Tenant, threshold.ID), threshold, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPFilters(filters []*utils.TPFilterProfile) (err error) { if len(filters) == 0 { return nil } for _, filter := range filters { iDB.db.Set(utils.CacheTBLTPFilters, utils.ConcatenatedKey(filter.TPid, filter.Tenant, filter.ID), filter, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPRoutes(routes []*utils.TPRouteProfile) (err error) { if len(routes) == 0 { return nil } for _, route := range routes { iDB.db.Set(utils.CacheTBLTPRoutes, utils.ConcatenatedKey(route.TPid, route.Tenant, route.ID), route, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPAttributes(attributes []*utils.TPAttributeProfile) (err error) { if len(attributes) == 0 { return nil } for _, attribute := range attributes { iDB.db.Set(utils.CacheTBLTPAttributes, utils.ConcatenatedKey(attribute.TPid, attribute.Tenant, attribute.ID), attribute, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPChargers(cpps []*utils.TPChargerProfile) (err error) { if len(cpps) == 0 { return nil } for _, cpp := range cpps { iDB.db.Set(utils.CacheTBLTPChargers, utils.ConcatenatedKey(cpp.TPid, cpp.Tenant, cpp.ID), cpp, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPDispatcherProfiles(dpps []*utils.TPDispatcherProfile) (err error) { if len(dpps) == 0 { return nil } for _, dpp := range dpps { iDB.db.Set(utils.CacheTBLTPDispatchers, utils.ConcatenatedKey(dpp.TPid, dpp.Tenant, dpp.ID), dpp, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } func (iDB *InternalDB) SetTPDispatcherHosts(dpps []*utils.TPDispatcherHost) (err error) { if len(dpps) == 0 { return nil } for _, dpp := range dpps { iDB.db.Set(utils.CacheTBLTPDispatcherHosts, utils.ConcatenatedKey(dpp.TPid, dpp.Tenant, dpp.ID), dpp, nil, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return } // implement CdrStorage interface func (iDB *InternalDB) SetCDR(cdr *CDR, allowUpdate bool) (err error) { if cdr.OrderID == 0 { cdr.OrderID = iDB.cnter.Next() } cdrKey := utils.ConcatenatedKey(cdr.CGRID, cdr.RunID, cdr.OriginID) if !allowUpdate { if _, has := iDB.db.Get(utils.CacheCDRsTBL, cdrKey); has { return utils.ErrExists } } idxs := make(utils.StringSet) iDB.indexedFieldsMutex.RLock() if len(iDB.stringIndexedFields) == 0 && len(iDB.prefixIndexedFields) == 0 { // add default indexes idxs.Add(utils.ConcatenatedKey(utils.CGRID, cdr.CGRID)) idxs.Add(utils.ConcatenatedKey(utils.RunID, cdr.RunID)) idxs.Add(utils.ConcatenatedKey(utils.OriginHost, cdr.OriginHost)) idxs.Add(utils.ConcatenatedKey(utils.Source, cdr.Source)) idxs.Add(utils.ConcatenatedKey(utils.OriginID, cdr.OriginID)) idxs.Add(utils.ConcatenatedKey(utils.ToR, cdr.ToR)) idxs.Add(utils.ConcatenatedKey(utils.RequestType, cdr.RequestType)) idxs.Add(utils.ConcatenatedKey(utils.Tenant, cdr.Tenant)) idxs.Add(utils.ConcatenatedKey(utils.Category, cdr.Category)) idxs.Add(utils.ConcatenatedKey(utils.AccountField, cdr.Account)) idxs.Add(utils.ConcatenatedKey(utils.Subject, cdr.Subject)) idxs.Add(utils.ConcatenatedKey(utils.Destination, cdr.Destination)) // include the whole Destination for i := len(cdr.Destination) - 1; i > 0; i-- { // add destination as prefix idxs.Add(utils.ConcatenatedKey(utils.Destination, cdr.Destination[:i])) } } else { // add user indexes mpCDR := cdr.AsMapStringIface() for _, cdrIdx := range iDB.stringIndexedFields { idxs.Add(utils.ConcatenatedKey(cdrIdx, utils.IfaceAsString(mpCDR[cdrIdx]))) } for _, cdrIdx := range iDB.prefixIndexedFields { strVal := utils.IfaceAsString(mpCDR[cdrIdx]) idxs.Add(utils.ConcatenatedKey(cdrIdx, strVal)) for i := len(strVal) - 1; i > 0; i-- { idxs.Add(utils.ConcatenatedKey(cdrIdx, strVal[:i])) } } } iDB.indexedFieldsMutex.RUnlock() var saveCDR any = cdr if config.CgrConfig().CdrsCfg().CompressStoredCost && cdr.CostDetails != nil { saveCDR, err = cdr.AsCompressedCostCDR(iDB.ms) if err != nil { return } } iDB.db.Set(utils.CacheCDRsTBL, cdrKey, saveCDR, idxs.AsSlice(), cacheCommit(utils.NonTransactional), utils.NonTransactional) return } func (iDB *InternalDB) RemoveSMCost(smc *SMCost) (err error) { iDB.db.Remove(utils.CacheSessionCostsTBL, utils.ConcatenatedKey(smc.CGRID, smc.RunID, smc.OriginHost, smc.OriginID), cacheCommit(utils.NonTransactional), utils.NonTransactional) return } func (iDB *InternalDB) RemoveSMCosts(qryFltr *utils.SMCostFilter) error { var smMpIDs utils.StringMap // Apply string filter for _, fltrSlc := range []struct { key string ids []string }{ {utils.CGRID, qryFltr.CGRIDs}, {utils.RunID, qryFltr.RunIDs}, {utils.OriginID, qryFltr.OriginIDs}, {utils.OriginHost, qryFltr.OriginHosts}, {utils.CostSource, qryFltr.CostSources}, } { if len(fltrSlc.ids) == 0 { continue } grpMpIDs := make(utils.StringMap) for _, id := range fltrSlc.ids { grpIDs := iDB.db.GetGroupItemIDs(utils.CacheSessionCostsTBL, utils.ConcatenatedKey(fltrSlc.key, id)) for _, id := range grpIDs { grpMpIDs[id] = true } } if len(grpMpIDs) == 0 { return utils.ErrNotFound } if smMpIDs == nil { smMpIDs = grpMpIDs } else { for id := range smMpIDs { if !grpMpIDs.HasKey(id) { delete(smMpIDs, id) if len(smMpIDs) == 0 { return utils.ErrNotFound } } } } } if smMpIDs == nil { smMpIDs = utils.StringMapFromSlice(iDB.db.GetItemIDs(utils.CacheSessionCostsTBL, utils.EmptyString)) } // check for Not filters for _, fltrSlc := range []struct { key string ids []string }{ {utils.CGRID, qryFltr.NotCGRIDs}, {utils.RunID, qryFltr.NotRunIDs}, {utils.OriginID, qryFltr.NotOriginIDs}, {utils.OriginHost, qryFltr.NotOriginHosts}, {utils.CostSource, qryFltr.NotCostSources}, } { if len(fltrSlc.ids) == 0 { continue } for _, id := range fltrSlc.ids { grpIDs := iDB.db.GetGroupItemIDs(utils.CacheCDRsTBL, utils.ConcatenatedKey(fltrSlc.key, id)) for _, id := range grpIDs { if smMpIDs.HasKey(id) { delete(smMpIDs, id) if len(smMpIDs) == 0 { return utils.ErrNotFound } } } } } if len(smMpIDs) == 0 { return utils.ErrNotFound } for key := range smMpIDs { iDB.db.Remove(utils.CacheSessionCostsTBL, key, cacheCommit(utils.NonTransactional), utils.NonTransactional) } return nil } // GetCDRs returns the CDRs from DB based on given filters func (iDB *InternalDB) GetCDRs(filter *utils.CDRsFilter, remove bool) (cdrs []*CDR, count int64, err error) { // filterPair used only for GetCDRs for internalDB type filterPair struct { key string ids []string } var pairSlice []filterPair var notPairSlice []filterPair // get indexed fields if len(iDB.stringIndexedFields) != 0 || len(iDB.prefixIndexedFields) != 0 { for _, cdrIdx := range iDB.stringIndexedFields { switch cdrIdx { case utils.CGRID: pairSlice = append(pairSlice, filterPair{utils.CGRID, filter.CGRIDs}) notPairSlice = append(notPairSlice, filterPair{utils.CGRID, filter.NotCGRIDs}) filter.NotCGRIDs = nil filter.CGRIDs = nil case utils.RunID: pairSlice = append(pairSlice, filterPair{utils.RunID, filter.RunIDs}) notPairSlice = append(notPairSlice, filterPair{utils.RunID, filter.NotRunIDs}) filter.NotRunIDs = nil filter.RunIDs = nil case utils.OriginID: pairSlice = append(pairSlice, filterPair{utils.OriginID, filter.OriginIDs}) notPairSlice = append(notPairSlice, filterPair{utils.OriginID, filter.NotOriginIDs}) filter.NotOriginIDs = nil filter.OriginIDs = nil case utils.OriginHost: pairSlice = append(pairSlice, filterPair{utils.OriginHost, filter.OriginHosts}) notPairSlice = append(notPairSlice, filterPair{utils.OriginHost, filter.NotOriginHosts}) filter.NotOriginHosts = nil filter.OriginHosts = nil case utils.Source: pairSlice = append(pairSlice, filterPair{utils.Source, filter.Sources}) notPairSlice = append(notPairSlice, filterPair{utils.Source, filter.NotSources}) filter.NotSources = nil filter.Sources = nil case utils.ToR: pairSlice = append(pairSlice, filterPair{utils.ToR, filter.ToRs}) notPairSlice = append(notPairSlice, filterPair{utils.ToR, filter.NotToRs}) filter.NotToRs = nil filter.ToRs = nil case utils.RequestType: pairSlice = append(pairSlice, filterPair{utils.RequestType, filter.RequestTypes}) notPairSlice = append(notPairSlice, filterPair{utils.RequestType, filter.NotRequestTypes}) filter.NotRequestTypes = nil filter.RequestTypes = nil case utils.Tenant: pairSlice = append(pairSlice, filterPair{utils.Tenant, filter.Tenants}) notPairSlice = append(notPairSlice, filterPair{utils.Tenant, filter.NotTenants}) filter.NotTenants = nil filter.Tenants = nil case utils.Category: pairSlice = append(pairSlice, filterPair{utils.Category, filter.Categories}) notPairSlice = append(notPairSlice, filterPair{utils.Category, filter.NotCategories}) filter.NotCategories = nil filter.Categories = nil case utils.AccountField: pairSlice = append(pairSlice, filterPair{utils.AccountField, filter.Accounts}) notPairSlice = append(notPairSlice, filterPair{utils.AccountField, filter.NotAccounts}) filter.NotAccounts = nil filter.Accounts = nil case utils.Subject: pairSlice = append(pairSlice, filterPair{utils.Subject, filter.Subjects}) notPairSlice = append(notPairSlice, filterPair{utils.Subject, filter.NotSubjects}) filter.NotSubjects = nil filter.Subjects = nil default: if val, has := filter.ExtraFields[cdrIdx]; has && val != utils.MetaExists { // if the filter value is *exist it should not be treated as a indexed field pairSlice = append(pairSlice, filterPair{cdrIdx, []string{val}}) delete(filter.ExtraFields, cdrIdx) } if val, has := filter.NotExtraFields[cdrIdx]; has && val != utils.MetaExists { // if the filter value is *exist it should not be treated as a indexed field notPairSlice = append(notPairSlice, filterPair{cdrIdx, []string{val}}) delete(filter.NotExtraFields, cdrIdx) } } } for _, cdrIdx := range iDB.prefixIndexedFields { switch cdrIdx { case utils.Destination: pairSlice = append(pairSlice, filterPair{utils.Destination, filter.DestinationPrefixes}) notPairSlice = append(notPairSlice, filterPair{utils.Destination, filter.NotDestinationPrefixes}) filter.DestinationPrefixes = nil filter.NotDestinationPrefixes = nil default: if val, has := filter.ExtraFields[cdrIdx]; has && val != utils.MetaExists { // if the filter value is *exist it should not be treated as a indexed field pairSlice = append(pairSlice, filterPair{cdrIdx, []string{val}}) delete(filter.ExtraFields, cdrIdx) } if val, has := filter.NotExtraFields[cdrIdx]; has && val != utils.MetaExists { // if the filter value is *exist it should not be treated as a indexed field notPairSlice = append(notPairSlice, filterPair{cdrIdx, []string{val}}) delete(filter.NotExtraFields, cdrIdx) } } } } else { pairSlice = []filterPair{ {utils.CGRID, filter.CGRIDs}, {utils.RunID, filter.RunIDs}, {utils.OriginID, filter.OriginIDs}, {utils.OriginHost, filter.OriginHosts}, {utils.Source, filter.Sources}, {utils.ToR, filter.ToRs}, {utils.RequestType, filter.RequestTypes}, {utils.Tenant, filter.Tenants}, {utils.Category, filter.Categories}, {utils.AccountField, filter.Accounts}, {utils.Subject, filter.Subjects}, {utils.Destination, filter.DestinationPrefixes}, } notPairSlice = []filterPair{ {utils.CGRID, filter.NotCGRIDs}, {utils.RunID, filter.NotRunIDs}, {utils.OriginID, filter.NotOriginIDs}, {utils.OriginHost, filter.NotOriginHosts}, {utils.Source, filter.NotSources}, {utils.ToR, filter.NotToRs}, {utils.RequestType, filter.NotRequestTypes}, {utils.Tenant, filter.NotTenants}, {utils.Category, filter.NotCategories}, {utils.AccountField, filter.NotAccounts}, {utils.Subject, filter.NotSubjects}, {utils.Destination, filter.NotDestinationPrefixes}, } filter.CGRIDs = nil filter.RunIDs = nil filter.OriginIDs = nil filter.OriginHosts = nil filter.Sources = nil filter.ToRs = nil filter.RequestTypes = nil filter.Tenants = nil filter.Categories = nil filter.Accounts = nil filter.Subjects = nil filter.DestinationPrefixes = nil filter.NotCGRIDs = nil filter.NotRunIDs = nil filter.NotOriginIDs = nil filter.NotOriginHosts = nil filter.NotSources = nil filter.NotToRs = nil filter.NotRequestTypes = nil filter.NotTenants = nil filter.NotCategories = nil filter.NotAccounts = nil filter.NotSubjects = nil filter.NotDestinationPrefixes = nil } // find indexed fields var cdrMpIDs utils.StringSet // Apply string filter for _, fltrSlc := range pairSlice { if len(fltrSlc.ids) == 0 { continue } grpMpIDs := make(utils.StringSet) for _, id := range fltrSlc.ids { grpMpIDs.AddSlice(iDB.db.GetGroupItemIDs(utils.CacheCDRsTBL, utils.ConcatenatedKey(fltrSlc.key, id))) } if grpMpIDs.Size() == 0 { if filter.Count { return nil, 0, nil } return nil, 0, utils.ErrNotFound } if cdrMpIDs == nil { cdrMpIDs = grpMpIDs continue } cdrMpIDs.Intersect(grpMpIDs) if cdrMpIDs.Size() == 0 { if filter.Count { return nil, 0, nil } return nil, 0, utils.ErrNotFound } } if cdrMpIDs == nil { cdrMpIDs = utils.NewStringSet(iDB.db.GetItemIDs(utils.CacheCDRsTBL, utils.EmptyString)) } // check for Not filters for _, fltrSlc := range notPairSlice { if len(fltrSlc.ids) == 0 { continue } for _, id := range fltrSlc.ids { for _, id := range iDB.db.GetGroupItemIDs(utils.CacheCDRsTBL, utils.ConcatenatedKey(fltrSlc.key, id)) { if !cdrMpIDs.Has(id) { continue } cdrMpIDs.Remove(id) if cdrMpIDs.Size() == 0 { if filter.Count { return nil, 0, nil } return nil, 0, utils.ErrNotFound } } } } if cdrMpIDs.Size() == 0 { if filter.Count { return nil, 0, nil } return nil, 0, utils.ErrNotFound } // check non indexed fields var minUsage time.Duration var maxUsage time.Duration if len(filter.MinUsage) != 0 { if minUsage, err = utils.ParseDurationWithNanosecs(filter.MinUsage); err != nil { return nil, 0, err } } if len(filter.MaxUsage) != 0 { if maxUsage, err = utils.ParseDurationWithNanosecs(filter.MaxUsage); err != nil { return nil, 0, err } } var offset int if filter.Paginator.Offset != nil { offset = *filter.Paginator.Offset } filter.Prepare() for key := range cdrMpIDs { x, ok := iDB.db.Get(utils.CacheCDRsTBL, key) if !ok || x == nil { return nil, 0, utils.ErrNotFound } var cdr *CDR if config.CgrConfig().CdrsCfg().CompressStoredCost { cdrCompres, ok := x.(*CompressedCostCDR) if !ok { return nil, 0, utils.ErrNotFound } cdr, err = NewCDRFromCompressedCDR(cdrCompres, iDB.ms) if err != nil { return } } else { cdr = x.(*CDR) } // default indexed filters if (len(filter.CGRIDs) > 0 && !slices.Contains(filter.CGRIDs, cdr.CGRID)) || (len(filter.RunIDs) > 0 && !slices.Contains(filter.RunIDs, cdr.RunID)) || (len(filter.OriginIDs) > 0 && !slices.Contains(filter.OriginIDs, cdr.OriginID)) || (len(filter.OriginHosts) > 0 && !slices.Contains(filter.OriginHosts, cdr.OriginHost)) || (len(filter.Sources) > 0 && !slices.Contains(filter.Sources, cdr.Source)) || (len(filter.ToRs) > 0 && !slices.Contains(filter.ToRs, cdr.ToR)) || (len(filter.RequestTypes) > 0 && !slices.Contains(filter.RequestTypes, cdr.RequestType)) || (len(filter.Tenants) > 0 && !slices.Contains(filter.Tenants, cdr.Tenant)) || (len(filter.Categories) > 0 && !slices.Contains(filter.Categories, cdr.Category)) || (len(filter.Accounts) > 0 && !slices.Contains(filter.Accounts, cdr.Account)) || (len(filter.Subjects) > 0 && !slices.Contains(filter.Subjects, cdr.Subject)) || (len(filter.NotCGRIDs) > 0 && slices.Contains(filter.NotCGRIDs, cdr.CGRID)) || (len(filter.NotRunIDs) > 0 && slices.Contains(filter.NotRunIDs, cdr.RunID)) || (len(filter.NotOriginIDs) > 0 && slices.Contains(filter.NotOriginIDs, cdr.OriginID)) || (len(filter.NotOriginHosts) > 0 && slices.Contains(filter.NotOriginHosts, cdr.OriginHost)) || (len(filter.NotSources) > 0 && slices.Contains(filter.NotSources, cdr.Source)) || (len(filter.NotToRs) > 0 && slices.Contains(filter.NotToRs, cdr.ToR)) || (len(filter.NotRequestTypes) > 0 && slices.Contains(filter.NotRequestTypes, cdr.RequestType)) || (len(filter.NotTenants) > 0 && slices.Contains(filter.NotTenants, cdr.Tenant)) || (len(filter.NotCategories) > 0 && slices.Contains(filter.NotCategories, cdr.Category)) || (len(filter.NotAccounts) > 0 && slices.Contains(filter.NotAccounts, cdr.Account)) || (len(filter.NotSubjects) > 0 && slices.Contains(filter.NotSubjects, cdr.Subject)) || (len(filter.Costs) > 0 && !slices.Contains(filter.Costs, cdr.Cost)) || (len(filter.NotCosts) > 0 && slices.Contains(filter.NotCosts, cdr.Cost)) || (len(filter.DestinationPrefixes) > 0 && !utils.HasPrefixSlice(filter.DestinationPrefixes, cdr.Destination)) || (len(filter.NotDestinationPrefixes) > 0 && utils.HasPrefixSlice(filter.NotDestinationPrefixes, cdr.Destination)) || (filter.OrderIDStart != nil && cdr.OrderID < *filter.OrderIDStart) || (filter.OrderIDEnd != nil && cdr.OrderID >= *filter.OrderIDEnd) || (filter.AnswerTimeStart != nil && !filter.AnswerTimeStart.IsZero() && cdr.AnswerTime.Before(*filter.AnswerTimeStart)) || (filter.AnswerTimeEnd != nil && !filter.AnswerTimeEnd.IsZero() && cdr.AnswerTime.After(*filter.AnswerTimeEnd)) || (filter.SetupTimeStart != nil && !filter.SetupTimeStart.IsZero() && cdr.SetupTime.Before(*filter.SetupTimeStart)) || (filter.SetupTimeEnd != nil && !filter.SetupTimeEnd.IsZero() && cdr.SetupTime.Before(*filter.SetupTimeEnd)) || (len(filter.MinUsage) != 0 && cdr.Usage < minUsage) || (len(filter.MaxUsage) != 0 && cdr.Usage > maxUsage) { continue } // normal filters if filter.MinCost != nil { if filter.MaxCost == nil { if cdr.Cost < *filter.MinCost { continue } } else if *filter.MinCost == 0.0 && *filter.MaxCost == -1.0 { // Special case when we want to skip errors if cdr.Cost < 0 { continue } } else if cdr.Cost < *filter.MinCost || cdr.Cost >= *filter.MaxCost { continue } } else if filter.MaxCost != nil { if *filter.MaxCost == -1.0 { // Non-rated CDRs if cdr.Cost < 0 { continue } } else if cdr.Cost >= *filter.MaxCost { // Above limited CDRs, since MinCost is empty, make sure we query also NULL cost continue } } if len(filter.ExtraFields) != 0 { passFilter := true for extFldID, extFldVal := range filter.ExtraFields { val, has := cdr.ExtraFields[extFldID] passFilter = val == extFldVal if extFldVal == utils.MetaExists { passFilter = has } if !passFilter { break } } if !passFilter { continue } } if len(filter.NotExtraFields) != 0 { passFilter := true for notExtFldID, notExtFldVal := range filter.NotExtraFields { val, has := cdr.ExtraFields[notExtFldID] passFilter = val != notExtFldVal if notExtFldVal == utils.MetaExists { passFilter = !has } if !passFilter { break } } if !passFilter { continue } } if filter.OrderBy == utils.EmptyString { // if do not have to order exit early if offset > 0 { offset-- continue } if filter.Paginator.Limit != nil && len(cdrs) >= *filter.Paginator.Limit { break } } //pass all filters and append to slice cdrs = append(cdrs, cdr) } if len(cdrs) <= offset { // if we have offset populated but not enough cdrs return if filter.Count { return nil, 0, nil } return nil, 0, utils.ErrNotFound } if filter.Count { cdrs = cdrs[offset:] if filter.Paginator.Limit != nil && len(cdrs) >= *filter.Paginator.Limit { return nil, int64(*filter.Paginator.Limit), nil } return nil, int64(len(cdrs)), nil } if len(cdrs) == 0 { if remove { return nil, 0, nil } return nil, 0, utils.ErrNotFound } if filter.OrderBy != utils.EmptyString { separateVals := strings.Split(filter.OrderBy, utils.InfieldSep) ascendent := !(len(separateVals) == 2 && separateVals[1] == "desc") switch separateVals[0] { case utils.OrderID: if ascendent { sort.Slice(cdrs, func(i, j int) bool { return cdrs[i].OrderID < cdrs[j].OrderID }) } else { sort.Slice(cdrs, func(i, j int) bool { return cdrs[i].OrderID > cdrs[j].OrderID }) } case utils.AnswerTime: if ascendent { sort.Slice(cdrs, func(i, j int) bool { return cdrs[i].AnswerTime.Before(cdrs[j].AnswerTime) }) } else { sort.Slice(cdrs, func(i, j int) bool { return cdrs[i].AnswerTime.After(cdrs[j].AnswerTime) }) } case utils.SetupTime: if ascendent { sort.Slice(cdrs, func(i, j int) bool { return cdrs[i].SetupTime.Before(cdrs[j].SetupTime) }) } else { sort.Slice(cdrs, func(i, j int) bool { return cdrs[i].SetupTime.After(cdrs[j].SetupTime) }) } case utils.Usage: if ascendent { sort.Slice(cdrs, func(i, j int) bool { return cdrs[i].Usage < cdrs[j].Usage }) } else { sort.Slice(cdrs, func(i, j int) bool { return cdrs[i].Usage > cdrs[j].Usage }) } case utils.Cost: if ascendent { sort.Slice(cdrs, func(i, j int) bool { return cdrs[i].Cost < cdrs[j].Cost }) } else { sort.Slice(cdrs, func(i, j int) bool { return cdrs[i].Cost > cdrs[j].Cost }) } default: return nil, 0, fmt.Errorf("Invalid value : %s", separateVals[0]) } cdrs = cdrs[offset:] if filter.Paginator.Limit != nil { if len(cdrs) > *filter.Paginator.Limit { cdrs = cdrs[:*filter.Paginator.Limit] } } } if remove { for _, cdr := range cdrs { iDB.db.Remove(utils.CacheCDRsTBL, utils.ConcatenatedKey(cdr.CGRID, cdr.RunID, cdr.OriginID), cacheCommit(utils.NonTransactional), utils.NonTransactional) } return nil, 0, nil } return } func (iDB *InternalDB) GetSMCosts(cgrid, runid, originHost, originIDPrfx string) (smCosts []*SMCost, err error) { var smMpIDs utils.StringMap for _, fltrSlc := range []struct { key string id string }{ {utils.CGRID, cgrid}, {utils.RunID, runid}, {utils.OriginHost, originHost}, } { if fltrSlc.id == utils.EmptyString { continue } grpMpIDs := make(utils.StringMap) grpIDs := iDB.db.GetGroupItemIDs(utils.CacheSessionCostsTBL, utils.ConcatenatedKey(fltrSlc.key, fltrSlc.id)) for _, id := range grpIDs { grpMpIDs[id] = true } if len(grpMpIDs) == 0 { return nil, utils.ErrNotFound } if smMpIDs == nil { smMpIDs = grpMpIDs } else { for id := range smMpIDs { if !grpMpIDs.HasKey(id) { delete(smMpIDs, id) if len(smMpIDs) == 0 { return nil, utils.ErrNotFound } } } } } if smMpIDs == nil { smMpIDs = utils.StringMapFromSlice(iDB.db.GetItemIDs(utils.CacheSessionCostsTBL, utils.EmptyString)) } if len(smMpIDs) == 0 { return nil, utils.ErrNotFound } for key := range smMpIDs { x, ok := iDB.db.Get(utils.CacheSessionCostsTBL, key) if !ok || x == nil { return nil, utils.ErrNotFound } smCost := x.(*SMCost) if originIDPrfx != utils.EmptyString && !strings.HasPrefix(smCost.OriginID, originIDPrfx) { continue } smCosts = append(smCosts, smCost) } return } func (iDB *InternalDB) SetSMCost(smCost *SMCost) (err error) { if smCost.CostDetails == nil { return nil } idxs := make(utils.StringSet) idxs.Add(utils.ConcatenatedKey(utils.CGRID, smCost.CGRID)) idxs.Add(utils.ConcatenatedKey(utils.RunID, smCost.RunID)) idxs.Add(utils.ConcatenatedKey(utils.OriginHost, smCost.OriginHost)) idxs.Add(utils.ConcatenatedKey(utils.OriginID, smCost.OriginID)) idxs.Add(utils.ConcatenatedKey(utils.CostSource, smCost.CostSource)) iDB.db.Set(utils.CacheSessionCostsTBL, utils.ConcatenatedKey(smCost.CGRID, smCost.RunID, smCost.OriginHost, smCost.OriginID), smCost, idxs.AsSlice(), cacheCommit(utils.NonTransactional), utils.NonTransactional) return err } // Will dump everything inside stordb to files func (iDB *InternalDB) DumpStorDB() (err error) { return iDB.db.DumpAll() } // Will rewrite every dump file of StorDB func (iDB *InternalDB) RewriteStorDB() (err error) { return iDB.db.RewriteAll() } // BackupStorDB will momentarely stop any dumping and rewriting until all dump folder is backed up in folder path backupFolderPath, making zip true will create a zip file in the path instead func (iDB *InternalDB) BackupStorDB(backupFolderPath string, zip bool) (err error) { return iDB.db.BackupDumpFolder(backupFolderPath, zip) }