Files
cgrates/cmd/cgr-loader/cgr-loader.go
2012-07-07 11:04:17 +03:00

99 lines
3.4 KiB
Go

/*
Rating system designed to be used in VoIP Carriers World
Copyright (C) 2012 Radu Ioan Fericean
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 main
import (
"flag"
"github.com/cgrates/cgrates/timespans"
"log"
)
var (
separator = flag.String("separator", ",", "Default field separator")
redisserver = flag.String("redisserver", "127.0.0.1:6379", "redis server address (tcp:127.0.0.1:6379)")
redisdb = flag.Int("rdb", 10, "redis database number (10)")
redispass = flag.String("pass", "", "redis database password")
flush = flag.Bool("flush", false, "Flush the database before importing")
monthsFn = flag.String("month", "Months.csv", "Months file")
monthdaysFn = flag.String("monthdays", "MonthDays.csv", "Month days file")
weekdaysFn = flag.String("weekdays", "WeekDays.csv", "Week days file")
destinationsFn = flag.String("destinations", "Destinations.csv", "Destinations file")
ratesFn = flag.String("rates", "Rates.csv", "Rates file")
timingsFn = flag.String("timings", "Timings.csv", "Timings file")
ratestimingsFn = flag.String("ratestimings", "RatesTimings.csv", "Rates timings file")
ratingprofilesFn = flag.String("ratingprofiles", "RatingProfiles.csv", "Rating profiles file")
actionsFn = flag.String("actions", "Actions.csv", "Actions file")
actiontimingsFn = flag.String("actiontimings", "ActionTimings.csv", "Actions timings file")
actiontriggersFn = flag.String("actiontriggers", "ActionTriggers.csv", "Actions triggers file")
accountactionsFn = flag.String("accountactions", "AccountActions.csv", "Account actions file")
sep rune
)
func writeToDatabase() {
storage, err := timespans.NewRedisStorage(*redisserver, *redisdb)
if err != nil {
log.Fatalf("Could not open database connection: %v", err)
}
if *flush {
storage.Flush()
}
log.Print("Destinations")
for _, d := range destinations {
storage.SetDestination(d)
log.Print(d.Id, " : ", d.Prefixes)
}
log.Print("Rating profiles")
for _, cds := range ratingProfiles {
for _, cd := range cds {
err = storage.SetActivationPeriodsOrFallback(cd.GetKey(), cd.ActivationPeriods, cd.FallbackKey)
log.Print(cd.GetKey())
}
}
log.Print("Action timings")
for k, ats := range actionsTimings {
storage.SetActionTimings(timespans.ACTION_TIMING_PREFIX+":"+k, ats)
log.Println(k)
}
log.Print("Actions")
for k, as := range actions {
storage.SetActions(k, as)
log.Println(k)
}
log.Print("Account actions")
for _, ub := range accountActions {
storage.SetUserBalance(ub)
log.Println(ub.Id)
}
}
func main() {
flag.Parse()
sep = []rune(*separator)[0]
loadDestinations()
loadRates()
loadTimings()
loadRatesTimings()
loadRatingProfiles()
loadActions()
loadActionTimings()
loadActionTriggers()
loadAccountActions()
writeToDatabase()
}