Implement config drv funcs for redis, mongo and internal dbs

This commit is contained in:
ionutboangiu
2022-03-01 18:44:20 +02:00
committed by Dan Christian Bogos
parent eb22903fd4
commit b2f11e7f42
6 changed files with 489 additions and 168 deletions

View File

@@ -652,14 +652,35 @@ func (iDB *InternalDB) RemoveAccountDrv(_ *context.Context, tenant, id string) (
return
}
func (iDB *InternalDB) GetConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionIDs []string) (map[string][]byte, error) {
return nil, utils.ErrNotImplemented
func (iDB *InternalDB) GetConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionIDs []string) (sectionMap map[string][]byte, err error) {
sectionMap = make(map[string][]byte)
for _, sectionID := range sectionIDs {
x, ok := iDB.db.Get(utils.CacheConfig, utils.ConcatenatedKey(tenant, nodeID, sectionID))
if !ok || x == nil {
utils.Logger.Warning(fmt.Sprintf("<%+v> Could not find any data for section <%+v>",
utils.ConcatenatedKey(tenant, nodeID), sectionID))
continue
}
sectionMap[sectionID] = x.([]byte)
}
if len(sectionMap) == 0 {
err = utils.ErrNotFound
return
}
return
}
func (iDB *InternalDB) SetConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionsData map[string][]byte) error {
return utils.ErrNotImplemented
func (iDB *InternalDB) SetConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionsData map[string][]byte) (err error) {
for sectionID, sectionData := range sectionsData {
iDB.db.Set(utils.CacheConfig, utils.ConcatenatedKey(tenant, nodeID, sectionID),
sectionData, nil, true, utils.NonTransactional)
}
return
}
func (iDB *InternalDB) RemoveConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionIDs []string) error {
return utils.ErrNotImplemented
func (iDB *InternalDB) RemoveConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionIDs []string) (err error) {
for _, sectionID := range sectionIDs {
iDB.db.Remove(utils.CacheConfig, utils.ConcatenatedKey(tenant, nodeID, sectionID), true, utils.NonTransactional)
}
return
}

391
engine/storage_it_test.go Normal file
View File

