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

Added fixes for mongo driver

Browse Source
This commit is contained in:
Trial97
2018-12-13 18:57:48 +02:00
committed by Dan Christian Bogos
parent 8e2f70a1aa
commit b1e091346e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
engine/storage_mongo_datadb.go
View File

@@ -128,7 +128,7 @@ func TimeDecodeValue1(dc bsoncodec.DecodeContext, vr bsonrw.ValueReader, val ref
// NewMongoStorage givese new mongo driver
func NewMongoStorage(host, port, db, user, pass, storageType string,
cdrsIndexes []string, cacheCfg config.CacheCfg) (ms *MongoStorage, err error) {
url := "mongodb://" + host
url := host
if port != "" {
url += ":" + port
}
@@ -141,7 +141,7 @@ func NewMongoStorage(host, port, db, user, pass, storageType string,
dbName = strings.Split(db, "?")[0] // remove extra info after ?
}
ctx := context.Background()
url = "mongodb://" + url
reg := bson.NewRegistryBuilder().RegisterDecoder(tTime, bsoncodec.ValueDecoderFunc(TimeDecodeValue1)).Build()
opt := &options.ClientOptions{
Registry: reg,
@@ -174,7 +174,7 @@ func NewMongoStorage(host, port, db, user, pass, storageType string,
cdrsIndexes: cdrsIndexes,
}
if err = ms.client.UseSession(ms.ctx, func(sctx mongo.SessionContext) error {
if col, err := ms.client.Database(db).ListCollections(sctx, nil, options.ListCollections().SetNameOnly(true)); err != nil {
if col, err := ms.client.Database(dbName).ListCollections(sctx, nil, options.ListCollections().SetNameOnly(true)); err != nil {
return err
} else {
empty := true