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

Implement GetResourceProfileIDsCount

Browse Source
This commit is contained in:
ionutboangiu
2021-05-25 14:34:08 +03:00
committed by Dan Christian Bogos
parent c8d69b882a
commit 20b23d0ce8
2 changed files with 21 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
apis/attributes.go
View File

@@ -77,8 +77,8 @@ func (admS *AdminSv1) GetAttributeProfileIDsCount(ctx *context.Context, args *ut
tnt = admS.cfg.GeneralCfg().DefaultTenant
}
var keys []string
prfx := utils.AttributeProfilePrefix + tnt + utils.ConcatenatedKeySep
if keys, err = admS.dm.DataDB().GetKeysForPrefix(ctx, prfx); err != nil {
if keys, err = admS.dm.DataDB().GetKeysForPrefix(ctx,
utils.AttributeProfilePrefix+tnt+utils.ConcatenatedKeySep); err != nil {
return err
}
if len(keys) == 0 {

19
apis/resources.go
View File

@@ -62,6 +62,25 @@ func (adms *AdminSv1) GetResourceProfileIDs(ctx *context.Context, args *utils.Pa
return nil
}
// GetResourceProfileIDsCount returns the total number of ResourceProfileIDs registered for a tenant
// returns ErrNotFound in case of 0 ResourceProfileIDs
func (admS *AdminSv1) GetResourceProfileIDsCount(ctx *context.Context, args *utils.TenantWithAPIOpts, reply *int) (err error) {
tnt := args.Tenant
if tnt == utils.EmptyString {
tnt = admS.cfg.GeneralCfg().DefaultTenant
}
var keys []string
if keys, err = admS.dm.DataDB().GetKeysForPrefix(ctx,
utils.ResourceProfilesPrefix+tnt+utils.ConcatenatedKeySep); err != nil {
return err
}
if len(keys) == 0 {
return utils.ErrNotFound
}
*reply = len(keys)
return
}
//SetResourceProfile adds a new resource configuration
func (adms *AdminSv1) SetResourceProfile(ctx *context.Context, arg *engine.ResourceProfileWithAPIOpts, reply *string) (err error) {
if missing := utils.MissingStructFields(arg.ResourceProfile, []string{utils.ID}); len(missing) != 0 {