@@ -0,0 +1,391 @@
//go:build integration
// +build integration
/*
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 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>
*/
package engine
import (
"reflect"
"testing"
"time"
"github.com/cgrates/birpc/context"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/utils"
)
func TestSetGetRemoveConfigSectionsDrvRedis(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
db, err := NewRedisStorage(cfg.DataDbCfg().Host+":"+cfg.DataDbCfg().Port, 10, cfg.DataDbCfg().User,
cfg.DataDbCfg().Password, cfg.GeneralCfg().DBDataEncoding, utils.RedisMaxConns, utils.RedisMaxAttempts,
utils.EmptyString, false, 0, 0, false, utils.EmptyString, utils.EmptyString, utils.EmptyString)
if err != nil {
t.Error(err)
}
defer db.Close()
sectionIDs := []string{"thresholds", "resources"}
expected := make(map[string][]byte)
// Try to retrieve the values before setting them (should receive an empty map)
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ErrNotFound, err)
} else if !reflect.DeepEqual(rcv, expected) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(expected), utils.ToJSON(rcv))
}
ms, err := utils.NewMarshaler(utils.JSON)
if err != nil {
t.Error(err)
}
thCfg := &config.ThresholdSJsonCfg{
Enabled: utils.BoolPointer(true),
Indexed_selects: utils.BoolPointer(true),
Store_interval: utils.StringPointer("2s"),
String_indexed_fields: &[]string{"req.index11"},
Prefix_indexed_fields: &[]string{"req.index22"},
Suffix_indexed_fields: &[]string{"req.index33"},
Actions_conns: &[]string{"*internal"},
Nested_fields: utils.BoolPointer(true),
Opts: &config.ThresholdsOptsJson{
ProfileIDs: []*utils.DynamicStringSliceOpt{
{
Tenant: "cgrates.org",
Value: []string{"value1"},
},
},
ProfileIgnoreFilters: []*utils.DynamicBoolOpt{
{
Tenant: "cgrates.org",
Value: true,
},
},
},
}
thJsnCfg, err := ms.Marshal(thCfg)
if err != nil {
t.Error(err)
}
rsCfg := &config.ResourceSJsonCfg{
Enabled: utils.BoolPointer(true),
Indexed_selects: utils.BoolPointer(true),
Thresholds_conns: &[]string{"*birpc"},
Store_interval: utils.StringPointer("2s"),
String_indexed_fields: &[]string{"*req.index11"},
Prefix_indexed_fields: &[]string{"*req.index22"},
Suffix_indexed_fields: &[]string{"*req.index33"},
Nested_fields: utils.BoolPointer(true),
Opts: &config.ResourcesOptsJson{
UsageID: []*utils.DynamicStringOpt{
{
Value: "usg2",
},
},
UsageTTL: []*utils.DynamicStringOpt{
{
Value: "1m0s",
},
},
Units: []*utils.DynamicFloat64Opt{
{
Value: 2,
},
},
},
}
rsJsnCfg, err := ms.Marshal(rsCfg)
if err != nil {
t.Error(err)
}
sectData := map[string][]byte{
"thresholds": thJsnCfg,
"resources": rsJsnCfg,
}
if err := db.SetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectData); err != nil {
t.Error(err)
}
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(rcv, sectData) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(sectData), utils.ToJSON(rcv))
} else {
rcvThCfg := &config.ThresholdSJsonCfg{}
ms.Unmarshal(rcv["thresholds"], &rcvThCfg)
if !reflect.DeepEqual(rcvThCfg, thCfg) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(thCfg), utils.ToJSON(rcvThCfg))
}
rcvRsCfg := &config.ResourceSJsonCfg{}
ms.Unmarshal(rcv["resources"], &rcvRsCfg)
if !reflect.DeepEqual(rcvRsCfg, rsCfg) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(rsCfg), utils.ToJSON(rcvRsCfg))
}
}
if err := db.RemoveConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err != nil {
t.Error(err)
}
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ErrNotFound, err)
} else if !reflect.DeepEqual(rcv, expected) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(expected), utils.ToJSON(rcv))
}
}
func TestSetGetRemoveConfigSectionsDrvMongo(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
db, err := NewMongoStorage(cfg.DataDbCfg().Host, "27017", "10", cfg.DataDbCfg().User,
cfg.DataDbCfg().Password, cfg.GeneralCfg().DBDataEncoding, utils.DataDB, nil, 10*time.Second)
if err != nil {
t.Error(err)
}
defer db.Close()
sectionIDs := []string{"thresholds", "resources"}
expected := make(map[string][]byte)
// Try to retrieve the values before setting them (should receive an empty map)
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ErrNotFound, err)
} else if !reflect.DeepEqual(rcv, expected) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(expected), utils.ToJSON(rcv))
}
ms, err := utils.NewMarshaler(utils.JSON)
if err != nil {
t.Error(err)
}
thCfg := &config.ThresholdSJsonCfg{
Enabled: utils.BoolPointer(true),
Indexed_selects: utils.BoolPointer(true),
Store_interval: utils.StringPointer("2s"),
String_indexed_fields: &[]string{"req.index11"},
Prefix_indexed_fields: &[]string{"req.index22"},
Suffix_indexed_fields: &[]string{"req.index33"},
Actions_conns: &[]string{"*internal"},
Nested_fields: utils.BoolPointer(true),
Opts: &config.ThresholdsOptsJson{
ProfileIDs: []*utils.DynamicStringSliceOpt{
{
Tenant: "cgrates.org",
Value: []string{"value1"},
},
},
ProfileIgnoreFilters: []*utils.DynamicBoolOpt{
{
Tenant: "cgrates.org",
Value: true,
},
},
},
}
thJsnCfg, err := ms.Marshal(thCfg)
if err != nil {
t.Error(err)
}
rsCfg := &config.ResourceSJsonCfg{
Enabled: utils.BoolPointer(true),
Indexed_selects: utils.BoolPointer(true),
Thresholds_conns: &[]string{"*birpc"},
Store_interval: utils.StringPointer("2s"),
String_indexed_fields: &[]string{"*req.index11"},
Prefix_indexed_fields: &[]string{"*req.index22"},
Suffix_indexed_fields: &[]string{"*req.index33"},
Nested_fields: utils.BoolPointer(true),
Opts: &config.ResourcesOptsJson{
UsageID: []*utils.DynamicStringOpt{
{
Value: "usg2",
},
},
UsageTTL: []*utils.DynamicStringOpt{
{
Value: "1m0s",
},
},
Units: []*utils.DynamicFloat64Opt{
{
Value: 2,
},
},
},
}
rsJsnCfg, err := ms.Marshal(rsCfg)
if err != nil {
t.Error(err)
}
sectData := map[string][]byte{
"thresholds": thJsnCfg,
"resources": rsJsnCfg,
}
if err := db.SetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectData); err != nil {
t.Error(err)
}
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(rcv, sectData) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(sectData), utils.ToJSON(rcv))
} else {
rcvThCfg := &config.ThresholdSJsonCfg{}
ms.Unmarshal(rcv["thresholds"], &rcvThCfg)
if !reflect.DeepEqual(rcvThCfg, thCfg) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(thCfg), utils.ToJSON(rcvThCfg))
}
rcvRsCfg := &config.ResourceSJsonCfg{}
ms.Unmarshal(rcv["resources"], &rcvRsCfg)
if !reflect.DeepEqual(rcvRsCfg, rsCfg) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(rsCfg), utils.ToJSON(rcvRsCfg))
}
}
if err := db.RemoveConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err != nil {
t.Error(err)
}
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ErrNotFound, err)
} else if !reflect.DeepEqual(rcv, expected) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(expected), utils.ToJSON(rcv))
}
}
func TestSetGetRemoveConfigSectionsDrvInternal(t *testing.T) {
db := NewInternalDB(nil, nil, nil)
if err != nil {
t.Error(err)
}
defer db.Close()
sectionIDs := []string{"thresholds", "resources"}
expected := make(map[string][]byte)
// Try to retrieve the values before setting them (should receive an empty map)
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ErrNotFound, err)
} else if !reflect.DeepEqual(rcv, expected) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(expected), utils.ToJSON(rcv))
}
ms, err := utils.NewMarshaler(utils.JSON)
if err != nil {
t.Error(err)
}
thCfg := &config.ThresholdSJsonCfg{
Enabled: utils.BoolPointer(true),
Indexed_selects: utils.BoolPointer(true),
Store_interval: utils.StringPointer("2s"),
String_indexed_fields: &[]string{"req.index11"},
Prefix_indexed_fields: &[]string{"req.index22"},
Suffix_indexed_fields: &[]string{"req.index33"},
Actions_conns: &[]string{"*internal"},
Nested_fields: utils.BoolPointer(true),
Opts: &config.ThresholdsOptsJson{
ProfileIDs: []*utils.DynamicStringSliceOpt{
{
Tenant: "cgrates.org",
Value: []string{"value1"},
},
},
ProfileIgnoreFilters: []*utils.DynamicBoolOpt{
{
Tenant: "cgrates.org",
Value: true,
},
},
},
}
thJsnCfg, err := ms.Marshal(thCfg)
if err != nil {
t.Error(err)
}
rsCfg := &config.ResourceSJsonCfg{
Enabled: utils.BoolPointer(true),
Indexed_selects: utils.BoolPointer(true),
Thresholds_conns: &[]string{"*birpc"},
Store_interval: utils.StringPointer("2s"),
String_indexed_fields: &[]string{"*req.index11"},
Prefix_indexed_fields: &[]string{"*req.index22"},
Suffix_indexed_fields: &[]string{"*req.index33"},
Nested_fields: utils.BoolPointer(true),
Opts: &config.ResourcesOptsJson{
UsageID: []*utils.DynamicStringOpt{
{
Value: "usg2",
},
},
UsageTTL: []*utils.DynamicStringOpt{
{
Value: "1m0s",
},
},
Units: []*utils.DynamicFloat64Opt{
{
Value: 2,
},
},
},
}
rsJsnCfg, err := ms.Marshal(rsCfg)
if err != nil {
t.Error(err)
}
sectData := map[string][]byte{
"thresholds": thJsnCfg,
"resources": rsJsnCfg,
}
if err := db.SetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectData); err != nil {
t.Error(err)
}
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(rcv, sectData) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(sectData), utils.ToJSON(rcv))
} else {
rcvThCfg := &config.ThresholdSJsonCfg{}
ms.Unmarshal(rcv["thresholds"], &rcvThCfg)
if !reflect.DeepEqual(rcvThCfg, thCfg) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(thCfg), utils.ToJSON(rcvThCfg))
}
rcvRsCfg := &config.ResourceSJsonCfg{}
ms.Unmarshal(rcv["resources"], &rcvRsCfg)
if !reflect.DeepEqual(rcvRsCfg, rsCfg) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(rsCfg), utils.ToJSON(rcvRsCfg))
}
}
if err := db.RemoveConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err != nil {
t.Error(err)
}
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ErrNotFound, err)
} else if !reflect.DeepEqual(rcv, expected) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(expected), utils.ToJSON(rcv))
}
}

