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

Added list/count rates for mongo driver

Browse Source
This commit is contained in:
porosnicuadrian
2022-02-08 14:13:28 +02:00
committed by Dan Christian Bogos
parent 3432bbd853
commit e4e7282939
2 changed files with 21 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
apis/rates_it_test.go
View File

@@ -75,7 +75,6 @@ var (
testRateProfileRemoveMultipleRates,
testRateProfileSetMultipleRatesInProfile,
testRateProfileUpdateProfile,
testRateSKillEngine,
}
)

22
engine/storage_mongo_datadb.go
View File

@@ -1221,7 +1221,27 @@ func (ms *MongoStorage) GetRateProfileDrv(ctx *context.Context, tenant, id strin
return utils.NewRateProfileFromMapDataDBMap(tenant, id, mapRP, ms.ms)
}
func (ms *MongoStorage) GetRateProfileRateIDsDrv(ctx *context.Context, key, prefix, argsPrefix string) (rateIDs []string, err error) {
func (ms *MongoStorage) GetRateProfileRateIDsDrv(ctx *context.Context, tenant, profileID, prefixArgs string) (rateIDs []string, err error) {
prefix := utils.Rates + utils.ConcatenatedKeySep
if prefixArgs != utils.EmptyString {
prefix = utils.ConcatenatedKey(utils.Rates, prefixArgs)
}
mapRP := make(map[string]interface{})
err = ms.query(ctx, func(sctx mongo.SessionContext) (err error) {
cur := ms.getCol(ColRpp).FindOne(sctx, bson.M{"tenant": tenant, "id": profileID})
if err := cur.Decode(mapRP); err != nil {
if err == mongo.ErrNoDocuments {
return utils.ErrNotFound
}
return err
}
return nil
})
for key := range mapRP {
if strings.HasPrefix(key, prefix) {
rateIDs = append(rateIDs, strings.TrimPrefix(key, utils.Rates+utils.ConcatenatedKeySep))
}
}
return
}