diff --git a/cmd/loader/kyoto/kyoto_loader.go b/cmd/loader/loader.go similarity index 66% rename from cmd/loader/kyoto/kyoto_loader.go rename to cmd/loader/loader.go index e59fcd6a2..5ba3d1c61 100644 --- a/cmd/loader/kyoto/kyoto_loader.go +++ b/cmd/loader/loader.go @@ -3,19 +3,22 @@ package main import ( "flag" "fmt" + "github.com/simonz05/godis" "github.com/fsouza/gokabinet/kc" "github.com/rif/cgrates/timespans" "time" ) -var ( - filename = flag.String("filename", "storage.kch", "kyoto storage file") +var ( + storage = flag.String("storage", "kyoto", "kyoto | redis") + filename = flag.String("filename", "storage.kch", "kyoto storage file (storage.kch)") + redisserver = flag.String("server", "tcp:127.0.0.1:6379", "redis server address (tcp:127.0.0.1:6379)") + redisdb = flag.Int("db", 10, "redis database number (10)") + redispass = flag.String("pass", "", "redis database password") ) func main() { flag.Parse() - db, _ := kc.Open(*filename, kc.WRITE) - defer db.Close() t1 := time.Date(2012, time.January, 1, 0, 0, 0, 0, time.UTC) cd1 := ×pans.CallDescriptor{CstmId: "vdf", Subject: "rif", DestinationPrefix: "0256"} @@ -42,6 +45,14 @@ func main() { value := cd1.EncodeValues() - db.Set(key, string(value)) + if *storage == "kyoto" { + db, _ := kc.Open(*filename, kc.WRITE) + db.Set(key, string(value)) + db.Close() + } else { + db := godis.New(*redisserver, *redisdb, *redispass) + db.Set(key, string(value)) + db.Quit() + } fmt.Println("Done!") } diff --git a/cmd/loader/redis/redis_loader.go b/cmd/loader/redis/redis_loader.go deleted file mode 100644 index fff219dc4..000000000 --- a/cmd/loader/redis/redis_loader.go +++ /dev/null @@ -1,38 +0,0 @@ -package main - -import ( - "fmt" - "github.com/simonz05/godis" - "github.com/rif/cgrates/timespans" - "time" -) - -func main() { - r := godis.New("", 10, "") - t1 := time.Date(2012, time.January, 1, 0, 0, 0, 0, time.UTC) - cd1 := ×pans.CallDescriptor{CstmId: "vdf", Subject: "rif", DestinationPrefix: "0256"} - ap1 := ×pans.ActivationPeriod{ActivationTime: t1} - ap1.AddInterval(×pans.Interval{ - WeekDays: []time.Weekday{time.Monday, time.Tuesday, time.Wednesday, time.Thursday, time.Friday}, - EndTime: "18:00:00", - ConnectFee: 0, - Price: 0.2, - BillingUnit: 1.0}) - ap1.AddInterval(×pans.Interval{ - WeekDays: []time.Weekday{time.Monday, time.Tuesday, time.Wednesday, time.Thursday, time.Friday}, - StartTime: "18:00:00", - ConnectFee: 0, - Price: 0.1, - BillingUnit: 1.0}) - ap1.AddInterval(×pans.Interval{ - WeekDays: []time.Weekday{time.Saturday, time.Sunday}, - ConnectFee: 0, - Price: 0.1, - BillingUnit: 1.0}) - cd1.AddActivationPeriod(ap1) - key := cd1.GetKey() - - value := cd1.EncodeValues() - r.Set(key, string(value)) - fmt.Println("Done!") -}