View File

@@ -1497,16 +1497,72 @@ func (ms *MongoStorage) RemoveAccountDrv(ctx *context.Context, tenant, id string
})
}
func (ms *MongoStorage) GetConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionIDs []string) (map[string][]byte, error) {
return nil, utils.ErrNotImplemented
func (ms *MongoStorage) GetConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionIDs []string) (sectionMap map[string][]byte, err error) {
sectionMap = make(map[string][]byte)
for _, sectionID := range sectionIDs {
if err = ms.query(context.TODO(), func(sctx mongo.SessionContext) (err error) {
cur := ms.getCol(ColCfg).FindOne(sctx, bson.M{
"tenant": tenant,
"nodeID": nodeID,
"section": sectionID,
}, options.FindOne().SetProjection(bson.M{"cfgData": 1, "_id": 0}))
cfgMap := make(map[string][]byte)
if err = cur.Decode(&cfgMap); err != nil {
if err == mongo.ErrNoDocuments {
err = nil
return
}
return
}
sectionMap[sectionID] = cfgMap["cfgData"]
return
}); err != nil {
return
}
}
if len(sectionMap) == 0 {
err = utils.ErrNotFound
return
}
return
}
func (ms *MongoStorage) SetConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionsData map[string][]byte) error {
return utils.ErrNotImplemented
func (ms *MongoStorage) SetConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionsData map[string][]byte) (err error) {
for sectionID, sectionData := range sectionsData {
if err = ms.query(ctx, func(sctx mongo.SessionContext) (err error) {
_, err = ms.getCol(ColCfg).UpdateOne(sctx, bson.M{
"tenant": tenant,
"nodeID": nodeID,
"section": sectionID,
}, bson.M{"$set": bson.M{
"tenant": tenant,
"nodeID": nodeID,
"section": sectionID,
"cfgData": sectionData}},
options.Update().SetUpsert(true),
)
return err
}); err != nil {
return
}
}
return
}
func (ms *MongoStorage) RemoveConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionIDs []string) error {
return utils.ErrNotImplemented
func (ms *MongoStorage) RemoveConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionIDs []string) (err error) {
for _, sectionID := range sectionIDs {
if err = ms.query(ctx, func(sctx mongo.SessionContext) (err error) {
_, err = ms.getCol(ColCfg).DeleteOne(sctx, bson.M{
"tenant": tenant,
"nodeID": nodeID,
"section": sectionID,
})
return err
}); err != nil {
return
}
}
return
}
func newAggregateStages(profileID, tenant, prefix string) (match, query bson.D) {

View File

@@ -939,28 +939,30 @@ func (rs *RedisStorage) GetConfigSectionsDrv(ctx *context.Context, tenant, nodeI
sectionMap = make(map[string][]byte)
for _, sectionID := range sectionIDs {
var value []byte
if err = rs.Cmd(&value, redisGET, utils.ConfigPrefix+sectionID); err != nil {
if err = rs.Cmd(&value, redisHGET, utils.ConfigPrefix+utils.ConcatenatedKey(tenant, nodeID), sectionID); err != nil {
return
}
if value != nil {
sectionMap[sectionID] = value
}
}
if len(sectionMap) == 0 {
err = utils.ErrNotFound
return
}
return
}
func (rs *RedisStorage) SetConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionsData map[string][]byte) (err error) {
for sectionID, sectionData := range sectionsData {
if err = rs.Cmd(nil, redisSET, utils.ConfigPrefix+sectionID, string(sectionData)); err != nil {
return
}
if err = rs.FlatCmd(nil, redisHSET, utils.ConfigPrefix+utils.ConcatenatedKey(tenant, nodeID), sectionsData); err != nil {
return
}
return
}
func (rs *RedisStorage) RemoveConfigSectionsDrv(ctx *context.Context, tenant, nodeID string, sectionIDs []string) (err error) {
for _, sectionID := range sectionIDs {
if err = rs.Cmd(nil, redisDEL, utils.ConfigPrefix+sectionID); err != nil {
if err = rs.Cmd(nil, redisHDEL, utils.ConfigPrefix+utils.ConcatenatedKey(tenant, nodeID), sectionID); err != nil {
return
}
}

View File

@@ -1,150 +0,0 @@
//go:build integration
// +build integration
/*
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 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>
*/
package engine
import (
"reflect"
"testing"
"github.com/cgrates/birpc/context"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/utils"
)
func TestSetGetRemoveConfigSectionsDrv(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
db, err := NewRedisStorage(cfg.DataDbCfg().Host+":"+cfg.DataDbCfg().Port, 10, cfg.DataDbCfg().User,
cfg.DataDbCfg().Password, cfg.GeneralCfg().DBDataEncoding, utils.RedisMaxConns, utils.RedisMaxAttempts,
utils.EmptyString, false, 0, 0, false, utils.EmptyString, utils.EmptyString, utils.EmptyString)
if err != nil {
t.Fatal(err)
}
defer db.Close()
sectionIDs := []string{"thresholds", "resources"}
expected := make(map[string][]byte)
// Try to retrieve the values before setting them (should receive an empty map)
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err != nil {
t.Fatal(err)
} else if !reflect.DeepEqual(rcv, expected) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(expected), utils.ToJSON(rcv))
}
ms, err := utils.NewMarshaler(utils.JSON)
if err != nil {
t.Fatal(err)
}
thCfg := &config.ThresholdSJsonCfg{
Enabled: utils.BoolPointer(true),
Indexed_selects: utils.BoolPointer(true),
Store_interval: utils.StringPointer("2s"),
String_indexed_fields: &[]string{"req.index11"},
Prefix_indexed_fields: &[]string{"req.index22"},
Suffix_indexed_fields: &[]string{"req.index33"},
Actions_conns: &[]string{"*internal"},
Nested_fields: utils.BoolPointer(true),
Opts: &config.ThresholdsOptsJson{
ProfileIDs: []*utils.DynamicStringSliceOpt{
{
Tenant: "cgrates.org",
Value: []string{"value1"},
},
},
ProfileIgnoreFilters: []*utils.DynamicBoolOpt{
{
Tenant: "cgrates.org",
Value: true,
},
},
},
}
thJsnCfg, err := ms.Marshal(thCfg)
if err != nil {
t.Fatal(err)
}
rsCfg := &config.ResourceSJsonCfg{
Enabled: utils.BoolPointer(true),
Indexed_selects: utils.BoolPointer(true),
Thresholds_conns: &[]string{"*birpc"},
Store_interval: utils.StringPointer("2s"),
String_indexed_fields: &[]string{"*req.index11"},
Prefix_indexed_fields: &[]string{"*req.index22"},
Suffix_indexed_fields: &[]string{"*req.index33"},
Nested_fields: utils.BoolPointer(true),
Opts: &config.ResourcesOptsJson{
UsageID: []*utils.DynamicStringOpt{
{
Value: "usg2",
},
},
UsageTTL: []*utils.DynamicStringOpt{
{
Value: "1m0s",
},
},
Units: []*utils.DynamicFloat64Opt{
{
Value: 2,
},
},
},
}
rsJsnCfg, err := ms.Marshal(rsCfg)
if err != nil {
t.Fatal(err)
}
sectData := map[string][]byte{
"thresholds": thJsnCfg,
"resources": rsJsnCfg,
}
if err := db.SetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectData); err != nil {
t.Fatal(err)
}
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err != nil {
t.Fatal(err)
} else if !reflect.DeepEqual(rcv, sectData) {
t.Fatalf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(sectData), utils.ToJSON(rcv))
} else {
rcvThCfg := &config.ThresholdSJsonCfg{}
ms.Unmarshal(rcv["thresholds"], &rcvThCfg)
if !reflect.DeepEqual(rcvThCfg, thCfg) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(thCfg), utils.ToJSON(rcvThCfg))
}
rcvRsCfg := &config.ResourceSJsonCfg{}
ms.Unmarshal(rcv["resources"], &rcvRsCfg)
if !reflect.DeepEqual(rcvRsCfg, rsCfg) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(rsCfg), utils.ToJSON(rcvRsCfg))
}
}
if err := db.RemoveConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err != nil {
t.Fatal(err)
}
if rcv, err := db.GetConfigSectionsDrv(context.Background(), "cgrates.org", "1234", sectionIDs); err != nil {
t.Fatal(err)
} else if !reflect.DeepEqual(rcv, expected) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ToJSON(expected), utils.ToJSON(rcv))
}
}

View File

@@ -1629,6 +1629,7 @@ const (
// Cache Name
const (
CacheConfig = "*config"
CacheResources = "*resources"
CacheResourceProfiles = "*resource_profiles"
CacheEventResources = "*event_resources"