fixed mongo storage

This commit is contained in:
Radu Ioan Fericean
2013-05-17 14:46:33 +03:00
parent 6e2b741028
commit 28bca0efcc
6 changed files with 13 additions and 11 deletions

View File

@@ -148,14 +148,14 @@ func main() {
var getter rater.DataStorage
switch *db_type {
case REDIS:
db_nb, err := strconv.Atoi(*db_name)
db_nb, err := strconv.Atoi(*db_name)
if err != nil {
log.Fatal("Redis db name must be an integer!")
}
if *db_port != "" {
*db_host += ":" + *db_port
}
getter, err = rater.NewRedisStorage(*db_host, db_nb, *db_pass)
getter, err = rater.NewGosexyStorage(*db_host, db_nb, *db_pass)
case MONGO:
getter, err = rater.NewMongoStorage(*db_host, *db_port, *db_name, *db_user, *db_pass)
case POSTGRES:

View File

@@ -170,7 +170,7 @@ func startSessionManager(responder *rater.Responder, loggerDb rater.DataStorage)
}
time.Sleep(time.Duration(i/2) * time.Second)
}
}
if err != nil {
rater.Logger.Crit(fmt.Sprintf("Could not connect to rater: %v", err))
@@ -225,7 +225,7 @@ func checkConfigSanity() error {
if cfg.BalancerEnabled {
if cfg.BalancerRPCEncoding != cfg.MediatorRPCEncoding {
rater.Logger.Crit("If you are connecting the mediator via the loopback to the balancer use the same type of rpc encoding!")
return errors.New("Balancer and Mediator using different encoding")
return errors.New("Balancer and Mediator using different encoding")
}
}
if cfg.RaterEnabled {
@@ -250,7 +250,7 @@ func configureDatabase(db_type, host, port, name, user, pass string) (getter rat
if port != "" {
host += ":" + port
}
getter, err = rater.NewRedisStorage(host, db_nb, pass)
getter, err = rater.NewGosexyStorage(host, db_nb, pass)
case MONGO:
getter, err = rater.NewMongoStorage(host, port, name, user, pass)
case POSTGRES:
@@ -261,7 +261,7 @@ func configureDatabase(db_type, host, port, name, user, pass string) (getter rat
}
if err != nil {
return nil, err
return nil, err
}
return getter, nil
}
@@ -282,7 +282,7 @@ func main() {
// some consitency checks
errCfg := checkConfigSanity()
if errCfg != nil {
rater.Logger.Crit( errCfg.Error() )
rater.Logger.Crit(errCfg.Error())
return
}

View File

@@ -59,7 +59,7 @@ func main() {
t2 := time.Date(2012, time.February, 02, 18, 30, 0, 0, time.UTC)
cd := rater.CallDescriptor{Direction: "OUT", TOR: "0", Tenant: "vdf", Subject: "rif", Destination: "0256", TimeStart: t1, TimeEnd: t2}
getter, err := rater.NewRedisStorage("localhost:6379", 10, "")
getter, err := rater.NewGosexyStorage("localhost:6379", 10, "")
//getter, err := rater.NewMongoStorage("localhost", "cgrates")
defer getter.Close()

View File

@@ -50,8 +50,8 @@ var (
//storageGetter, _ = NewRedisStorage(db_server+":6379", 11, "")
//storageGetter, _ = NewRedigoStorage(db_server+":6379", 11, "")
//storageGetter, _ = NewGosexyStorage(db_server+":6379", 11, "")
storageLogger = storageGetter
debitPeriod = 10 * time.Second
storageLogger = storageGetter
debitPeriod = 10 * time.Second
)
/*

View File

@@ -124,7 +124,7 @@ type LogErrEntry struct {
func (ms *MongoStorage) GetRatingProfile(key string) (rp *RatingProfile, err error) {
rp = new(RatingProfile)
err = ms.db.C("ratingprofiles").Find(bson.M{"_id": key}).One(&rp)
err = ms.db.C("ratingprofiles").Find(bson.M{"id": key}).One(&rp)
return
}

View File

@@ -1,7 +1,9 @@
#!/usr/bin/env sh
go get -v -u github.com/fzzy/radix/redis
go get -v -u code.google.com/p/goconf/conf
go get -v -u github.com/bmizerany/pq
go get -v -u github.com/vmihailenco/msgpack
go get -v -u github.com/ugorji/go-msgpack
go get -v -u labix.org/v2/mgo
go get -v -u github.com/cgrates/fsock
go get -u -v github.com/go-sql-driver/mysql